main.cpp

/*
    Copyright (C) 2014 Luis M. Santos
    Contact: luismigue1234@hotmail.com

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU Lesser General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    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 Lesser General Public License for more details.

    You should have received a copy of the GNU Lesser General Public License
    along with This program.  If not, see <http://www.gnu.org/licenses/>.
*/
#include <iostream>
#include <fstream>
#include "html2faxcover.h"
#include "data_base.h"

enum Modes
{
    STAND_ALONE, CALL_HYLAFAX_FAXCOVER, CALL_HYLAFAX_FAXCOVER_AS_IS, HASH_TEST
};


//Prototypes
void getNameAndEmail(MySQL& db, std::string criteria, std::string& name, std::string& email);
std::string generateCommand(const h2fax::faxcover_args& args);

int main(int argc, char* argv[])
{
    //Check if the internal API is the right version
    if(h2fax::PROG_VERSION < 1.1f)
        return EXIT_FAILURE;

    //Let's make sure we have our tmp dir
    //h2fax::exec_cmd(tmp.c_str(), "mkdir -p ", "");
    //Let's redirect cerr
    std::ofstream cerrLog("/tmp/cerrlog");
    std::streambuf *cerr_copy = std::cerr.rdbuf();
    if(cerrLog.is_open())
       std::cerr.rdbuf(cerrLog.rdbuf());

    //Let's redirect cout
    std::ofstream coutLog("/tmp/coutlog");
    std::streambuf *cout_copy = std::cout.rdbuf();
    if(coutLog.is_open())
       std::cout.rdbuf(coutLog.rdbuf());

    //Grab command line arguments in a nicely packaged structure!
    h2fax::faxcover_args args = h2fax::getFaxcoverParameters(argc, argv);
    h2fax::uint execMode;
    std::string root_prog_path = std::string(args.appName).substr(0, std::string(args.appName).rfind("/")) + PATH_SLASH;
    //data_base settings((root_prog_path + "settings").c_str());//Works when php scripts call the prog directly but not when the script calls a program that calls this program. Use the method below!
    //Make sure you have an etc/html2faxcover configuration directory that has the apache owner and group permissions!
    data_base settings("/etc/html2faxcover/settings");
    std::string name, email;
    std::string prefix = settings.GetStrFromData("prefix");
    std::string html2ps_path = settings.GetStrFromData("html2ps");
    std::string tmp = settings.GetStrFromData("tmp");
    std::string hylafax = settings.GetStrFromData("hylafax");
    std::string lock = settings.GetStrFromData("lock_file");

    //Let's only execute the rest of the program if we have all of the settings
    if(args.argNum > 1)
    {
     execMode = cStrToInt(args.execMode);

    //Let's prep the file path for later usage
    std::string path = args.template_path;
    path = replaceCharInStr(path, '\\', '/');
    std::string root_path = path.substr(0, path.rfind("/")) + PATH_SLASH; 
    std::string file = path.substr(root_path.size());

    //Setup date
    args.date = h2fax::getDate();
    //Let's check we have the information for connecting to a database
    if(!args.database || !args.host || !args.username || !args.password)
   {
        //Let's store the database connection info
        args.Database = settings.GetStrFromData("database").c_str();
        args.Host = settings.GetStrFromData("host").c_str();
        args.Username = settings.GetStrFromData("username").c_str();
        args.Password = settings.GetStrFromData("password").c_str();
        //Let's save the c_string pointers in the regular spot for arguments
        args.database = args.Database.c_str();
        args.host = args.Host.c_str();
        args.username = args.Username.c_str();
        args.password = args.Password.c_str();
   }

    /*This is more for debugging purposes. It is a way to force the program to get the db
    connection info from the settings*/
   if(settings.GetIntFromData("ignore_db_args"))
   {
       //Let's store the database connection info
        args.Database = settings.GetStrFromData("database").c_str();
        args.Host = settings.GetStrFromData("host").c_str();
        args.Username = settings.GetStrFromData("username").c_str();
        args.Password = settings.GetStrFromData("password").c_str();
        //Let's save the c_string pointers in the regular spot for arguments
        args.database = args.Database.c_str();
        args.host = args.Host.c_str();
        args.username = args.Username.c_str();
        args.password = args.Password.c_str();
   }

    //Now, let's create a database connection!
    MySQL db(args.database, args.username, args.password, args.host);
    //MySQL db("avantfax\0","avantfax\0","d58fe49\0","localhost\0");//Hardcoded test version
    if(db.getStatus())
        return EXIT_FAILURE;

    /*Let's lock the tmp directory*/
    while(h2fax::LockFile(lock.c_str()) != 0){}
    //Let's clean our temporary location!
    h2fax::removeDir(tmp.c_str());

    //Restore cout
    if(coutLog.is_open())
      std::cout.rdbuf(cout_copy);

    switch(execMode)
    {
    case STAND_ALONE:
        {
            //Get name and address from the database
            getNameAndEmail(db, args.from, name, email);
            //Update args
            args.from = name.c_str();
            args.email = email.c_str();
            /*Copy folder containing coversheet and supplemental materials like
            images to a temporary location*/
            h2fax::copyDir(root_path.c_str(), tmp.c_str());
            //Take away the AvantFax prefix and substitute the markers like to-company for their actual value
            h2fax::removeAvantFaxPrefix((tmp + file).c_str(), prefix);
            h2fax::substituteMarkers((tmp + file).c_str(), args);
            //Convert file to PS for HylaFax!
            h2fax::convertToPS((tmp + file).c_str(), html2ps_path, "");
            /*By now, HylaFax should be receiving the processed html template via
            stdin or the pipe. I am not sure how it reads the output since I have
            not looked at the code for the program.*/
            //Let's clean our temporary location!
            h2fax::removeDir(tmp.c_str());

            break;
        }
    case CALL_HYLAFAX_FAXCOVER:
        {
            //Get name and address from the database
            getNameAndEmail(db, args.from, name, email);
            //Update args
            args.from = name.c_str();
            args.email = email.c_str();
            /*Copy folder containing coversheet and supplemental materials like
            images to a temporary location*/
            h2fax::copyDir(root_path.c_str(), tmp.c_str());
            //Take away the AvantFax prefix and substitute the markers like to-company for their actual value
            h2fax::removeAvantFaxPrefix((tmp + file).c_str(), prefix);
            //Convert file to PS for HylaFax!
            h2fax::convertToPS((tmp + file).c_str(), html2ps_path, " > " + tmp + "tmp.ps");
            //Now, we execute faxcover with all of the switches.
            args.template_path = (tmp + "tmp.ps").c_str();
            h2fax::convertToPS(generateCommand(args).c_str(), hylafax, "");
            //By now, HylaFax should be receiving the processed html template via stdin.
            break;
        }
    case CALL_HYLAFAX_FAXCOVER_AS_IS:
        {
            //In this case, the program is asked to strip the template of the AvantFax prefix and send it to faxcover.
            //Copy template file to a temporary local file
            h2fax::copyFile(args.template_path, (tmp + "tmp.ps").c_str());
            //Take away the AvantFax prefix and substitute the markers like to-company for their actual value
            h2fax::removeAvantFaxPrefix((tmp + "tmp.ps").c_str(), prefix);
            //Now, we execute faxcover with all of the switches.
            args.template_path = (tmp + "tmp.ps").c_str();
            h2fax::convertToPS(generateCommand(args).c_str(), hylafax, "");
            //By now, HylaFax should be receiving the processed html template via stdin.
            break;
        }
    case HASH_TEST:
        {
          for(int i = 0; i < h2fax::MARKERSNUM; i++)
          {
            std::cerr << std::hex << args.hashFunction(h2fax::MARKERS[i].c_str()) << ": " << h2fax::MARKERS[i] << std::endl;
          }
          break;
        }
    default:
        std::cerr << "The execution mode flag that was provided does not match any case in the program. The program will "
            << "close! Thanks for using me? << Sounds a bit echi even for a machine! :3" << std::endl;
    }
    }

    /*Let's unlock the tmp directory*/
    h2fax::UnLockFile(lock.c_str());

    //Restore cerr
    if(cerrLog.is_open())
      std::cerr.rdbuf(cerr_copy);

    return EXIT_SUCCESS;
}

void getNameAndEmail(MySQL& db, std::string criteria, std::string& name, std::string& email)
{
    //Let's do a small check!
    if(criteria[0] != '\'' || criteria[criteria.size() - 1] != '\'')
        criteria = "'" + criteria + "'";
    std::string tmp = db.prepareStatement("UserAccount", "name", "username = " + criteria);
    //Query Database for username
    db.queryDB(tmp);
    db.getResult(name);
    //Generate a new query for the email
    tmp = db.prepareStatement("UserAccount", "email", "username = " + criteria);
    db.queryDB(tmp);
    db.getResult(email);
}

std::string generateCommand(const h2fax::faxcover_args& args)
{
    return "-c " + std::string(args.comments) + " -C " + std::string(args.template_path) + " -l "
        + std::string(args.to_location) + " -L " + std::string(args.from_location) + " -M " + std::string(args.email)
        + " -n " + std::string(args.to_fax) + " -N " + std::string(args.from_fax) + " -p " + std::string(args.pageCount)
        + " -r " + std::string(args.regarding) + " -f " + std::string(args.from) + " -t " + std::string(args.to)
        + " -v " + std::string(args.to_voice) + " -V " + std::string(args.from_voice) + " -x "
        + std::string(args.to_company) + " -X " + std::string(args.from_company) + "-s" + std::string(args.pageSize) + " -u ";
}