/*
 * Copyright (C) 2008 Search Solution Corporation. All rights reserved by Search Solution. 
 *
 *   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; version 2 of the License. 
 *
 *  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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA 
 *
 */


/*
 * broker_log_converter.c - 
 */

#ident "$Id$"

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#ifndef WIN32
#include <unistd.h>
#endif

#include "cas_common.h"
#include "cas_cci.h"
#include "broker_log_util.h"

#ifdef WIN32
#include "broker_getopt.h"
#endif

#define CAS_LOG_MSG_INDEX 19
#define CAS_RUN_NEW_LINE_CHAR	1

#define GET_MSG_START_PTR(MSG_P, LINEBUF)               \
        do {                                            \
          char *tmp_ptr;                                \
          tmp_ptr = LINEBUF + CAS_LOG_MSG_INDEX;        \
          tmp_ptr = strchr(tmp_ptr, ' ');               \
          if (tmp_ptr == NULL)                          \
            MSG_P = "";                                 \
          else                                          \
            MSG_P = tmp_ptr + 1;                        \
        } while (0)

static int get_args (int argc, char *argv[]);
static int open_file (char *infilename, char *outfilename, FILE ** infp,
		      FILE ** outfp);
static int log_converter (FILE * infp, FILE * outfp);
static void close_file (FILE * infp, FILE * outfp);
static int is_cas_log_cmd (char *str);
static int log_bind_value (char *str, int lineno, FILE * outfp);
static char *get_execute_type (char *msg_p, int *prepare_flag,
			       int *execute_flag);

static char add_query_info = 0;
static char *infilename = NULL;

int
main (int argc, char *argv[])
{
  int start_arg;
  char *outfilename = NULL;
  FILE *outfp, *infp;
  int res;

  if ((start_arg = get_args (argc, argv)) < 0)
    return -1;

  infilename = argv[start_arg];
  if (start_arg + 1 <= argc)
    outfilename = argv[start_arg + 1];

  if (open_file (infilename, outfilename, &infp, &outfp) < 0)
    return -1;

  res = log_converter (infp, outfp);

  close_file (infp, outfp);
  return res;
}

static int
log_converter (FILE * infp, FILE * outfp)
{
  char *linebuf;
  char query_flag = 0;
  int lineno = 0;
  char *msg_p;
  T_STRING *linebuf_tstr = NULL;
  int execute_flag = 0;
  int prepare_flag = 0;
  char in_execute = 0;
  int exec_h_id = 0;

  linebuf_tstr = t_string_make (1000);
  if (linebuf_tstr == NULL)
    {
      fprintf (stderr, "malloc error\n");
      goto error;
    }

  while (1)
    {
      if (ut_get_line (infp, linebuf_tstr, NULL, NULL) < 0)
	{
	  fprintf (stderr, "malloc error\n");
	  goto error;
	}
      if (t_string_len (linebuf_tstr) <= 0)
	break;
      linebuf = t_string_str (linebuf_tstr);
      lineno++;

      if (linebuf[strlen (linebuf) - 1] == '\n')
	linebuf[strlen (linebuf) - 1] = '\0';

      if (is_cas_log_cmd (linebuf))
	{
	  if (query_flag)
	    {
	      fprintf (outfp, "\n");
	      fprintf (outfp, "P %d %d\n", exec_h_id, prepare_flag);
	    }
	  query_flag = 0;

	  GET_MSG_START_PTR (msg_p, linebuf);
	  if (strncmp (msg_p, "execute", 7) == 0)
	    {
	      msg_p = get_execute_type (msg_p, &prepare_flag, &execute_flag);
	      if (msg_p == NULL)
		{
		  in_execute = 0;
		  continue;
		}
	      if (strncmp (msg_p, "srv_h_id ", 9) == 0)
		{
		  char *endp;

		  in_execute = 1;
		  msg_p += 9;
		  exec_h_id = strtol (msg_p, &endp, 10);
		  if (msg_p == endp)
		    {
		      in_execute = 0;
		      continue;
		    }
		  msg_p = endp + 1;

		  fprintf (outfp, "Q ");
		  if (add_query_info == 1 && prepare_flag != 0x40)
		    fprintf (outfp, "/* %s */ ", infilename);
		  fprintf (outfp, "%s%c", msg_p, CAS_RUN_NEW_LINE_CHAR);
		  query_flag = 1;
		}
	      else
		{
		  if (in_execute == 1)
		    {
		      fprintf (outfp, "E %d %d\n", exec_h_id, execute_flag);
		      fprintf (outfp, "C %d\n", exec_h_id);
		      fprintf (outfp, "T\n");
		    }
		  in_execute = 0;
		}
	    }
	  else if (strncmp (msg_p, "bind ", 5) == 0)
	    {
	      if (log_bind_value (msg_p, lineno, outfp) < 0)
		goto error;
	    }
	}
      else if (query_flag)
	{
	  fprintf (outfp, "%s%c ", linebuf, CAS_RUN_NEW_LINE_CHAR);
	}
    }

  return 0;

error:
  return -1;
}

static int
open_file (char *infilename, char *outfilename, FILE ** infp, FILE ** outfp)
{
  if (strcmp (infilename, "-") == 0)
    {
      *infp = stdin;
    }
  else
    {
      *infp = fopen (infilename, "r");
      if (*infp == NULL)
	{
	  fprintf (stderr, "fopen error[%s]\n", infilename);
	  return -1;
	}
    }

  if (outfilename == NULL)
    {
      *outfp = stdout;
    }
  else
    {
      *outfp = fopen (outfilename, "w");
      if (*outfp == NULL)
	{
	  fprintf (stderr, "fopen error[%s]\n", outfilename);
	  return -1;
	}
    }

  return 0;
}

static void
close_file (FILE * infp, FILE * outfp)
{
  if (infp != stdin)
    fclose (infp);

  fflush (outfp);
  if (outfp != stdout)
    fclose (outfp);
}

static int
is_cas_log_cmd (char *str)
{
  if (strlen (str) < CAS_LOG_MSG_INDEX)
    return 0;

  if (str[2] == '/' && str[5] == ' ' && str[8] == ':' && str[11] == ':'
      && str[18] == ' ')
    return 1;
  return 0;
}

static char *
get_execute_type (char *msg_p, int *prepare_flag, int *execute_flag)
{
  if (strncmp (msg_p, "execute ", 8) == 0)
    {
      *prepare_flag = 0;
      *execute_flag = 0;
      return (msg_p += 8);
    }
  else if (strncmp (msg_p, "execute_call ", 13) == 0)
    {
      *prepare_flag = 0x40;	/* CCI_PREPARE_CALL */
      *execute_flag = 0;
      return (msg_p += 13);
    }
  else if (strncmp (msg_p, "execute_all ", 12) == 0)
    {
      *prepare_flag = 0;
      *execute_flag = 2;	/* CCI_EXEC_QUERY_ALL */
      return (msg_p += 12);
    }
  else
    {
      return NULL;
    }
}

static int
log_bind_value (char *str, int lineno, FILE * outfp)
{
  char *p, *q;
  char *value_p;
  int type;

  p = strchr (str, ':');
  if (p == NULL)
    {
      fprintf (stderr, "log error [line:%d]\n", lineno);
      return -1;
    }
  p += 2;
  q = strchr (p, ' ');
  if (q == NULL)
    {
      if (strcmp (p, "NULL") == 0)
	{
	  value_p = "";
	}
      else
	{
	  fprintf (stderr, "log error [line:%d]\n", lineno);
	  return -1;
	}
    }
  else
    {
      *q = '\0';
      value_p = q + 1;
    }

  if (strcmp (p, "NULL") == 0)
    type = CCI_U_TYPE_NULL;
  else if (strcmp (p, "CHAR") == 0)
    type = CCI_U_TYPE_CHAR;
  else if (strcmp (p, "VARCHAR") == 0)
    type = CCI_U_TYPE_STRING;
  else if (strcmp (p, "NCHAR") == 0)
    type = CCI_U_TYPE_NCHAR;
  else if (strcmp (p, "VARNCHAR") == 0)
    type = CCI_U_TYPE_VARNCHAR;
  else if (strcmp (p, "BIT") == 0)
    type = CCI_U_TYPE_BIT;
  else if (strcmp (p, "VARBIT") == 0)
    type = CCI_U_TYPE_VARBIT;
  else if (strcmp (p, "NUMERIC") == 0)
    type = CCI_U_TYPE_NUMERIC;
  else if (strcmp (p, "INT") == 0)
    type = CCI_U_TYPE_INT;
  else if (strcmp (p, "SHORT") == 0)
    type = CCI_U_TYPE_SHORT;
  else if (strcmp (p, "MONETARY") == 0)
    type = CCI_U_TYPE_MONETARY;
  else if (strcmp (p, "FLOAT") == 0)
    type = CCI_U_TYPE_FLOAT;
  else if (strcmp (p, "DOUBLE") == 0)
    type = CCI_U_TYPE_DOUBLE;
  else if (strcmp (p, "DATE") == 0)
    type = CCI_U_TYPE_DATE;
  else if (strcmp (p, "TIME") == 0)
    type = CCI_U_TYPE_TIME;
  else if (strcmp (p, "TIMESTAMP") == 0)
    type = CCI_U_TYPE_TIMESTAMP;
  else if (strcmp (p, "OBJECT") == 0)
    type = CCI_U_TYPE_OBJECT;
  else
    {
      fprintf (stderr, "log error [line:%d]\n", lineno);
      return -1;
    }

  fprintf (outfp, "B %d %s\n", type, value_p);
  return 0;
}

static int
get_args (int argc, char *argv[])
{
  int c;

  while ((c = getopt (argc, argv, "q")) != EOF)
    {
      switch (c)
	{
	case 'q':
	  add_query_info = 1;
	  break;
	default:
	  goto usage;
	}
    }

  if (optind + 1 >= argc)
    goto usage;

  return optind;

usage:
  fprintf (stderr, "%s infile outfile\n", argv[0]);
  return -1;
}