Template.cpp

#include "Template.h"

#include <QFile>

#include "FileName.h"
#include "IException.h"
#include "Project.h"
#include "XmlStackedHandlerReader.h"

namespace Isis{
  Template::Template(QString fileName, QString templateType, QString importName, QObject *parent) : QObject(parent) {
    m_fileName = fileName;
    m_templateType = templateType;
    m_importName = importName;

  }

  Template::Template(FileName templateFolder, XmlStackedHandlerReader *xmlReader, QObject *parent) :
      QObject(parent) {
    xmlReader->pushContentHandler(new XmlHandler(this, templateFolder));
  }


  Template::~Template() {
  }


  QString Template::fileName() const {
    return m_fileName;
  }


  QString Template::templateType() const {
    return m_templateType;
  }


  QString Template::importName() const {
    return m_importName;
  }


  void Template::updateFileName(Project *project) {
    m_fileName = project->templateRoot() + "/" + m_templateType + "/" + m_importName +"/" + FileName(m_fileName).name();
  }


  void Template::deleteFromDisk() {
    if (!QFile::remove(m_fileName)) {
      throw IException(IException::Io,
                       tr("Could not remove file [%1]").arg(m_fileName),
                       _FILEINFO_);
    }
  }


  void Template::save(QXmlStreamWriter &stream, const Project *project, FileName newProjectRoot)
      const {
        stream.writeStartElement("template");
        stream.writeAttribute("fileName", FileName(m_fileName).name());
        stream.writeAttribute("templateType", m_templateType);
        stream.writeAttribute("importName", m_importName);
        stream.writeEndElement();

  }


  Template::XmlHandler::XmlHandler(Template *currentTemplate, FileName templateFolder) {
    m_xmlHandlerTemplate = currentTemplate;
    m_xmlHandlerTemplateFolderName = templateFolder;
  }


  bool Template::XmlHandler::startElement(const QString &namespaceURI, const QString &localName,
                                         const QString &qName, const QXmlAttributes &atts) {
    if (XmlStackedHandler::startElement(namespaceURI, localName, qName, atts)) {
      if (localName == "template") {
        QString fileName = atts.value("fileName");
        QString templateType = atts.value("templateType");
        QString importName = atts.value("importName");

        if (!fileName.isEmpty()) {
          m_xmlHandlerTemplate->m_fileName = m_xmlHandlerTemplateFolderName.expanded() 
                                             + "/" + templateType 
                                             + "/" + importName
                                             + "/" + fileName;
        }
        
        if (!templateType.isEmpty()) {
          m_xmlHandlerTemplate->m_templateType = templateType;
        }
        
        if (!importName.isEmpty()) {
          m_xmlHandlerTemplate->m_importName = importName;
        }
      }
    }
    return true;
  }
}