#!/usr/bin/perl
# Samba.org buildfarm
# Copyright (C) 2008 Andrew Bartlett <abartlet@samba.org>
# Copyright (C) 2008 Jelmer Vernooij <jelmer@samba.org>
#   
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 3 of the License, or
# (at your option) any later version.
#   
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#   
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
#

use FindBin qw($RealBin);

use hostdb;
use Mail::Send;
use warnings;
use strict;
use Data::Dumper;
use util;

my $hostname;

my $db = new hostdb("$RealBin/hostdb.sqlite") or die("Unable to connect to host database: $!");
my $dry_run = 0;

print "Samba Build farm management tool\n";
print "================================\n";

my $op;

if ($#ARGV > -1) {
	$op = shift(@ARGV);
} else {
	print "Initialize host database:       init\n";
	print "Add Machine to build farm:      add\n";
	print "Remove Machine from build farm: remove\n";
	print "Modify build farm account:      modify\n";
	print "Print build farm host info:     info\n";
	print "Print build farm host list:     list\n";
	print "Select Operation: [add] ";

	$op = lc(<STDIN>);
	chomp($op);

	if ($op eq "") {
		$op = "add";
	}
}

if ($op eq "remove") {
	print "Please enter hostname to delete: ";
	$hostname = <>;
	chomp($hostname);
	$db->deletehost($hostname) or die("Unable to delete host $hostname");
} elsif ($op eq "modify") {
	print "Please enter hostname to modify: ";
	$hostname = <>;
	chomp($hostname);
	my $host = $db->host($hostname);
	print "Owner: $host->{owner} <$host->{owner_email}>\n";
	print "Platform: $host->{platform}\n";
	print "\n";
	print "Modify owner or platform: [platform] ";
	my $mod_op = <>;
	chomp($mod_op);
	if ($mod_op eq "") {
		$mod_op = "platform";
	}
	if ($mod_op eq "platform") {
		print "Enter new platform: ";
		my $platform = <>;
		chomp($platform);
		$db->update_platform($hostname, $platform) or die "Unable to update platform";
	} elsif ($mod_op eq "owner") {
		print "Enter new owner's name: ";
		my $owner = <>;
		chomp($owner);
		print "Enter new owner's e-mail address: ";
		my $owner_email = <>;
		chomp($owner_email);
		$db->update_owner($hostname, $owner, $owner_email) or die "Unable to update owner";
		
	}	
} elsif ($op eq "add") {
	print "Machine hostname: ";
	my $hostname = <>;
	chomp($hostname);
	print "Machine platform (eg Fedora 9 x86_64): ";
	my $platform = <>;
	chomp($platform);
	print "Machine Owner Name: ";
	my $owner = <>;
	chomp($owner);
	print "Machine Owner E-mail: ";
	my $owner_email = <>;
	chomp($owner_email);
	print "Enter password: [generate random] ";
	my $password = <>;
	chomp($password);
	if ($password eq "") {
		$password = `pwgen 16 1`;
		chomp($password);
		print "Password will be: $password\n";
	}
	print "Enter permission e-mail, finish with a .\n";
	my $permission;
	while (<>) {
		last if $_ eq ".\n";
		$permission = $_;
	}
	
	$db->createhost($hostname, $platform, $owner, $owner_email, $password, $permission) or die("Unable to create host $hostname");

	my ($fh, $msg);
	
	# send the password in an e-mail to that address
	my $subject = "Your new build farm host $hostname";
	if ($dry_run) {
		print "To: $owner <$owner_email>\n";
		print "Subject: $subject\n";
		open(MAIL,"|cat");
	} else {
		$msg = new Mail::Send(Subject=>$subject, To=>"\"$owner\" \<$owner_email\>", Bcc=>"build\@samba.org");
		$msg->set("From", "\"Samba Build Farm\" \<build\@samba.org\>");
		$fh = $msg->open; 
	}

	my $body = << "__EOF__";	
Welcome to the Samba.org build farm.  

Your host $hostname has been added to the Samba Build farm.  

We have recorded that it is running $platform.  

If you have not already done so, please read:
http://build.samba.org/instructions.html

The password for your rsync .password file is $password

An e-mail asking you to subscribe to the build-farmers mailing
list will arrive shortly.  Please ensure you maintain your 
subscription to this list while you have hosts in the build farm.

Thank you for your contribution to ensuring portability and quality
of Samba.org projects.


__EOF__
	if ($dry_run) {
		print MAIL $body;

		close(MAIL);
	} else {
		print $fh "$body";
		$fh->close;

		$msg = new Mail::Send(Subject=>'Subscribe to build-farmers mailing list', To=>'build-farmers-join@lists.samba.org');
		$msg->set("From", "\"$owner\" \<$owner_email\>");
		$fh = $msg->open; 
		print $fh "Please subscribe $owner to the build-farmers mailing list\n\n";
		print $fh "Thanks, your friendly Samba build farm administrator <build\@samba.org>";
		$fh->close;
	}
	

	
} elsif ($op eq "init") {
	$db->provision();
	print "Host database initialized successfully.\n";
} elsif ($op eq "info") {
	print "Hostname: ";
	$hostname = <>;
	chomp($hostname);
	my $host = $db->host($hostname);
	die ("No such host $host") unless ($host);
	print "Host: $host->{name}";
	if ($host->{fqdn}) { print " ($host->{fqdn})"; }
	print "\n";
	print "Platform: $host->{platform}\n";
	print "Owner: $host->{owner} <$host->{owner_email}>\n";
	
	# Don't run the update of the text files
	exit(0);
} elsif ($op eq "list") {
    my $hosts = $db->host_ages();
    foreach (@$hosts) {
	my $h = $_;
	my $age = util::dhm_time(time() - $h->{last_update});
	printf "%-12s $h->{host}\n", $age;
    }
#    print Dumper($hosts);
} else {
	die("Unknown command $op");
}

open(RSYNC_SECRETS, ">$RealBin/../rsyncd.secrets.new") or die("Unable to open rsyncd.secrets file: $!");
print RSYNC_SECRETS $db->create_rsync_secrets();
close(RSYNC_SECRETS);

rename("$RealBin/../rsyncd.secrets.new", "../rsyncd.secrets");

open(HOSTS, ">$RealBin/web/hosts.list.new") or die("Unable to open hosts file: $!");
print HOSTS $db->create_hosts_list();
close(HOSTS);

rename("$RealBin/web/hosts.list.new", "$RealBin/web/hosts.list");

1;