/*

  * $Id$

  *
  * Copyright (C) 2002-2003 Fhg Fokus

  * Copyright (C) 2008 Juha Heinanen (USE_MYSQL parts)

  *
  * This file is part of sems, a free SIP media server.
  *
  * sems 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.
  *
  * For a license to use the sems software under conditions
  * other than those described here, or to purchase support for this
  * software, please contact iptel.org by e-mail at the following addresses:
  *    info@iptel.org
  *
  * sems 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
  */
 

 #include "EarlyAnnounce.h"

 #include "AmConfig.h"
 #include "AmUtils.h"
 
 #include "sems.h"
 #include "log.h"
 
 #define MOD_NAME "early_announce"
 

 #ifdef USE_MYSQL
 #include <mysql++/mysql++.h>
 #include <stdio.h>
 #define DEFAULT_AUDIO_TABLE "default_audio"
 #define DOMAIN_AUDIO_TABLE "domain_audio"
 #define USER_AUDIO_TABLE "user_audio"
 #endif
 

 EXPORT_SESSION_FACTORY(EarlyAnnounceFactory,MOD_NAME);

 #ifdef USE_MYSQL
 string EarlyAnnounceFactory::AnnounceApplication;
 string EarlyAnnounceFactory::AnnounceMessage;
 string EarlyAnnounceFactory::DefaultLanguage;
 #else

 string EarlyAnnounceFactory::AnnouncePath;
 string EarlyAnnounceFactory::AnnounceFile;

 #endif

 EarlyAnnounceFactory::ContB2B EarlyAnnounceFactory::ContinueB2B = 
   EarlyAnnounceFactory::Never;

 EarlyAnnounceFactory::EarlyAnnounceFactory(const string& _app_name)

   : AmSessionFactory(_app_name)
 {
 }
 

 #ifdef USE_MYSQL
 mysqlpp::Connection EarlyAnnounceFactory::Connection(mysqlpp::use_exceptions);
 
 int get_announce_msg(string application, string message, string user,
 		     string domain, string language, string *audio_file)
 {
     string query_string;
 
     if (!user.empty()) {
 	*audio_file = string("/tmp/") +  application + "_" + 
 	    message + "_" + domain + "_" + user + ".wav";
 	query_string = "select audio from " + string(USER_AUDIO_TABLE) +
 	    " where application='" + application + "' and message='" +
 	    message + "' and userid='" + user + "' and domain='" +
 	    domain + "'";
     } else if (!domain.empty()) {
 	*audio_file = string("/tmp/") +  application + "_" +
 	    message + "_" + domain + "_" + language + ".wav";
 	query_string = "select audio from " + string(DOMAIN_AUDIO_TABLE) +
 	    " where application='" + application + "' and message='" +
 	    message + "' and domain='" + domain + "' and language='" +
 	    language + "'";
     } else {
 	*audio_file = string("/tmp/") +  application  + "_" +
 	    message + "_" + language + ".wav";
 	query_string = "select audio from " + string(DEFAULT_AUDIO_TABLE) +
 	    " where application='" + application + "' and message='" +
 	    message + "' and language='" + language + "'";
     }
 
     try {
 
 	mysqlpp::Query query = EarlyAnnounceFactory::Connection.query();
 	    
 	DBG("Query string <%s>\n", query_string.c_str());
 
 	query << query_string;

 
 #ifdef VERSION2

 	mysqlpp::Result res = query.store();

 #else
     	mysqlpp::StoreQueryResult res = query.store();
 #endif

 
 	mysqlpp::Row row;
 
 	if (res) {
 	    if ((res.num_rows() > 0) && (row = res.at(0))) {
 		FILE *file;
 		file = fopen((*audio_file).c_str(), "wb");

 #ifdef VERSION2
 		unsigned long length = row.raw_string(0).size();

 		fwrite(row.at(0).data(), 1, length, file);

 #else
 		mysqlpp::String s = row[0];
 		fwrite(s.data(), 1, s.length(), file);
 #endif

 		fclose(file);
 		return 1;
 	    } else {
 		*audio_file = "";
 		return 1;
 	    }
 	} else {
 	    ERROR("Database query error\n");
 	    *audio_file = "";
 	    return 0;
 	}
     }
 
     catch (const mysqlpp::Exception& er) {
 	// Catch-all for any MySQL++ exceptions
 	ERROR("MySQL++ error: %s\n", er.what());
 	*audio_file = "";
 	return 0;
     }
 }
 
 #endif
 

 int EarlyAnnounceFactory::onLoad()

 {

   AmConfigReader cfg;
   if(cfg.loadFile(AmConfig::ModConfigPath + string(MOD_NAME ".conf")))
     return -1;

   // get application specific global parameters
   configureModule(cfg);

   if (cfg.hasParameter("continue_b2b")) { 
     if (cfg.getParameter("continue_b2b") == "yes") {
       ContinueB2B = Always;
       DBG("early_announce in b2bua mode.\n");
     }
     else if (cfg.getParameter("continue_b2b") == "app-param") {
       ContinueB2B = AppParam;
       DBG("early_announce in b2bua/final reply mode "
 	  "(depends on app-param).\n");
     } else {
       DBG("early_announce sends final reply.\n");
     }

   }
 

 #ifdef USE_MYSQL
 
   /* Get default audio from MySQL */
 
   string mysql_server, mysql_user, mysql_passwd, mysql_db;
 
   mysql_server = cfg.getParameter("mysql_server");
   if (mysql_server.empty()) {
     mysql_server = "localhost";
   }
 
   mysql_user = cfg.getParameter("mysql_user");
   if (mysql_user.empty()) {

     ERROR("conference.conf parameter 'mysql_user' is missing.\n");

     return -1;
   }
 
   mysql_passwd = cfg.getParameter("mysql_passwd");
   if (mysql_passwd.empty()) {

     ERROR("conference.conf parameter 'mysql_passwd' is missing.\n");

     return -1;
   }
 
   mysql_db = cfg.getParameter("mysql_db");
   if (mysql_db.empty()) {
     mysql_db = "sems";
   }
 
   AnnounceApplication = cfg.getParameter("application");
   if (AnnounceApplication.empty()) {
     AnnounceApplication = MOD_NAME;
   }
 
   AnnounceMessage = cfg.getParameter("message");
   if (AnnounceMessage.empty()) {
     AnnounceMessage = "greeting_msg";
   }
 
   DefaultLanguage = cfg.getParameter("default_language");
   if (DefaultLanguage.empty()) {
     DefaultLanguage = "en";
   }
 
   try {
 

 #ifdef VERSION2

     Connection.set_option(Connection.opt_reconnect, true);

 #else
     Connection.set_option(new mysqlpp::ReconnectOption(true));
 #endif

     Connection.connect(mysql_db.c_str(), mysql_server.c_str(),
 		       mysql_user.c_str(), mysql_passwd.c_str());
     if (!Connection) {
       ERROR("Database connection failed: %s\n", Connection.error());
       return -1;
     }
   }

 
   catch (const mysqlpp::BadOption& er) {
     ERROR("MySQL++ set_option error: %s\n", er.what());
     return -1;
   }
  	

   catch (const mysqlpp::Exception& er) {
     // Catch-all for any MySQL++ exceptions
     ERROR("MySQL++ error: %s\n", er.what());
     return -1;
   }
 
   string announce_file;
   if (!get_announce_msg(AnnounceApplication, AnnounceMessage, "", "",
 			DefaultLanguage, &announce_file)) {
     return -1;
   }
   if (announce_file.empty()) {
     ERROR("default announce for " MOD_NAME " module does not exist.\n");
     return -1;
   }
 
 #else 
 
   /* Get default audio from file system */
 

   AnnouncePath = cfg.getParameter("announce_path",ANNOUNCE_PATH);
   if( !AnnouncePath.empty() 
       && AnnouncePath[AnnouncePath.length()-1] != '/' )
     AnnouncePath += "/";

   AnnounceFile = cfg.getParameter("default_announce",ANNOUNCE_FILE);

   string announce_file = AnnouncePath + AnnounceFile;
   if(!file_exists(announce_file)){
     ERROR("default file for " MOD_NAME " module does not exist ('%s').\n",
 	  announce_file.c_str());
     return -1;
   }

 #endif
 

   return 0;

 }
 
 

 void EarlyAnnounceDialog::onInvite(const AmSipRequest& req) 

 {

   try {

     string sdp_reply;
     acceptAudio(req.body,req.hdrs,&sdp_reply);

     if(dlg.reply(req,183,"Session Progress",
 		 "application/sdp",sdp_reply) != 0){

       throw AmSession::Exception(500,"could not reply");
     }

     else {	    
       invite_req = req;

     }

   } catch(const AmSession::Exception& e) {

     ERROR("%i %s\n",e.code,e.reason.c_str());
     setStopped();
     AmSipDialog::reply_error(req,e.code,e.reason);
   }

 }
 
 

 AmSession* EarlyAnnounceFactory::onInvite(const AmSipRequest& req)

 {

 
 #ifdef USE_MYSQL
 
     string iptel_app_param = getHeader(req.hdrs, PARAM_HDR);
     string language = get_header_keyvalue(iptel_app_param,"Language");
     string announce_file = "";
 
     if (language.empty()) language = DefaultLanguage;
 
     get_announce_msg(AnnounceApplication, AnnounceMessage, req.user,
 		     req.domain, "", &announce_file);
     if (!announce_file.empty()) goto end;
     get_announce_msg(AnnounceApplication, AnnounceMessage, "", req.domain,
 		     language, &announce_file);
     if (!announce_file.empty()) goto end;
     get_announce_msg(AnnounceApplication, AnnounceMessage, "", "", language,
 		     &announce_file);
 
 #else
 

   string announce_path = AnnouncePath;
   string announce_file = announce_path + req.domain 

     + "/" + req.user + ".wav";

   DBG("trying '%s'\n",announce_file.c_str());
   if(file_exists(announce_file))
     goto end;

   announce_file = announce_path + req.user + ".wav";
   DBG("trying '%s'\n",announce_file.c_str());
   if(file_exists(announce_file))
     goto end;

   announce_file = AnnouncePath + AnnounceFile;

 
 #endif
 

  end:
   return new EarlyAnnounceDialog(announce_file);

 }
 

 EarlyAnnounceDialog::EarlyAnnounceDialog(const string& filename)

   : filename(filename)

 {

   set_sip_relay_only(false);

 }
 

 EarlyAnnounceDialog::~EarlyAnnounceDialog()

 {
 }
 

 void EarlyAnnounceDialog::onSessionStart(const AmSipRequest& req)

 {

   // we can drop all received packets
   // this disables DTMF detection as well
   setReceiving(false);

   DBG("EarlyAnnounceDialog::onSessionStart\n");
   if(wav_file.open(filename,AmAudioFile::Read))
     throw string("EarlyAnnounceDialog::onSessionStart: Cannot open file\n");

   setOutput(&wav_file);

 }
 

 void EarlyAnnounceDialog::onBye(const AmSipRequest& req)

 {

   DBG("onBye: stopSession\n");
   setStopped();

 }
 

 void EarlyAnnounceDialog::onCancel()

 {

   dlg.reply(invite_req,487,"Call terminated");

   setStopped();

 }
 

 void EarlyAnnounceDialog::process(AmEvent* event)

 {
 

   AmAudioEvent* audio_event = dynamic_cast<AmAudioEvent*>(event);
   if(audio_event && 

      (audio_event->event_id == AmAudioEvent::cleared)) {

       DBG("AmAudioEvent::cleared\n");

       bool continue_b2b = false;
       if (EarlyAnnounceFactory::ContinueB2B == 
 	  EarlyAnnounceFactory::Always) {
 	continue_b2b = true;
       } else if (EarlyAnnounceFactory::ContinueB2B == 
 		 EarlyAnnounceFactory::AppParam) {

 	string iptel_app_param = getHeader(invite_req.hdrs, PARAM_HDR);

 	if (iptel_app_param.length()) {
 	  continue_b2b = get_header_keyvalue(iptel_app_param,"B2B")=="yes";
 	} else {

 	  continue_b2b = getHeader(invite_req.hdrs,"P-B2B")=="yes";

 	}
       }
       DBG("determined: continue_b2b = %s\n", continue_b2b?"true":"false");
 
       if (!continue_b2b) {

 	unsigned int code_i = 404;
 	string reason = "Not Found";
 	

 	string iptel_app_param = getHeader(invite_req.hdrs, PARAM_HDR);

 	if (iptel_app_param.length()) {
 	  string code = get_header_keyvalue(iptel_app_param,"Final-Reply-Code");
 	  if (code.length() && str2i(code, code_i)) {
 	    ERROR("while parsing Final-Reply-Code parameter\n");
 	  }
 	  reason = get_header_keyvalue(iptel_app_param,"Final-Reply-Reason");

 	  if (!reason.length())
 	    reason = "Not Found";

 	} else {

 	  string code = getHeader(invite_req.hdrs,"P-Final-Reply-Code");

 	  if (code.length() && str2i(code, code_i)) {
 	    ERROR("while parsing P-Final-Reply-Code\n");
 	  }

 	  string h_reason =  getHeader(invite_req.hdrs,"P-Final-Reply-Reason");

 	  if (h_reason.length()) {
 	    INFO("Use of P-Final-Reply-Code/P-Final-Reply-Reason is deprecated. ");
 	    INFO("Use '%s: Final-Reply-Code=<code>;"
 		 "Final-Reply-Reason=<rs>' instead.\n",PARAM_HDR);
 	    reason = h_reason;
 	  }

 	}
 

 	DBG("Replying with code %d %s\n", code_i, reason.c_str());

 	dlg.reply(invite_req, code_i, reason);

 	setStopped();
       } else {
 	set_sip_relay_only(true);

 	recvd_req.insert(std::make_pair(invite_req.cseq,invite_req));

 	relayEvent(new B2BSipRequestEvent(invite_req,true));

       }

       return;

     }
 

   AmB2BCallerSession::process(event);

 }