--output flag working

1 files changed, 44 insertions, 3 deletions

diff --git a/main.plx b/main.plx
index b4ba88c..f7d97d8 100755
--- a/main.plx
+++ b/main.plx
@@ -13,7 +13,7 @@ our %columns;
 
 # Variables to be set by user (TODO)
 our $music_dir = File::HomeDir->my_home . "/Music/";
-our $dbname = "library.db";
+our $dbname;
 our $table_name = "LIBRARY";
 our %extensions = (
 	flac => '1',
@@ -21,6 +21,11 @@ our %extensions = (
 	ogg => '1'
 );
 
+# Keep track of options that have been set
+our %options = (
+	output => 0
+);
+
 my %data;      #Hold info from Audio::Scan
 my $statement; #Hold statements for sqlite
 
@@ -85,6 +90,20 @@ sub get_files {
 	return @file_list;
 }
 
+# Print a help message
+sub print_help {
+	print
+"Usage:
+  $0 [OPTION]... [DIRECTORY]
+
+Generate a database for audio files in DIRECTORY (by default ~/Music).
+
+Options:
+  -h, --help		display this help and exit
+  -o, --output FILE	specify output file for database (default is library.db at the root of DIRECTORY)
+";
+}
+
 # Test scan for Audio::Scan module
 sub scan_test {
 	my $data = Audio::Scan->scan("/home/louie/Music/Bjork/Debut/01 Human Behaviour.flac");
@@ -95,7 +114,29 @@ sub scan_test {
 }
 
 
-# TODO parse flags and arguments
+# parse flags and arguments
+for my $i (0..$#ARGV-1){
+	if ($ARGV[$i] =~ /-h|--help/){
+		print_help();
+		exit;
+	}
+
+	elsif ($ARGV[$i] =~ /-o|--output/){
+		$i++;
+		$dbname = "$ARGV[$i]";
+		$options{output} = 1;
+	}
+
+	elsif ($ARGV[$i] =~ /^[^-]/){
+		$music_dir = "$ARGV[$i]";
+		last;
+	}
+}
+
+$music_dir =~ s/\/$//;
+if (!$options{output}){
+	$dbname = $music_dir . "/library.db";
+}
 
 # Test to ensure $music_dir is a valid directory
 if (! -d $music_dir){
@@ -134,7 +175,7 @@ for my $i (keys %columns){
 }
 
 # 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;
+my $dbh = DBI->connect("DBI:SQLite:dbname=$dbname", "", "", { RaiseError => 1}) or die $DBI::errstr;
 print "Opened database successfully\n";
 
 # Overwrite $table_name if it exists (TODO alert user to overwrite)