/*

 * Copyright (C) 2016  J. Ali Harlow <ali@juiblex.co.uk>

 *

 * 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 2 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, write to the Free Software Foundation, Inc.,

 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.

 */

#include <stdlib.h>

#include <sys/time.h>

#include <glib.h>

#include <razor.h>

#include <plover/plover.h>

static gchar *log_file_path(const char *directory,const char *name,

  const char *stamp)

{

    gchar *s,*path;

    if (name)

	s=g_strdup(name);

    else

	s=g_strconcat("test-",stamp,NULL);

    if (directory)

	path=g_build_filename(directory,s,NULL);

    else

	path=g_strdup(s);

    g_free(s);

    return path;

}

static GDateTime *date_days_ago(int age)

{

    GDateTime *now,*then;

    now=g_date_time_new_now_utc();

    then=g_date_time_add_days(now,-age);

    g_date_time_unref(now);

    return then;

}

static gchar *date_stamp(int age)

{

    gchar *stamp;

    GDateTime *then;

    then=date_days_ago(age);

    stamp=g_date_time_format(then,"%Y%m%d");

    g_date_time_unref(then);

    return stamp;

}

static void simulate_log_file(const char *directory,const char *name,int age)

{

    gchar *path,*contents,*stamp;

    GDateTime *then;

    GError *err=NULL;

    struct timeval times[2]={{0,},{0,}};

    stamp=date_stamp(age);

    path=log_file_path(directory,name,stamp);

    contents=g_strdup_printf("Log for %s\n",stamp);

    if (!g_file_set_contents(path,contents,-1,&err))

	g_error("%s: %s",path,err->message);

    g_free(contents);

    then=date_days_ago(age);

    times[0].tv_sec=times[1].tv_sec=g_date_time_to_unix(then);

    utimes(path,times);

    g_date_time_unref(then);

    g_free(path);

    g_free(stamp);

}

static void verify_log_file(const char *directory,const char *name,int age,

  gboolean should_exist)

{

    gchar *path,*contents,*stamp,*s;

    GError *err=NULL;

    stamp=date_stamp(age);

    path=log_file_path(directory,name,stamp);

    if (!should_exist)

	g_assert(!g_file_test(path,G_FILE_TEST_EXISTS));

    else

    {

	if (!g_file_get_contents(path,&contents,NULL,&err))

	    g_error("%s: %s",path,err->message);

	s=g_strdup_printf("Log for %s\n",stamp);

	g_assert_cmpstr(contents,==,s);

	g_free(s);

	g_free(contents);

    }

    g_free(path);

    g_free(stamp);

}

static int safe_log_open(const char *s)

{

    int fd1,fd2,result;

    fflush(stdout);

    fflush(stderr);

    fd1=dup(1);

    fd2=dup(2);

    result=plover_log_open(s);

    fflush(stdout);

    fflush(stderr);

    dup2(fd1,1);

    dup2(fd2,2);

    close(fd1);

    close(fd2);

    return result;

}

static void verify_log_open(const char *directory)

{

    int result;

    gchar *stamp,*s;

    simulate_log_file(directory,"test",1);

    simulate_log_file(directory,NULL,1);

    simulate_log_file(directory,NULL,2);

    simulate_log_file(directory,NULL,3);

    simulate_log_file(directory,NULL,4);

    if (directory)

	s=g_build_filename(directory,"test",NULL);

    else

	s=g_strdup("test");

    result=safe_log_open(s);

    g_free(s);

    g_assert_cmpint(result,==,0);

    stamp=date_stamp(1);

    s=g_strconcat("test-",stamp,"a",NULL);

    g_free(stamp);

    verify_log_file(directory,s,1,TRUE);

    g_free(s);

    verify_log_file(directory,NULL,1,TRUE);

    verify_log_file(directory,NULL,2,TRUE);

    verify_log_file(directory,NULL,3,TRUE);

    verify_log_file(directory,NULL,4,FALSE);

}

static void test_rotate(void)

{

    system("rm -rf test-log-rotate");

    g_assert(mkdir("test-log-rotate",0777)==0);

    verify_log_open("test-log-rotate");

}

static void test_rotate_cwd(void)

{

    system("rm -rf test-log-rotate");

    g_assert(mkdir("test-log-rotate",0777)==0);

    g_assert(chdir("test-log-rotate")==0);

    verify_log_open(NULL);

    g_assert(chdir("..")==0);

}

static void test_rotate_root(void)

{

    gchar *cwd,*s;

    system("rm -rf test-log-rotate");

    g_assert(mkdir("test-log-rotate",0777)==0);

    cwd=g_get_current_dir();

    s=g_strconcat("file:",cwd,"/",NULL);

    g_free(cwd);

    g_setenv("RAZOR_ROOT",s,TRUE);

    g_free(s);

    verify_log_open("test-log-rotate");

    g_unsetenv("RAZOR_ROOT");

}

static void test_readonly_dir(void)

{

    system("rm -rf test-log-rotate");

    g_assert(mkdir("test-log-rotate",0)==0);

    g_assert(safe_log_open("test-log-rotate/test")!=0);

}

static void test_directory(void)

{

    system("rm -rf test-log-rotate");

    g_assert(mkdir("test-log-rotate",0777)==0);

    g_assert(mkdir("test-log-rotate/test",0777)==0);

    g_assert(safe_log_open("test-log-rotate/test")!=0);

}

static void test_readonly_file(void)

{

    GError *err=NULL;

    system("rm -rf test-log-rotate");

    g_assert(mkdir("test-log-rotate",0777)==0);

    if (!g_file_set_contents("test-log-rotate/test","",0,&err))

	g_error("test-log-rotate/test: %s",err->message);

    g_assert(chmod("test-log-rotate/test",0)==0);

    g_assert(safe_log_open("test-log-rotate/test")!=0);

}

int main(int argc,char **argv)

{

    int retval;

    g_test_init(&argc,&argv,NULL);

    g_test_bug_base("mailto:ali@juiblex.co.uk");

    g_unsetenv("RAZOR_ROOT");

    g_test_add_func("/log/rotate",test_rotate);

    g_test_add_func("/log/rotate-cwd",test_rotate_cwd);

    g_test_add_func("/log/rotate-root",test_rotate_root);

    g_test_add_func("/log/readonly-dir",test_readonly_dir);

    g_test_add_func("/log/directory",test_directory);

    g_test_add_func("/log/readonly-file",test_readonly_file);

    retval=g_test_run();

    return retval;

}