zmb 0.2.1 → 0.3.0

data/README.markdown

@@ -26,12 +26,27 @@ You can run zmb in a shell mode to test plugins without even connecting to any i
 
 ### Included plugins
 
-- IRC
-- Quote
+- IRC - Connect to a IRC server
+- Quote - Quotes
+- Poll - Voting system
 - Relay - Relay between servers and/or channels
 - Users - User management
+- Log - Log everything said in a channel
+- GCalc - Execute a expression using google calculator
+- Announce - Send message to a channel automatically in a certain amount of time
+- DNS - Perform DNS, RDNS and whois lookups
+- NickServ - Log into NickServ
+- Security - Hashes, rot13, and morse code
+- Random - Pick a random value from a list, yes or no, coinflip
+- URL - Dpaste, pastie, bitly, tinyurl, isgd
 - Bank - Points system
 
+#### Other features
+
+- Piping commands together
+
+    .help | pastie
+
 ### Support
 
 You can find support at #zmb @ efnet.

data/VERSION

@@ -1 +1 @@
-0.2.1
+0.3.0

data/lib/zmb.rb

@@ -7,6 +7,7 @@ rescue LoadError
   gem 'json'
 end
 
+require 'zmb/utils'
 require 'zmb/plugin'
 require 'zmb/settings'
 require 'zmb/event'

data/lib/zmb/timer.rb

@@ -14,10 +14,14 @@ class Timer
   end
 
   def fire(sender)
-    if @data then
-      @delegate.send @symbol, @data
-    else
-      @delegate.send @symbol
+    begin
+      if @data then
+        @delegate.send @symbol, @data
+      else
+        @delegate.send @symbol
+      end
+    rescue Exception
+      
     end
 
     if not @repeat

data/lib/zmb/utils.rb

@@ -0,0 +1,144 @@
+class String
+  def self.random(len=8)
+    characters = ('a'..'z').to_a + ('1'..'9').to_a
+    (1..len).map{ characters[rand(characters.size)] }.join
+  end
+  
+  def split_seperators
+    if include?("\n") then
+      split("\n").map{ |arg| arg.strip }
+    elsif include?(',') then
+      split(',').map{ |arg| arg.strip }
+    elsif include?(' ') then
+      split(' ')
+    else
+      [self]
+    end
+  end
+  
+  def truncate_words(num)
+    return [self] if size <= num
+    
+    lines = Array.new
+    line = Array.new
+    len = 0
+    
+    split(' ').each do |word|
+      len += 1 unless len == 0
+      len += word.size
+      
+      if not len <= num then
+        lines << line.join(' ')
+        line = Array.new
+        len = word.size
+      end
+      
+      line << word
+    end
+    
+    lines << line.join(' ') if lines.size != 0
+    
+    lines
+  end
+  
+  def plural(amount=2)
+    if amount == 1 then
+      self
+    else
+      self + 's'
+    end
+  end
+  
+  def http(type, q=nil)
+    u = URI.parse(self)
+    
+    http = Net::HTTP.new(u.host, u.port)
+    q = q.to_query_string if q.class == Hash
+    q = u.query unless q
+    
+    http.start do |h|
+      case type
+        when 'get' then h.get(u.path + '?' + q)
+        when 'post' then h.post(u.path, q)
+        when 'head' then h.head(u.path + '?' + q)
+      end
+    end
+  end
+  
+  def get(q={})
+    http('get', q)
+  end
+  
+  def post(q={})
+    http('post', q)
+  end
+end
+
+class Array
+  def split_seperators
+    self
+  end
+  
+  def word_count(seperator='')
+    join(seperator).size
+  end
+  
+  def list_join
+    (size > 2 ? [slice(0..-2).join(', '), last] : self).join(' and ')
+  end
+  
+  def list_count
+    items = {}
+    
+    each do |i|
+      items[i] = 0 unless items.has_key?(i)
+      items[i] += 1
+    end
+    
+    items.map{ |i, c| c == 1 ? i : "#{i} (#{c})" }.join(', ')
+  end
+end
+
+class Hash
+  def to_query_string
+    map { |k, v| 
+      if v.instance_of?(Hash)
+        v.map { |sk, sv|
+          "#{k}[#{sk}]=#{sv}"
+        }.join('&')
+      else
+        "#{k}=#{v}"
+      end
+    }.join('&')
+  end
+end
+
+class Time
+  def since
+    Time.now - self
+  end
+  
+  def since_words
+    s = since
+    m = Array.new
+    s = s * -1 if future = (s < 0) # Is the time in the future?
+    
+    [
+      ['year',   60 * 60 * 24 * 365],
+      ['month',  60 * 60 * 24 * 30],
+      ['week',   60 * 60 * 24 * 7],
+      ['day',    60 * 60 * 24],
+      ['hour',   60 * 60],
+      ['minute', 60],
+      ['second', 1],
+    ].each do |word, t|
+      amount = (s/t).floor
+      s -= amount * t
+      m << "#{amount} #{word.plural(amount)}" unless amount == 0
+    end
+    
+    m << 'now' if m.size == 0
+    
+    m.list_join + (future ? ' left' : '')
+  end
+end

data/plugins/alias.rb

@@ -12,9 +12,17 @@ class Alias
   
   def commands
     {
-      'alias' => [:add, 2, { :permission => 'admin' }],
-      'unalias' => [:del, 1, { :permission => 'admin' }],
-      'aliases' => [:aliases, 0, { :permission => 'admin' }]
+      'alias' => [:add, 2, {
+        :permission => 'admin',
+        :help => 'Create a alias',
+        :example => 'hello? .echo Hello!' }],
+      'unalias' => [:del, 1, {
+        :permission => 'admin',
+        :help => 'Remove a alias',
+        :usage => 'hello?' }],
+      'aliases' => [:aliases, 0, {
+        :permission => 'admin',
+        :help => 'List all aliases' }],
     }
   end
   

data/plugins/announce.rb

@@ -55,13 +55,21 @@ class Announce
       'announcements' => [:announcements, 0, {
         :permission => 'admin',
         :help => 'List all the id\'s for' }],
-      'announcement' => [:announcement, 1, { :permission => 'admin' }],
+      'announcement' => [:announcement, 1, {
+        :permission => 'admin',
+        :help => 'Display infomation about a announcement',
+        :usage => 'id',
+        :example => '1' }],
       'announce' => [:announce, 4, {
         :permission => 'admin',
         :help => 'Add a announcement',
         :usage => 'instance location interval message',
         :example => 'efnet #zmb 600 Check github for the latest updates!' }],
-      'announce-del' => [:announce_del, 1, { :permission => 'admin' }],
+      'announce-del' => [:announce_del, 1, {
+        :permission => 'admin',
+        :help => 'Delete a announcement',
+        :usage => 'id',
+        :example => '1' }],
     }
   end
   

data/plugins/commands.rb

@@ -21,22 +21,47 @@ class Commands
     }
   end
   
+  def escape
+    {
+      '"' => "\000d\000",
+      "'" => "\000s\000",
+      '|' => "\000p\000",
+    }
+  end
+  
   def event(sender, e)
     return if not e.message?
     
     if e.message[0, @cc.length] == @cc then
       line = e.message[@cc.length..-1].clone
+    elsif e.delegate.respond_to?('nick') and e.message[0, (e.delegate.nick.length+2)] == (e.delegate.nick + ': ') then
+      line = e.message[(e.delegate.nick.length+2)..-1].clone
     elsif e.private? then
       line = e.message.clone
+    else
+      return
     end
     
+    return if e.name[0..0] == '*'
+    
+    line.sub!('{time}', Time.now.strftime('%H:%M:%S'))
+    line.sub!('{day}', Time.now.strftime('%d'))
+    line.sub!('{weekday}', Time.now.strftime('%A'))
+    line.sub!('{timezone}', Time.now.strftime('%Z'))
+    line.sub!('{month}', Time.now.strftime('%B'))
+    line.sub!('{year}', Time.now.strftime('%Y'))
+    line.sub!('{username}', e.user.username) if e.respond_to?('user') and e.user.respond_to?('username')
+    line.sub!('{points}', "#{e.bank.balance}") if e.respond_to?('bank') and e.bank.respond_to?('balance')
+    line.sub!('{channel}', e.channel) if e.respond_to?('channel')
+    line.sub!('{name}', e.name) if e.respond_to?('name')
+    line.sub!('{userhost}', e.userhost) if e.respond_to?('userhost')
+    line.sub!('{rand}', String.random)
+    
     # Encode escaped quotation marks and pipes
-    line.gsub!('\"', "\000d\000")
-    line.gsub!("\\'", "\000s\000")
-    line.gsub!('\|', "\000p\000")
+    escape.each{ |k,v| line.gsub!("\\" + k, v) }
     
     # Check there are a even amount of "" and ''
-    if ((line.count("'") % 2) == 1) and ((line.count('""') % 2) == 1) then
+    if ((line.count("'") % 2) == 1) and ((line.count('"') % 2) == 1) then
       return e.reply('Incorrect amount of quotation marks\'s')
     end
     
@@ -51,11 +76,7 @@ class Commands
       args = command.split(/"([^"]*)"|'([^']*)'|\s/).reject{ |x| x.empty? }
       
       # Decode escape quotation marks and pipes inside the args
-      args.each do |arg|
-        arg.gsub!("\000d\000", '"')
-        arg.gsub!("\000s\000", "'")
-        arg.gsub!("\000p\000", '|')
-      end
+      args.each{ |arg| escape.each{ |k,v| arg.gsub!(v, k) } }
       
       cmd = args.delete_at(0)
       args << input if input
@@ -65,8 +86,8 @@ class Commands
     e.reply(input) if input
   end
   
-  def execute(cmd, e, args)
-    return if not @cmds.has_key?(cmd)
+  def execute(cmd, e, args=[])
+    return "#{cmd}: command not found" if not @cmds.has_key?(cmd)
     
     c = @cmds[cmd]
     
@@ -90,7 +111,13 @@ class Commands
     end
     
     begin
-      c[0].send(c[1], e, *args)
+      if c[1].class == Symbol then
+        c[0].send(c[1], e, *args)
+      elsif c[1].class == Proc then
+          c[1].call(e, *args)
+      else
+        "Bad command definition"
+      end
     rescue ArgumentError
       'incorrect arguments'
     rescue Exception
@@ -117,18 +144,6 @@ class Commands
     @cmds = @cmds.reject{ |k,v| v[0] == instance }
   end
   
-  def split_seperators(data)
-    if data.include?("\n") then
-      data.split("\n").map{ |arg| arg.strip }
-    elsif data.include?(',') then
-      data.split(',').map{ |arg| arg.strip }
-    elsif data.include?(' ') then
-      data.split(' ')
-    else
-      data
-    end
-  end
-  
   def commands
     {
       'help' => :help,
@@ -136,12 +151,27 @@ class Commands
       'which' => [:which, 1, { :help => 'Find which plugin handles a command' }],
       'cc' => [:control_command, 1, { :permission => 'admin' }],
       'eval' => [:evaluate, 1, { :permission => 'admin' }],
-      'count' => :count,
+      'ieval' => [:instance_evaluate, 2, { :permission => 'admin' }],
+      'count' => lambda { |e, data| "#{data.split_seperators.size}" },
       'grep' => [:grep, 2],
       'not' => [:not_command, 2],
       'tail' => :tail,
-      'echo' => :echo,
-      'reverse' => :reverse,
+      'echo' => [:echo, 1, { :example => 'Hello, {username}' }],
+      'reverse' => lambda { |e, data| data.reverse },
+      'first' => lambda { |e, data| data.split_seperators.first },
+      'last' => lambda { |e, data| data.split_seperators.last },
+      'sub' => [:sub, 3, {
+        :help => 'Replace all occurances of a pattern',
+        :usage => 'pattern replacement data',
+        :example => 'l * Hello World!' }],
+      'tr' => [:tr, 3, {
+        :help => 'Returns a copy of str with the characters in from_str replaced by the corresponding characters in to_str',
+        :usage => 'from_str to_str data',
+        :example => 'aeiou * hello' }],
+      'downcase' => lambda { |e, data| data.downcase },
+      'upcase' => lambda { |e, data| data.upcase },
+      'swapcase' => lambda { |e, data| data.swapcase },
+      'capitalize' => lambda { |e, data| data.capitalize },
     }
   end
   
@@ -209,28 +239,40 @@ class Commands
     end
   end
   
-  def count(e, data)
-    "#{split_seperators(data).size}"
+  def instance_evaluate(e, inst, string)
+    begin
+      if @delegate.instances.has_key?(inst) then
+        "#{@delegate.instances[inst].instance_eval string}"
+      else
+        "#{inst}: No such instance"
+      end
+    rescue Exception
+      "#{$!.message}\n#{$!.inspect}"
+    end
   end
   
   def grep(e, search, data)
-    split_seperators(data).reject{ |d| not d.include?(search) }.join(', ')
+    data.split_seperators.reject{ |d| not d.include?(search) }.join(', ')
   end
   
   def not_command(e, search, data)
-    split_seperators(data).reject{ |d| d.include?(search) }.join(', ')
+    data.split_seperators.reject{ |d| d.include?(search) }.join(', ')
   end
   
   def tail(e, data)
-    split_seperators(data).reverse[0..2].join(', ')
+    data.split_seperators.reverse[0..2].join(', ')
   end
   
   def echo(e, data)
     "#{data}"
   end
   
-  def reverse(e, data)
-    data.reverse
+  def sub(e, pattern, replacement, data)
+    data.gsub(pattern, replacement)
+  end
+  
+  def tr(e, from_str, to_str, data)
+    data.tr(from_str, to_str)
   end
 end
 

data/plugins/countdown.rb

@@ -0,0 +1,46 @@
+require 'time'
+
+class Countdown
+  attr_accessor :settings
+  
+  def initialize(sender, settings)
+    @settings = { 'countdown' => {} }
+    settings['countdown'].each{ |k,v| @settings['countdown'][k] = Time.parse(v) } if settings.has_key?('countdown')
+  end
+  
+  def commands
+    {
+      'countdown' => [:countdown, 1, { :usage => 'key' }],
+      'countdowns' => [:countdowns, 0],
+      'add-countdown' => [:add, 2, { :permission => 'countdown', :usage => 'key date/time', :example => 'xmas 25-12-2010' }],
+      'rm-countdown' => [:remove, 1, { :permission => 'countdown', :usage => 'key' }]
+    }
+  end
+  
+  def countdown(e, key)
+    if @settings['countdown'].has_key?(key) then
+      @settings['countdown'][key].since_words
+    else
+      "No such countdown"
+    end
+  end
+  
+  def countdowns(e)
+    @settings['countdown'].keys.join(', ')
+  end
+  
+  def add(e, key, time)
+    @settings['countdown'][key] = Time.parse(time)
+    "Countdown #{key} added"
+  end
+  
+  def remove(e, key)
+    @settings['countdown'].delete(key)
+    "Countdown #{key} removed"
+  end
+end
+
+Plugin.define do
+  name 'countdown'
+  object Countdown
+end