Add local script loading

butterfly/__init__.py

@@ -46,13 +46,18 @@ class Route(tornado.web.RequestHandler):
     @property
     def builtin_themes_dir(self):
         return os.path.join(
-                os.path.dirname(__file__), 'themes')
+            os.path.dirname(__file__), 'themes')
 
     @property
     def themes_dir(self):
         return os.path.join(
             self.application.butterfly_dir, 'themes')
 
+    @property
+    def local_js_dir(self):
+        return os.path.join(
+            self.application.butterfly_dir, 'js')
+
     def get_theme_dir(self, theme):
         if theme.startswith('built-in-'):
             return os.path.join(

butterfly/routes.py

@@ -340,7 +340,7 @@ class ThemesList(Route):
                 'built-in-%s' % theme
                 for theme in os.listdir(self.builtin_themes_dir)
                 if os.path.isdir(os.path.join(
-                        self.builtin_themes_dir, theme)) and
+                    self.builtin_themes_dir, theme)) and
                 not theme.startswith('.')]
         else:
             builtin_themes = []
@@ -351,3 +351,22 @@ class ThemesList(Route):
             'builtin_themes': sorted(builtin_themes),
             'dir': self.themes_dir
         }))
+
+
+@url('/local.js')
+class LocalJsStatic(Route):
+    def get(self):
+        self.set_header("Content-Type", 'application/javascript')
+        if os.path.exists(self.local_js_dir):
+            for fn in os.listdir(self.local_js_dir):
+                if not fn.endswith('.js'):
+                    continue
+                with open(os.path.join(self.local_js_dir, fn), 'rb') as s:
+                    while True:
+                        data = s.read(16384)
+                        if data:
+                            self.write(data)
+                        else:
+                            self.write(';')
+                            break
+        self.finish()

butterfly/templates/index.html

@@ -23,5 +23,6 @@
                  '' if options.unminified else 'min.')) }}"></script>
     <script src="{{ static_url('ext.%sjs' % (
                  '' if options.unminified else 'min.')) }}"></script>
+    <script src="{{ reverse_url('LocalJsStatic') }}"></script>
   </body>
 </html>

butterfly/utils.py

@@ -216,6 +216,7 @@ def get_socket_env(inode, user):
                         'gnome-session',
                         'gnome-session-binary',
                         'startkde',
+                        'startdde',
                         'xfce4-session']:
                     with open('/proc/%s/status' % pid) as e:
                         uid = None