// Original code taken from the example webkit-gtk+ application. see notice below.
// Modified code is licensed under the GPL 3.  See LICENSE file.


/*
 * Copyright (C) 2006, 2007 Apple Inc.
 * Copyright (C) 2007 Alp Toker <alp@atoker.com>
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY
 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE COMPUTER, INC. OR
 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
 * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include <gtk/gtk.h>
#include <gdk/gdk.h>
#include <gdk/gdkx.h>
#include <gdk/gdkkeys.h>
#include <gdk/gdkkeysyms.h>
#include <webkit/webkit.h>

#include <pthread.h>
#include <stdio.h>
#include <string.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <unistd.h>
#include <stdlib.h>

static GtkWidget*     main_window;
static GtkWidget*     modeline;
static WebKitWebView* web_view;

static gchar*   history_file;
static gchar*   home_page;
static gchar*   uri       = NULL;
static gchar*   fifodir   = NULL;
static char     fifopath[64];
static bool     modevis = FALSE;
static gboolean verbose = FALSE;
static Window   xwin    = NULL;

static GOptionEntry entries[] =
{
  { "uri",      'u', 0, G_OPTION_ARG_STRING, &uri,      "Uri to load",                                   NULL },
  { "fifo-dir", 'd', 0, G_OPTION_ARG_STRING, &fifodir,  "Directory to place FIFOs",                      NULL },
  { NULL }
};

struct command
{
  char command[256];
  void (*func)(WebKitWebView*);
};

static struct command commands[256];
static int            numcmds = 0;

struct alias
{
  char alias[256];
  char command[256];
};

static struct alias aliases[256];
static int          numalias = 0;

static void parse_command(char*);

static bool parse_modeline (GtkWidget* mode, GdkEventKey* event)
{
  if ((event->type==GDK_KEY_PRESS) && (event->keyval==GDK_Return))
    parse_command (gtk_entry_get_text (modeline));
 
  return false;
}

static void log_history_cb (WebKitWebView* page, WebKitWebFrame* frame, gpointer data)
{
  strncpy (uri, webkit_web_frame_get_uri (frame), strlen (webkit_web_frame_get_uri (frame)));
  
  FILE * output_file = fopen(history_file, "a");
  if (output_file == NULL)
    {
      fprintf(stderr, "Cannot open %s for logging\n", history_file);
    }
  else
    {
      time_t rawtime;
      struct tm * timeinfo;
      char buffer [80];
      time ( &rawtime );
      timeinfo = localtime ( &rawtime );
      strftime (buffer,80,"%Y-%m-%d %H:%M:%S",timeinfo);
      
      fprintf(output_file, "%s %s\n",buffer, uri);
      fclose(output_file);
    }
}

static void toggle_command_mode ()
{
  if (modevis)
    {
      gtk_widget_hide (modeline);
      gtk_widget_grab_focus (web_view);
    }
  else
    {
      gtk_widget_show (modeline);
      gtk_widget_grab_focus (modeline);
    }
  modevis = ! modevis;
}

static gboolean key_press_cb (WebKitWebView* page, GdkEventKey* event)
{
  gboolean result=FALSE; //TRUE to stop other handlers from being invoked for the event. FALSE to propagate the event further.
  if ((event->type==GDK_KEY_PRESS) && (event->keyval==GDK_Escape))
    {
      toggle_command_mode ();
      result=TRUE;
    }
 
  return(result);
}

static GtkWidget* create_browser ()
{
  GtkWidget* scrolled_window = gtk_scrolled_window_new (NULL, NULL);
  gtk_scrolled_window_set_policy (GTK_SCROLLED_WINDOW (scrolled_window), GTK_POLICY_NEVER, GTK_POLICY_NEVER);

  web_view = WEBKIT_WEB_VIEW (webkit_web_view_new ());
  gtk_container_add (GTK_CONTAINER (scrolled_window), GTK_WIDGET (web_view));

  g_signal_connect (G_OBJECT (web_view), "load-committed", G_CALLBACK (log_history_cb), web_view);

  return scrolled_window;
}

static GtkWidget* create_window ()
{
  GtkWidget* window = gtk_window_new (GTK_WINDOW_TOPLEVEL);
  gtk_window_set_default_size (GTK_WINDOW (window), 800, 600);
  gtk_widget_set_name (window, "Uzbl Browser");
  g_signal_connect (G_OBJECT (window), "destroy", G_CALLBACK (gtk_main_quit), NULL);
  g_signal_connect (G_OBJECT (window), "key-press-event", G_CALLBACK(key_press_cb), NULL);

  return window;
}

static GtkWidget* create_modeline ()
{
  GtkWidget* modeline = gtk_entry_new ();
  g_signal_connect (G_OBJECT (modeline), "key-press-event", G_CALLBACK(parse_modeline), modeline);

  return modeline;
}

static void parse_command(char *command)
{
  int  i    = 0;
  bool done = false;
  char *cmdstr = command;
  void (*func)(WebKitWebView*);

  printf("Checking aliases\n");
  for (i = 0; i < numalias && ! done; i++)
    {
      if (!strncmp (cmdstr, aliases[i].alias, strlen (aliases[i].alias)))
        {
          strcpy(cmdstr, aliases[i].command);
          done = true;
        }
    }

  done = false;
  printf("Checking commands\n");
  for (i = 0; i < numcmds && ! done; i++)
    {
      if (!strncmp (cmdstr, commands[i].command, strlen (commands[i].command)))
        {
          func = commands[i].func;
          done = true;
        }
    }

  printf("Command identified as \"%s\"\n", cmdstr);

  if (done)
    {
      func (web_view);
    }
  else
    {
      if (!strncmp ("http://", command, 7))
        {
          printf ("Loading URI \"%s\"\n", command);
          uri = command;
          webkit_web_view_load_uri (web_view, uri);
        }
    }
}

static void *control_fifo()
{
  if (fifodir)
    {
      sprintf (fifopath, "%s/uzbl_%d", fifodir, getpid ());
    }
  else
    {
      sprintf (fifopath, "/tmp/uzbl_%d", getpid ());
    }

  if (mkfifo (fifopath, 0666) == -1)
    {
      printf ("Possible error creating fifo\n");
    }

    printf ("Opened control fifo in %s\n", fifopath);

    while (true)
      {
        FILE *fifo = fopen(fifopath, "r");
        if (!fifo)
          {
            printf("Could not open %s for reading\n", fifopath);
            return NULL;
          }
        
        char buffer[256];
        memset (buffer, 0, sizeof (buffer));
        while (!feof (fifo) && fgets (buffer, sizeof (buffer), fifo))
          {
            if (strcmp (buffer, "\n"))
              {
                buffer[strlen (buffer) - 1] = '\0'; // Remove newline
                parse_command (buffer);
              }
          }
      }
    
    return NULL;
}

static void add_command (char* cmdstr, void* function)
{
  strncpy (commands[numcmds].command, cmdstr, strlen (cmdstr));
  commands[numcmds].func = function;
  numcmds++;
}

static void add_command_alias (char* alias, char* command)
{
  strncpy (aliases[numalias].alias,   alias,   strlen (alias));
  strncpy (aliases[numalias].command, command, strlen (command));
  numalias++;
}

static void setup_gtk (int argc, char* argv[])
{
  gtk_init (&argc, &argv);

  GtkWidget* vbox = gtk_vbox_new (FALSE, 0);
  gtk_box_pack_start (GTK_BOX (vbox), create_browser (), TRUE, TRUE, 0);
  modeline = create_modeline ();
  gtk_box_pack_start (GTK_BOX (vbox), modeline, FALSE, FALSE, 0);

  main_window = create_window ();
  gtk_container_add (GTK_CONTAINER (main_window), vbox);
  GError *error = NULL;

  GOptionContext* context = g_option_context_new ("- The Usable Browser, controlled entirely through a FIFO");
  g_option_context_add_main_entries (context, entries, NULL);
  g_option_context_add_group (context, gtk_get_option_group (TRUE));
  g_option_context_parse (context, &argc, &argv, &error);

  if (uri)
    {
      webkit_web_view_load_uri (web_view, uri);
    }

  gtk_widget_grab_focus (GTK_WIDGET (web_view));
  gtk_widget_show_all (main_window);
  gtk_widget_hide(modeline);
  gtk_widget_grab_focus (GTK_WIDGET (web_view));
}

static void setup_commands ()
{
  //This func. is nice but currently it cannot be used for functions that require arguments or return data. --sentientswitch

  add_command("back",     &webkit_web_view_go_back);
  add_command("forward",  &webkit_web_view_go_forward);
  add_command("refresh",  &webkit_web_view_reload); //Buggy
  add_command("stop",     &webkit_web_view_stop_loading);
  add_command("zoom in",  &webkit_web_view_zoom_in); //Can crash (when max zoom reached?).
  add_command("zoom out", &webkit_web_view_zoom_out); //Crashes as zoom +
  //add_command("get uri", &webkit_web_view_get_uri);
}

static void setup_threading ()
{
  pthread_t control_thread;
  pthread_create(&control_thread, NULL, control_fifo, NULL);
}

static void setup_settings ()
{
  GKeyFile* config = g_key_file_new ();
  gboolean  res    = g_key_file_load_from_file (config, "./sampleconfig", G_KEY_FILE_NONE, NULL); //TODO: pass config file as argument

  if (res)
    {
      printf ("Config loaded\n");
    }
  else
    {
      fprintf (stderr, "config loading failed\n"); //TODO: exit codes with gtk? 
    }

  history_file = g_key_file_get_value (config, "behavior", "history_file", NULL);
  if (history_file)
    {
      printf ("Setting history file to: %s\n", history_file);
    }
  else
    {
      printf ("History logging disabled\n");
    }

  home_page = g_key_file_get_value (config, "behavior", "home_page", NULL);
  if (home_page)
    {
      printf ("Setting home page to: %s\n", home_page);
    }
  else
    {
      printf ("Home page disabled\n");
    }

  /*GError *error = 0;
  char   *keys  = g_key_file_get_keys (config, "alias", NULL, &error);

  if (error) 
    {
      printf("Error: %n\n", error);
    }
  else
    {
      printf("Loading aliases\n");
      while (keys != NULL &&  (*keys) != NULL)
        {
          char* value = g_key_file_get_value (config, (gchar *)"alias", (*keys), NULL);
          add_command_alias((*keys), value);
          ++keys;
        }
    }

    Until segfaults is fixed, manually add aliases to test the rest of it. */
  add_command_alias("b",  "back");
  add_command_alias("f",  "forward");
  add_command_alias("z+", "zoom in");
  add_command_alias("z-", "zoom out");
  add_command_alias("r",  "refresh");
  add_command_alias("s",  "stop");
}

int main (int argc, char* argv[])
{
  if (!g_thread_supported ())
    g_thread_init (NULL);

  xwin = GDK_WINDOW_XID (GTK_WIDGET (main_window)->window);
  printf("My X window id is %i\n",(int) xwin);

  setup_settings ();
  setup_gtk (argc, argv);
  setup_commands ();
  setup_threading ();
  gtk_main ();

  printf ("Shutting down...\n");

  unlink (fifopath);

  return 0;
}