lib/ui/web/log.rb in arachni-0.2.2.1 vs lib/ui/web/log.rb in arachni-0.2.2.2

@@ -18,17 +18,21 @@
 # A simple logger using DataMapper
 #
 # @author: Tasos "Zapotek" Laskos
 #                                      <[e-mail]>
 #                                      <[e-mail]>
-# @version: 0.1
+# @version: 0.1.1
 #
 class Log
 
     class Entry
         include DataMapper::Resource
 
+       def self.default_repository_name
+         :log
+       end
+
         property :id,           Serial
         property :action,       String
         property :object,       String
         property :client_addr,  String
         property :client_host,  String
@@ -40,18 +44,19 @@
     def initialize( opts, settings )
 
         @opts     = opts
         @settings = settings
 
-        DataMapper::setup( :default, "sqlite3://#{@settings.db}/log.db" )
-        DataMapper.finalize
-
-        Entry.auto_upgrade!
+        DataMapper::setup( :log, "sqlite3://#{@settings.db}/log.db" )
+        DataMapper.repository( :log ) {
+            DataMapper.finalize
+            Entry.auto_upgrade!
+        }
     end
 
     def entry
-        Entry
+        DataMapper.repository( :log ) { Entry }
     end
 
     def method_missing( sym, *args, &block )
 
         owner, action = sym.to_s.split( '_', 2 )
@@ -63,17 +68,19 @@
         if env = args[0]
             addr = env['REMOTE_ADDR']
             host = env['REMOTE_HOST']
         end
 
-        Entry.create(
-            :action => action,
-            :owner  => owner,
-            :object => object,
-            :client_addr => addr,
-            :client_host => host,
-            :datestamp   => Time.now
-        )
+        DataMapper.repository( :log ) {
+            Entry.create(
+                :action => action,
+                :owner  => owner,
+                :object => object,
+                :client_addr => addr,
+                :client_host => host,
+                :datestamp   => Time.now.asctime
+            )
+        }
     end
 
 end
 
 end