#!/usr/bin/perl -w
###############################################################################
# Licensed to the Apache Software Foundation (ASF) under one or more
# contributor license agreements.  See the NOTICE file distributed with
# this work for additional information regarding copyright ownership.
# The ASF licenses this file to You under the Apache License, Version 2.0
# (the "License"); you may not use this file except in compliance with
# the License.  You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
###############################################################################

use Getopt::Long;
use YAML;
use DBI;
use Data::Dumper;

my $getnames = 0;
my $dumpmessage = '';
my $setmessage = '';
my $resetmessage = '';
my $htmlfile = '';
my $subject = '';
my $help = 0;

GetOptions ('getmessagenames' => \$getnames,
            'dumpmessage:s' => \$dumpmessage,
            'resetmessage:s' => \$resetmessage,
            'setmessage:s' => \$setmessage,
            'htmlfile:s' => \$htmlfile,
            'subject:s' => \$subject,
            'help|?' => \$help);

if ($help ||
	($getnames == 1 && ($dumpmessage ne '' || $setmessage ne '' || $resetmessage ne '' || $htmlfile ne '' || $subject ne '')) ||
	($getnames == 0 && $dumpmessage ne '' && ($setmessage ne '' || $resetmessage ne '' || $htmlfile ne '' || $subject ne '')) ||
	($getnames == 0 && $resetmessage ne '' && ($setmessage ne '' || $dumpmessage ne '' || $htmlfile ne '' || $subject ne '')) ||
	($getnames == 0 && $setmessage ne '' && ($dumpmessage ne '' || $resetmessage ne '' || $htmlfile eq '' || $subject eq '')) ||
	($getnames == 0 && $setmessage eq '' && $dumpmessage eq '' && $resetmessage eq ''))
{
	print "Usage:\n\n";
	print "vclmessages.pl --getmessagenames\n";
	print "vclmessages.pl --dumpmessage '<name of message>'\n";
	print "vclmessages.pl --setmessage '<name of message>' --htmlfile <filename> --subject <message subject>\n";
	print "vclmessages.pl --resetmessage '<name of message>'\n\n";
	print "vclmessages.pl --help|-?\n\n";
	print "Where\n\n";
	print "--getmessagenames displays a list of all available names that can be used\n";
	print "--dumpmessage displays the current value of a message\n";
	print "--setmessage sets the value of a message to the contents of the specified file\n";
	print "--resetmessage sets the value of a message back to the original value as distributed with VCL\n\n";
	print "<name of message> = the name of the message from the database (enclose in single quotes)\n";
	print "\tuse --getmessagenames to get a list of message names\n\n";
	print "<filename> = filename (including path) of file containing html contents for email message\n\n";
	print "<message subject> = subject for email message (enclose in single quotes)\n\n";
	exit 0;
}

my $mode = 'getnames';
$mode = 'dumpmessage' if($dumpmessage ne '');
$mode = 'setmessage' if($setmessage ne '');
$mode = 'resetmessage' if($resetmessage ne '');

my $messagename = $dumpmessage;
$messagename = $setmessage if($mode eq 'setmessage');
$messagename = $resetmessage if($mode eq 'resetmessage');

my $database = `grep ^database /etc/vcl/vcld.conf | awk -F '=' '{print \$2}'`;
my $hostname = `grep ^server /etc/vcl/vcld.conf | awk -F '=' '{print \$2}'`;
my $user = `grep ^LockerWrtUser /etc/vcl/vcld.conf | awk -F '=' '{print \$2}'`;
my $password = `grep ^wrtPass /etc/vcl/vcld.conf | awk -F '=' '{print \$2}'`;

chomp $database;
chomp $hostname;
chomp $user;
chomp $password;

my $dsn = "DBI:mysql:database=$database;host=$hostname";
my $dbh = DBI->connect($dsn, $user, $password);

# ================= get names ================
if($mode eq 'getnames')
{
	my $sth = $dbh->prepare(
		"SELECT name FROM variable WHERE name LIKE 'usermessage%' OR name LIKE 'adminmessage%' ORDER BY name")
		or die "Error: Failed to prepare database query: $dbh->errstr()";
	$sth->execute();
	while (my $ref = $sth->fetchrow_hashref()) {
		print "$ref->{'name'}\n";
	}
	$sth->finish;
	$sth->finish;
	$dbh->disconnect;
	exit 0;
}
# ================ dump message ===============
elsif($mode eq 'dumpmessage')
{
	my $sth = $dbh->prepare(
		"SELECT value FROM variable WHERE name = ?")
		or die "Error: Failed to prepare database query: $dbh->errstr()";
	$sth->execute($messagename);
	if($sth->rows == 0)
	{
		print "Error: Failed to find message with name $messagename\n";
		$sth->finish;
		$dbh->disconnect;
		exit 0;
	}
	if($sth->rows > 1)
	{
		print "Error: Found multiple messages with name $messagename\n";
		$sth->finish;
		$dbh->disconnect;
		exit 0;
	}
	my $ref = $sth->fetchrow_hashref();
	my $data = YAML::Load($ref->{'value'});
	#print Dumper($data);
	print "Subject: $data->{'subject'}\n";
	print "Message:\n";
	print "$data->{'message'}\n";

	$sth->finish;
	$dbh->disconnect;
	exit 0;
}
# ================= reset message ===============
elsif($mode eq 'resetmessage')
{
	my $sth = $dbh->prepare(
		'SELECT value FROM messagereset WHERE name = ?')
		or die "Error: Failed to prepare database query: $dbh->errstr()";
	$sth->execute($messagename) or die "Error: failed to query database: $dbh->errstr()";
	if($sth->rows == 0)
	{
		print "Error: Failed to find message with name $messagename\n";
		$sth->finish;
		$dbh->disconnect;
		exit 0;
	}
	if($sth->rows > 1)
	{
		print "Error: Found multiple messages with name $messagename\n";
		$sth->finish;
		$dbh->disconnect;
		exit 0;
	}

	my $ref = $sth->fetchrow_hashref();
	my $message = $ref->{'value'};

	$sth = $dbh->prepare(
		"UPDATE variable SET value = ?, setby = 'setemail script', timestamp = NOW() WHERE name = ?")
		or die "Error: Failed to prepare database query: $dbh->errstr()";

	$sth->bind_param(1, $message);
	$sth->bind_param(2, $messagename);
	$sth->execute() or die "Error: Failed to update value for $messagename\n";

	$sth->finish;
	$dbh->disconnect;
	print "Success: Value reset for $messagename\n";
}
# ================= set message ===============
elsif($mode eq 'setmessage')
{
	my $htmlemail;
	open(my $fh, '<', $htmlfile) or die "Error: failed to open $htmlfile for reading";
	{
		local $/;
		$htmlemail = <$fh>;
	}
	close($fh);

	my $sth = $dbh->prepare(
		'SELECT value FROM variable WHERE name = ?')
		or die "Error: Failed to prepare database query: $dbh->errstr()";
	$sth->execute($messagename) or die "Error: failed to query database: $dbh->errstr()";
	if($sth->rows == 0)
	{
		print "Error: Failed to find message with name $messagename\n";
		$sth->finish;
		$dbh->disconnect;
		exit 0;
	}
	if($sth->rows > 1)
	{
		print "Error: Found multiple messages with name $messagename\n";
		$sth->finish;
		$dbh->disconnect;
		exit 0;
	}

	my $ref = $sth->fetchrow_hashref();
	my $data = YAML::Load($ref->{'value'});
	$data->{'message'} = $htmlemail;
	$data->{'subject'} = $subject;
	my $yaml = YAML::Dump($data);

	$sth = $dbh->prepare(
		"UPDATE variable SET value = ?, setby = 'setemail script', timestamp = NOW() WHERE name = ?")
		or die "Error: Failed to prepare database query: $dbh->errstr()";

	$sth->bind_param(1, $yaml);
	$sth->bind_param(2, $messagename);
	$sth->execute() or die "Error: Failed to update value for $messagename\n";

	$sth->finish;
	$dbh->disconnect;
	print "Success: Value for $messagename updated from contents of $htmlfile\n";
}