ircbot 0.1.0 → 0.1.1

data/Rakefile

@@ -29,7 +29,7 @@ spec = Gem::Specification.new do |s|
   s.require_path = 'lib'
   s.add_dependency('extlib', '>= 0.9.14')
   s.add_dependency('net-irc', '>= 0.0.9')
-  s.files = %w(MIT-LICENSE README Rakefile) + Dir.glob("{lib,spec,plugins}/**/*")
+  s.files = %w(MIT-LICENSE README Rakefile) + Dir.glob("{lib,spec,plugins,config}/**/*")
 end
 
 Rake::GemPackageTask.new(spec) do |pkg|

data/bin/ircbot

@@ -4,19 +4,60 @@ require 'ircbot'
 
 config = nil
 plugin = Pathname(Dir.getwd) + "plugins"
+create = false
 
 while (arg = ARGV.shift)
   case arg
   when /^-f$/		; config = ARGV.shift
-  when /^-c$/		; cpi    = ARGV.shift
+  when /^-c$/		; create = true
   else
     puts "invalid argument: `#{arg}'"
     exit
   end
 end
 
-unless config
-  raise "Specify your config file\nusage: #{$0} -f config/xxx.dat"
+config = Pathname(config.to_s)
+
+
+######################################################################
+### Create config file
+
+if create
+  require 'ircbot/client/config/generator'
+  nick      = config.basename(".*").to_s
+  generator = Ircbot::Client::Config::Generator.new(:nick => nick)
+
+  if nick.empty?
+    nick   = generator.nick
+    config = Pathname("#{nick}.yml")
+  end
+
+  unless config.extname == ".yml"
+    config = Pathname(config.to_s + ".yml")
+  end
+
+  if config.exist?
+    puts "ERROR: Config file (#{config}) already exists."
+    puts "Delete it first, or specify another filename by -f option."
+    exit
+  end
+
+  config.open("w+") do |f|
+    f.print generator.execute
+  end
+  puts "Created #{config}. Run following command.\nircbot -f #{config}"
+  exit
+end
+
+######################################################################
+### Run bot
+
+unless config.extname == ".yml"
+  config = Pathname(config.to_s + ".yml")
+end
+
+unless config.exist?
+  raise "Specify your config file\nusage: #{$0} -f xxx.yml"
 end
 
 Ircbot.push_path(:plugin, plugin)

data/config/sama-zu.yml

@@ -0,0 +1,8 @@
+host: localhost
+port: 6667
+nick: sama-zu
+user: samazu
+real: sama~zu
+channels: "#test"
+plugins: irc plugins
+multiline: true

data/config/wota.yml

@@ -0,0 +1,8 @@
+host: irc.haun.org
+port: 6667
+nick: sama-zu
+user: samazu
+real: sama~zu
+channels: "#wota"
+plugins: irc plugins which
+multiline: true

data/config/yml.erb

@@ -0,0 +1,8 @@
+host: localhost
+port: 6666
+nick: <%= nick %>
+user: <%= nick %>
+real: <%= nick %>
+channels: "#test"
+plugins: irc plugins
+multiline: true

data/lib/ircbot/client/config/generator.rb

@@ -0,0 +1,65 @@
+# -*- coding: utf-8 -*-
+
+module Ircbot
+  class Client
+    class Config
+      class Generator
+        attr_reader :nick
+
+        def initialize(options = {})
+          @nick = options[:nick]
+          @code = options[:code] || read_template
+
+          @nick = current_user + "_bot" if @nick.to_s.empty?
+        end
+
+        def execute
+          require 'erb'
+          erb = ERB.new(@code)
+          erb.result(binding)
+        end
+
+        private
+          def read_template
+            path = Ircbot.paths[:config].first + "yml.erb"
+            return path.read{}
+          end
+
+          def current_user
+            require 'etc'
+            Etc.getlogin
+          rescue Exception
+            "unknown"
+          end
+      end
+      def channels
+        case (val = super)
+        when Array
+          val
+        else
+          val.to_s.split
+        end
+      end
+    end
+
+    ######################################################################
+    ### Event
+
+    def on_rpl_welcome(m)
+      super
+
+      config.channels.each do |channel|
+        post JOIN, channel
+      end
+    end
+
+    ######################################################################
+    ### Command
+
+    def broadcast(text)
+      config.channels.each do |channel|
+        privmsg channel, text
+      end
+    end
+  end
+end

data/lib/ircbot/client/plugins.rb

@@ -49,7 +49,8 @@ module Ircbot
         m.reply(self, reply) if opts[:reply]
       rescue Exception => e
         type = (e.class == RuntimeError) ? 'Error' : "#{e.class}"
-        m.reply(self, "#{type}: #{e.message}")
+        text = (type == e.message) ? '' : e.message
+        m.reply(self, "#{type}: #{text}")
         plugins_rescue_action type, plugin, e
       end
 

data/lib/ircbot/plugins.rb

@@ -27,14 +27,21 @@ module Ircbot
     ######################################################################
     ### Accessors
 
-    def plugin!(name)
-      find_plugin!(name)
+    def plugin!(plugin)
+      case plugin
+      when String, Symbol
+        @plugins[plugin.to_s] or raise PluginNotFound, plugin.to_s
+      when Plugin
+        plugin
+      else
+        raise PluginNotFound, "#{plugin.class}(#{plugin})"
+      end
     end
+    alias :[] :plugin!
 
     def plugin(name)
       plugin!(name) rescue nil
     end
-    alias :[] :plugin!
 
     def bot
       client
@@ -44,15 +51,11 @@ module Ircbot
     ### Operations
 
     def start(plugin)
-      find_plugin(plugin).running = true
+      plugin!(plugin).running = true
     end
 
     def stop(plugin)
-      find_plugin(plugin).running = false
-    end
-
-    def load(plugin)
-      load_plugins(plugin)
+      plugin!(plugin).running = false
     end
 
     def delete(plugin)
@@ -83,9 +86,7 @@ module Ircbot
       when String, Symbol
         begin
           name = plugin.to_s
-          unless @plugins[name]
-            self << load_plugin(name)
-          end
+          self << load(name)
         rescue Exception => e
           broadcast "Plugin error(#{name}): #{e}[#{e.class}]"
         end
@@ -96,7 +97,7 @@ module Ircbot
     end
     alias :<< :load_plugins
 
-    def load_plugin(name)
+    def load(name)
       path = Ircbot.glob_for(:plugin, name).first or
         raise PluginNotFound, name.to_s
 
@@ -106,10 +107,6 @@ module Ircbot
       class_name = Extlib::Inflection.camelize(name) + "Plugin"
       return Object.const_get(class_name).new
 
-#       Object.subclasses_of(Plugin).each do |k|
-#         return k.new if Extlib::Inflection.demodulize(k.name) =~ /^#{class_name}(Plugin)?$/
-#       end
-
     rescue NameError => e
       raise LoadError, "Expected #{path} to define #{class_name} (#{e})"
     end
@@ -123,17 +120,6 @@ module Ircbot
     end
 
     private
-      def find_plugin!(plugin)
-        case plugin
-        when String, Symbol
-          @plugins[plugin.to_s] or raise PluginNotFound, plugin.to_s
-        when Plugin
-          plugin
-        else
-          raise PluginNotFound, "#{plugin.class}(#{plugin})"
-        end
-      end
-
       def broadcast(text)
         p text
       end

data/lib/ircbot/version.rb

@@ -1,4 +1,4 @@
 module Ircbot
-  VERSION  = '0.1.0' unless defined?(Ircbot::VERSION)
+  VERSION  = '0.1.1' unless defined?(Ircbot::VERSION)
   HOMEPAGE = "http://github.com/maiha/ircbot"
 end

data/lib/ircbot.rb

@@ -18,6 +18,7 @@ require "ircbot/framework"
 require "ircbot/version"
 
 Ircbot.push_path(:plugin, Ircbot.system_root + 'plugins')
+Ircbot.push_path(:config, Ircbot.system_root + 'config')
 Ircbot.toplevel_binding = binding
 
 ######################################################################

data/plugins/irc.rb

@@ -4,6 +4,13 @@ require 'rubygems'
 require 'ircbot'
 
 class IrcPlugin < Ircbot::Plugin
+  def help
+    <<-EOF
+[Irc plugin] make bot send irc native commands
+ex) send "!JOIN #test" message to #{config.nick}
+    EOF
+  end
+
   def reply(text)
     if direct?
       if text[0] == ?!

data/plugins/plugins.rb

@@ -20,11 +20,11 @@ commands: load, start, stop, delete
 
       case command.to_s
       when "load", "register"
-        plugins << Array(arg.split)
+        plugins.load arg
         throw :halt, "Loaded #{arg}"
 
       when "delete", "remove"
-        plugins.remove arg
+        plugins.delete arg
         throw :halt, "Removed #{arg}"
 
       when "start"

data/plugins/which.rb

@@ -0,0 +1,27 @@
+#!/usr/bin/env ruby
+# -*- coding: utf-8 -*-
+
+require 'rubygems'
+require 'ircbot'
+require 'chawan'
+
+class WhichPlugin < Ircbot::Plugin
+  def reply(text)
+    case text
+    when /どっち.*(？|\?)/
+      nouns = Chawan.parse($`).compact.noun
+      noun = nouns[rand(nouns.size)]
+      return "#{noun}に大決定や！"
+    else
+      return nil
+    end
+  rescue Chawan::CannotAnalyze
+    return nil
+  end
+end
+
+if $0 == __FILE__
+  plugin = WhichPlugin.new
+  puts plugin.reply("AKBとハロはどっちがいい？")
+  puts plugin.reply("どっち？")
+end

data/spec/plugins_spec.rb

@@ -11,7 +11,9 @@ describe Ircbot::Plugins do
   provide :plugins
   provide :active
   provide :load_plugins
-  provide :load_plugin
+  provide :load
+  provide :plugin!
+  provide :plugin
   provide :[]
   provide :<<
   provide :bot

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: ircbot
 version: !ruby/object:Gem::Version 
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors: 
 - maiha
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-01-09 00:00:00 +09:00
+date: 2010-01-10 00:00:00 +09:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -50,6 +50,7 @@ files:
 - lib/ircbot/client/config.rb
 - lib/ircbot/client/config/channels.rb
 - lib/ircbot/client/config/plugins.rb
+- lib/ircbot/client/config/generator.rb
 - lib/ircbot/client/encoding.rb
 - lib/ircbot/client/commands.rb
 - lib/ircbot/client/plugins.rb
@@ -70,7 +71,11 @@ files:
 - spec/plugins_spec.rb
 - plugins/echo.rb
 - plugins/plugins.rb
+- plugins/which.rb
 - plugins/irc.rb
+- config/wota.yml
+- config/sama-zu.yml
+- config/yml.erb
 has_rdoc: true
 homepage: http://github.com/maiha/ircbot
 licenses: []