/*
 * Asterisk -- A telephony toolkit for Linux.
 *
 * MySQL CDR logger 
 * 
 * James Sharp <jsharp@psychoses.org>
 *
 * This program is free software, distributed under the terms of
 * the GNU General Public License.
 *
 */

#include <sys/types.h>
#include <asterisk/config.h>
#include <asterisk/options.h>
#include <asterisk/channel.h>
#include <asterisk/cdr.h>
#include <asterisk/module.h>
#include <asterisk/logger.h>
#include "../asterisk.h"

#include <stdio.h>
#include <string.h>

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

#include <mysql.h>

#define DATE_FORMAT "%Y-%m-%d %T"

static char *desc = "MySQL CDR Backend";
static char *name = "mysql";
static char *config = "cdr_mysql.conf";

static MYSQL *mysql;

static char * copy_date(char *buf, struct timeval tv, int len)
{
	struct tm st_tm;
	time_t t;
	t = tv.tv_sec;
	localtime_r(&t,&st_tm);
	strftime(buf, len, DATE_FORMAT, &st_tm);
	return buf;
}


static int mysql_log(struct ast_cdr *cdr)
{
  char *sqlcmd, tm_start[128], tm_answer[128], tm_end[128];
  int iMemSize = 2048;

  sqlcmd = (char *)malloc(iMemSize);

  if ( sqlcmd == NULL )
      return (-1);

  memset(sqlcmd,0,iMemSize);

  ast_log(LOG_DEBUG,"cdr_mysql: inserting a CDR record.\n");
  snprintf(sqlcmd,iMemSize,"INSERT INTO cdr (accountcode,src,dst,dcontext,clid,channel,dstchannel,lastapp,lastdata,starttime,answertime,endtime,duration,billsec,disposition,amaflags) VALUES ('%s','%s','%s','%s','%s','%s','%s','%s','%s','%s','%s','%s',%i,%i,'%s','%s')",
                  cdr->accountcode,cdr->src, cdr->dst, cdr->dcontext,cdr->clid,cdr->channel, cdr->dstchannel,
                  cdr->lastapp, cdr->lastdata,
                  copy_date(tm_start,cdr->start,sizeof(tm_start)),
                  copy_date(tm_answer,cdr->answer,sizeof(tm_answer)),
                  copy_date(tm_end,cdr->end,sizeof(tm_end)),
                  cdr->duration,cdr->billsec,ast_cdr_disp2str(cdr->disposition),ast_cdr_flags2str(cdr->amaflags));

  ast_log(LOG_DEBUG,"cdr_mysql: SQL command as follows:  %s\n",sqlcmd);
  printf("cdr_mysql: SQL command as follows:  %s\n",sqlcmd);
  
  if (mysql_real_query(mysql,sqlcmd,strlen(sqlcmd)))
  {
      ast_log(LOG_ERROR,"Failed to insert into database.  SQL = %s", sqlcmd);
      free(sqlcmd);
      return -1;
  }
  free(sqlcmd);
  return 0;
}



char *description(void)
{
  return desc;
}

int unload_module(void)
{ 

  mysql_close(mysql);
  ast_cdr_unregister(name);
  return 0;
}

int load_module(void)
{
  int res;
  struct ast_config *cfg;
  struct ast_variable *var;

  char *hostname, *dbname, *dbuser, *password;

  cfg = ast_load(config);
  if (!cfg) {
    ast_log(LOG_WARNING, "Unable to load config for mysql CDR's: %s\n", config);
    return 0;
  }
  
  var = ast_variable_browse(cfg, "global");
  if (!var) {
    /* nothing configured */
    return 0;
  }

  hostname = ast_variable_retrieve(cfg,"global","hostname");
  dbname = ast_variable_retrieve(cfg,"global","dbname");
  dbuser = ast_variable_retrieve(cfg,"global","user");
  password = ast_variable_retrieve(cfg,"global","password");
  ast_log(LOG_DEBUG,"cdr_mysql: got hostname of %s\n",hostname);
  ast_log(LOG_DEBUG,"cdr_mysql: got user of %s\n",dbuser);
  ast_log(LOG_DEBUG,"cdr_mysql: got dbname of %s\n",dbname);
  ast_log(LOG_DEBUG,"cdr_mysql: got password of %s\n",password);

  if (hostname == NULL)
    {
      ast_log(LOG_ERROR,"Database server hostname not specified.\n");
      return -1;
    }
  if (dbuser == NULL)
    {
      ast_log(LOG_ERROR,"Database dbuser not specified.\n");
      return -1;
    }
  if (dbname == NULL)
    {
      ast_log(LOG_ERROR,"Database dbname not specified.\n");
      return -1;
    }
  if (password == NULL)
    {
      ast_log(LOG_ERROR,"Database password not specified.\n");
      return -1;
    }


  mysql = mysql_init(NULL);

  mysql = mysql_real_connect(mysql, hostname, dbuser, password, dbname, 0, NULL, 0);

  if (mysql == NULL) {
    ast_log(LOG_ERROR, "Failed to connect to mysql database.\n");
    return -1;
  } else {
    ast_log(LOG_DEBUG,"Successfully connected to MySQL database.\n");
  }
  

  res = ast_cdr_register(name, desc, mysql_log);
  if (res) {
    ast_log(LOG_ERROR, "Unable to register MySQL CDR handling\n");
  }
  return res;
}

int reload(void)
{

  struct ast_config *cfg;
  struct ast_variable *var;

  char *hostname, *dbname, *password, *dbuser;
  
  mysql_close(mysql);


  cfg = ast_load(config);
  if (!cfg) {
    ast_log(LOG_WARNING, "Unable to load MySQL CDR config %s\n", config);
    return 0;
  }
  
  var = ast_variable_browse(cfg, "global");
  if (!var) {
    /* nothing configured */
    return 0;
  }

  hostname = ast_variable_retrieve(cfg,"global","hostname");
  dbname = ast_variable_retrieve(cfg,"global","dbname");
  dbuser = ast_variable_retrieve(cfg,"global","user");
  password = ast_variable_retrieve(cfg,"global","password");
  ast_log(LOG_DEBUG,"cdr_mysql: got hostname of %s\n",hostname);
  ast_log(LOG_DEBUG,"cdr_mysql: got dbname of %s\n",dbname);
  ast_log(LOG_DEBUG,"cdr_mysql: got dbuser of %s\n",dbuser);
  ast_log(LOG_DEBUG,"cdr_mysql: got password of %s\n",password);

  if (hostname == NULL)
    {
      ast_log(LOG_ERROR,"Database server hostname not specified.\n");
      return -1;
    }
  if (dbname == NULL)
    {
      ast_log(LOG_ERROR,"Database dbname not specified.\n");
      return -1;
    }
  if (dbuser == NULL)
    {
      ast_log(LOG_ERROR,"Database dbuser not specified.\n");
      return -1;
    }
  
  if (password == NULL)
    {
      ast_log(LOG_ERROR,"Database password not specified.\n");
      return -1;
    }
  
 mysql = mysql_init(NULL);
 
 mysql = mysql_real_connect(mysql, hostname, dbuser, password, dbname, 0, NULL, 0);
 
 if (mysql == NULL) {
   ast_log(LOG_ERROR, "Failed to connect to mysql database.\n");
   return -1;
 } else {
   ast_log(LOG_DEBUG,"Successfully connected to MySQL database.\n");
 }
 



  return 0;
}

int usecount(void)
{
  return 0;
}

char *key()
{
  return ASTERISK_GPL_KEY;
}