get-licensed/get-licensed.pl

#!/bin/env perl

use strict;
use warnings;
use FindBin;
use lib "$FindBin::Bin";
use File::Find;
use File::Basename;
use File::Copy;
use FileHandle;
use File::Glob ':globally';
use Data::Dumper;
use Getopt::Long;
use Cwd;
use Licenses;

# Application configuration and state storage
my $app = {};

# Defaults
$app->{'options'}->{'directory'}  = getcwd;
$app->{'options'}->{'license'}    = 'mit';
$app->{'options'}->{'user'}       = $ENV{'USER'};
$app->{'options'}->{'date'}       = (localtime(time))[5] + 1900;
$app->{'options'}->{'about'}      = "This is an open-source software";
$app->{'options'}->{'dry'}        = 0;

$app->{'state'}->{'license_stub'} = "NON EXISTANT";
$app->{'state'}->{'app_dir'}      = dirname($0);

# Command line options
my $correct_usage = GetOptions (
    'directory=s' => \$app->{'options'}->{'directory'},
    'license=s' => \$app->{'options'}->{'license'},
    'about=s' => \$app->{'options'}->{'about'},
    'user=s' => \$app->{'options'}->{'user'},
    'date=s' => \$app->{'options'}->{'date'},
    'dry' => \$app->{'options'}->{'dry'},
  );

setup_licenses($app);
setup_filetype_comment_mapping($app);

print "Applying '"
      . $app->{'licenses'}->{$app->{'options'}->{'license'}}->{'pretty_name'}
      . "' license to files in '"
      . $app->{'options'}->{'directory'}
      . "' directory\n\n";

print "Copyright: Copyright (C) "
      . $app->{'options'}->{'date'}
      . " " . $app->{'options'}->{'user'}
      . "\n\n";

print "About: " . $app->{'options'}->{'about'}
      . "\n\n";

print "Stub:\n-----\n"
      . $app->{'licenses'}->{$app->{'options'}->{'license'}}->{'header_stub'}
      . "\n-----\n";

# print Data::Dumper->Dump([$app]);

find (sub {
    find_files($app, ${File::Find::name}, $_)
  }, $app->{'options'}->{'directory'});

#
# Subroutines
#
sub find_files {
  my ($app, $file, $name) = @_;

  #
  # Opportunity to filter any file types befrore any processing starts
  #
  return if ($file =~ /\.git\b/);

  handle_files ($app, $file, $name);
}

sub handle_files {
  my ($app, $file, $name) = @_;

  return unless -f $name;

  my($filename, $directories, $suffix) = fileparse($file, qr/\.[^.]*/);

  if (exists $app->{'filetype_comments'}->{$suffix}) {
    my $rh = FileHandle->new($name);
    my @contents = <$rh>;

    print "Handling... $filename - $directories - $suffix ----- " .
          ref($app->{'filetype_comments'}->{$suffix}) . "\n";

    my $stub = get_license_stub($app, $suffix);

    if ((defined $contents[0]) && ($contents[0] =~ /^\s*#!/)) {
      print "   ... has #! line\n";

      $contents[0] =  $contents[0] . "\n" . $stub . "\n";
    } else {
      $contents[0] =  $stub . "\n" . $contents[0] . "\n";
    }

    if ($app->{'options'}->{'dry'} == 0) {
      copy($name, $name . ".licensor_backup") or die "Could not create backup file";

      my $wh = FileHandle->new($name, O_WRONLY);
      print $wh @contents;
    }
  }
  else {
    print "[ALERT] Don't know how to comment in file with '$suffix' extension\n";
  }
}

sub get_license_stub {
  my ($app, $suffix) = @_;

  my $stub = $app->{'licenses'}->{$app->{'options'}->{'license'}}->{'header_stub'};
  if (ref($app->{'filetype_comments'}->{$suffix}) eq '') {
    $stub =~ s/^(.)/$app->{'filetype_comments'}->{$suffix} $1/msg;
    $stub = $app->{'filetype_comments'}->{$suffix} . " -----\n" .
            "$stub\n" .
            $app->{'filetype_comments'}->{$suffix} . " -----\n";
  } elsif (ref($app->{'filetype_comments'}->{$suffix}) eq 'ARRAY') {
    $stub =~ s/^(.)/  $1/msg;
    $stub = $app->{'filetype_comments'}->{$suffix}[0] .
            " -----\n$stub\n ----- " .
            $app->{'filetype_comments'}->{$suffix}[1];
  } else {
    # not an option
  }

  $stub =~ s/\s+$//msg;

  return $stub;
}

sub setup_licenses {
  my ($app) = @_;

  my $licenses = new Licenses;
  $licenses->get_licenses($app);

  print "[INFO] Imported "
        . join (', ', keys %{$app->{'licenses'}})
        . " licenses\n";
}

sub setup_filetype_comment_mapping {
  my ($app) = @_;

  $app->{'filetype_comments'} = {
    '.pl'         => '#',
    '.pm'         => '#',
    '.rb'         => '#',
    '.erb'        => ['<!--', '-->'],
    '.html'       => ['<!--', '-->'],
    '.css'        => ['/*', '*/'],
    '.coffee'     => '#',
    '.js'         => '//',
    '.scss'       => '//',
    '.v'          => '//',
    '.sv'         => '//',
    '.vr'         => '//',
    '.gitignore'  => '#',
    '.gitkeep'    => '#'

  };
}