From 86cdc6777da1c43f7b2def0368c624f43b9e9048 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Marc=20W=C3=A4ckerlin?= <marc@mrw.sh>
Date: Wed, 28 Jun 2017 08:22:23 +0000
Subject: [PATCH] new command while, similar to if

---
 scripts/wt-mode.el |   2 +-
 src/commands.hxx   | 103 ++++++++++++++++++++++++++++++++++++++++++++-
 src/editor.hxx     |   2 +-
 3 files changed, 104 insertions(+), 3 deletions(-)

diff --git a/scripts/wt-mode.el b/scripts/wt-mode.el
index 58f9633..e17b8b0 100644
--- a/scripts/wt-mode.el
+++ b/scripts/wt-mode.el
@@ -25,7 +25,7 @@
 ;;   echo $(webrunner -h | sed -n 's,  COMMAND: ,,p') | sed 's, ,\\\\|,g'
 (defconst wt-font-lock-keywords
   (list
-   '("^ *\\(auth\\|ca-certificate\\|call\\|case\\|check\\|clear-cookies\\|click\\|clicktype\\|client-certificate\\|do\\|download\\|echo\\|execute\\|exists\\|exit\\|expect\\|fail\\|for\\|function\\|if\\|ignoreto\\|include\\|label\\|load\\|not\\|offline-storage-path\\|open\\|screenshot\\|set\\|setvalue\\|sleep\\|testcase\\|testsuite\\|timeout\\|unset\\|upload\\) " . font-lock-builtin-face)
+   '("^ *\\(auth\\|ca-certificate\\|call\\|case\\|check\\|clear-cookies\\|click\\|clicktype\\|client-certificate\\|do\\|download\\|echo\\|execute\\|exists\\|exit\\|expect\\|fail\\|for\\|function\\|if\\|while\\|ignoreto\\|include\\|label\\|load\\|not\\|offline-storage-path\\|open\\|screenshot\\|set\\|setvalue\\|sleep\\|testcase\\|testsuite\\|timeout\\|unset\\|upload\\) " . font-lock-builtin-face)
    '("^ *#.*$" . font-lock-comment-face))
   "Highlighting expressions for Webtester")
 
diff --git a/src/commands.hxx b/src/commands.hxx
index 70c7dc1..fe9f569 100644
--- a/src/commands.hxx
+++ b/src/commands.hxx
@@ -2234,7 +2234,8 @@ class If: public CommandContainer {
     }
     QString command() const {
       return tag()+" "+_variable+" "+_cmp+" "+_value
-        +(_script.get()?"\n  "+_script->print().join("\n  "):"");
+        +(_script.get()?"\n  "+_script->print().join("\n  "):"")
+        +(_else.get()?"\nelse\n  "+_else->print().join("\n  "):"");
     }
     std::shared_ptr<Command> parse(Script* script, QString args,
                                    QStringList& in, QString file, int line,
@@ -2313,6 +2314,105 @@ class If: public CommandContainer {
     std::shared_ptr<Script> _else;
 };
 
+class While: public CommandContainer {
+  public:
+    QString tag() const {
+      return "while";
+    }
+    QString description() const {
+      return
+        tag()+" <variable> <cmp> <value>\n"
+        "  <command1>\n"
+        "  <command2>\n"
+        "  <...>\n"
+        "\n\n"+
+        tag()+" <selector> -> <text>\n"
+        "  <command1>\n"
+        "  <command2>\n"
+        "  <...>\n"
+        "\n\n"
+        "Repeats commands conditionally. "
+        "The first variant compares a variable to a value. "
+        "The comparision <cmp> can be = ! . ^ ~ < >, "
+        "which means equal, different, contains, contains not, match, "
+        "less (as integer), bigger (as integer). "
+        "Match allows a regular expression. "
+        "The second variant checks for a text in a selector, "
+        "similar to command exists. The text can be empty to just "
+        "check for the existence of a selector.";
+    }
+    QString command() const {
+      return tag()+" "+_variable+" "+_cmp+" "+_value
+        +(_script.get()?"\n  "+_script->print().join("\n  "):"");
+    }
+    std::shared_ptr<Command> parse(Script* script, QString args,
+                                   QStringList& in, QString file, int line,
+                                   int indent) {
+      std::shared_ptr<While> cmd(new While());
+      int pos(args.indexOf(QRegularExpression("[=!.^~<>]")));
+      int len(1);
+      if (args.contains("->")) {
+        pos = args.indexOf("->");
+        len = 2;
+        cmd->_cmp = "->";
+      } else {
+        if (pos<0) throw BadArgument(tag()+" needs a comparision, not: "+args);
+        cmd->_cmp = args[pos];
+      }
+      cmd->_variable = args.left(pos).trimmed();
+      cmd->_value = args.mid(pos+len).trimmed();
+      cmd->_script = cmd->subParser(script, subCommandBlock(in), file, line, indent);
+      return cmd;
+    }
+    bool execute(Script* script, QWebFrame* frame) {
+      Logger log(this, script, false);
+      QString value(script->replacevars(_value));
+      QString selector(script->replacevars(_variable));
+      for (bool check(true); check;) {
+        if (_cmp=="->") {
+          check = false;
+          Q_FOREACH(QWebElement element, frame->findAllElements(selector)) {
+            if (value.isEmpty() || // just find element
+                element.toOuterXml().indexOf(value)!=-1 ||
+                element.toPlainText().indexOf(value)!=-1) {
+              check = true;
+              break;
+            }
+          }
+          log(QString("evaluated expression to ")+(check?"true":"false")+": "
+              +selector+" "+_cmp+" "+value);
+        } else {
+          switch (_cmp[0].toLatin1()) {
+            case '=': check = script->variable(_variable)==value;
+              break;
+            case '!': check = script->variable(_variable)!=value;
+              break;
+            case '.': check = script->variable(_variable).contains(value);
+              break;
+            case '^': check = !script->variable(_variable).contains(value);
+              break;
+            case '~': check =
+                script->variable(_variable).contains(QRegularExpression(value));
+              break;
+            case '<': check = script->variable(_variable).toInt()<value.toInt();
+              break;
+            case '>': check = script->variable(_variable).toInt()>value.toInt();
+              break;
+            default:;
+          }
+          log(QString("evaluated expression to ")+(check?"true":"false")+": "
+              +script->variable(_variable)+" "+_cmp+" "+value);
+        }
+        if (check) if (!runScript(log, this, _script, script, frame)) return false;
+      }
+      return true;
+    }
+  private:
+    QString _variable;
+    QString _cmp;
+    QString _value;
+};
+
 class TestSuite: public Command {
   public:
     QString tag() const {
@@ -3006,6 +3106,7 @@ inline void Script::initPrototypes() {
   add(new Function);
   add(new Call);
   add(new If);
+  add(new While);
   add(new TestSuite);
   add(new TestCase);
   add(new Check);
diff --git a/src/editor.hxx b/src/editor.hxx
index dbb4d28..57e6cf6 100644
--- a/src/editor.hxx
+++ b/src/editor.hxx
@@ -19,7 +19,7 @@ class Highlighter: public QSyntaxHighlighter {
     void include(QString);
   public:
     Highlighter(QTextDocument *parent): QSyntaxHighlighter(parent) {
-      QString commands="auth|ca-certificate|call|case|check|clear-cookies|click|clicktype|client-certificate|do|download|echo|execute|exists|exit|expect|fail|for|function|if|ignoreto|include|label|load|not|offline-storage-path|open|screenshot|set|setvalue|sleep|testcase|testsuite|timeout|unset|upload";
+      QString commands="auth|ca-certificate|call|case|check|clear-cookies|click|clicktype|client-certificate|do|download|echo|execute|exists|exit|expect|fail|for|function|if|while|ignoreto|include|label|load|not|offline-storage-path|open|screenshot|set|setvalue|sleep|testcase|testsuite|timeout|unset|upload";
       _expressions<<Expression("^ *("+commands+")\\b").weight(QFont::Bold).fg(Qt::darkBlue)
                   <<Expression("^ *#.*$").weight(QFont::Bold).fg(Qt::black);
     }