wip

pglablib/sqlast/BuildStandardItemTreeModelNodeVisitor.cpp

@@ -0,0 +1,69 @@
+#include "BuildStandardItemTreeModelNodeVisitor.h"
+
+using namespace sqlast;
+
+namespace {
+
+    template <class T>
+    class AutoRevert {
+    public:
+        AutoRevert(T& var, const T newValue)
+            : variable(var)
+            , previousValue(var)
+        {
+            variable = std::move(newValue);
+        }
+
+        AutoRevert(T& var, const T&& newValue)
+            : variable(var)
+            , previousValue(var)
+        {
+            variable = std::move(newValue);
+        }
+
+        ~AutoRevert()
+        {
+            variable = std::move(previousValue);
+        }
+
+        AutoRevert(const AutoRevert&) = delete;
+        AutoRevert operator=(const AutoRevert&) = delete;
+
+    private:
+        T& variable;
+        T previousValue;
+    };
+
+    template<class T>
+    AutoRevert<T> MakeAutoRevert(T& var, const T newValue)
+    {
+        return AutoRevert<T>(var, newValue);
+    }
+}
+
+BuildStandardItemTreeModelNodeVisitor::BuildStandardItemTreeModelNodeVisitor()
+    : model(std::make_unique<QStandardItemModel>())
+    , currentParent(model->invisibleRootItem())
+{
+}
+
+void BuildStandardItemTreeModelNodeVisitor::Visit(SelectItem &selectItem)
+{
+}
+
+void BuildStandardItemTreeModelNodeVisitor::Visit(SelectList &selectList)
+{
+}
+
+void BuildStandardItemTreeModelNodeVisitor::Visit(SelectStatement &selectStatement)
+{
+    auto item = new QStandardItem("SELECT");
+    currentParent->appendRow(item);
+    auto guard = MakeAutoRevert(currentParent, item);
+
+
+}
+
+void sqlast::BuildStandardItemTreeModelNodeVisitor::Visit(StatementList &statementList)
+{
+}

pglablib/sqlast/BuildStandardItemTreeModelNodeVisitor.h

@@ -0,0 +1,28 @@
+#pragma once
+
+#include "NodeVisitor.h"
+
+#include <QStandardItemModel>
+#include <memory>
+
+namespace sqlast {
+
+    class BuildStandardItemTreeModelNodeVisitor : public NodeVisitor
+    {
+    public:
+        BuildStandardItemTreeModelNodeVisitor();
+
+        // NodeVisitor interface
+    public:
+        virtual void Visit(SelectItem &selectItem) override;
+        virtual void Visit(SelectList &selectList) override;
+        virtual void Visit(SelectStatement &selectStatement) override;
+        virtual void Visit(StatementList &statementList) override;
+        virtual void Visit(StringLiteral &stringLiteral) override;
+    private:
+        std::unique_ptr<QStandardItemModel> model;
+        QStandardItem *currentParent;
+
+    };
+
+}

pglablib/sqlast/Node.cpp

@@ -4,5 +4,9 @@ using namespace sqlast;
 
 Node::Node()
 {
-
+}
+
+QString Node::ToString() const
+{
+    return QString::fromUtf8(typeid(*this).name());
 }

pglablib/sqlast/Node.h

@@ -1,12 +1,25 @@
 #pragma once
 
+#include <QString>
+#include <stdexcept>
+#include <typeinfo>
+
 namespace sqlast {
 
+    class NodeVisitor;
+
     class Node {
     public:
         Node();
         virtual ~Node() = default;
 
+        virtual void Accept(NodeVisitor &visitor) = 0;
+        /// Every derived class that has child nodes should override these
+        /// to facilitate
+//        virtual int ChildCount() const { return 0; }
+//        virtual const Node* GetChild(int index) const { throw std::out_of_range("GetChild"); }
+
+        virtual QString ToString() const;
     };
 
 }

pglablib/sqlast/NodeVisitor.cpp

@@ -0,0 +1,8 @@
+#include "NodeVisitor.h"
+#include "sqlast/SelectList.h"
+
+void sqlast::NodeVisitor::VisitSelectListItems(SelectList &selectList)
+{
+    for (int idx = 0; idx < selectList.Count(); ++idx)
+        Visit(selectList.Get(idx));
+}

pglablib/sqlast/NodeVisitor.h

@@ -0,0 +1,27 @@
+#pragma once
+
+namespace sqlast {
+
+    class SelectItem;
+    class SelectList;
+    class SelectStatement;
+    class StatementList;
+    class StringLiteral;
+
+    class NodeVisitor
+    {
+    public:
+        virtual ~NodeVisitor() = default;
+
+        virtual void Visit(SelectItem &selectItem) = 0;
+        virtual void Visit(SelectList &selectList) = 0;
+        virtual void Visit(SelectStatement &selectStatement) = 0;
+        virtual void Visit(StatementList &statementList) = 0;
+        virtual void Visit(StringLiteral &stringLiteral) = 0;
+
+    protected:
+        void VisitSelectListItems(SelectList &selectList);
+
+    };
+
+}

pglablib/sqlast/SelectItem.cpp

@@ -1,5 +1,6 @@
 #include "SelectItem.h"
 #include "Expression.h"
+#include "NodeVisitor.h"
 
 using namespace sqlast;
 
@@ -11,3 +12,8 @@ void SelectItem::SetAlias(const std::string &alias)
 {
     this->alias = alias;
 }
+
+void SelectItem::Accept(NodeVisitor &visitor)
+{
+    visitor.Visit(*this);
+}

pglablib/sqlast/SelectItem.h

@@ -17,6 +17,8 @@ public:
 
     void SetAlias(const std::string &alias);
     std::string GetAlias() const { return alias; }
+
+    void Accept(NodeVisitor &visitor) override;
 private:
     std::unique_ptr<Expression> expression;
     std::string alias;

pglablib/sqlast/SelectList.cpp

@@ -1,5 +1,6 @@
 #include "SelectList.h"
 #include "SelectItem.h"
+#include "sqlast/NodeVisitor.h"
 
 using namespace sqlast;
 
@@ -9,10 +10,20 @@ SelectList::SelectList()
 
 void SelectList::Add(std::unique_ptr<SelectItem> select_item)
 {
-    list.push_back(std::move(select_item));
+    selectItems.push_back(std::move(select_item));
 }
 
 int SelectList::Count() const
 {
-    return static_cast<int>(list.size());
+    return static_cast<int>(selectItems.size());
+}
+
+SelectItem &SelectList::Get(int index)
+{
+    return *selectItems.at(index);
+}
+
+void SelectList::Accept(NodeVisitor &visitor)
+{
+    visitor.Visit(*this);
 }

pglablib/sqlast/SelectList.h

@@ -6,6 +6,8 @@
 
 namespace sqlast {
 
+
+
     class SelectItem;
 
     class SelectList : public Node
@@ -16,14 +18,13 @@ namespace sqlast {
         void Add(std::unique_ptr<SelectItem> select_item);
         int Count() const;
 
-        SelectItem& Get(int index)
-        {
-            return *list.at(index);
-        }
+        SelectItem& Get(int index);
+
+        void Accept(NodeVisitor &visitor) override;
     private:
         using List = std::vector<std::unique_ptr<SelectItem>>;
 
-        List list;
+        List selectItems;
     };
 
 }

pglablib/sqlast/SelectStatement.cpp

@@ -1,5 +1,6 @@
 #include "SelectStatement.h"
 #include "SelectList.h"
+#include "NodeVisitor.h"
 
 using namespace sqlast;
 
@@ -17,3 +18,8 @@ void SelectStatement::SetSelectList(std::unique_ptr<SelectList> value)
 {
     selectList = std::move(value);
 }
+
+void SelectStatement::Accept(NodeVisitor &visitor)
+{
+    visitor.Visit(*this);
+}

pglablib/sqlast/SelectStatement.h

@@ -15,6 +15,7 @@ namespace sqlast {
         SelectList* GetSelectList();
         void SetSelectList(std::unique_ptr<SelectList> value);
 
+        void Accept(NodeVisitor &visitor) override;
     private:
         std::unique_ptr<SelectList> selectList;
     };

pglablib/sqlast/StatementList.cpp

@@ -1,6 +1,6 @@
 #include "StatementList.h"
 #include "Statement.h"
-
+#include "sqlast/NodeVisitor.h"
 
 using namespace sqlast;
 
@@ -22,3 +22,8 @@ int StatementList::Count() const
     return static_cast<int>(statements.size());
 }
 
+void StatementList::Accept(NodeVisitor &visitor)
+{
+    visitor.Visit(*this);
+}
+

pglablib/sqlast/StatementList.h

@@ -18,6 +18,7 @@ namespace sqlast {
         Statement &Get(int index);
         int Count() const;
 
+        void Accept(NodeVisitor &visitor) override;
     private:
         using Statements = std::vector<std::unique_ptr<Statement>>;
 

pglablib/sqlast/StringLiteral.cpp

@@ -1,7 +1,13 @@
 #include "StringLiteral.h"
+#include "NodeVisitor.h"
 
 using namespace sqlast;
 
 StringLiteral::StringLiteral(const std::string s)
     : value(QString::fromStdString(s))
 {}
+
+void StringLiteral::Accept(NodeVisitor &visitor)
+{
+    visitor.Visit(*this);
+}

pglablib/sqlast/StringLiteral.h

@@ -12,6 +12,7 @@ namespace sqlast {
         explicit StringLiteral(const std::string s);
 
         QString GetValue() const { return value; }
+        void Accept(NodeVisitor &visitor) override;
     private:
         QString value;
     };