/*
**
** Copyright (C) 2008-2020 Dirk-Jan C. Binnema <djcb@djcbsoftware.nl>
**
** 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, 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, write to the Free Software Foundation,
** Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
**
*/

#include "config.h"

#include <glib.h>
#include <glib/gstdio.h>

#include <stdlib.h>
#include <unistd.h>
#include <string.h>

#include "test-mu-common.hh"
#include "mu-store.hh"
#include "mu-query.hh"

static gchar *CONTACTS_CACHE = NULL;

static gchar*
fill_contacts_cache (void)
{
	gchar *cmdline, *tmpdir;
	GError *err;

	tmpdir = test_mu_common_get_random_tmpdir();
	cmdline = g_strdup_printf (
		"/bin/sh -c '"
		"%s init  --muhome=%s --maildir=%s --quiet; "
		"%s index --muhome=%s  --quiet'",
		MU_PROGRAM,  tmpdir, MU_TESTMAILDIR,
		MU_PROGRAM,  tmpdir);

	if (g_test_verbose())
		g_print ("%s\n", cmdline);

	err  = NULL;
	if (!g_spawn_command_line_sync (cmdline, NULL, NULL,
					NULL, &err)) {
		g_printerr ("Error: %s\n", err ? err->message : "?");
		g_assert (0);
	}

	g_free (cmdline);
	return tmpdir;
}

static void
test_mu_cfind_plain (void)
{
	gchar *cmdline, *output, *erroutput;

	cmdline = g_strdup_printf ("%s cfind --muhome=%s --format=plain "
				   "'testmu\\.xxx?'",
				   MU_PROGRAM, CONTACTS_CACHE);
	if (g_test_verbose())
		g_print ("%s\n", cmdline);

	output = erroutput = NULL;
	g_assert (g_spawn_command_line_sync (cmdline, &output, &erroutput,
					     NULL, NULL));

	/* note, output order is unspecified */
	g_assert (output);
	if (output[0] == 'H')
		g_assert_cmpstr (output,
				 ==,
				 "Helmut Kröger hk@testmu.xxx\n"
				 "Mü testmu@testmu.xx\n");
	else
		g_assert_cmpstr (output,
				 ==,
				 "Mü testmu@testmu.xx\n"
				 "Helmut Kröger hk@testmu.xxx\n");
	g_free (cmdline);
	g_free (output);
	g_free (erroutput);
}

static void
test_mu_cfind_bbdb (void)
{
	gchar *cmdline, *output, *erroutput, *expected;
	gchar today[12];
	struct tm *tmtoday;
	time_t now;
	const char *old_tz;

	old_tz = set_tz ("Europe/Helsinki");

	cmdline = g_strdup_printf ("%s cfind --muhome=%s --format=bbdb "
				   "'testmu\\.xxx?'",
				   MU_PROGRAM, CONTACTS_CACHE);

	output = erroutput = NULL;
	g_assert (g_spawn_command_line_sync (cmdline, &output, &erroutput,
					     NULL, NULL));

#define frm1   ";; -*-coding: utf-8-emacs;-*-\n"			\
		";;; file-version: 6\n"					\
		"[\"Helmut\" \"Kröger\" nil nil nil nil (\"hk@testmu.xxx\") " \
		"((creation-date . \"%s\") "				\
		"(time-stamp . \"1970-01-01\")) nil]\n"			\
		"[\"Mü\" \"\" nil nil nil nil (\"testmu@testmu.xx\") "	\
		"((creation-date . \"%s\") "				\
		"(time-stamp . \"1970-01-01\")) nil]\n"


#define	frm2   ";; -*-coding: utf-8-emacs;-*-\n"			\
		";;; file-version: 6\n"					\
		"[\"Mü\" \"\" nil nil nil nil (\"testmu@testmu.xx\") "	\
		"((creation-date . \"%s\") "				\
		"(time-stamp . \"1970-01-01\")) nil]\n"			\
		"[\"Helmut\" \"Kröger\" nil nil nil nil (\"hk@testmu.xxx\") " \
		"((creation-date . \"%s\") "				\
		"(time-stamp . \"1970-01-01\")) nil]\n"

	g_assert (output);

	now = time(NULL);
	tmtoday = localtime(&now);
	strftime(today,sizeof(today),"%Y-%m-%d", tmtoday);

	expected = g_strdup_printf (output[52] == 'H' ? frm1 : frm2,
				    today, today);

	/* g_print ("\n%s\n", output); */

	g_assert_cmpstr (output, ==, expected);

	g_free (cmdline);
	g_free (output);
	g_free (erroutput);
	g_free (expected);

	set_tz (old_tz);
}


static void
test_mu_cfind_wl (void)
{
	gchar *cmdline, *output, *erroutput;

	cmdline = g_strdup_printf ("%s cfind --muhome=%s --format=wl "
				   "'testmu\\.xxx?'",
				   MU_PROGRAM, CONTACTS_CACHE);

	output = erroutput = NULL;
	g_assert (g_spawn_command_line_sync (cmdline, &output, &erroutput,
					     NULL, NULL));

	g_assert (output);
	if (output[0] == 'h')
		g_assert_cmpstr (output,
				 ==,
				 "hk@testmu.xxx \"HelmutK\" \"Helmut Kröger\"\n"
				 "testmu@testmu.xx \"Mü\" \"Mü\"\n");
	else
		g_assert_cmpstr (output,
				 ==,
				 "testmu@testmu.xx \"Mü\" \"Mü\"\n"
				 "hk@testmu.xxx \"HelmutK\" \"Helmut Kröger\"\n");

	g_free (cmdline);
	g_free (output);
	g_free (erroutput);
}


static void
test_mu_cfind_mutt_alias (void)
{
	gchar *cmdline, *output, *erroutput;

	cmdline = g_strdup_printf ("%s cfind --muhome=%s --format=mutt-alias "
				   "'testmu\\.xxx?'",
				   MU_PROGRAM, CONTACTS_CACHE);

	output = erroutput = NULL;
	g_assert (g_spawn_command_line_sync (cmdline, &output, &erroutput,
					     NULL, NULL));

	/* both orders are possible... */
	g_assert (output);

	if (output[6] == 'H')
		g_assert_cmpstr (output,
				 ==,
				 "alias HelmutK Helmut Kröger <hk@testmu.xxx>\n"
				 "alias Mü Mü <testmu@testmu.xx>\n");
	else
		g_assert_cmpstr (output,
				 ==,
				 "alias Mü Mü <testmu@testmu.xx>\n"
				 "alias HelmutK Helmut Kröger <hk@testmu.xxx>\n");


	g_free (cmdline);
	g_free (output);
	g_free (erroutput);
}

static void
test_mu_cfind_mutt_ab (void)
{
	gchar *cmdline, *output, *erroutput;

	cmdline = g_strdup_printf ("%s cfind --muhome=%s --format=mutt-ab "
				   "'testmu\\.xxx?'",
				   MU_PROGRAM, CONTACTS_CACHE);

	if (g_test_verbose())
		g_print("%s\n", cmdline);

	output = erroutput = NULL;
	g_assert (g_spawn_command_line_sync (cmdline, &output, &erroutput,
					     NULL, NULL));
	g_assert (output);

	if (output[39] == 'h')
		g_assert_cmpstr (output,
				 ==,
				 "Matching addresses in the mu database:\n"
				 "hk@testmu.xxx\tHelmut Kröger\t\n"
				 "testmu@testmu.xx\tMü\t\n");
	else
		g_assert_cmpstr (output,
				 ==,
				 "Matching addresses in the mu database:\n"
				 "testmu@testmu.xx\tMü\t\n"
				 "hk@testmu.xxx\tHelmut Kröger\t\n");

	g_free (cmdline);
	g_free (output);
	g_free (erroutput);
}


static void
test_mu_cfind_org_contact (void)
{
	gchar*cmdline, *output, *erroutput;

	cmdline = g_strdup_printf ("%s cfind --muhome=%s --format=org-contact "
				   "'testmu\\.xxx?'",
				   MU_PROGRAM, CONTACTS_CACHE);

	output = erroutput = NULL;
	g_assert (g_spawn_command_line_sync (cmdline, &output, &erroutput,
					     NULL, NULL));

	g_assert (output);

	if (output[2] == 'H')
		g_assert_cmpstr (output,
				 ==,
				 "* Helmut Kröger\n"
				 ":PROPERTIES:\n"
				 ":EMAIL: hk@testmu.xxx\n"
				 ":END:\n\n"
				 "* Mü\n"
				 ":PROPERTIES:\n"
				 ":EMAIL: testmu@testmu.xx\n"
				 ":END:\n\n");
	else
		g_assert_cmpstr (output,
				 ==,
				 "* Mü\n"
				 ":PROPERTIES:\n"
				 ":EMAIL: testmu@testmu.xx\n"
				 ":END:\n\n"
				 "* Helmut Kröger\n"
				 ":PROPERTIES:\n"
				 ":EMAIL: hk@testmu.xxx\n"
				 ":END:\n\n");


	g_free (cmdline);
	g_free (output);
	g_free (erroutput);
}



static void
test_mu_cfind_csv (void)
{
	gchar *cmdline, *output, *erroutput;

	cmdline = g_strdup_printf ("%s cfind --muhome=%s --format=csv "
				   "'testmu\\.xxx?'",
				   MU_PROGRAM, CONTACTS_CACHE);

	if (g_test_verbose())
		g_print("%s\n", cmdline);

	output = erroutput = NULL;
	g_assert (g_spawn_command_line_sync (cmdline, &output, &erroutput,
					     NULL, NULL));
	g_assert (output);
	if (output[1] == 'H')
		g_assert_cmpstr (output,
				 ==,
				 "\"Helmut Kröger\",\"hk@testmu.xxx\"\n"
				 "\"Mü\",\"testmu@testmu.xx\"\n");
	else
		g_assert_cmpstr (output,
				 ==,
				 "\"Mü\",\"testmu@testmu.xx\"\n"
				 "\"Helmut Kröger\",\"hk@testmu.xxx\"\n");
	g_free (cmdline);
	g_free (output);
	g_free (erroutput);
}


int
main (int argc, char *argv[])
{
	int rv;
	g_test_init (&argc, &argv, NULL);

	if (!set_en_us_utf8_locale())
		return 0; /* don't error out... */

	CONTACTS_CACHE = fill_contacts_cache ();

	g_test_add_func ("/mu-cmd-cfind/test-mu-cfind-plain", test_mu_cfind_plain);
	g_test_add_func ("/mu-cmd-cfind/test-mu-cfind-bbdb",  test_mu_cfind_bbdb);
	g_test_add_func ("/mu-cmd-cfind/test-mu-cfind-wl",  test_mu_cfind_wl);
	g_test_add_func ("/mu-cmd-cfind/test-mu-cfind-mutt-alias",
			 test_mu_cfind_mutt_alias);
	g_test_add_func ("/mu-cmd-cfind/test-mu-cfind-mutt-ab",
			 test_mu_cfind_mutt_ab);
	g_test_add_func ("/mu-cmd-cfind/test-mu-cfind-org-contact",
			 test_mu_cfind_org_contact);
	g_test_add_func ("/mu-cmd-cfind/test-mu-cfind-csv",
			 test_mu_cfind_csv);

	g_log_set_handler (NULL,
			   (GLogLevelFlags)(
			   G_LOG_LEVEL_MASK | G_LOG_LEVEL_WARNING|
			   G_LOG_FLAG_FATAL| G_LOG_FLAG_RECURSION),
			   (GLogFunc)black_hole, NULL);

	rv = g_test_run ();

	g_free (CONTACTS_CACHE);
	CONTACTS_CACHE = NULL;

	return rv;
}
