#!/usr/bin/env perl
#
# vim: set ts=4:noet
#
# sboconfig
# script to handle sbotools configuration
#
# authors:	Jacob Pipkin <j@dawnrazor.net>
#			Luke Williams <xocel@iquidus.org>
# license: WTFPL <http://sam.zoy.org/wtfpl/COPYING>

use 5.16.0;
use strict;
use warnings FATAL => 'all';
use SBO::Lib;
use File::Basename;
use Getopt::Long;
use File::Copy;
use File::Path qw(make_path);
use File::Temp qw(tempfile);;

my $self = basename($0);

sub show_usage {
	print <<EOF
Usage: $self [options] [arguments]

Options:
  -h: this screen.
  -v: version information.
  -l: show current options.

Config options (defaults shown):
  -c|--clean FALSE:
      NOCLEAN: if TRUE, do NOT clean up after building by default.
  -d|--distclean FALSE:
      DISTCLEAN: if TRUE, DO clean distfiles by default after building.
  -j|--jobs FALSE:
      JOBS: numeric -j setting to feed to make for multicore systems.
  -p|--pkg-dir FALSE:
      PKG_DIR: set a directory to store packages in. 
  -s|--sbo-home /usr/sbo:
      SBO_HOME: set the SBo directory.

EOF
}

my %options;

GetOptions(\%options, 'help|h', 'version|v', 'list|l', 'noclean|c=s',
	'distclean|d=s', 'jobs|j=s', 'pkg-dir|p=s', 'sbo-home|s=s');

show_usage() and exit 0 if exists $options{help};
show_version() and exit 0 if exists $options{version};

my %valid_confs = (
	noclean		=> 'NOCLEAN',
	distclean	=> 'DISTCLEAN',
	jobs		=> 'JOBS',
	'pkg-dir'	=> 'PKG_DIR',
	'sbo-home'	=> 'SBO_HOME',
);

my %params = (
	NOCLEAN		=> 'c|--noclean',
	DISTCLEAN	=> 'd|--distclean',
	JOBS		=> 'j|--jobs',
	PKG_DIR		=> 'p|--pkg-dir',
	SBO_HOME	=> 's|--sbo-home',
);

if (exists $options{list}) {
	my @keys = sort {$a cmp $b} keys %config;
	say "sboconfig -$params{$_}:\n    $_=$config{$_}" for @keys;
	exit 0;
}

show_usage() and exit 0 unless keys %options > 0;

# setup what's being changed, sanity check.
my %changes;
for my $key (keys %valid_confs) {
	my $value = $valid_confs{$key};
	$changes{$value} = $options{$key} if exists $options{$key};
}

my $warn = 'You have provided an invalid parameter for';

if (exists $changes{NOCLEAN}) {
	usage_error("$warn -c") unless $changes{NOCLEAN} =~ /^(TRUE|FALSE)$/;
}
if (exists $changes{DISTCLEAN}) {
	usage_error("$warn -d") unless $changes{DISTCLEAN} =~ /^(TRUE|FALSE)$/;
}
if (exists $changes{JOBS}) {
	usage_error("$warn -j") unless $changes{JOBS} =~ /^(\d+|FALSE)$/;
}
if (exists $changes{PKG_DIR}) {
	usage_error("$warn -p") unless $changes{PKG_DIR} =~ qr#^(/|FALSE$)#;
}
if (exists $changes{SBO_HOME}) {
	usage_error("$warn -s") unless $changes{SBO_HOME} =~ qr#^/#;
}

# safely modify our conf file; write its contents to a temp file, modify the
# temp file, write the contents of the temp file back to the conf file
# TODO: if multiple options are provided to this script, this sub should write
# them all at once, instead of only a single one and having to call it once for
# each option specified to the script.
sub config_write {
	exists $_[1] or script_error('config_write requires two arguments.');
	my ($key, $val) = @_;
	if (! -d $conf_dir) {
		mkdir $conf_dir or usage_error("Unable to create $conf_dir. Exiting.");
	}
	if (-f $conf_file) {
		my $tempfh = tempfile(DIR => $tempdir);
		my ($conffh, $exit) = open_read($conf_file);
		if ($exit) {
			warn $conffh;
			exit $exit;
		}
		my $conftents = do {local $/; <$conffh>};
		print {$tempfh} $conftents;
		# tie the temp file so that if $key is already there, we just change 
		# that line and untie it
		tie my @temp, 'Tie::File', $tempfh;
		my $has;
		my $regex = qr/\A\Q$key\E=/;
		FIRST: for my $tmpline (@temp) {
			$has++, $tmpline = "$key=$val", last FIRST if $tmpline =~ $regex;
		}
		untie @temp;
		# otherwise, append our new $key=$value pair
		print {$tempfh} "$key=$val\n" unless $has;
		# then overwrite the conf file with the contents of the temp file
		seek $tempfh, 0, 0;
		my $contents = do {local $/; <$tempfh>};
		close $conffh;
		($conffh, $exit) = open_fh($conf_file, '>');
		if ($exit) {
			warn $conffh;
			exit $exit;
		}
		print {$conffh} $contents or return();
		close $conffh, close $tempfh;
	} else {
		# no config file, easiest case of all.
		my ($fh, $exit) = open_fh($conf_file, '>') or return();
		if ($exit) {
			warn $fh;
			exit $exit;
		}
		print {$fh} "$key=$val\n";
		close $fh;
	}
	return 1;
}

for my $key (keys %changes) {
	my $value = $changes{$key};
	say "Setting $key to $value...";
	config_write($key, $value);
}

exit 0;
