vrobot4/source/servers/discord.rb

require 'discordrb'

# A server implementation for Discord using discordrb.
class Sv_Discord < Vrobot4::Server::AudioServer
   # The server type name.
   def self.type() "Discord" end
   Vrobot4::Server.add_server_type self

   attr_reader :bot # The Discordrb::Bot instance.

   # (see Vrobot4::Server::Server#initialize)
   def initialize info
      super

      @id = info["client"]

      @ops = info["admins"] || []
      @hop = info["halfop"] || []

      @bot = Discordrb::Bot.new \
         token:     info["apikey"],
         client_id: info["client"]

      @bot.message do |evt|
         m = Vrobot4::Server::Message.new \
            msg:  evt.message.content,
            user: User.new(evt.user, @ops, @hop),
            chan: Channel.new(evt.channel),
            serv: self,
            reply:   -> (text) {evt.respond text},
            reply_b: -> (text) {evt.respond "```\n#{text}```"}

         handle_text_cmd m
      end
   end

   # (see Vrobot4::Server::AudioServer#voice_quit)
   def voice_quit m
      @bot.voice_destroy m.chan.real.server.id
   end

   # (see Vrobot4::Server::AudioServer#voice_play)
   def voice_play m, fname
      vc = @bot.voice m.chan.real
      raise RuntimeError, "I'm not in a voice channel" unless vc
      vc.play_file fname
   end

   # (see Vrobot4::Server::AudioServer#voice_play_io)
   def voice_play_io m, io
      vc = @bot.voice m.chan.real
      raise ArgumentError, "Invalid IO stream" unless io
      raise RuntimeError, "I'm not in a voice channel" unless vc
      Thread.new do
         vc.play_io io
      end
      # HACK
      sleep 1
      oldst = nil
      while vc.stream_time != oldst
         oldst = vc.stream_time
         sleep 5
      end
   end

   # (see Vrobot4::Server::Server#connect)
   def connect
      @bot.run
   end

   # (see Vrobot4::Server::Server#flags)
   def flags
      ?A
   end

   protected

   # (see Vrobot4::Server::Server#load_permissions)
   def load_permissions pinf
      @mprm = {chan: {}, role: {}, glob: {}}
      pinf.each do |pr|
         mod = Vrobot4::Module.get_module_type(pr["module"])[:type]
         if pr.key? "channel"
            @mprm[:chan][mod] = ChannelPerms.new unless @mprm[:chan].key? mod
            @mprm[:chan][mod][pr["channel"]] = true
         elsif pr.key? "roles"
            @mprm[:role][mod] = pr["roles"]
         elsif pr.key? "enable"
            @mprm[:glob][mod] = pr["enable"]
         end
      end if pinf
   end

   # Helper for channel permissions handling.
   class ChannelPerms
      def initialize() @cprm = {} end

      # @param chan [Sv_Discord::Channel] the channel for this permission
      # @return [Boolean] true if the channel is enabled, false otherwise.
      def [](chan) @cprm[chan.name] or @cprm[chan.real.id] end

      # Sets a channel's permission on/off.
      # @param chan [Sv_Discord::Channel] the channel for this permission
      # @param set [Boolean] if this permission should be set
      # @return [void]
      def []=(chan, set) @cprm[chan] = set end
   end
   private_constant :ChannelPerms

   # A Discord user.
   class User < Vrobot4::Server::User
      attr_reader :real # The Discordrb::User instance.

      # @param user [Discordrb::User] the discord user
      # @param ops [Array] list of operator role IDs
      # @param hop [Array] list of half-operator role IDs
      def initialize user, ops, hop
         @real  = user
         @name  = user.name
         @id    = user.id
         @roles = "v"
         if user.is_a? Discordrb::Member
            if user.owner?
               @roles += "Ooh"
            elsif ops and ops.any? do |role| user.role? role end
               @roles += "oh"
            elsif hop and hop.any? do |role| user.role? role end
               @roles += "h"
            end
         end
      end
   end

   # A Discord channel.
   class Channel < Vrobot4::Server::Channel
      attr_reader :real # The Discordrb::Channel instance.

      # @param chan [Discordrb::Channel] the discord channel
      def initialize chan
         @real = chan
         @name = ?# + chan.name
         @id   = chan.id
      end
   end
end

## EOF