Load/Save jobject

1 files changed, 10 insertions, 13 deletions

diff --git a/activity.py b/activity.py
index c6b8655..9f3f2d5 100644
--- a/activity.py
+++ b/activity.py
@@ -39,20 +39,10 @@ class InfoslicerActivity(SharedActivity):
         self.connect('init', self._init_cb)
         self.connect('tube', self._tube_cb)
 
-    def read_file(self, filepath):
-        pass
-
-    def write_file(self, filepath):
-        # XXX this method will be invoked on every props.status changing
-        # thus ignore it and use can_close() instead to save data on exit
-        pass
-
-    def can_close(self):
-        book.teardown()
-        return True
-
     def _init_cb(self, sender):
-        book.init()
+        book.wiki = book.WikiBook()
+        if not book.custom:
+            book.custom = book.CustomBook()
 
         self.edit_page = 1
         self.edit = edit.View()
@@ -76,6 +66,13 @@ class InfoslicerActivity(SharedActivity):
         toolbox.set_current_toolbar(1)
         self.show_all()
 
+    def read_file(self, filepath):
+        book.custom = book.CustomBook(filepath)
+
+    def write_file(self, filepath):
+        book.wiki.sync()
+        book.custom.sync(filepath)
+
     def set_edit_sensitive(self, enable):
         self.edit_bar.props.sensitive = enable
         self.edit_page = (enable and 1 or 2)