Changeset 265

Timestamp:

05/20/06 18:42:53 (3 years ago)

Author:

stillflame

Message:

- started plugging BogswiddleBot? up to the database
- initial schema and configuration file to use postgres and ActiveRecord?
- an attempt at getting logging to the database working

Location:

trunk/tools/irc

Files:

• .irbrc (added)
• README (added)
• bogsdb.rb (added)
• bogswiddle.rb (modified) (2 diffs)
• channel.rb (added)
• database.yml-example (added)
• message.rb (added)
• nick.rb (added)
• schema.pg.sql (added)
• trac.rb (modified) (2 diffs)

trunk/tools/irc/bogswiddle.rb

@@ -24 +24 @@
 # 
 
+$:.unshift File.dirname(__FILE__)
+require 'bogsdb'
 require 'rbot/ircbot'
 
@@ -41 +43 @@
     }
 
+    # Where our database config file is
+    ConfigFilename = File.join( File.dirname(__FILE__), 'database.yml')
+
+    # Some default config settings for the database
+    DefaultDatabaseConfig = {
+        'hostname' => 'localhost',
+    }
+
+    # Keep track of whether the database is open
+    @@db = nil
+
 
     ### Load in our custom plugin
     def initialize( *args )
-        super
-        require File.join( File.dirname(__FILE__), 'trac' )
+        super 
+        # :MC: can't load a plugin until the bot has been created
+        require 'trac'
+
+        BogswiddleBot.init_db
+        integrate_with_db
     end
 
 
-    ### Log to the database instead of a file
-    def log( message, where="server" )
-        # :TODO: Log to the database, creating a table for the 'where' if it
-        # doesn't exist already. Or maybe only for public messages?
+    ### Initialize the database
+    def BogswiddleBot.init_db
+        File.open( ConfigFilename ) {|cfg|
+            config = DefaultDatabaseConfig.merge( YAML.load( cfg.read ) )
+            @@db = ActiveRecord::Base.establish_connection( config )
+        } unless @@db
+    end
+
+
+    ### Load saved plugin data, make sure default records exist
+    def integrate_with_db
+        #:TODO:
+    end
+
+
+    ### Keep our log in the database instead of a file
+    def log( m, where="server" )
+        message = Message.new
+        message.channel = Channel.find_or_create( "#{@nick}_log" )
+        message.author = Nick.find_or_create( @nick )
+        message.happened_at = Time.now
+        message.content = m
+        message.save
+    end
+
+
+    ### Log an actual message
+    def log_message( m )
+        message = Message.new
+        message.channel = Channel.find_or_create( m.replyto )   # replyto will be either the channel or the sourcenick
+        message.author = Nick.find_or_create( m.sourcenick )
+        message.content = m.message
+        message.happened_at = m.time
+        message.save
     end
 

trunk/tools/irc/trac.rb

@@ -99 +99 @@
             :last_change => Time.at(0),
             :where => m.channel
-        } #:TODO: these may want to be persistant
+        } #:TODO: these want to be persistant
+    end
+
+    def do_not_feed( m, params )
+        @feeds.delete_if {|url,args|
+            url.match( params[:which].strip )
+        }
     end
 
@@ -106 +112 @@
 plugin = TracPlugin.new
 plugin.map 'trac :what', :action => 'feed_on'
+plugin.map 'stop tracing :which', :action => 'do_not_feed'