1 files changed, 53 insertions, 0 deletions
diff --git a/SQLiteStudio3/coreSQLiteStudio/viewmodifier.h b/SQLiteStudio3/coreSQLiteStudio/viewmodifier.h
new file mode 100644
index 0000000..6d7c770
--- /dev/null
+++ b/SQLiteStudio3/coreSQLiteStudio/viewmodifier.h
@@ -0,0 +1,53 @@
+#ifndef VIEWMODIFIER_H
+#define VIEWMODIFIER_H
+
+#include "db/db.h"
+#include "parser/ast/sqlitecreateview.h"
+#include "parser/ast/sqlitecreatetrigger.h"
+#include <QString>
+
+class API_EXPORT ViewModifier
+{
+    public:
+        ViewModifier(Db* db, const QString& view);
+        ViewModifier(Db* db, const QString& database, const QString& view);
+
+        void alterView(const QString& newView);
+        void alterView(SqliteCreateViewPtr newView);
+
+        QStringList generateSqls() const;
+        QList<bool> getMandatoryFlags() const;
+        QStringList getWarnings() const;
+        QStringList getErrors() const;
+        bool hasMessages() const;
+
+    private:
+        void handleTriggers();
+        void collectNewColumns();
+        void addMandatorySql(const QString& sql);
+        void addOptionalSql(const QString& sql);
+        bool handleNewColumns(SqliteCreateTriggerPtr trigger);
+
+        Db* db = nullptr;
+        Dialect dialect;
+        QString database;
+        QString view;
+
+        /**
+         * @brief sqls Statements to be executed to make changes real.
+         */
+        QStringList sqls;
+        QList<bool> sqlMandatoryFlags;
+
+        QStringList warnings;
+        QStringList errors;
+
+        /**
+         * @brief createView Original DDL.
+         */
+        SqliteCreateViewPtr createView;
+
+        QStringList newColumns;
+};
+
+#endif // VIEWMODIFIER_H