1 files changed, 42 insertions, 3 deletions

diff --git a/js/document/models/base.js b/js/document/models/base.js
index c7e2de69..c44123c3 100755
--- a/js/document/models/base.js
+++ b/js/document/models/base.js
@@ -88,9 +88,26 @@ exports.BaseDocumentModel = Montage.create(Component, {
     },
     ////////////////////////////////////////////////////////////////////
     //
+    _currentViewIdentifier: {
+        value: ""
+    },
+    ////////////////////////////////////////////////////////////////////
+    //
+    currentViewIdentifier: {
+        get: function() {
+            return this._currentViewIdentifier;
+        },
+        set: function(value) {
+            this._currentViewIdentifier = value;
+        }
+    },
+    ////////////////////////////////////////////////////////////////////
+    //
     _selection: {
         value: []
     },
+    ////////////////////////////////////////////////////////////////////
+    //
     domContainer: {
         value: null
     },
@@ -120,7 +137,8 @@ exports.BaseDocumentModel = Montage.create(Component, {
     libs: {
         value: null
     },
-
+    ////////////////////////////////////////////////////////////////////
+    //
     _mObjects: {
             value: []
         },
@@ -138,6 +156,23 @@ exports.BaseDocumentModel = Montage.create(Component, {
     switchViewTo: {
         value: function (view) {
             //
+            switch (view) {
+                case 'design':
+                    //
+                    if (this.views.design) this.views.design.show();
+                    if (this.views.code) this.views.code.hide();
+                    this.currentView = this.views.design;
+                    break;
+                case 'code':
+                    //
+                    if (this.views.code) this.views.code.show();
+                    if (this.views.design) this.views.design.hide();
+                    this.currentView = this.views.code;
+                    break;
+                default:
+                    //Error
+                    break;
+            }
         }
     },
     ////////////////////////////////////////////////////////////////////
@@ -233,8 +268,10 @@ exports.BaseDocumentModel = Montage.create(Component, {
                         this.libs.canvas = true;
                     }
                 }
+            } else if (this.currentView === this.views.code) {
+                //TODO: Add save logic for code view
             } else {
-                //TODO: Add logic to save code view data
+                //TODO: Error handle
             }
         }
     },
@@ -274,8 +311,10 @@ exports.BaseDocumentModel = Montage.create(Component, {
                         this.libs.canvas = true;
                     }
                 }
+            } else if (this.currentView === this.views.code) {
+                //TODO: Add save logic for code view
             } else {
-                //TODO: Add logic to save code view data
+                //TODO: Error handle
             }
 
         }