add .css and image file support

README.markdown

@@ -17,6 +17,11 @@ Double bonus: `~/.js/default.js`  is loaded on every
 request,  meaning you  can stick  plugins  or helper
 functions in it.
 
+Triple bonus:  Include CSS and  image files  in your
+.js files and serve them from  your local dirs.  CSS
+files  in  `~/.css` and  .gif, .jpg, and .png  files
+from `~/images`.
+
 GreaseMonkey user scripts are great, but you need to
 publish them  somewhere and re-publish  after making
 modifications. With dotjs, just add or edit files in
@@ -33,6 +38,7 @@ modifications. With dotjs, just add or edit files in
 
 ![](https://bit.ly/gAHTbC)
 
+
 ## How It Works
 
 Chrome extensions can't access the local filesystem,

Rakefile

@@ -60,12 +60,20 @@ namespace :install do
     cp "bin/djsd", DAEMON_INSTALL_DIR, :verbose => true, :preserve => true
   end
 
-  desc "Create ~/.js"
+  desc "Create ~/.js, ~/.css, and ~/.images"
   task :create_dir do
     if !File.directory? js_dir = File.join(ENV['HOME'], ".js")
       mkdir js_dir
       chmod 0755, js_dir
     end
+    if !File.directory? css_dir = File.join(ENV['HOME'], ".css")
+      mkdir css_dir
+      chmod 0755, css_dir
+    end
+    if !File.directory? images_dir = File.join(ENV['HOME'], ".images")
+      mkdir images_dir
+      chmod 0755, images_dir
+    end
   end
 
   desc "Install Google Chrome extension"
@@ -90,6 +98,8 @@ namespace :uninstall do
     puts "3. The 'dotjs' Google Chrome Extension",""
     puts "I will not remove:", ""
     puts "1. ~/.js", ""
+    puts "2. ~/.css", ""
+    puts "3. ~/.images", ""
     print "Ok? (y/n) "
 
     begin

bin/djsd

@@ -14,17 +14,36 @@ end
 require 'webrick'
 
 dotjs = Class.new(WEBrick::HTTPServlet::AbstractServlet) do
-  def do_GET(request, response)
-    file    = File.expand_path("#{request.path.gsub('/','')}")
-    default = File.expand_path("default.js")
+  IMAGE_TYPES = %w(.png .jpg .gif)
 
-    body = "// dotjs is working! //\n"
-    body << File.read(default) + "\n" if File.file?(default)
-    body << File.read(file) if File.file?(file)
+  def do_GET(request, response)
+    files = []
+    ext = File.extname(request.path)
+    case
+      when ext == '.js'
+        files << File.expand_path("~/.js/default.js")
+        files << File.expand_path("~/.js/#{request.path.gsub('/','')}")
+        mime_type = 'text/javascript'
+      when ext == '.css'
+        files <<  File.expand_path("~/.css/default.css")
+        files <<  File.expand_path("~/.css/#{request.path.gsub('/','')}")
+        mime_type = 'text/css'
+      when IMAGE_TYPES.include?(ext)
+        files << File.expand_path("~/.images/#{request.path.gsub('/','')}")
+        mime_type = "image/#{ext[1 .. -1]}"
+      else
+        puts "oops"
+    end
+
+    body = ''
+    files.each do |f|
+      body << File.read(f) if File.exists?(f)
+      body << "\n"
+    end
 
     response.status = body.empty? ? 204 : 200
     response['Access-Control-Allow-Origin'] = '*'
-    response['Content-Type'] = 'text/javascript'
+    response['Content-Type'] = mime_type
     response.body = body
   end
 end