deploy.pl/notify.pl

205 lines
7.1 KiB
Perl
Raw Normal View History

#!/usr/bin/env perl
#
2022-11-13 21:14:11 +01:00
# Copyright (c) 2019-2022 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
2021-05-09 09:55:58 +02:00
# 2021/05/09 complete restructure and KISS
2022-11-13 21:14:11 +01:00
# 2022/11/13 added price structure for notifications
#
use 5.024;
use strict;
use warnings;
use autodie;
use POSIX qw(strftime);
use HTTP::Tiny;
2021-08-04 18:08:29 +02:00
use File::Basename;
2021-05-09 09:55:58 +02:00
# get function and function_variable (vmid) from arguments
my $function = $ARGV[0] || "empty";
my $function_variable = $ARGV[1] || "empty";
2022-11-13 21:14:11 +01:00
# define the prices for the different components of the VM
my $base_price = '60';
my %memory_prices = ('2G' => '10', '4G' => '30', '8G' => '70');
my %hdd_prices = ('50G' => '50', '100G' => '100', '150G' => '150', '200G' => '200');
# 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;
}
2021-05-09 09:55:58 +02:00
sub mailout {
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'};
2021-05-02 19:39:49 +02:00
my $_ip4netmask = $conf{'conf'}{'NETMASK'};
my $_ip4gw = $conf{'conf'}{'ROUTER'};
2021-05-09 09:55:58 +02:00
my $template = "$_tmpl/email-$function.txt";
2021-05-02 19:39:49 +02:00
my $server_number = $1 if $_server =~ /([0-9]+)/;
my $evenodd = $server_number % 2;
my $year = strftime("%Y", localtime);
my $month = strftime("%m", localtime);
2021-02-17 09:25:20 +01:00
my $response = HTTP::Tiny->new->get('https://openbsd.amsterdam/index.html');
my $total_donated = $1 if $response->{'content'} =~ /([0-9,\.]+) donated to the OpenBSD/;
2021-02-17 09:25:20 +01:00
my $total_vms = $1 if $response->{'content'} =~ /([0-9]+) VMs deployed/;
$response = HTTP::Tiny->new->get('https://openbsd.amsterdam/servers.html');
my $total_hosts = () = $response->{'content'} =~ /(\>Server )/g;
for my $vm_name (sort keys %vms) {
my $_date = $vms{$vm_name}{'date'};
my $_payment = $vms{$vm_name}{'payment'};
my $_subscription = $vms{$vm_name}{'subscription'} || "no";
my $_donated = $vms{$vm_name}{'donated'};
my $_name = $vms{$vm_name}{'name'};
my $_email = $vms{$vm_name}{'email'};
my $_hostname = $vms{$vm_name}{'hostname'};
2021-05-09 09:55:58 +02:00
my $_username = $vms{$vm_name}{'username'};
2022-11-13 21:14:11 +01:00
my $_memory = $vms{$vm_name}{'memory'} || '';
my $_disk2 = $vms{$vm_name}{'disk2'} || '';
my $memory_price = $memory_prices{$_memory} || '0';
my $hdd_price = $hdd_prices{$_disk2} || '0';
my $price = $base_price + $memory_price + $hdd_price;
2021-02-17 09:25:20 +01:00
2021-05-09 09:55:58 +02:00
(my $_firstname, my $_lastname) = split(/ /, $_name, 2);
(my $_year, my $_month, my $_day) = split(/\//, $_date, 3);
my $ip4address = qx(grep -A2 $vm_name $_etc/dhcpd.conf | awk '/fixed-address/{print \$2}' | tr -d ';\n');
my $ip6address = qx(grep -A3 $vm_name $_etc/dhcpd.conf | awk '/fixed-address-ipv6/{print \$2}' | tr -d ';\n');
(my $ip6gw = $ip6address) =~ s/::.*/::1/;
2021-05-02 19:39:49 +02:00
2021-05-09 09:55:58 +02:00
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/FIRSTNAME/$_firstname/g;
2021-05-09 09:55:58 +02:00
$row =~ s/IPV4$/$ip4address/g;
$row =~ s/IPV4NETMASK$/$_ip4netmask/g;
$row =~ s/IPV4GW$/$_ip4gw/g;
$row =~ s/IPV6$/$ip6address/g;
$row =~ s/IPV6GW$/$ip6gw/g;
$row =~ s/VMID/$vm_name/g;
$row =~ s/SERVER/$_server/g;
$row =~ s/HOSTNAME/$_hostname/g;
$row =~ s/USERNAME/$_username/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;
2022-11-13 21:14:11 +01:00
$row =~ s/PRICE/$price/g;
2021-05-09 09:55:58 +02:00
if ($row =~ /TIME\((.*)\)/) {
my @TIMES = split(/,/, $1);
$row =~ s/TIME\(.*\)/$TIMES[$evenodd]/g;
}
2021-05-09 09:55:58 +02:00
print $fh_email "$row\n";
}
2021-05-09 09:55:58 +02:00
close $fh_email;
2022-11-13 21:14:11 +01:00
print "$function: $_date, $_payment ($price), $_name, $_email, $_hostname, $_server ($vm_name), $ip4address\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);
2021-05-09 09:55:58 +02:00
if ($function =~ /notify/) {
mailout(\%conf, \%vms);
2022-04-09 10:57:44 +02:00
} elsif ($function =~ /(deployed|redeployed|cpu|msg|thanx)/ and $function_variable !~ /empty/) {
2021-05-09 09:55:58 +02:00
my %slice = %vms{$function_variable};
mailout(\%conf, \%slice);
2021-05-09 16:28:05 +02:00
} elsif ($function =~ /(renewal|subscription|deprovision)/) {
2021-05-09 09:55:58 +02:00
my $year = strftime("%Y", localtime);
my $month = strftime("%m", localtime);
for my $vm_name (sort keys %vms) {
if ($vms{$vm_name}{'donated'} =~ /(done|expire|sponsor|renewal)/) { delete $vms{$vm_name}; next; }
if ($vms{$vm_name}{'date'} =~ /$year\//) { delete $vms{$vm_name}; next; }
if ($vms{$vm_name}{'date'} !~ /\/$month\//) { delete $vms{$vm_name}; next; }
if ($function =~ /(renewal|deprovision)/) {
if (defined $vms{$vm_name}{'subscription'} and $vms{$vm_name}{'subscription'} =~ /yes/) { delete $vms{$vm_name}; next; }
}
if ($function =~ /subscription/) {
if (!defined $vms{$vm_name}{'subscription'} or $vms{$vm_name}{'subscription'} eq "" or $vms{$vm_name}{'subscription'} =~ /no/) { delete $vms{$vm_name}; next; }
2021-05-09 09:55:58 +02:00
}
}
mailout(\%conf, \%vms);
2021-05-09 16:28:05 +02:00
} elsif ($function =~ /stopped/) {
2021-05-09 09:55:58 +02:00
my @stopped_vms = qx(vmctl show | grep stopped | awk '{print \$9}');
for my $vm_name (sort keys %vms) {
if (!grep(/$vm_name/, @stopped_vms)) { delete $vms{$vm_name}; next; }
}
mailout(\%conf, \%vms);
2021-05-09 16:28:05 +02:00
} else {
2022-04-09 10:57:44 +02:00
print "usage: " . basename($0) . " [stopped | renewal | notify | deprovision] | [deployed | redeployed | cpu | msg | thanx] <vmid>\n";
2021-05-09 09:55:58 +02:00
}