tcpbot/plugins/DynPlug.rb

require 'cinch'
require 'open-uri'
require_relative '../classes/Util.rb'
class DynPlug
  include Cinch::Plugin
  include Hooks::ACLHook
  include Util::PluginHelper
  set :prefix, /^:/
  @clist = %w{dynload reload unload load coreload}
  @@commands["dynload"] = ":dynload <plugname> <url> - dynamically load a plugin from <url> using <plugname> as plugin name"
  @@commands["reload"] = ":reload <plugname> - reload plugin source"
  @@commands["unload"] = ":unload <plugname> - unload plugin source"
  @@commands["load"] = ":load <plugname> - load plugin source"
  @@commands["coreload"] = ":coreload - reload core class files"
  @@levelRequired = 10
  @fail = false
  match /dynload ([a-zA-Z][a-zA-Z0-9]+) (.+)/, method: :dynload;
  match /reload ([a-zA-Z][a-zA-Z0-9]+)/, method: :reload;
  match /unload ([a-zA-Z][a-zA-Z0-9]+)/, method: :unload;
  match /load ([a-zA-Z][a-zA-Z0-9]+)/, method: :mload;
  match /coreload/, method: :coreload


  def dynload(m, modname, url) 
    if(!aclcheck(m)) 
      m.reply("#{m.user.nick}: your access level is not high enough for this command.")
      return
    end
    begin
      if(File.exist?("./plugins/#{modname}.rb")) 
        m.reply("plugin with name #{modname} already exists")
        return false;
      end
      content = ""
      begin 
        open(url) do |f|
          content = f.read
          content.gsub!(%r{</?[^>]+?>}, '')
          open("./plugins/#{modname}.rb", "w") do |plugin|
            plugin.write content
          end
        end
      rescue Exception => error
        m.user.send("Error downloading #{modname}: #{error}")
      end
      load "./plugins/#{modname}.rb";
      ibot = Util::BotFamily.instance.get(Util::Util.instance.hton("#{m.bot.config.server}:#{m.bot.config.port}")).bot
      ibot.plugins.register_plugin(Object.const_get(modname))
      m.reply("#{modname} added successfully")
    rescue Exception => error
      m.user.send "Error loading #{modname}: #{error}"
      m.reply "Error loading, deleting downloaded module. Check privmsg for details"
      m.user.send "#{modname}.rb deleted" if File.unlink "./plugins/#{modname}.rb"
    end
  end

  def reload(m, modname) 
    if(!aclcheck(m)) 
      m.reply("#{m.user.nick}: your access level is not high enough for this command.")
      return
    end
    unload(m, modname)
    if(!@fail)
      mload(m, modname)
    else
      @fail = false
    end
  end
  def unload(m, modname) 
    if(!aclcheck(m)) 
      m.reply("#{m.user.nick}: your access level is not high enough for this command.")
      return
    end
    res = ""
    if(File.exist?("./plugins/#{modname}.rb")) 

      ibot = Util::BotFamily.instance.get(Util::Util.instance.hton("#{m.bot.config.server}:#{m.bot.config.port}")).bot
      i = ibot.plugins.find_index { |x| x.class == Kernel.const_get(modname) }
      if(i == nil) 
        res = "#{modname} not loaded currently"
        @fail = true
      else 
        ibot.plugins.unregister_plugin(ibot.bot.plugins[i])
        @fail = false
        res = "#{modname} unloaded successfully"
      end
    else 
      res = "#{modname} not found..."
      @fail = true
    end
    m.reply res
  end

  def mload(m, modname) 
    if(!aclcheck(m)) 
      m.reply("#{m.user.nick}: your access level is not high enough for this command.")
      return
    end
    res = ""
    if(File.exist?("./plugins/#{modname}.rb")) 	
      ibot = Util::BotFamily.instance.get(Util::Util.instance.hton("#{m.bot.config.server}:#{m.bot.config.port}")).bot
      i = ibot.plugins.find_index { |x| x.class == modname }
      if(i != nil) 
        res = "#{modname} already loaded; try :reload instead"
      else 
        begin
          load "./plugins/#{modname}.rb"
          ibot.plugins.register_plugin(Object.const_get(modname))
          res = "#{modname} loaded successfully"
        rescue Exception => error
          res =  "Error loading #{modname}. check privmsg for details"
          m.user.send "Error loading #{modname}: #{error}"
        end
      end	
    else 
      res = "#{modname} not found..."
    end
    m.reply res
  end

  def coreload(m)
    response = "#{m.user.nick}: "
    if(!aclcheck(m)) 
      response << "your access level is not high enough for this command."
    else
      response << "Loading...\n"
      cores = Dir.glob("*.rb", base: "classes/")
      total = cores.length
      count = 0
      msg = []
      cores.each { |core|
        begin
          load "classes/#{core}"
          msg.push("#{core} loaded successfully!")
          count += 1
        rescue Exception => e
          msg.push " failed to reload: #{e}"
        end
      }
      response << msg.join(" // ") << "\n#{count}/#{total} core files reloaded."
    end
    m.reply response
  end





end