#!/usr/bin/env perl
#
# Copyright (c) 2020 Mischa Peters <mischa @ openbsd.amsterdam>
#
# Permission to use, copy, modify, and distribute this software for any
# purpose with or without fee is hereby granted, provided that the above
# copyright notice and this permission notice appear in all copies.
#
# THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
# WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
# MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
# ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
# WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
# ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
# OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
#
# vmm(4)/vmd(8) VM notify script for OpenBSD Amsterdam
# 2020/05/17 initial release
#
# This is WIP, currently there is a lot of overlap in the functions.
#
use 5.024;
use strict;
use warnings;
use autodie;
use POSIX qw(strftime);

# Get function and function_variable (vmid) from arguments
my $function = $ARGV[0] || "empty";
my $function_variable = $ARGV[1] || "empty";

# fuction to parse _deploy.conf and vm*.txt files
# all variables are stripped and added to either %vms or %conf
sub get_variables {
	my ($hash_name, @files)	= @_;
	my %hash;
	my $filename;
	my $vm_name;
	my $vm_number;

	for my $file (@files) {
		# When hash is 'vms' use the vm_name as key
		# Otherwise use 'conf' as key
		if ($hash_name eq "vms") {
			($filename = $file) =~ s/.*\///;
			($vm_name = $filename) =~ s/\.txt//;
			($vm_number = $vm_name) =~ s/^vm//;
			$hash{$vm_name}{'vm_number'} = $vm_number;
		}

		open my $fh, "<", "$file";
		while (my $row = <$fh>) {
			next if ($row =~ /^\s*($|#)/);
			chomp ($row);
			(my $key, my $val) = split(/=/, $row, 2);
			if ($hash_name eq "vms") {
				($hash{$vm_name}{$key} .= $val) =~ s/^"+|"+$//g;
			} else {
				($hash{$hash_name}{$key} .= $val) =~ s/^"+|"+$//g;
			}
		}
		close $fh;
	}
	return %hash;
}

sub deployed {
	my %conf = %{$_[0]};
	my %vms = %{$_[1]};
	my $_etc = $conf{'conf'}{'ETC'};
	my $_vms = $conf{'conf'}{'VMS'};
	my $_tmpl = $conf{'conf'}{'TEMPLATES'};
	my $_server = $conf{'conf'}{'SERVER'};
	my $template = "$_tmpl/email-deployed.txt";

	for my $vm_name (sort keys %vms) {
		my $_date = $vms{$vm_name}{'date'};
		my $_payment = $vms{$vm_name}{'payment'};
		my $_donated = $vms{$vm_name}{'donated'};
		my $_name = $vms{$vm_name}{'name'};
		my $_email = $vms{$vm_name}{'email'};
		my $_hostname = $vms{$vm_name}{'hostname'};
		my $_boot = $vms{$vm_name}{'boot'} || "yes";

		if ($function_variable =~ /empty/) { print "No VMID provided\n"; last; }
		if ($function_variable =~ /$vm_name/) {
			(my $_firstname, my $_lastname) = split(/ /, $_name, 2);
			my $ipaddress = qx(grep -A2 $vm_name $_etc/dhcpd.conf | awk '/fixed-address/{print \$2}' | tr -d ';\n');

			open(my $fh, '<', $template);
			open my $fh_email, "|-", "/usr/sbin/sendmail -t";
			printf $fh_email "To: %s\n", $_email;
			while (my $row = <$fh>) {
				chomp $row;
				$row =~ s/NAME/$_firstname/g;
				$row =~ s/IP$/$ipaddress/g;
				$row =~ s/VMID/$vm_name/g;
				$row =~ s/SERVER/$_server/g;
				print $fh_email "$row\n";
			}
			close $fh_email;
			print "NOTIFIED: $_date, $_payment, $_name, $_email, $_hostname, $_server ($vm_name), $ipaddress\n";
		}
	}
}

# function to collect stopped vms
sub stopped {
	my %conf = %{$_[0]};
	my %vms = %{$_[1]};
	my $_etc = $conf{'conf'}{'ETC'};
	my $_vms = $conf{'conf'}{'VMS'};
	my $_tmpl = $conf{'conf'}{'TEMPLATES'};
	my $_server = $conf{'conf'}{'SERVER'};
	my $template = "$_tmpl/email-stopped.txt";

	my @stopped_vms = qx(vmctl show | grep stopped | awk '{print \$9}');

	for my $vm_name (sort keys %vms) {
		my $_date = $vms{$vm_name}{'date'};
		my $_payment = $vms{$vm_name}{'payment'};
		my $_donated = $vms{$vm_name}{'donated'};
		my $_name = $vms{$vm_name}{'name'};
		my $_email = $vms{$vm_name}{'email'};
		my $_hostname = $vms{$vm_name}{'hostname'};
		my $_boot = $vms{$vm_name}{'boot'} || "yes";

		if ($_boot =~ /no/) {
			print "NOT NOTIFIED: $_name, $_email, $_hostname, $_server ($vm_name)\n";
			next;
		}
		if (grep(/$vm_name/, @stopped_vms)) {
			(my $_firstname, my $_lastname) = split(/ /, $_name, 2);
			my $ipaddress = qx(grep -A2 $vm_name $_etc/dhcpd.conf | awk '/fixed-address/{print \$2}' | tr -d ';\n');

			open(my $fh, '<', $template);
			open my $fh_email, "|-", "/usr/sbin/sendmail -t";
			printf $fh_email "To: %s\n", $_email;
			while (my $row = <$fh>) {
				chomp $row;
				$row =~ s/NAME/$_firstname/g;
				$row =~ s/IP$/$ipaddress/g;
				$row =~ s/VMID/$vm_name/g;
				$row =~ s/SERVER/$_server/g;
				print $fh_email "$row\n";
			}
			close $fh_email;
			print "NOTIFIED: $_date, $_payment, $_name, $_email, $_hostname, $_server ($vm_name), $ipaddress\n";
		}
	}
}

sub renewal {
	my %conf = %{$_[0]};
	my %vms = %{$_[1]};
	my $_etc = $conf{'conf'}{'ETC'};
	my $_vms = $conf{'conf'}{'VMS'};
	my $_tmpl = $conf{'conf'}{'TEMPLATES'};
	my $_server = $conf{'conf'}{'SERVER'};
	my $template = "$_tmpl/email-renewal.txt";

	my $year = strftime("%Y", localtime);
	my $month = strftime("%m", localtime);

	my $total_donated = qx(ftp -Vo- https://obsda.ms/index.html | grep "donated to OpenBSD" | awk -F';' '{print \$4}' | awk '{printf \$1}');
	my $total_vms = qx(ftp -Vo- https://obsda.ms/index.html | grep "deployed" | awk -F '>' '{print \$3}' | awk '{printf \$1}');
	my $total_hosts = qx(ftp -Vo- https://obsda.ms/servers.html | grep -c ">Server " | tr -d '\n');

	for my $vm_name (sort keys %vms) {
		my $_date = $vms{$vm_name}{'date'};
		my $_payment = $vms{$vm_name}{'payment'};
		my $_donated = $vms{$vm_name}{'donated'};
		my $_name = $vms{$vm_name}{'name'};
		my $_email = $vms{$vm_name}{'email'};
		my $_hostname = $vms{$vm_name}{'hostname'};
		my $_boot = $vms{$vm_name}{'boot'} || "yes";

		if ($_donated =~ /done/) { next; }
		if ($_donated =~ /expire/) { next; }
		if ($_donated =~ /sponsor/) { next; }
		if ($_donated =~ /renewal/) { next; }
		if ($_date =~ /$year\//) { next; }
		if ($_date =~ /\/$month\//) {
			(my $_firstname, my $_lastname) = split(/ /, $_name, 2);
			(my $_year, my $_month, my $_day) = split(/\//, $_date, 3);
			my $ipaddress = qx(grep -A2 $vm_name $_etc/dhcpd.conf | awk '/fixed-address/{print \$2}' | tr -d ';\n');

			open(my $fh, '<', $template);
			open my $fh_email, "|-", "/usr/sbin/sendmail -t";
			printf $fh_email "To: %s\n", $_email;
			while (my $row = <$fh>) {
				chomp $row;
				$row =~ s/NAME/$_firstname/g;
				$row =~ s/IP$/$ipaddress/g;
				$row =~ s/VMID/$vm_name/g;
				$row =~ s/SERVER/$_server/g;

				$row =~ s/YEAR/$year/g;
				$row =~ s/TOTAL_DONATED/$total_donated/g;
				$row =~ s/TOTAL_VMS/$total_vms/g;
				$row =~ s/TOTAL_HOSTS/$total_hosts/g;
				$row =~ s/PAYMENT/$_payment/g;

				print $fh_email "$row\n";
			}
			close $fh_email;
			print "NOTIFIED: $_date, $_payment, $_name, $_email, $_hostname, $_server ($vm_name), $ipaddress\n";
		}
	}
}

# function to notify people
sub notify {
	my %conf = %{$_[0]};
	my %vms = %{$_[1]};
	my $_etc = $conf{'conf'}{'ETC'};
	my $_vms = $conf{'conf'}{'VMS'};
	my $_tmpl = $conf{'conf'}{'TEMPLATES'};
	my $_server = $conf{'conf'}{'SERVER'};
	my $template = "$_tmpl/email-notify.txt";

	for my $vm_name (sort keys %vms) {
		my $_date = $vms{$vm_name}{'date'};
		my $_payment = $vms{$vm_name}{'payment'};
		my $_donated = $vms{$vm_name}{'donated'};
		my $_name = $vms{$vm_name}{'name'};
		my $_email = $vms{$vm_name}{'email'};
		my $_hostname = $vms{$vm_name}{'hostname'};
		my $_boot = $vms{$vm_name}{'boot'} || "yes";

		if (-e "$template") {
		#if ($_boot =~ /yes/) {
			(my $_firstname, my $_lastname) = split(/ /, $_name, 2);
			my $ipaddress = qx(grep -A2 $vm_name $_etc/dhcpd.conf | awk '/fixed-address/{print \$2}' | tr -d ';\n');

			open(my $fh, '<', $template);
			open my $fh_email, "|-", "/usr/sbin/sendmail -t";
			printf $fh_email "To: %s\n", $_email;
			while (my $row = <$fh>) {
				chomp $row;
				$row =~ s/NAME/$_firstname/g;
				$row =~ s/IP$/$ipaddress/g;
				$row =~ s/VMID/$vm_name/g;
				$row =~ s/SERVER/$_server/g;
				print $fh_email "$row\n";
			}
			close $fh_email;
			print "NOTIFIED: $_date, $_payment, $_name, $_email, $_hostname, $_server ($vm_name), $ipaddress\n";
		}
	}
}

# function to print all keys & values for debug purposes
sub debug_parse {
	my %conf = %{$_[0]};
	my %vms = %{$_[1]};
	my $_etc = $conf{'conf'}{'ETC'};
	my $_vms = $conf{'conf'}{'VMS'};
	my $_tmpl = $conf{'conf'}{'TEMPLATES'};
	my $_server = $conf{'conf'}{'SERVER'};
	my $template = "$_tmpl/email-debug.txt";
	print "All VMs\n##\n";
	for my $vm_name (sort keys %vms) {
		my $_date = $vms{$vm_name}{'date'};
		my $_payment = $vms{$vm_name}{'payment'};
		my $_donated = $vms{$vm_name}{'donated'};
		my $_name = $vms{$vm_name}{'name'};
		my $_email = $vms{$vm_name}{'email'};
		my $_hostname = $vms{$vm_name}{'hostname'};
		my $_boot = $vms{$vm_name}{'boot'} || "yes";
		print "$_date, $_payment, $_name, $_email, $_hostname, $_server ($vm_name)\n";
	}
	print "##\n";
}

# check if _deploy.conf exists
my $dev = $ENV{'HOME'} . "/openbsd.amsterdam/deploy.pl";
my $prod = $ENV{'HOME'};
my $dir;
my $debug;
my %conf;
my %vms;
if (-d "$dev") {
	$dir = $dev;
	$debug = 1;
	%conf = get_variables('conf',  "$dir/_deploy.conf");
} elsif (-d "$prod") {
	$dir = $prod;
	%conf = get_variables('conf',  "$dir/_deploy.conf");
} else {
	printf "Unable to find config file\n";
	exit 1;
}

# parse all vm*.txt files in the VMS directory
my @files = glob "$conf{'conf'}{'VMS'}/*.txt";
%vms = get_variables('vms', @files);

# run functions
if ($debug) { debug_parse(\%conf, \%vms); }
if ($function =~ /deployed/) { deployed(\%conf, \%vms); }
elsif ($function =~ /stopped/) { stopped(\%conf, \%vms); }
elsif ($function =~ /renewal/) { renewal(\%conf, \%vms); }
elsif ($function =~ /notify/) { notify(\%conf, \%vms); }
else { print "Specify function: deployed, stopped, renewal, notify\n"; }