1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
|
#!/usr/bin/perl
use strict;
use warnings;
use utf8;
use DBI;
use Audio::Scan;
use File::HomeDir;
# Keep track of columns that need to be created in the database
our %columns;
# Variables to be set by user (TODO)
our $music_dir = File::HomeDir->my_home . "/Music/";
our $dbname = "library.db";
our $table_name = "LIBRARY";
our %extensions = (
flac => '1',
mp3 => '1',
ogg => '1'
);
my %data; #Hold info from Audio::Scan
my $statement; #Hold statements for sqlite
# Wrapper to handle calls to Audio::Scan->scan(); returns tags hash
# @_[0] -> file to scan
sub audio_scan {
my $data = Audio::Scan->scan("$_[0]");
$data = $data->{tags};
return %$data;
}
# Wrapper to handle sqlite commands
# @_[0] -> database handle
# @_[1] -> command/statement
# @_[2] (optional) -> output statement
sub db_cmd {
my $rv = $_[0]->do($_[1]);
if ($rv < 0){
die $DBI::errstr;
}
if (defined $_[2]){
print "$_[2]\n";
}
}
# Scan a directory recursively, return an array of files (optionally, matching a certain file extension or extensions)
# @_[0] -> $music_dir
# @_[1] -> hash of file extensions to scan for
sub get_files {
my @file_list;
my @file_split;
# Remove extra /'s from the end of $_[0]
$_[0] =~ s/\/+$//g;
my $dir_path = $_[0];
my $extensions = $_[1];
opendir my $dh, "$dir_path" or die "$!";
while (my $file = readdir($dh)) {
# Skip . and .. directories
if ($file eq "." or $file eq ".."){
next;
}
if (-d "$dir_path/$file"){
push(@file_list, get_files("$dir_path/$file", %extensions));
}
elsif (-f "$dir_path/$file" and -r "$dir_path/$file"){
# Check that the extension matches
@file_split = split /\./, "$dir_path/$file";
if (defined $extensions{"$file_split[-1]"} and $extensions{"$file_split[-1]"} == 1){
push(@file_list, "$dir_path/$file");
}
}
}
closedir $dh;
return @file_list;
}
# Test scan for Audio::Scan module
sub scan_test {
my $data = Audio::Scan->scan("/home/louie/Music/Bjork/Debut/01 Human Behaviour.flac");
$data = $data->{tags};
for (keys %$data){
print "$_ -> $data->{$_}\n";
}
}
# TODO parse flags and arguments
# Test to ensure $music_dir is a valid directory
if (! -d $music_dir){
die "Error: \"$music_dir\" is not a directory\n";
}
# Get a list of files in $music_dir
print "Looking through files in $music_dir\n";
my @file_list = get_files($music_dir, %extensions);
# DEBUG
for my $i (sort @file_list){
print "$i\n";
}
# Get tags for each file
#for my $file (sort @file_list){
# $data = Audio::Scan->scan("$file");
# $data = $data->{tags};
# for my $i (keys %$data){
# print "$i -> $data->{$i}\n";
# }
#}
# Append tags to %columns
for my $file (@file_list){
#$data = Audio::Scan->scan("$file");
#$data = $data->{tags};
%data = audio_scan("$file");
for my $i (keys %data){
$columns{$i} = '1';
}
}
# DEBUG
for my $i (keys %columns){
print "$i\n";
}
# Connect to sqlite database created in the base of $music_dir
my $dbh = DBI->connect("DBI:SQLite:dbname=$music_dir/$dbname", "", "", { RaiseError => 1}) or die $DBI::errstr;
print "Opened database successfully\n";
# Overwrite $table_name if it exists (TODO alert user to overwrite)
# TODO create option to append to table, instead of overwriting
$statement = "DROP TABLE if EXISTS $table_name";
db_cmd($dbh, $statement, "Overwriting table \"$table_name\"");
# Create table $table_name in the database with columns from %columns
# Need to create additional columns for ID and PATH
$statement = "CREATE TABLE $table_name
(ID INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,
PATH TEXT NOT NULL";
for my $i (sort(keys %columns)){
$statement = $statement . ",
\"$i\" TEXT";
}
$statement = $statement . ");";
db_cmd($dbh, $statement, "Created table successfully");
# Add each file from @file_list to the table
$statement = "INSERT INTO $table_name(PATH)
VALUES";
for my $file (@file_list){
$statement = $statement . "(\"$file\"),";
}
$statement =~ s/[,]$/;/g;
db_cmd($dbh, $statement);
# Set each file's tags in the table
# TODO handle array tags (such as genre)
for my $file (@file_list){
$statement = "UPDATE $table_name
SET ";
%data = audio_scan("$file");
# Loop to add all the columns for $statement
for my $i (sort(keys %data)){
next if $i eq "MCDI"; #FIXME MCDI field creates issues
$data{$i} =~ s/\"/\'\'/g;
$statement = $statement . "\"$i\" = \"$data{$i}\",";
}
$statement =~ s/[,]$/\n/g;
# Specify this insertion is for $file only
$statement = $statement . "WHERE PATH = \"$file\";";
#FIXME MCDI tag is binary. This should be considered and handled in a secondary file
db_cmd($dbh, $statement, "Updated tags for $file");
}
# Disconnect from sqlite database
$dbh->disconnect();
|
