added excel processor

2 files changed, 178 insertions, 2 deletions

diff --git a/ninka-excel.pl b/ninka-excel.pl
new file mode 100644
index 0000000..90f9457
--- /dev/null
+++ b/ninka-excel.pl
@@ -0,0 +1,175 @@
+#!/usr/bin/perl
+
+use strict;
+use Switch;
+use File::Temp;
+use File::Find;
+use File::Basename;
+use Scalar::Util qw(looks_like_number);
+use Spreadsheet::WriteExcel;
+
+
+
+if(scalar(@ARGV) != 2){
+    print STDERR "Incorrect number of arguments\n";
+    print STDERR "Correct usage is: perl ninka-wrapper <path to package file> <database name>\n";
+    exit 1;
+}
+
+my $path = $0;
+
+$path =~ s/\/+[^\/]+$//;
+if ($path eq "") {
+    $path = "./";
+}
+
+my ($pack, $excelFile) = @ARGV;
+
+my $workbook = Spreadsheet::WriteExcel->new($excelFile);
+my $worksheet = $workbook->add_worksheet();
+my $format = $workbook->add_format(); # Add a format
+$format->set_bold();
+$format->set_color('blue');
+$format->set_align('center');
+
+$worksheet->set_column(0, 9,  30);
+$worksheet->write(0, 0, 'Container File', $format);
+$worksheet->write(0, 1, 'Path', $format);
+$worksheet->write(0, 2, 'Filename', $format);
+$worksheet->write(0, 3, 'Licenses', $format);
+$worksheet->write(0, 4, 'Num found', $format);
+$worksheet->write(0, 5, 'Lines', $format);
+$worksheet->write(0, 6, 'TokensIgnored', $format);
+$worksheet->write(0, 7, 'TokensUnmatched', $format);
+$worksheet->write(0, 8, 'TokensUnknown', $format);
+$worksheet->write(0, 9, 'Tokens', $format);
+
+my $tempdir = File::Temp->newdir();
+my $dirname = $tempdir->dirname;
+
+print "***** Extracting file [$pack] to temporary directory [$dirname] *****\n";
+my $packext = getExtension($pack);
+if ($packext eq ".bz2" || $packext eq ".gz") {
+    execute("tar -xvf '$pack' --directory '$dirname'");
+} elsif ($packext eq ".jar" || $packext eq ".zip") {
+    execute("unzip -d $dirname $pack");
+} else {
+    print "ninka-wrapper does not support packages with extension [$packext]\n";
+}
+
+my @files;
+find(
+    sub { push @files, $File::Find::name unless -d; }, 
+    $dirname
+);
+
+print "***** Beginning Execution of Ninka *****\n";
+foreach my $file (@files) {
+    print "Running ninka on file [$file]\n";
+    execute("perl ${path}/ninka.pl '$file'");
+}
+
+my @ninkafiles;
+find(
+    sub {
+	my $ext = getExtension($File::Find::name);
+	if($ext =~ m/(comments|sentences|goodsent|badsent|senttok|license)$/){
+	    push @ninkafiles, $File::Find::name;  
+	}
+    }, 
+    $dirname
+);
+
+print "***** Entering Ninka Data into excell file [$excelFile] *****\n";
+my $row = 1;
+
+foreach my $file (@ninkafiles) {
+    
+    my $filepath = dirname($file);
+    $filepath =~ s/$dirname//;
+    my $basefile = basename($file);
+    my $rootfile = removeExtension($basefile);
+    my $packname = basename($pack);
+
+    #Read entire file into a string
+    open (my $fh, '<', $file) or die "Can't open file $!";
+    my $filedata = do { local $/; <$fh> };
+
+    my $sth;
+    switch (getExtension($basefile)){
+	case ".comments" {
+	    ;
+	}
+	case ".sentences" {
+	    ;
+	}
+	case ".goodsent" {
+	    ;
+	}
+	case ".badsent" {
+	    ;
+	}
+	case ".senttok" {
+	    ;
+	}
+	case ".license" {
+	    print "Inserting [$basefile] into table licenses\n";
+	    my @columns = parseLicenseData($filedata);
+
+	    $worksheet->write($row, 0, $packname);
+	    $worksheet->write($row, 1, $filepath);
+	    $worksheet->write($row, 2, $rootfile);
+
+	    my $originalFile = $file;
+	    $originalFile =~ s/\.license$//;
+
+	    if (-T $originalFile) {
+		foreach my $i (0..7) {
+		    $worksheet->write($row, $i+3, $columns[$i]);
+		}
+	    } else {
+		$worksheet->write($row, 3, "Binary File");		
+	    }
+
+            $row++;
+	}
+    }
+    close($fh);
+}
+
+$workbook->close();
+
+sub parseLicenseData {
+    my ($data) = @_;
+    chomp($data);
+    my @columns;
+    my @fields = split(';', $data);
+    if($fields[0] eq "NONE\n"){
+	@columns = '' x 7;
+	@columns[0] = 'NONE';
+    } else {
+	@columns = @fields;
+    }
+    return @columns;
+}
+
+sub getExtension {
+    my ($file) = @_;
+    my $filename = basename($file);
+    my ($ext) = $filename =~ /(\.[^.]+)$/;
+    return $ext;
+}
+
+sub removeExtension {
+    my ($file) = @_;
+    (my $filename = $file) =~ s/\.[^.]+$//;
+    return $filename;
+}
+
+sub execute {
+    my ($command) = @_;
+    my $output = `$command`;
+    my $status = ($? >> 8);
+    die "execution of [$command] failed: status [$status]\n" if ($status != 0);
+    return $output;
+}
diff --git a/ninka-sqlite.pl b/ninka-sqlite.pl
index a2c6a76..ddecedd 100644
--- a/ninka-sqlite.pl
+++ b/ninka-sqlite.pl
@@ -23,7 +23,7 @@ if ($path eq "") {
 
 my ($pack, $db) = @ARGV;
 
-my $dbh = DBI->connect("DBI:SQLite:dbname=$db", "", "", {RaiseError => 1})
+my $dbh = DBI->connect("DBI:SQLite:dbname=$db", "", "", {RaiseError => 1, AutoCommit => 0})
     or die $DBI::errstr;
 $dbh->do("CREATE TABLE IF NOT EXISTS
           comments (filename TEXT, path TEXT, container TEXT, content TEXT,
@@ -136,7 +136,8 @@ foreach my $file (@ninkafiles) {
     close($fh);
 }
 
-$dbh->disconnect;
+$dbh->commit();
+$dbh->disconnect();
 
 sub parseLicenseData {
     my ($data) = @_;