simply-git/sg

#!/usr/bin/perl

use strict;
use warnings;

use Getopt::Long qw(GetOptions);
use Log::Log4perl qw(:easy);
# TODO: This needs to be scoped properly
use lib "/usr/local/lib";
use Shellex::Shellex qw(shellex findBin);
use SimplyGit::Git qw(
	readConfig getStatus returnState addFiles 
	commitChanges pushChanges stashAndReset resetFromUpstream 
	updateGitIgnore appendRepoUserConfig parseSGConfig 
	warnOnUser basicClone basicPull knocker
);

sub initSG($) {

	my $sgDir = shift;
	my $homeDir = shellex("echo \$HOME");
	chomp $homeDir;
	my $path = $homeDir . "/" . $sgDir;
	my $logFile = $homeDir . "/" . $sgDir . "/" . "sgLog.txt";
	my $configFile = $homeDir . "/" . $sgDir . "/" . "sg.config";
	if ( ! -d $path ) {
		print "Creating $path\n";
		shellex("mkdir $path");
	}

	if ( ! -f $logFile ) {
		print "Creating $logFile\n";
		shellex("touch $logFile");
	}

	if ( ! -f $configFile ) {
		print "Creating $configFile\n";
		shellex("touch $configFile");
	}

	return ( $path, $logFile, $configFile );
	
}

my ( $sgPath, $sgLogFile, $sgConfigFile ) = initSG(".sg");
sub getLogName { return $sgLogFile; };
my $log_conf = q(
        log4perl.rootLogger              = ERROR, LOG1, screen
        log4perl.appender.LOG1           = Log::Log4perl::Appender::File
        log4perl.appender.LOG1.filename  = sub { getLogName(); }
        log4perl.appender.LOG1.mode      = append
        log4perl.appender.LOG1.layout    = Log::Log4perl::Layout::PatternLayout
        log4perl.appender.LOG1.layout.ConversionPattern = %d %p >> %m %n

	log4perl.appender.screen = Log::Log4perl::Appender::Screen
	log4perl.appender.screen.stderr = 0
	log4perl.appender.screen.layout = PatternLayout
	log4perl.appender.screen.layout.ConversionPattern = %d %p >> %m %n
);

Log::Log4perl::init(\$log_conf);
my $logger = get_logger();
my $gitCmd = findBin("git",$logger);
# Removed .sg from repo dir to home, don't need this sub right now
# updateGitIgnore(".","/.sg",$logger);

my %args;
GetOptions(
	\%args,
	'push-all',
	'interactive',
	'view',
	'reset-from-master',
	'reset-from-upstream',
	'upstream-url=s',
	'commit-msg=s',
	'dump-config',
	'configure-local-user',
	'user=s',
	'email=s',
	'config-file=s',
	'knock',
	'knock-clone=s',
	'help',
	'knock-pull',
);

# TODO: This should maybe be more robust?
if ( ! -d ".git" && ( ! defined $args{'knock-clone'} && ! defined $args{'knock'} && ! defined $args{'help'} ) ) {
	print "Not a git dir, exiting...\n";
	exit 1;
}

sub printHelp {

	my $help = <<EOF
simply-git
Usage:
	--view
	Display git status of files and other information

	--dump-config
	Dump .git/config to STDOUT. Not really useful but exposed for testing of reading config into internal data structure

	--push-all [--commit-msg]
	Push all untracked and modified files 
		* Can be used with interactive mode
		* Can provide a commit msg with --commit-msg (otherwise a generic will be provided)

	--interactive
	Enable interactive mode with supported opts

	--reset-from-master
	Reset all current changes so that the file tree matches origin master

	--reset-from-upstream [ --upstream-url ]
	If upstream is defined will reset local branch to match upstream ( does not push changes to origin )
		* Assumes you have an upstream configured
		* Pass SSH/HTTPS URL to --upstream-url to add an upstream

	--configure-local-user [--user,--email]
	Configure local git user
		* Can be used with interactive mode

	--config-file
	Default is ~/.sg/sg.config, can use this opt to use another file
		* See example.config

	--knock
	Will try and knock the defined git server at the defined ports before any operation
		* See example.config
		* Can pass this by itself to perform a knock and exit
	
	--knock-clone 
	Will try and knock the defined git server and clone the provided repo
		* Will not check if you're in a git dir

	--knock-pull
	Will try and knock the defined git server and git pull

EOF
;

	print "$help\n";

}

if ( scalar keys %args < 1 ) {
	printHelp();
}

###
# TODO: This args processing is better and more predictable than it was, bit there is still
# likely a lot of room for improvement...
###

sub parseArgs {

	if ( defined $args{'help'} ) {
		printHelp();
		exit 0;
	}

	if ( defined $args{'view'} && scalar keys %args > 1 ) {
		print "Can't pass other args with --view\n";
		exit 1;
	}

	if ( defined $args{'dump-config'} && scalar keys %args > 1 ) {
		print "Can't pass other args with --dump-config\n";
		exit 1;
	}

	if ( defined $args{'reset-from-master'} && scalar keys %args > 1 ) {
		print "Can't pass other args with --reset-from-master\n";
		exit 1;
	}

	if ( defined $args{'push-all'} ) {
		foreach my $arg ( keys %args ) {
			if ( $arg eq "interactive" || $arg eq "commit-msg" || $arg eq "push-all" || $arg eq "knock" ) {
				next;
			} else {
				print "Can only pass --interactive and --commit-msg with --push-all\n";
				exit 1;
			}
		}
	}

	if ( defined $args{'configure-local-user'} ) {
		if ( scalar keys %args < 2 ) {
			print "Must pass either --interactive or --user AND --email to --configure-local-user\n";
			exit 1;
		}

		foreach my $arg ( keys %args ) {
			if ( $arg eq "interactive" || $arg eq "user" || $arg eq "email" || $arg eq "configure-local-user" ) {
				next;
			} else {
				print "Must/can only pass --interactive, OR --user AND --email with --configure-local-user\n";
				exit 1;
			}
		}

		if ( ! defined $args{'interactive'} && ! defined $args{'user'} || ! defined $args{'interactive'} && ! defined $args{'email'} ) {
			print "If not using --interactive with --configure-local-user, --user and --email MUST be defined\n";
			exit 1;
		}
	}

	if ( defined $args{'reset-from-upstream'} ) {
		if ( scalar keys %args > 2  ) {
			print "Can only pass --upstream-url with --reset-from-upstream\n";
			exit 1;
		}

		foreach my $arg ( keys %args ) {
			if ( $arg eq "reset-from-upstream" || $arg eq "upstream-url" ) {
				next;
			} else {
				print "Can only pass --upstream-url with --reset-from-upstream\n";
				exit 1;
			}
		}
	}

	if ( ! defined $args{'config-file'} ) {
		$args{'config-file'} = $sgConfigFile;
	}

	if ( defined $args{'knock-clone'} ) {
		if ( scalar keys %args > 2 ) {
			print "--knock-clone accepts no other args\n";
			exit 1;
		}
	}
	
	if ( defined $args{'knock-pull'} ) {
		if ( scalar keys %args > 2 ) {
			print "--knock-pull accepts no other args\n";
			exit 1;
		}
	}

}

sub color_print($$) {


	#echo -e "\033[0mNC (No color)"
	#echo -e "\033[1;37mWHITE\t\033[0;30mBLACK"
	#echo -e "\033[0;34mBLUE\t\033[1;34mLIGHT_BLUE"
	#echo -e "\033[0;32mGREEN\t\033[1;32mLIGHT_GREEN"
	#echo -e "\033[0;36mCYAN\t\033[1;36mLIGHT_CYAN"
	#echo -e "\033[0;31mRED\t\033[1;31mLIGHT_RED"
	#echo -e "\033[0;35mPURPLE\t\033[1;35mLIGHT_PURPLE"
	#echo -e "\033[0;33mYELLOW\t\033[1;33mLIGHT_YELLOW"
	#echo -e "\033[1;30mGRAY\t\033[0;37mLIGHT_GRAY"
	#
	# Doing it this way likely hurts portability but
	# it's better than nothing which is what I'm currently doing
	# and would like to avoid additional module deps, as this
	# isn't too hard to implement
	
	my $print_string = shift;
	my $color = shift;

	if ( $color ne "BLUE" && $color ne "GREEN" && $color ne "RED" ) {
		$logger->error("Bad color passed to color_print");
		exit 1;
	}

	my %color_map = (
		BLUE  => "\033[0;34m",
		GREEN => "\033[0;32m",
		RED   => "\033[0;31m",
		RESET => "\033[0m",
	);

	printf "$color_map{$color}$print_string$color_map{'RESET'}";

}


parseArgs();
my %sgConfig = parseSGConfig($args{'config-file'},$logger);
if ( defined $sgConfig{'UserWarn'}  && -d ".git" ) {
	warnOnUser($sgConfig{'user.name'},$sgConfig{'user.email'},$logger);
}

sub knock() {
	if ( defined $sgConfig{'Knock'} && ( defined $args{'knock'} || defined $args{'knock-clone'} || defined $args{'knock-pull'} ) ) {
		knocker($sgConfig{'knock.target'},$sgConfig{'ports'},$logger);
	}
}

if ( defined $args{'knock'} && scalar keys %args == 2 ) {
	print "Just knocking then exiting...\n";
	knock();
	exit 1;
}

# TODO: This sub could be more concise with a sub to print array refs
if ( defined $args{'view'} ) {
	my ( $untrackedRef, $modifiedRef, $addedRef, $deletedRef ) = returnState($logger);
	#my $refs = shellex("$gitCmd show-ref",$logger);
	my $branch = shellex("$gitCmd show-branch",$logger);
	my $name = shellex("$gitCmd config --get user.name",$logger);
	chomp $name;
	my $email = shellex("$gitCmd config --get user.email",$logger);
	chomp $email;
	color_print("-->Username: $name\n-->Email: $email\n","BLUE");
	print "Branches:\n";
	color_print("$branch\n","GREEN");
	#print "$refs\n";
	print "Files:\n";
	my $swpWarning = "\t# Likely a Vi .swp file";

	my $untrackedTotal = scalar @$untrackedRef;
	print "* $untrackedTotal untracked file(s):\n";
	foreach my $file ( @$untrackedRef ) {
		if ( $file =~ m/.swp/ ) {
			color_print("\t$file $swpWarning\n","GREEN");
		} else {
			color_print("\t$file\n","GREEN");
		}
	}

	my $modifiedTotal = scalar @$modifiedRef;
	print "* $modifiedTotal modified file(s):\n";
	foreach my $file ( @$modifiedRef ) {
		if ( $file =~ m/.swp/ ) {
			color_print("\t$file $swpWarning\n","GREEN");
		} else {
			color_print("\t$file\n","GREEN");
		}
	}

	my $commitTotal = scalar @$addedRef;
	print "* $commitTotal file(s) added to commit:\n";
	foreach my $file ( @$addedRef ) {
		if ( $file =~ m/.swp/ ) {
			print "\t$file $swpWarning\n";
		} else {
			print "\t$file\n";
		}
	}

	my $deletedTotal = scalar @$deletedRef;
	print "* $deletedTotal file(s) to be deleted from commit:\n";
	foreach my $file ( @$deletedRef ) {
		if ( $file =~ m/.swp/ ) {
			color_print("\t$file $swpWarning\n","RED");
		} else {
			color_print("\t$file\n","RED");
		}
	}
}

if ( defined $args{'push-all'} ) {

	my ( $untrackedRef, $modifiedRef ) = returnState($logger);
	my @files;
	push(@files,@$untrackedRef); push(@files,@$modifiedRef);
	my @filesToCommit;
	if ( defined $args{'interactive'} ) {
		foreach my $file ( @files ) {
			print "Add $file to commit (y/n): ";
			my $input = <STDIN>;
			chomp $input;
			if ( $input =~ m/^Y|^y/ ) {
				push(@filesToCommit,$file);
			} else {
				next;
			}
		}

	} else {
		@filesToCommit = @files;
	}
		
	print "Commiting the following files:\n";
	foreach my $file ( @filesToCommit ) {
		print "\t$file\n";
	}

	if ( defined $args{'interactive'} ) {
		print "Does this look correct (y/n) : ";
		my $input = <STDIN>;
		chomp $input;
		if ( $input !~ m/^Y|^y/ ) {
			print "Canceling...\n";
			exit 1;
		}
	}
	
	addFiles(\@filesToCommit,$logger);

	if ( defined $args{'interactive'} && ! defined $args{'commit-msg'}) {
		print "Enter a commit message: ";
		my $input = <STDIN>;
		chomp $input;
		commitChanges($input,$logger);
	} elsif ( defined $args{'commit-msg'} ) { 
		my $commitMsg = "$args{'commit-msg'}";
		commitChanges($commitMsg,$logger);
	} else {
		my $epoch = time();
		my $commitMsg = "Generic Commit at $epoch";
		commitChanges($commitMsg,$logger);
	}

	if ( defined $args{'interactive'} ) {
		print "Push changes? (y/n): ";
		my $input = <STDIN>;
		chomp $input;
		if ( $input !~ m/^Y|^y/ ) {
			# TODO: Unstage changes?
			print "Canceling...\n";
			exit 1;
		}
		
		knock();
		my $gitOutput = pushChanges($logger);
		print "Git returned:\n$gitOutput\n";

	} 
	
	else {

		knock();
		pushChanges($logger);
		my $gitOutput = pushChanges($logger);
		print "Git returned:\n$gitOutput\n";

	}


}

if ( defined $args{'reset-from-master'} ) {

	knock();
	stashAndReset($logger);

}

if ( defined $args{'reset-from-upstream'} ) {

	if ( defined $args{'upstream-url'} ) {
		print "Setting upstream to $args{'upstream-url'}\n";
		chomp $args{'upstream-url'};
		shellex("$gitCmd remote add upstream $args{'upstream-url'}",$logger);
		shellex("$gitCmd fetch upstream",$logger);
		knock();
		resetFromUpstream($logger);
	} else {
		knock();
		resetFromUpstream($logger);
	}
}

if ( defined $args{'dump-config'} ) {

	my %configHash = readConfig(".",$logger);
	foreach my $key ( keys %configHash ) {
		my $hRef = $configHash{$key};
		print "[$key]\n";
		foreach my $ckey ( keys %$hRef ) {
			print "\t$ckey = ${$hRef}{$ckey}\n";
		}
	}

}

if ( defined $args{'configure-local-user'} ) {

	if ( defined $args{'interactive'} ) {
		print "Enter user to set: ";
		my $desiredName = <STDIN>;
		chomp $desiredName;
		print "Enter email to set: ";
		my $desiredEmail = <STDIN>;
		chomp $desiredEmail;

		appendRepoUserConfig($desiredName,$desiredEmail,$logger);

	} else {

		appendRepoUserConfig($args{'user'},$args{'email'},$logger);

	}

}

if ( defined $args{'knock-clone'} ) {
	knock();
	basicClone($args{'knock-clone'},$logger);
}

if ( defined $args{'knock-pull'} ) {
	knock();
	basicPull($logger);
}