zmb 0.1.1

data/plugins/irc.rb

@@ -0,0 +1,230 @@
+require 'socket'
+
+require 'zmb/timer'
+
+class Event
+  attr_accessor :delegate, :command, :args, :name, :userhost, :message
+  
+  def initialize(sender, line)
+    puts line
+    
+    if line[0,1] == ':' then
+      line = line[1..-1] # Remove the :
+      hostname, command, args = line.split(' ', 3)
+      args = "#{hostname} #{args}"
+    else
+      command, args = line.split(' ', 3)
+    end
+    
+    @delegate = sender
+    @command = command.downcase
+    @args = args
+    
+    case @command
+      when 'privmsg'
+        @userhost, @channel, @message = args.split(' ', 3)
+        @name, @userhost = @userhost.split('!', 2)
+        @message = @message[1..-1]
+    end
+  end
+  
+  def private?
+    @channel == @delegate.nick
+  end
+  
+  def sender
+    private? ? @name : @channel
+  end
+  
+  def message?
+    @message != nil
+  end
+  
+  def reply(m)
+    if message? then
+      m = m.split("\n") if not m.respond_to?('each')
+      m.each{ |mess| @delegate.message(sender, mess) }
+    else
+      @delegate.write m
+    end
+  end
+end
+
+class IrcConnection
+  attr_accessor :host, :port, :channels, :nick, :name, :realname, :password, :throttle
+  
+  def initialize(sender, settings={})
+    @delegate = sender
+    
+    @host = settings['host'] if settings.has_key?('host')
+    begin
+      @port = Integer(settings['port']) if settings.has_key?('port')
+    rescue Exception
+      @port = 6667
+    end
+    
+    @channels = Array.new
+    @channels = settings['channels'] if settings.has_key?('channels')
+    
+    @nick = settings['nick'] if settings.has_key?('nick')
+    @name = settings['name'] if settings.has_key?('name')
+    @realname = settings['realname'] if settings.has_key?('realname')
+    @password = settings['password'] if settings.has_key?('password')
+    
+    @throttle = 10
+    @throttle = settings['throttle'] if settings.has_key?('throttle')
+    
+    sender.timer_add(Timer.new(self, :connect, 0.1, false)) if sender.running?
+  end
+  
+  def socket=(value)
+    @socket = value
+  end
+  
+  def to_json(*a)
+    {
+      'host' => @host,
+      'port' => @port,
+      
+      'channels' => @channels,
+      
+      'nick' => @nick,
+      'name' => @name,
+      'realname' => @realname,
+      'password' => @password,
+      
+      'throttle' => @throttle,
+      'plugin' => 'irc',
+    }.to_json(*a)
+  end
+  
+  def self.wizard
+    {
+      'host' => { 'help' => 'Hostname', 'default' => 'localhost' },
+      'port' => { 'help' => 'Port', 'default' => 6667 },
+      'nick' => { 'help' => 'Nickname', 'default' => 'zmb' },
+      'name' => { 'help' => 'Name', 'default' => 'zmb' },
+      'realname' => { 'help' => 'Realname', 'default' => 'zmb' },
+      'password' => { 'help' => 'If the ircd requires a password, enter this here.', 'default' => nil },
+    }
+  end
+  
+  def commands
+    require 'zmb/commands'
+    {
+      'join' => PermCommand.new('admin', self, :join_command),
+      'part' => PermCommand.new('admin', self, :part_command),
+      'raw' => PermCommand.new('admin', self, :raw_command),
+      'tell' => PermCommand.new('admin', self, :tell_command, 2),
+    }
+  end
+  
+  def unloaded
+    write "Quit :ZMB"
+    @socket.close if @socket
+  end
+  
+  def nick=(value)
+    @nick = value
+    write "NICK #{@nick}"
+  end
+  
+  def connected?
+    @socket != nil
+  end
+  
+  def connect
+    if @host and @port and not connected? then
+      Thread.new do
+        @socket = TCPSocket.new(@host, @port)
+        @delegate.socket_add(self, @socket)
+        perform
+      end
+    end
+  end
+  
+  def disconnected(sender, socket)
+    @socket = nil
+    sender.timer_add(Timer.new(self, :connect, @throttle))
+  end
+  
+  def perform
+    write "PASS #{@password}" if @password
+    write "NICK #{@nick}" if @nick
+    write "USER #{@name} 0 0 :#{@realname}" if @name and @realname
+  end
+  
+  def write(line)
+    @socket.write line + "\r\n" if @socket
+  end
+  
+  def message(recipient, msg)
+    write "PRIVMSG #{recipient} :#{msg}"
+  end
+  
+  def join(channel)
+    @channels << channel if not @channels.include?(channel)
+    write "JOIN #{channel}"
+  end
+  
+  def part(channel)
+    @channels.delete(channel) if @channels.include?(channel)
+    write "PART #{channel}"
+  end
+  
+  def received(sender, socket, data)
+    @buffer = '' if @buffer == nil
+    @buffer += data
+    line, @buffer = @buffer.split("\r\n", 2)
+    @buffer = '' if @buffer == nil
+    
+    while line != nil do
+      e = Event.new(self, line)
+      
+      # Catch some events
+      case e.command
+        when 'ping' then write "PONG #{e.args[1..-1]}"
+        when '001' then @channels.each{ |channel| write "JOIN #{channel}" }
+        when 'nick' then tmp, @nick = e.args.split(' :', 2)
+        when '433' then
+          @nick="#{@nick}_"
+          write "NICK #{@nick}"
+      end
+      
+      @delegate.event(self, e)
+      line, @buffer = @buffer.split("\r\n", 2)
+    end
+  end
+  
+  def join_command(e, channel)
+    join channel
+    "#{channel} joined"
+  end
+  
+  def part_command(e, channel)
+    part channel
+    "left #{channel}"
+  end
+  
+  def raw_command(e, line)
+    write line
+    nil
+  end
+  
+  def tell_command(e, to, message)
+    msg = msg.split("\n") if not msg.respond_to?('each')
+    msg.each{ |m| message(to, m) }
+    nil
+  end
+  
+  def running(sender)
+    connect
+  end
+end
+
+Plugin.define do
+  name "irc"
+  description "A plugin which allows you to connect to irc servers."
+  object IrcConnection
+  multi_instances true
+end

data/plugins/quote.rb

@@ -0,0 +1,100 @@
+class Quote
+  attr_accessor :quotes, :autoindex
+  
+  def initialize(sender, settings={})
+    @quotes = Hash.new
+    @quotes = settings['quotes'] if settings.has_key?('quotes')
+    @autoindex = 1
+    @autoindex = settings['autoindex'] if settings.has_key?('autoindex')
+  end
+  
+  def to_json(*a)
+    {
+      'plugin' => 'quote',
+      'quotes' => @quotes,
+      'autoindex' => @autoindex,
+    }.to_json(*a)
+  end
+  
+  def add(quote, username=nil)
+    @quotes["#{@autoindex}"] = {
+      'quote' => quote,
+      'time' => Time.now,
+      'username' => username,
+    }
+    
+    @autoindex += 1
+    @autoindex - 1
+  end
+  
+  def count
+    @quotes.keys.count
+  end
+  
+  def commands
+    require 'zmb/commands'
+    {
+      'quote' => Command.new(self, :quote_command, 1, 'Show a random quote or the quote with matching id'),
+      'quote-add' => Command.new(self, :add_command, 1, 'Add a quote'),
+      'quote-del' => PermCommand.new('quote', self, :del_command, 1, 'Delete a quote by id'),
+      'quote-count' => Command.new(self, :count_command, 0, 'Show amount of quotes'),
+      'quote-last' => Command.new(self, :last_command, 0, 'Show the last quote'),
+      'quote-search' => Command.new(self, :search_command, 1, 'Search to find a quote'),
+    }
+  end
+  
+  def quote_command(e, id=nil)
+    return "quote \##{id} not found" if id and not @quotes.has_key?(id)
+    return "\"#{@quotes[id]['quote']}\" by #{@quotes[id]['username']} at #{@quotes[id]['time']}" if id
+    return "no quotes" if count < 1
+    
+    id = "#{rand(autoindex - 1) + 1}"
+    while not @quotes.has_key?(id)
+      id = "#{rand(autoindex - 1) + 1}"
+    end
+    
+    "\"#{@quotes[id]['quote']}\" by #{@quotes[id]['username']} at #{@quotes[id]['time']}"
+  end
+  
+  def add_command(e, quote)
+    if e.user and e.user.respond_to?('authenticated?') and e.user.authenticated? then
+      "quote added \##{add(quote, e.user.username)}"
+    else
+      'permission denied'
+    end
+  end
+  
+  def del_command(e, id)
+    if @quotes.has_key?(id) then
+      @quotes.delete id
+      "quote #{id} deleted"
+    else
+      "no quote found with id=#{id}"
+    end
+  end
+  
+  def count_command(e)
+    "#{count} quotes"
+  end
+  
+  def last_command(e)
+    return "no quotes" if count < 1
+    quote_command(e, @quotes.keys.sort.reverse[0])
+  end
+  
+  def search_command(e, search)
+    result = @quotes.map{ |id, quote| "#{id}: #{quote['quote']}" if quote['quote'].include?(search) }.reject{ |q| not q }
+    
+    if result.count then
+      result.join("\n")
+    else
+      "no quotes found"
+    end
+  end
+end
+
+Plugin.define do
+  name "quote"
+  description "quote database"
+  object Quote
+end

data/plugins/relay.rb

@@ -0,0 +1,70 @@
+class Relay
+  attr_accessor :relays
+  
+  def initialize(sender, settings={})
+    @relays = settings['relays'] if settings.has_key?('relays')
+    @relays = Hash.new if not @relays
+    
+    @delegate = sender
+  end
+  
+  def to_json(*a)
+    { 'relays' => @relays, 'plugin' => 'relay' }.to_json(*a)
+  end
+  
+  def event(sender, e)
+    if e.message? and @delegate.instances.has_value?(sender) then
+      relay = "#{@delegate.instances.invert[sender]}:#{e.sender}"
+      
+      if @relays.has_key?(relay) then
+        instance, recipient = @relays[relay].split(':', 2)
+        @delegate.instances[instance].message(recipient, "<#{e.name}> #{e.message}") if @delegate.instances.has_key?(instance)
+      end
+    end
+  end
+  
+  def commands
+    require 'zmb/commands'
+    {
+      'relays' => PermCommand.new('admin', self, :relays_command, 0, 'Show all relays'),
+      'relay-add' => PermCommand.new('admin', self, :add_command, 2, 'Add a relay'),
+      'relay-del' => PermCommand.new('admin', self, :del_command, 1, 'Delete a relay'),
+    }
+  end
+  
+  def relays_command(e)
+    if @relays.count < 1 then
+      "no relays"
+    else
+      @relays.map{ |relay, location| "#{relay} > #{location}"}.join("\n")
+    end
+  end
+  
+  def add_command(e, relay, location)
+    if relay.include? ':' and location.include? ':' then
+      if @relays.has_key?(relay) then
+        "relays must be unique"
+      else
+        @relays[relay] = location
+        "relay setup #{relay} => #{location}"
+      end
+    else
+      "relays must be in the format: plugin:location, example (efnet:\#zmb)"
+    end
+  end
+  
+  def del_command(e, relay)
+    if @relays.has_key?(relay) then
+      @relays.delete(relay)
+      "#{relay}: relay deleted"
+    else
+      "no such relay"
+    end
+  end
+end
+
+Plugin.define do
+  name "relay"
+  description "This plugin allows you to relay messages from one channel/server to another."
+  object Relay
+end

data/plugins/users.rb

@@ -0,0 +1,280 @@
+class Event
+  attr_accessor :user
+end
+
+class User
+  require 'digest/sha1'
+  
+  attr_accessor :username, :password, :userhosts, :permissions, :seen
+  
+  def to_json(*a)
+    {'username' => @username, 'password' => @password, 'userhosts' => @userhosts, 'permissions' => @permissions, 'seen' => @seen}.to_json(*a)
+  end
+  
+  def self.create_settings(data)
+    require 'time'
+    user = new(data['username'], data['password'])
+    user.userhosts = data['userhosts'] if data.has_key?('userhosts')
+    user.permissions = data['permissions'] if data.has_key?('permissions')
+    user.seen = Time.parse(data['seen']) if data.has_key?('seen') and data['seen']
+    user
+  end
+  
+  def initialize(username=nil, password=nil, userhost=nil)
+    @username = username
+    @password = Digest::SHA1.hexdigest(password) if password
+    @userhosts = Array.new
+    @userhosts << userhost if userhost
+    @permissions = Array.new
+  end
+  
+  def concat(other_user)
+    @permissions += other_user.permissions
+    @userhosts += other_user.userhosts
+  end
+  
+  def password=(new_password)
+    @password = Digest::SHA1.hexdigest(new_password)
+  end
+  
+  def password?(check_password)
+    @password != nil and @password == Digest::SHA1.hexdigest(check_password)
+  end
+  
+  def authenticated?
+    true
+  end
+  
+  def admin?
+    @permissions.include?("owner") or @permissions.include?("admin")
+  end
+  
+  def permission?(permission)
+    admin? or @permissions.include?(permission)
+  end
+  
+  def permit(permission)
+    @permissions << permission
+  end
+  
+  def deny(permission)
+    @permissions.delete(permission)
+  end
+end
+
+class AnonymousUser
+  def authenticated?
+    false
+  end
+  
+  def permission?(permission)
+    false
+  end
+  
+  def admin?
+    false
+  end
+end
+
+class Users
+  attr_accessor :users
+  
+  def initialize(sender, settings={})
+    @delegate = sender
+    @users = settings['users'].map{ |user| User.create_settings(user) } if settings.has_key?('users')
+    @users = Array.new if not @users
+  end
+  
+  def to_json(*a)
+    {'users' => @users, 'plugin' => 'users'}.to_json(*a)
+  end
+  
+  def user!(search)
+    @users.find{|user| user.userhosts.include?(search) or user.username == search}
+  end
+  
+  def user(search)
+    user!(search) or AnonymousUser.new
+  end
+  
+  def pre_event(sender, e)
+    e.user = user(e.userhost) if e.respond_to?('userhost') and not e.user
+    e.user.seen = Time.now if e.user.respond_to?('seen')
+  end
+  
+  def create_user(username, password=nil, userhost=nil)
+    user = User.new(username, password, userhost)
+    @users << user
+    user
+  end
+  
+  def commands
+    require 'zmb/commands'
+    {
+      'meet' => Command.new(self, :meet, 1, 'Meet a new user'),
+      'forget' => AuthCommand.new(self, :forget, 0, "Forget about a user"),
+      'permit' => PermCommand.new('admin', self, :permit, 2),
+      'deny' => PermCommand.new('admin', self, :deny, 2),
+      'perms' => AuthCommand.new(self, :perms, 1, 'List all permissions a user has.'),
+      'merge' => PermCommand.new('admin', self, :merge, 2, 'Merge two users together'),
+      'password' => AuthCommand.new(self, :password, 1, 'Set the password for your account'),
+      'login' => Command.new(self, :login, 2, 'Add your current host to the username and password provided.'),
+      'logout' => AuthCommand.new(self, :logout, 0, 'Remove your current host from your account.'),
+      'whoami' => Command.new(self, :whoami, 0, 'Who are you logged in as?'),
+      'userhosts' => AuthCommand.new(self, :userhosts, 0, 'List all the hosts associated with your account'),
+      'adduserhost' => AuthCommand.new(self, :adduserhost, 1, 'Add a host to your account'),
+      'deluserhost' => AuthCommand.new(self, :deluserhost, 1, 'Remove a host to your account'),
+      'names' => Command.new(self, :names, 1, 'List all the users'),
+      'seen' => Command.new(self, :seen, 1, 'When was a user last seen'),
+      'sudo' => PermCommand.new('admin', self, :sudo, 2, 'Execute a command as another user.'),
+    }
+  end
+  
+  def meet(e, username=nil)
+    if e.user.admin? and username then
+      return "#{username} already exists" if user!(username)
+      user = create_user(username)
+      "Hello #{username}"
+    elsif not e.user.authenticated? then
+      return "#{e.sender} already exists" if user!(e.sender)
+      if e.respond_to?('userhost') and e.respond_to?('user') then
+        user = create_user(e.name, nil, e.userhost)
+      end
+      "Hello #{user.username}"
+    else
+      "You already have an account #{e.user.username}"
+    end
+  end
+  
+  def forget(e)
+    "user #{@e.users.delete(e.user).username} deleted"
+  end
+  
+  def permit(e, username, permission)
+    if user = user!(username) then
+      user.permit(permission)
+      "permission added"
+    else
+      "#{username} does not exist"
+    end
+  end
+  
+  def deny(e, username, permission)
+    if user = user!(username) then
+      user.deny(permission)
+      'permission removed'
+    else
+      "#{username} does not exist"
+    end
+  end
+  
+  def perms(e, username=nil)
+    e.user.permissions.empty? ? "#{e.user.username} has no permissions" : e.user.permissions.join(', ')
+  end
+  
+  def merge(e, username, other_username)
+    user = user!(username)
+    other_user = user!(other_username)
+    
+    if user and other_user then
+      user.concat other_user 
+      @users.delete other_user
+      "#{other_user.username} now merged into #{user.username}"
+    else
+      'User(s) do not exist'
+    end
+  end
+  
+  def password(e, password)
+    e.user.password = password
+    "#{e.user.username} password has been set to #{password}"
+  end
+  
+  def login(e, username, password)
+    user = user!(username)
+    
+    if e.user.authenticated? then
+      'already logged in'
+    elsif user and user.password?(password) then
+      user.hosts << e.userhost
+      "#{request.hostname} added to your account #{user.username}"
+    else
+      'user and/or password is incorrect'
+    end
+  end
+  
+  def logout(e)
+    e.user.hosts.delete(e.userhost)
+    "userhost #{e.hostname} removed from your account."
+  end
+  
+  def whoami(e)
+    e.user.authenticated? ? "#{e.user.username}" : 'nobody'
+  end
+  
+  def userhosts(e)
+    e.user.userhosts.empty? ?  "#{e.user.username} has no userhosts" : e.user.userhosts.join(', ')
+  end
+  
+  def adduserhost(e, userhost)
+    e.user.userhosts << userhost
+    "#{userhost} added to #{e.user.username}"
+  end
+  
+  def deluserhost(e, userhost)
+    if e.user.userhosts.delete(userhost) then
+      "#{userhost} deleted from #{e.user.username}"
+    else
+      "#{e.user.username} doesn't have #{userhost}"
+    end
+  end
+  
+  def names(e, search=nil)
+    users = @users.map{|user| user.username}
+    users = users.grep(/#{search}/i) if search
+    
+    if users.empty? then
+      "no users found"
+    else
+      users.join(', ')
+    end
+  end
+  
+  def seen(e, username=nil)
+    if user = user!(username) and user.seen then
+      diff = Time.now - user.seen
+      
+      if diff < 60 then
+        msg = "#{Integer(diff)} seconds ago"
+      elsif diff < 3600 then
+        msg = "#{Integer(diff/60)} minutes ago"
+      elsif diff < 86400 then
+        msg = "about #{Integer(diff/3600)} hours ago"
+      else
+        msg = "#{Integer(diff/86400)} days ago"
+      end
+      
+      "#{username} last seen #{msg}"
+    else
+      "#{username} has never been seen"
+    end
+  end
+  
+  def sudo(e, search, command)
+    if user = user!(search) then
+      new_event = e.clone
+      new_event.user = user
+      new_event.message = @delegate.instances['commands'].cc + command
+      @delegate.event(self, new_event)
+      nil
+    else
+      "#{search}: Username or userhost not found"
+    end
+  end
+end
+
+Plugin.define do 
+  name "users"
+  description "users manager"
+  object Users
+end