if.cpp

/*
  This file is part of the Cutelee template system.
 
  Copyright (c) 2009,2010 Stephen Kelly <steveire@gmail.com>
 
  This library 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
  2.1 of the Licence, or (at your option) any later version.
 
  This library 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 library.  If not, see <http://www.gnu.org/licenses/>.
 
*/
 
#include "if.h"
#include "if_p.h"
 
#include "../lib/exception.h"
#include "parser.h"
 
IfNodeFactory::IfNodeFactory() {}
 
Node *IfNodeFactory::getNode(const QString &tagContent, Parser *p) const
{
  auto expr = smartSplit(tagContent);
 
  std::vector<std::pair<std::shared_ptr<IfToken>, NodeList>> nodelistConditions;
 
  auto n = new IfNode(p);
 
  IfParser ip(p, expr);
  auto cond = ip.parse();
  auto nodelist = p->parse(n, {QStringLiteral("elif"), QStringLiteral("else"),
                               QStringLiteral("endif")});
  nodelistConditions.push_back({cond, nodelist});
 
  auto token = p->takeNextToken();
 
  while (token.content.startsWith(QLatin1String("elif"))) {
    expr = smartSplit(token.content);
 
    IfParser ep(p, expr);
    cond = ep.parse();
    nodelist = p->parse(n, {QStringLiteral("elif"), QStringLiteral("else"),
                            QStringLiteral("endif")});
    nodelistConditions.push_back({cond, nodelist});
 
    token = p->takeNextToken();
  }
  if (token.content == QLatin1String("else")) {
    nodelist = p->parse(n, QStringLiteral("endif"));
    nodelistConditions.push_back({nullptr, nodelist});
    p->takeNextToken();
  }
 
  n->setNodelistConditions(nodelistConditions);
 
  auto commandName = expr.takeAt(0);
  if (expr.size() <= 0) {
    throw Cutelee::Exception(
        TagSyntaxError,
        QStringLiteral("'%1' statement requires at least one argument")
            .arg(commandName));
  }
 
  return n;
}
 
IfNode::IfNode(QObject *parent) : Node(parent) {}
 
void IfNode::setNodelistConditions(
    const std::vector<std::pair<std::shared_ptr<IfToken>, NodeList>> &conditionNodelists)
{
  mConditionNodelists = conditionNodelists;
}
 
void IfNode::render(OutputStream *stream, Context *c) const
{
  // Evaluate the expression. rendering variables with the context as needed.
  // and processing nodes recursively
 
  for (const auto &pair : mConditionNodelists) {
    bool match = false;
    if (pair.first) {
      try {
        match = Cutelee::variantIsTrue(pair.first->evaluate(c));
      } catch (const Cutelee::Exception &) {
      }
    } else {
      match = true;
    }
    if (match) {
      pair.second.render(stream, c);
      return;
    }
  }
}