grammar: Add stub for the trigger class

yunzixun · Jan 23, 2017 · 9266d20 · 9266d20
1 parent f8165d4
commit 9266d20
Show file tree

Hide file tree

Showing 3 changed files with 40 additions and 1 deletion.
diff --git a/src/sqlitedb.cpp b/src/sqlitedb.cpp
@@ -1253,7 +1253,7 @@ void DBBrowserDB::updateSchema( )
                 continue;
 
             DBBrowserObject obj(val_name, val_sql, type, val_tblname);
-            if((type == sqlb::Object::Types::Table || type == sqlb::Object::Types::Index || type == sqlb::Object::Types::View) && !val_sql.isEmpty())
+            if(!val_sql.isEmpty())
             {
                 obj.object = sqlb::Object::parseSQL(type, val_sql);
                 if(val_temp == "1")
@@ -1274,6 +1274,10 @@ void DBBrowserDB::updateSchema( )
                         foreach(const auto& column, columns)
                             view->addField(sqlb::FieldPtr(new sqlb::Field(column.first, column.second)));
                     }
+                } else if(type == sqlb::Object::Types::Trigger) {
+                    // For triggers set the name of the table the trigger operates on here because we don't have a parser for trigger statements yet.
+                    sqlb::TriggerPtr trg = obj.object.dynamicCast<sqlb::Trigger>();
+                    trg->setTable(val_tblname);
                 }
             }
             objMap.insert(val_type, obj);

diff --git a/src/sqlitetypes.cpp b/src/sqlitetypes.cpp
@@ -79,6 +79,9 @@ ObjectPtr Object::parseSQL(Object::Types type, const QString& sSQL)
     case Object::Types::View:
         result = View::parseSQL(sSQL);
         break;
+    case Object::Types::Trigger:
+        result = Trigger::parseSQL(sSQL);
+        break;
     default:
         return ObjectPtr(nullptr);
     }
@@ -1191,4 +1194,13 @@ FieldInfoList View::fieldInformation() const
     return result;
 }
 
+
+
+ObjectPtr Trigger::parseSQL(const QString& /*sSQL*/)
+{
+    // TODO
+
+    return TriggerPtr(new Trigger(""));
+}
+
 } //namespace sqlb
diff --git a/src/sqlitetypes.h b/src/sqlitetypes.h
@@ -16,6 +16,7 @@ class Object;
 class Table;
 class Index;
 class View;
+class Trigger;
 class Field;
 class Constraint;
 class IndexedColumn;
@@ -24,6 +25,7 @@ typedef QSharedPointer<Object> ObjectPtr;
 typedef QSharedPointer<Table> TablePtr;
 typedef QSharedPointer<Index> IndexPtr;
 typedef QSharedPointer<View> ViewPtr;
+typedef QSharedPointer<Trigger> TriggerPtr;
 typedef QSharedPointer<Field> FieldPtr;
 typedef QSharedPointer<Constraint> ConstraintPtr;
 typedef QVector<FieldPtr> FieldVector;
@@ -406,6 +408,27 @@ class View : public Object
     FieldVector m_fields;
 };
 
+class Trigger : public Object
+{
+public:
+    explicit Trigger(const QString& name): Object(name) {}
+    virtual ~Trigger() {}
+
+    virtual Types type() const { return Object::Trigger; }
+
+    QString sql() const { /* TODO */ return m_originalSql; }
+
+    static ObjectPtr parseSQL(const QString& sSQL);
+
+    virtual QString baseTable() const { return m_table; }
+
+    void setTable(const QString& table) { m_table = table; }
+    QString table() const { return m_table; }
+
+private:
+    QString m_table;
+};
+
 } //namespace sqlb
 
 #endif // SQLITETYPES_H