0
@@ -11,7 +11,6 @@ Usage:
0
 install wiki_mapper.user.js into greasemonkey
0
 ruby wiki_log.rb
0
 
0
-when you want output, go to http://localhost:9999/?exit=1
0
+when you want output, go to http://localhost:9999/show
0
 
0
-there will now be a file called wiki_map.dot and it can be abused in the usual
0
-ways.
0
+The program handles db creation and maintenance on its own

0
@@ -1,37 +1,73 @@
0
 require 'rubygems'
0
 require 'mongrel'
0
 require 'logger'
0
+require 'active_record'
0
 
0
-def stripdot(txt)
0
-  CGI::unescape(txt).gsub(/[^A-Za-z0-9_]/, '_')
0
+$db_name = 'wiki_dot.db'
0
+$dot_fn = 'wiki_map.tmp.dot'
0
+$png_fn = 'wiki_map.tmp.png'
0
+
0
+if !File.exists? $db_name
0
+  require 'sqlite3'
0
+  db = SQLite3::Database.new($db_name)
0
+  db.execute %q{ 
0
+    CREATE TABLE lines(
0
+      id INTEGER PRIMARY KEY,
0
+      created_at DATETIME,
0
+      line TEXT);
0
+    }
0
+  db.close
0
 end
0
 
0
-$file = File.open("wiki_map.dot", "w")
0
 
0
-class HeaderHandler < Mongrel::HttpHandler
0
 
0
+ActiveRecord::Base.establish_connection({ :adapter => 'sqlite3', 
0
+  :dbfile => $db_name})
0
+
0
+class Line < ActiveRecord::Base
0
+end
0
+
0
+
0
+def stripdot(txt)
0
+  CGI::unescape(txt).gsub(/[^A-Za-z0-9_]/, '_')
0
+end
0
 
0
+
0
+class WikiLoggerHandler < Mongrel::HttpHandler
0
   def process(request,response)
0
     response.start(200) do |head, out|
0
       head["Content-Type"] = "text/plain"
0
       vars = Mongrel::HttpRequest.query_parse(request.params['QUERY_STRING'])
0
-      if vars['exit'] == '1'
0
-        $file.puts '}'
0
-        $file.close
0
-        exit! 
0
-      elsif vars['single']
0
-        $file.puts "  #{stripdot vars['single']};"
0
+
0
+      if vars['single']
0
+        Line.new(:line => "#{stripdot vars['single']};").save
0
       else
0
-        $file.puts "  #{stripdot vars['from']} -> #{stripdot vars['to']};"
0
+        Line.new(
0
+          :line => "#{stripdot vars['from']} -> #{stripdot vars['to']};").save
0
       end
0
     end
0
   end
0
 end
0
 
0
+class WikiMapDisplayHandler < Mongrel::HttpHandler
0
+  def process(request,response)
0
+    response.start(200) do |head, out|
0
+      head["Content-Type"] = "image/png"
0
+      vars = Mongrel::HttpRequest.query_parse(request.params['QUERY_STRING'])
0
+      File.open($dot_fn, 'w') do |f|
0
+        f.write((["digraph wiki_map {"] +
0
+        Line.find(:all).collect{|l| "  #{l.line}"} + ["}"]).join("\n"))
0
+      end
0
+      `dot -Tpng #{$dot_fn} -o #{$png_fn}`
0
+      out.write File.read($png_fn)
0
+    end
0
+  end
0
+end
0
+      
0
 
0
-$file.puts "digraph wiki_map {"
0
 server = Mongrel::HttpServer.new("127.0.0.1", "9999")
0
-server.register("/", HeaderHandler.new)
0
+server.register("/", WikiLoggerHandler.new)
0
+server.register("/show", WikiMapDisplayHandler.new)
0
 server.register("/favicon.ico", Mongrel::Error404Handler.new(""))
0
 server.run.join
0