Ticket #971: keep_offline.patch

downloadmanager.py

@@ -20 +20 @@
 from gettext import gettext as _
 import time
 import tempfile
+import zipfile
+import shutil
 import urlparse
 import urllib
 
@@ -382 +384 @@
 
     def onDataAvailable(self, request, context, inputStream, offset, count):
         self._external_listener.onDataAvailable(request, context, inputStream,
-                                                offset, count);
-
+                                                offset, count)
+                                                
+def save_document(browser, activity):
+    cls = components.classes[ \
+                    '@mozilla.org/embedding/browser/nsWebBrowserPersist;1']
+    persist = cls.createInstance(interfaces.nsIWebBrowserPersist)
+                                     
+    local = components.classes["@mozilla.org/file/local;1"]
+    local_file = local.createInstance(interfaces.nsILocalFile)
+    local_data = local.createInstance(interfaces.nsILocalFile)
+
+    temp_dir = tempfile.mkdtemp()
+
+    local_file.initWithPath(os.path.join(temp_dir, 'index.html'))
+    local_data.initWithPath(os.path.join(temp_dir, 'data'))
+    
+    persist.persistFlags = interfaces.nsIWebBrowserPersist \
+                                     .PERSIST_FLAGS_REPLACE_EXISTING_FILES
+    persist.progressListener = _SaveDocumentProgressListener(activity,
+                                    temp_dir, browser.props.title)._wrapped
+    persist.saveDocument(browser.dom_window.document, local_file, local_data,
+                         None, 0, 0)
+                                 
+class _SaveDocumentProgressListener(object):
+    _com_interfaces_ = interfaces.nsIWebProgressListener
+    
+    def __init__(self, activity, temp_dir, title):
+        self._activity = activity
+        self._temp_dir = temp_dir
+        self._title = title
+        self._zip_path = os.path.join(activity.get_activity_root(), 
+                                      'instance', self._title+'.zip')
+        self._jobject = None
+        
+        self._wrapped = xpcom.server.WrapObject(self,
+                                interfaces.nsIWebProgressListener)
+                                
+    def onStateChange(self, web_progress, request, state_flags, status):
+        if state_flags & interfaces.nsIWebProgressListener.STATE_STOP:
+            if NS_FAILED(status):
+                alert = TimeoutAlert(9)
+                alert.props.title = _('Keep offline')
+                alert.props.msg = _('Saving "%s" for offline use failed') % \
+                                  self._title
+                self._activity.add_alert(alert)
+                alert.connect('response', self.__fail_response_cb)
+                alert.show()
+            else:
+                try: # HACK because NS_FAILED(status) is wrong
+                    self._create_zip()
+                    self._create_journal_object()
+                except OSError:
+                    return
+                
+                alert = Alert()
+                alert.props.title = _('Keep offline')
+                
+                cancel_icon = Icon(icon_name='dialog-cancel') 
+                alert.add_button(gtk.RESPONSE_CANCEL, _('Cancel'), cancel_icon) 
+                cancel_icon.show()
+
+                open_icon = Icon(icon_name='filesave') 
+                alert.add_button(gtk.RESPONSE_OK, _('Show in Journal'),
+                                 open_icon) 
+                open_icon.show()
+                
+                self._activity.add_alert(alert)
+                alert.connect('response', self.__success_response_cb)
+                
+    def onStatusChange(self, web_progress, request, status, message):
+        pass
+                    
+    def onProgressChange(self, web_progress, request, cur_self_progress,
+                           max_self_progress, cur_total_progress,
+                           max_total_progress):
+        pass
+        
+    def onLocationChange(self, web_progress, request, location):
+        pass
+    
+    def onSecurityChange(self, web_progress, request, state):
+        pass
+                               
+    def __fail_response_cb(self, alert, response_id):
+        self._activity.remove_alert(alert)
+        
+    def __success_response_cb(self, alert, response_id):
+        self._activity.remove_alert(alert)
+        
+        if response_id is gtk.RESPONSE_OK:
+            activity.show_object_in_journal(self._jobject.object_id)
+            
+                
+    def _create_zip(self):
+        bundle = zipfile.ZipFile(self._zip_path, 'w', zipfile.ZIP_DEFLATED)
+
+        bundle.write(os.path.join(self._temp_dir, 'index.html'), 'index.html')
+        data_path = os.path.join(self._temp_dir, 'data')
+        for i in os.listdir(data_path):
+            bundle.write(os.path.join(data_path, i),
+                         os.path.join('data', i))
+        bundle.close()
+                
+    def _create_journal_object(self):
+        self._jobject = datastore.create()
+        self._jobject.metadata['title'] = self._title
+        self._jobject.metadata['icon-color'] = \
+                profile.get_color().to_string()
+        self._jobject.metadata['mime_type'] = 'application/zip'
+        self._jobject.metadata['activity'] = 'org.laptop.WebActivity'
+        self._jobject.file_path = self._zip_path
+        datastore.write(self._jobject)

webactivity.py

@@ -30 +30 @@
 import sqlite3
 import cjson
 import gconf
+import zipfile
+import tempfile
 
 # HACK: Needed by http://dev.sugarlabs.org/ticket/456
 import gnome
@@ -395 +397 @@
             else:
                 _logger.error('Open uri-list: Does not support' 
                               'list of multiple uris by now.') 
+        elif self.metadata['mime_type'] == 'application/zip':
+            temp_dir = tempfile.mkdtemp()
+            
+            z = zipfile.ZipFile(file_path, 'r')
+            os.mkdir(os.path.join(temp_dir, 'data'))
+            for i in z.namelist():
+                open(os.path.join(temp_dir, i), 'wb').write(z.read(i))
+                
+            html_path = 'file://' + os.path.join(temp_dir, 'index.html')
+            self._tabbed_view.props.current_browser.load_uri(html_path)
         else:
             self._tabbed_view.props.current_browser.load_uri(file_path)
         
@@ -559 +571 @@
         browser = self._tabbed_view.props.current_browser
         browser.get_source(async_cb, async_err_cb)
 
+    def save_current_document(self):
+        downloadmanager.save_document(self._tabbed_view.props.current_browser,
+                                      self)
+ No newline at end of file

webtoolbar.py

@@ -272 +272 @@
         self._link_add.connect('clicked', self._link_add_clicked_cb)
         self.toolbar.insert(self._link_add, -1)
         self._link_add.show()
+        
+        self._save_document = ToolButton('filesave')
+        self._save_document.set_tooltip(_('Keep offline'))
+        self._save_document.connect('clicked', self._save_document_clicked_cb)
+        self.toolbar.insert(self._save_document, -1)
+        self._save_document.show()
 
         stop_button = StopButton(self._activity)
         self.toolbar.insert(stop_button, -1)
@@ -458 +464 @@
     def _link_add_clicked_cb(self, button):
         self.emit('add-link')
 
+    def _save_document_clicked_cb(self, button):
+        self._activity.save_current_document()
+ No newline at end of file