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
|
#!/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 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};
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 (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
# Disconnect from sqlite database
$dbh->disconnect();
|
