// ww1ipump.cpp
//
// Replicator implementation for the Westwood One iPump Replicator
//
//   (C) Copyright 2010-2023 Fred Gleason <fredg@paravelsystems.com>
//
//   This program is free software; you can redistribute it and/or modify
//   it under the terms of the GNU General Public License version 2 as
//   published by the Free Software Foundation.
//
//   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., 675 Mass Ave, Cambridge, MA 02139, USA.
//

#include <unistd.h>
#include <stdlib.h>
#include <stdio.h>
#include <syslog.h>
#include <errno.h>

#include <rdapplication.h>
#include <rdaudioconvert.h>
#include <rdcart.h>
#include <rdcut.h>
#include <rdconf.h>
#include <rddb.h>
#include <rddelete.h>
#include <rdescape_string.h>
#include <rdstringlist.h>
#include <rdtempdirectory.h>
#include <rdupload.h>

#include "ww1ipump.h"

#define RD_MAX_CART_NUMBER 999999

Ww1Ipump::Ww1Ipump(ReplConfig *repl_config)
  : ReplFactory(repl_config)
{
  QString sql;
  RDSqlQuery *q;

  sql="select `LAST_ISCI_XREFERENCE` from `VERSION`";
  q=new RDSqlQuery(sql);
  if(q->first()) {
    xds_isci_datetime=q->value(0).toDateTime();
  }
  delete q;
}


void Ww1Ipump::startProcess()
{
  CheckIsciXreference();
  CheckCarts();
}


bool Ww1Ipump::processCart(const unsigned cartnum)
{
  QString sql;
  RDSqlQuery *q;
  bool ret=false;

  sql=QString::asprintf("select `FILENAME` from `ISCI_XREFERENCE` \
                         where (`CART_NUMBER`=%u)&&(`LATEST_DATE`>=now())&&\
                         ((`TYPE`='R')||(`TYPE`='B'))",cartnum);
  q=new RDSqlQuery(sql);
  if(q->first()) {
    ret=PostCut(RDCut::cutName(cartnum,1),q->value(0).toString());
  }
  delete q;
  return ret;
}


void Ww1Ipump::CheckIsciXreference()
{
  QString sql;

  QFileInfo *fi=new QFileInfo(rda->system()->isciXreferencePath());
  if(fi->exists()) {
    if(fi->lastModified()>xds_isci_datetime) {
      if(LoadIsciXreference(rda->system()->isciXreferencePath())) {
	sql="update `VERSION` set `LAST_ISCI_XREFERENCE`=now()";
	RDSqlQuery::apply(sql);
	xds_isci_datetime=QDateTime(QDate::currentDate(),QTime::currentTime());
      }
    }
  }
  else {
    rda->syslog(LOG_WARNING,"unable to load ISCI cross reference file \"%s\"",
		(const char *)rda->system()->isciXreferencePath().toUtf8());
  }
  delete fi;
}


bool Ww1Ipump::LoadIsciXreference(const QString &filename)
{
  FILE *f=NULL;
  char line[1024];
  QString sql;
  RDSqlQuery *q;
  RDStringList fields;
  unsigned cartnum;
  QStringList datelist;
  QDate date;
  bool ok=false;
  unsigned linenum=3;

  if((f=fopen(filename.toUtf8(),"r"))==NULL) {
    rda->syslog(LOG_WARNING,
		"unable to load ISCI cross reference file \"%s\" [%s]",
		(const char *)rda->system()->isciXreferencePath().toUtf8(),
		strerror(errno));
    return false;
  }

  //
  // Purge Old Data
  //
  sql="delete from `ISCI_XREFERENCE`";
  q=new RDSqlQuery(sql);
  delete q;

  //
  // Skip Header
  //
  if(fgets(line,1024,f)==NULL) {
    rda->syslog(LOG_WARNING,"fgets() error reading ISCI xreference data");
  }
  if(fgets(line,1024,f)==NULL) {
    rda->syslog(LOG_WARNING,"fgets() error reading ISCI xreference data");
  }

  //
  // Load Records
  //
  while(fgets(line,1024,f)!=NULL) {
    fields=fields.split(',',line,"\"");
    if(fields.size()==9) {
      for(int i=0;i<fields.size();i++) {
	fields[i]=fields[i].replace("\"","").trimmed();
      }
      cartnum=fields[3].right(fields[3].length()-1).toUInt(&ok);
      if(ok&&(cartnum<=RD_MAX_CART_NUMBER)) {
	datelist=fields[6].split("/");
	if(datelist.size()==3) {
	  date.setDate(datelist[2].toInt()+2000,datelist[0].toInt(),
		       datelist[1].toInt());
	  if(ValidateFilename(fields[8])) {
	    if(date.isValid()) {
	      sql=QString("insert into `ISCI_XREFERENCE` set ")+
	      "`CART_NUMBER`="+QString::asprintf("%u",cartnum)+","+
	      "`ISCI`='"+RDEscapeString(fields[4])+"',"+
	      "`FILENAME`='"+RDEscapeString(fields[8])+"',"+
	      "`LATEST_DATE`='"+date.toString("yyyy/MM/dd")+"',"+
	      "`TYPE`='"+RDEscapeString(fields[0])+"',"+
	      "`ADVERTISER_NAME`='"+RDEscapeString(fields[1])+"',"+
	      "`PRODUCT_NAME`='"+RDEscapeString(fields[2])+"',"+
	      "`CREATIVE_TITLE`='"+RDEscapeString(fields[5])+"',"+
	      "`REGION_NAME`='"+RDEscapeString(fields[7])+"'";
	      RDSqlQuery::apply(sql);
	    }
	    else {
	      rda->syslog(LOG_WARNING,"invalid date in line %d of \"%s\"",
			  linenum,(const char *)filename.toUtf8());
	    }
	  }
	  else {
	    rda->syslog(LOG_WARNING,
			"invalid FILENAME field \"%s\" in line %d of \"%s\"",
			(const char *)fields[8].toUtf8(),linenum,
			(const char *)filename.toUtf8());
	  }
	}
	else {
	  rda->syslog(LOG_WARNING,"invalid date in line %d of \"%s\"",
		      linenum,(const char *)filename.toUtf8());
	}
      }
      else {
	rda->syslog(LOG_DEBUG,
		    "missing/invalid cart number in line %d of \"%s\"",
		    linenum,(const char *)filename.toUtf8());
      }
    }
    else {
      rda->syslog(LOG_WARNING,"line %d malformed in \"%s\"",
		  linenum,(const char *)filename.toUtf8());
    }
    linenum++;
  }

  //
  // Clean Up
  //
  rda->syslog(LOG_INFO,"loaded ISCI cross reference file \"%s\"",
	      (const char *)rda->system()->isciXreferencePath().toUtf8());
  fclose(f);
  return true;
}


bool Ww1Ipump::ValidateFilename(const QString &filename)
{
  bool ret=true;

  //
  // List of illegal characters taken from 'Illegal Characters4.doc'
  // from Citadel
  //
  ret=ret&&(filename.indexOf(" ")<0);
  ret=ret&&(filename.indexOf("\"")<0);
  ret=ret&&(filename.indexOf("%")<0);
  ret=ret&&(filename.indexOf("*")<0);
  ret=ret&&(filename.indexOf("+")<0);
  ret=ret&&(filename.indexOf("/")<0);
  ret=ret&&(filename.indexOf(":")<0);
  ret=ret&&(filename.indexOf(";")<0);
  ret=ret&&(filename.indexOf("<")<0);
  ret=ret&&(filename.indexOf("=")<0);
  ret=ret&&(filename.indexOf(">")<0);
  ret=ret&&(filename.indexOf("?")<0);
  ret=ret&&(filename.indexOf("@")<0);
  ret=ret&&(filename.indexOf("[")<0);
  ret=ret&&(filename.indexOf("\\")<0);
  ret=ret&&(filename.indexOf("]")<0);
  ret=ret&&(filename.indexOf("^")<0);
  ret=ret&&(filename.indexOf("{")<0);
  ret=ret&&(filename.indexOf("|")<0);
  ret=ret&&(filename.indexOf("}")<0);

  return ret;
}


void Ww1Ipump::CheckCarts()
{
  QString sql;
  RDSqlQuery *q;
  RDSqlQuery *q1;
  RDSqlQuery *q2;
  QString now=QDateTime(QDate::currentDate(),QTime::currentTime()).addDays(-6).
    toString("yyyy-MM-dd hh:mm:ss");

  //
  // Generate Update List
  //
  sql=QString("select ")+
    "`CART_NUMBER`,"+  // 00
    "`FILENAME` "+     // 01
    "from `ISCI_XREFERENCE` where "+
    "(`LATEST_DATE`>=now())&&((`TYPE`='R')||(`TYPE`='B'))";
  q=new RDSqlQuery(sql);
  while(q->next()) {
    sql=QString("select `REPL_CART_STATE`.`ID` from ")+
      "`REPL_CART_STATE` left join `CUTS` "+
      "on `REPL_CART_STATE`.`CART_NUMBER`=`CUTS`.`CART_NUMBER` where "+
      "(`CUTS`.`ORIGIN_DATETIME`<`REPL_CART_STATE`.`ITEM_DATETIME`)&&"+
      "(`REPL_CART_STATE`.`REPLICATOR_NAME`='"+
      RDEscapeString(config()->name())+"')&&"+
      QString::asprintf("(`REPL_CART_STATE`.`CART_NUMBER`=%u)&&",
			q->value(0).toUInt())+
      "(`REPL_CART_STATE`.`POSTED_FILENAME`='"+
      RDEscapeString(q->value(1).toString())+"')&&"+
      "(`REPL_CART_STATE`.`ITEM_DATETIME`>'"+RDEscapeString(now)+"')&&"+
      "(`REPL_CART_STATE`.`REPOST`='N')";
    q1=new RDSqlQuery(sql);
    if(!q1->first()) {
      if(PostCut(RDCut::cutName(q->value(0).toUInt(),1),
		 q->value(1).toString())) {
	sql=QString("select `ID` from `REPL_CART_STATE` where ")+
	  "(`REPLICATOR_NAME`='"+RDEscapeString(config()->name())+"')&&"+
	  QString::asprintf("(`CART_NUMBER`=%u)&&",q->value(0).toUInt())+
	  "(`POSTED_FILENAME`='"+RDEscapeString(q->value(1).toString())+"')";
	q2=new RDSqlQuery(sql);
	if(q2->first()) {
	  sql=QString("update `REPL_CART_STATE` set ")+
	    "`ITEM_DATETIME`=now(),"+
	    "`REPOST`='N' where "+
	    "(`REPLICATOR_NAME`='"+RDEscapeString(config()->name())+"')&&"+
	    QString::asprintf("(`CART_NUMBER`=%u)&&",q->value(0).toUInt())+
	    "(`POSTED_FILENAME`='"+RDEscapeString(q->value(1).toString())+"')";
	}
	else {
	  sql=QString("insert into `REPL_CART_STATE` set ")+
	    "`ITEM_DATETIME`=now(),"+
	    "`REPOST`='N',"+
	    "`REPLICATOR_NAME`='"+RDEscapeString(config()->name())+"',"+
	    QString::asprintf("`CART_NUMBER`=%u,",q->value(0).toUInt())+
	    "`POSTED_FILENAME`='"+RDEscapeString(q->value(1).toString())+"'";
	}
	delete q2;
	RDSqlQuery::apply(sql);
      }
    }
    delete q1;
  }
  delete q;
}


bool Ww1Ipump::PostCut(const QString &cutname,const QString &filename)
{
  //
  // Export File
  //
  RDAudioConvert::ErrorCode conv_err;
  RDUpload::ErrorCode upload_err;
  float speed_ratio=1.0;
  RDCut *cut=new RDCut(cutname);
  if(!cut->exists()) {
    delete cut;
    return false;
  }
  if(cut->length()==0) {
    delete cut;
    return true;
  }
  RDCart *cart=new RDCart(cut->cartNumber());
  if(cart->enforceLength()) {
    speed_ratio=(float)cut->length()/(float)cart->forcedLength();
  }
  RDSettings *settings=new RDSettings();
  QString tempfile=RDTempDirectory::basePath()+"/"+filename;
  RDAudioConvert *conv=new RDAudioConvert();
  conv->setSourceFile(RDCut::pathName(cutname));
  conv->setDestinationFile(tempfile);
  conv->setRange(cut->startPoint(),cut->endPoint());
  conv->setSpeedRatio(speed_ratio);
  settings->setFormat(config()->format());
  settings->setChannels(config()->channels());
  settings->setSampleRate(config()->sampleRate());
  settings->setBitRate(config()->bitRate());
  settings->setQuality(config()->quality());
  settings->setNormalizationLevel(config()->normalizeLevel()/1000);
  conv->setDestinationSettings(settings);
  delete cart;
  delete cut;
  switch(conv_err=conv->convert()) {
  case RDAudioConvert::ErrorOk:
    break;

  default:
    rda->syslog(LOG_WARNING,
		"Ww1Ipump: audio conversion failed: %s, cutname: %s",
		(const char *)RDAudioConvert::errorText(conv_err).toUtf8(),
		(const char *)cutname.toUtf8());
    delete conv;
    delete settings;
    return false;
  }
  delete conv;
  delete settings;

  //
  // Upload File
  //
  QDate monday=RDReplicator::roundDownToDow(QDate::currentDate(),1);
  QString err_msg;
  RDUpload *upload=new RDUpload(rda->config());
  upload->setSourceFile(tempfile);
  QString dest_url=config()->url();
  if(dest_url.right(1)!="/") {
    dest_url+="/";
  }
  dest_url+="Spots/"+
    QString::asprintf("%d/WEEK OF ",monday.year())+
    monday.toString("MMddyy")+"/"+filename;
  upload->setDestinationUrl(dest_url);
  upload->createDestinationDirs(true);

  //
  // FIXME: Finish implementing ssh(1) id keys!
  //
  switch(upload_err=upload->runUpload(config()->urlUsername(),
				      config()->urlPassword(),"",false,&err_msg,
				      rda->config()->logXloadDebugData())) {
  case RDUpload::ErrorOk:
    break;

  default:
    rda->syslog(LOG_WARNING,"Ww1Ipump: audio upload failed: %s",
		err_msg.toUtf8().constData());
    unlink(tempfile.toUtf8());
    delete upload;
    return false;
  }
  unlink(tempfile.toUtf8());
  delete upload;
  rda->syslog(LOG_INFO,"Ww1Ipump: uploaded cut %s to %s",
	      cutname.toUtf8().constData(),
	      dest_url.toUtf8().constData());

  return true;
}