/****************************************************************************

  PmuppAdmin.cpp

  Author: Andreas Suter
  e-mail: andreas.suter@psi.ch

*****************************************************************************/

/***************************************************************************
 *   Copyright (C) 2010-2017 by Andreas Suter                              *
 *   andreas.suter@psi.ch                                                  *
 *                                                                         *
 *   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; either version 2 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 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 <cstdlib>
#include <iostream>
using namespace std;

#include <QMessageBox>
#include <QString>
#include <QFile>
#include <QTextStream>
#include <QVector>
#include <QDir>
#include <QProcessEnvironment>
#include <QSysInfo>

#include <QtDebug>

#include "PmuppAdmin.h"

//--------------------------------------------------------------------------
// implementation of PmuppColor class
//--------------------------------------------------------------------------
/**
 * @brief PmuppColor::PmuppColor
 */
PmuppColor::PmuppColor()
{
  fName  = "UnDef";
  fRed   = -1;
  fGreen = -1;
  fBlue  = -1;
}

//--------------------------------------------------------------------------
/**
 * @brief PmuppColor::setRGB
 * @param r
 * @param g
 * @param b
 */
void PmuppColor::setRGB(const int r, const int g, const int b)
{
  if ((r>=0) && (r<=255))
    fRed = r;
  if ((g>=0) && (g<=255))
    fGreen = g;
  if ((b>=0) && (b<=255))
    fBlue = b;
}

//--------------------------------------------------------------------------
// implementation of PmuppAdminXMLParser class
//--------------------------------------------------------------------------
/**
 * <p>XML Parser class for the mupp administration file.
 *
 * \param admin pointer to an admin class instance.
 */
PmuppAdminXMLParser::PmuppAdminXMLParser(PmuppAdmin *admin) : fAdmin(admin)
{
  fKeyWord = eEmpty;
}

//--------------------------------------------------------------------------
/**
 * <p>Routine called at the beginning of the XML parsing process.
 */
bool PmuppAdminXMLParser::startDocument()
{
  return true;
}

//--------------------------------------------------------------------------
/**
 * <p>Routine called when a new XML tag is found. Here it is used
 * to set a tag for filtering afterwards the content.
 *
 * \param qName name of the XML tag.
 */
bool PmuppAdminXMLParser::startElement( const QString&, const QString&,
                                        const QString& qName,
                                        const QXmlAttributes& )
{
  if (qName == "path_file_name") {
    fKeyWord = eRecentFile;
  } else if (qName == "marker") {
    fKeyWord = eMarker;
  } else if (qName == "color") {
    fKeyWord = eColor;
  }

  return true;
}

//--------------------------------------------------------------------------
/**
 * <p>Routine called when the end XML tag is found. It is used to
 * put the filtering tag to 'empty'. It also resets the fFunc flag in case
 * the entry was a theory function.
 *
 * \param qName name of the element.
 */
bool PmuppAdminXMLParser::endElement( const QString&, const QString&, const QString& )
{
  fKeyWord = eEmpty;

  return true;
}

//--------------------------------------------------------------------------
/**
 * <p>This routine delivers the content of an XML tag. It fills the
 * content into the load data structure.
 *
 * \param str keeps the content of the XML tag.
 */
bool PmuppAdminXMLParser::characters(const QString& str)
{
  bool ok;
  int  ival, r, g, b;
  double dval;
  QString name("");
  QStringList tok;

  switch (fKeyWord) {
    case eRecentFile:
      fAdmin->addRecentFile(QString(str.toLatin1()).trimmed());
      break;
    case eMarker:
      tok = str.split(",", QString::SkipEmptyParts);

      if ((tok.count() != 1) && (tok.count() != 2)) {
        return false;
      }

      ival = tok[0].toInt(&ok);
      if (!ok)
        return false;

      dval = 1.0;
      if (tok.count() == 2) {
        dval = tok[1].toDouble(&ok);
        if (!ok)
          return false;
      }
      fAdmin->setMarker(ival, dval);
      break;
    case eColor:
      tok = str.split(",", QString::SkipEmptyParts);

      if ((tok.count() != 3) && (tok.count() != 4)) {
        return false;
      }
      ival = tok[0].toInt(&ok);
      if (!ok)
        return false;
      r = ival;
      ival = tok[1].toInt(&ok);
      if (!ok)
        return false;
      g = ival;
      ival = tok[2].toInt(&ok);
      if (!ok)
        return false;
      b = ival;
      if (tok.count() == 4)
        name = tok[3];
      fAdmin->setColor(r, g, b, name);
      break;
    default:
      break;
  }

  return true;
}

//--------------------------------------------------------------------------
/**
 * <p>Called at the end of the XML parse process. It checks if default paths
 * contain system variables, and if so expand them for the further use.
 */
bool PmuppAdminXMLParser::endDocument()
{
  return true;
}

//--------------------------------------------------------------------------
/**
 * <p>Report XML warnings.
 *
 * \param exception holds the information of the XML warning
 */
bool PmuppAdminXMLParser::warning( const QXmlParseException & exception )
{
  QString msg;

  msg  = QString("**WARNING** while parsing mupp_startup.xml in line no %1\n").arg(exception.lineNumber());
  msg += QString("**WARNING MESSAGE** ") + exception.message();

  qWarning() << endl << msg << endl;

  QMessageBox::warning(0, "WARNING", msg, QMessageBox::Ok, QMessageBox::NoButton);

  return true;
}

//--------------------------------------------------------------------------
/**
 * <p>Report recoverable XML errors.
 *
 * \param exception holds the information of the XML recoverable errors.
 */
bool PmuppAdminXMLParser::error( const QXmlParseException & exception )
{
  QString msg;

  msg  = QString("**ERROR** while parsing mupp_startup.xml in line no %1\n").arg(exception.lineNumber());
  msg += QString("**ERROR MESSAGE** ") + exception.message();

  qWarning() << endl << msg << endl;

  QMessageBox::critical(0, "ERROR", msg, QMessageBox::Ok, QMessageBox::NoButton);

  return true;
}

//--------------------------------------------------------------------------
/**
 * <p>Report fatal XML errors.
 *
 * \param exception holds the information of the XML fatal errors.
 */
bool PmuppAdminXMLParser::fatalError( const QXmlParseException & exception )
{
  QString msg;

  msg  = QString("**FATAL ERROR** while parsing mupp_startup.xml in line no %1\n").arg(exception.lineNumber());
  msg += QString("**FATAL ERROR MESSAGE** ") + exception.message();

  qWarning() << endl << msg << endl;

  QMessageBox::critical(0, "FATAL ERROR", msg, QMessageBox::Ok, QMessageBox::NoButton);

  return true;
}

//--------------------------------------------------------------------------
// implementation of PmuppAdmin class
//--------------------------------------------------------------------------
/**
 * <p>Initializes that PmuppAdmin object, and calls the XML parser which feeds
 * the object variables.
 */
PmuppAdmin::PmuppAdmin() : QObject()
{
  // XML Parser part
  // 1st: check local directory
  QString path = QString("./");
  QString fln = QString("mupp_startup.xml");
  QString pathFln = path + fln;
  QProcessEnvironment procEnv = QProcessEnvironment::systemEnvironment();
  if (!QFile::exists(pathFln)) {
    // 2nd: check $HOME/.musrfit/mupp/mupp_startup.xml
    path = procEnv.value("HOME", "");
    pathFln = path + "/.musrfit/mupp/" + fln;
    if (!QFile::exists(pathFln)) {
      // 3rd: check $MUSRFITPATH/mupp_startup.xml
      path = procEnv.value("MUSRFITPATH", "");
      pathFln = path + "/" + fln;
      if (!QFile::exists(pathFln)) {
        // 4th: check $ROOTSYS/bin/mupp_startup.xml
        path = procEnv.value("ROOTSYS", "");
        pathFln = path + "/bin/" + fln;
        if (!QFile::exists(pathFln)) {
          // 5th: not found anywhere hence create it
          path = procEnv.value("HOME", "");
          pathFln = path + "/.musrfit/mupp/" + fln;
          createMuppStartupFile();
        }
      }
    }
  }

  if (QFile::exists(pathFln)) { // administration file present
    PmuppAdminXMLParser handler(this);
    QFile xmlFile(pathFln);
    QXmlInputSource source( &xmlFile );
    QXmlSimpleReader reader;
    reader.setContentHandler( &handler );
    reader.setErrorHandler( &handler );
    if (!reader.parse( source )) {
      QMessageBox::critical(0, "ERROR",
                            "Error parsing mupp_startup.xml settings file.\nProbably a few things will not work porperly.\nPlease fix this first.",
                            QMessageBox::Ok, QMessageBox::NoButton);
      return;
    }
  } else {
    QMessageBox::critical(0, "ERROR",
                          "Couldn't find the mupp_startup.xml settings file.\nProbably a few things will not work porperly.\nPlease fix this first.",
                          QMessageBox::Ok, QMessageBox::NoButton);
    return;
  }
}

//--------------------------------------------------------------------------
/**
 * <p>Destructor
 */
PmuppAdmin::~PmuppAdmin()
{
  saveRecentFiles();
}

//--------------------------------------------------------------------------
/**
 * <p>Add recent path-file name to the internal ring-buffer.
 *
 * \param str recent path-file name to be added
 */
void PmuppAdmin::addRecentFile(const QString str)
{
  // check if file name is not already present
  for (int i=0; i<fRecentFile.size(); i++) {
    if (str == fRecentFile[i])
      return;
  }

  fRecentFile.push_front(str);
  if (fRecentFile.size() > MAX_RECENT_FILES)
    fRecentFile.resize(MAX_RECENT_FILES);
}

//--------------------------------------------------------------------------
/**
 * @brief PmuppAdmin::getRecentFile
 * @param idx
 * @return
 */
QString PmuppAdmin::getRecentFile(int idx)
{
  QString str("");

  if ((idx >= 0) && (idx < fRecentFile.size()))
    str = fRecentFile[idx];

  return str;
}

//--------------------------------------------------------------------------
/**
 * @brief PmuppAdmin::getMarker
 * @param idx
 * @return
 */
PmuppMarker PmuppAdmin::getMarker(int idx) {
  PmuppMarker marker;

  if (idx >= fMarker.size())
    return marker;

  return fMarker[idx];
}

//--------------------------------------------------------------------------
/**
 * @brief PmuppAdmin::getColor
 * @param name
 * @param r
 * @param g
 * @param b
 */
void PmuppAdmin::getColor(QString name, int &r, int &g, int &b)
{
  int idx=-1;
  for (int i=0; i<fColor.size(); i++) {
    if (fColor[i].getName() == name) {
      idx = i;
      break;
    }
  }

  if (idx == -1) {
    r = -1;
    g = -1;
    b = -1;
  } else {
    fColor[idx].getRGB(r, g, b);
  }
}

//--------------------------------------------------------------------------
/**
 * @brief PmuppAdmin::getColor
 * @param idx
 * @param r
 * @param g
 * @param b
 */
void PmuppAdmin::getColor(int idx, int &r, int &g, int &b)
{
  if ((idx<=0) || (idx>fColor.size())) {
    r = -1;
    g = -1;
    b = -1;
  } else {
    fColor[idx].getRGB(r, g, b);
  }
}

//--------------------------------------------------------------------------
/**
 * @brief PmuppAdmin::setMaker
 * @param marker
 * @param size
 */
void PmuppAdmin::setMarker(int marker, double size)
{
  PmuppMarker markerObj;

  // make sure marker is in proper range
  if ((marker<1) || (marker>49)) {
    QMessageBox::warning(0, "WARNING", QString("Found Marker (%1) not in the expected range.\nWill ignore it.").arg(marker));
    return;
  }
  markerObj.setMarker(marker);
  markerObj.setMarkerSize(size);

  fMarker.push_back(markerObj);
}

//--------------------------------------------------------------------------
/**
 * @brief PmuppAdmin::setColor
 * @param r
 * @param g
 * @param b
 * @param name
 */
void PmuppAdmin::setColor(int r, int g, int b, QString name)
{
  if (((r<0) || (r>255)) ||
      ((g<0) || (g>255)) ||
      ((b<0) || (b>255))) {
    QMessageBox::warning(0, "WARNING", QString("Found Color (%1,%2,%3) not in the expected range.\nWill ignore it.").arg(r).arg(g).arg(b));
    return;
  }

  PmuppColor color;
  color.setName(name);
  color.setRGB(r,g,b);

  fColor.push_back(color);
}

//--------------------------------------------------------------------------
/**
 * <p>Merges the recent file ring buffer into mupp_startup.xml and saves it.
 * If a local copy is present it will be saved there, otherwise the master file
 * will be used.
 */
void PmuppAdmin::saveRecentFiles()
{
  // check if mupp_startup.xml is present in the current directory, and if yes, use this file to
  // save the recent file names otherwise use the "master" mupp_startup.xml

  QString str("");
  QString fln = QString("./mupp_startup.xml");
  if (!QFile::exists(fln)) {
    QProcessEnvironment env = QProcessEnvironment::systemEnvironment();
    fln = QString("%1/.musrfit/mupp/mupp_startup.xml").arg(env.value("HOME"));
  }

  if (QFile::exists(fln)) { // administration file present
    QVector<QString> data;
    QFile file(fln);
    if (!file.open(QIODevice::ReadOnly | QIODevice::Text)) {
      cerr << endl << ">> PmuppAdmin::saveRecentFile: **ERROR** Cannot open " << fln.toLatin1().data() << " for reading." << endl;
      return;
    }
    QTextStream fin(&file);
    while (!fin.atEnd()) {
      data.push_back(fin.readLine());
    }
    file.close();

    // remove <path_file_name> from data
    for (QVector<QString>::iterator it = data.begin(); it != data.end(); ++it) {
      if (it->contains("<path_file_name>")) {
        it = data.erase(it);
        --it;
      }
    }

    // add recent files
    int i;
    for (i=0; i<data.size(); i++) {
      if (data[i].contains("<recent_files>"))
        break;
    }

    if (i == data.size()) {
      cerr << endl << ">> PmuppAdmin::saveRecentFile: **ERROR** " << fln.toLatin1().data() << " seems to be corrupt." << endl;
      return;
    }
    i++;
    for (int j=0; j<fRecentFile.size(); j++) {
      str = "    <path_file_name>" + fRecentFile[j] + "</path_file_name>";
      data.insert(i++, str);
    }

    if (!file.open(QIODevice::WriteOnly | QIODevice::Text)) {
      cerr << endl << ">> PmuppAdmin::saveRecentFile: **ERROR** Cannot open " << fln.toLatin1().data() << " for reading." << endl;
      return;
    }
    fin.setDevice(&file);
    for (int i=0; i<data.size(); i++)
      fin << data[i] << endl;
    file.close();
  } else {
    QString msg("Failed to write mupp_startup.xml. Neither a local nor a global copy found.");
    QMessageBox::warning(0, "WARNING", msg, QMessageBox::Ok, QMessageBox::NoButton);
  }
}

//--------------------------------------------------------------------------
/**
 * @brief PmuppAdmin::createMuppStartupFile
 */
void PmuppAdmin::createMuppStartupFile()
{
  // get $HOME
  QProcessEnvironment procEnv = QProcessEnvironment::systemEnvironment();
  QString pathName = procEnv.value("HOME", "");
  pathName += "/.musrfit/mupp";

  // check if the directory $HOME/.musrfit/mupp exists if not create it
  QDir dir(pathName);
  if (!dir.exists()) {
    // directory $HOME/.musrfit/mupp does not exist hence create it
    dir.mkpath(pathName);
  }

  // create default mupp_startup.xml file in $HOME/.musrfit/mupp
  pathName += "/mupp_startup.xml";

  // get the default mupp_startup.xml.in from the internal resources
  QFile fres(":/mupp_startup.xml.in");
  if (!fres.exists()) {
    QString msg = QString("Neither couldn't find nor create mupp_startup.xml. Things are likely not to work.");
    QMessageBox::critical(0, "ERROR", msg);
    return;
  }

  if (!fres.open(QIODevice::ReadOnly | QIODevice::Text)) {
    QString msg = QString("Couldn't open internal resource file mupp_startup.xml.in. Things are likely not to work.");
    QMessageBox::critical(0, "ERROR", msg);
    return;
  }
  // text stream for fres
  QTextStream fin(&fres);

  // mupp_startup.xml default file
  QFile file(pathName);

  if (!file.open(QIODevice::WriteOnly | QIODevice::Text))
    return;

  // text stream for file
  QTextStream fout(&file);

  QString line;
  while (!fin.atEnd()) {
    line = fin.readLine();
    fout << line << endl;
  }

  file.close();
  fres.close();
}

//--------------------------------------------------------------------------
// END
//--------------------------------------------------------------------------