mij-discord 1.0.0

data/lib/mij-discord/data/embed.rb

@@ -0,0 +1,115 @@
+# frozen_string_literal: true
+
+module MijDiscord::Data
+  class Embed
+    attr_reader :type
+
+    attr_reader :title
+
+    attr_reader :description
+
+    attr_reader :url
+
+    attr_reader :timestamp
+
+    attr_reader :color
+
+    attr_reader :footer
+
+    attr_reader :thumbnail
+
+    attr_reader :image
+
+    attr_reader :video
+
+    attr_reader :provider
+
+    attr_reader :author
+
+    attr_reader :fields
+
+    def initialize(data)
+      @type, @url = data['type'], data['url']
+      @title, @description = data['title'], data['description']
+
+      @color = ColorRGB.new(data['color']) if data['color']
+      @timestamp = Time.parse(data['timestamp']).utc if data['timestamp']
+
+      @footer = EmbedFooter.new(data['footer']) if data['footer']
+      @thumbnail = EmbedMedia.new(data['thumbnail']) if data['thumbnail']
+
+      @image = EmbedMedia.new(data['image']) if data['image']
+      @video = EmbedMedia.new(data['video']) if data['video']
+
+      @author = EmbedAuthor.new(data['author']) if data['author']
+      @provider = EmbedProvider.new(data['provider']) if data['provider']
+
+      @fields = data['fields']&.map {|x| EmbedField.new(x) }
+    end
+  end
+
+  class EmbedFooter
+    attr_reader :text
+
+    attr_reader :icon_url
+
+    attr_reader :proxy_icon_url
+
+    def initialize(data)
+      @text, @icon_url = data['text'], data['icon_url']
+      @proxy_icon_url =  data['proxy_icon_url']
+    end
+  end
+
+  class EmbedMedia
+    attr_reader :url
+
+    attr_reader :proxy_url
+
+    attr_reader :width
+
+    attr_reader :height
+
+    def initialize(data)
+      @url, @width, @height = data['url'], data['width'], data['height']
+      @proxy_url = data['proxy_url']
+    end
+  end
+
+  class EmbedAuthor
+    attr_reader :name
+
+    attr_reader :url
+
+    attr_reader :icon_url
+
+    attr_reader :proxy_icon_url
+
+    def initialize(data)
+      @name, @url, @icon_url = data['name'], data['url'], data['icon_url']
+      @proxy_icon_url = data['proxy_icon_url']
+    end
+  end
+
+  class EmbedProvider
+    attr_reader :name
+
+    attr_reader :url
+
+    def initialize(data)
+      @name, @url = data['name'], data['url']
+    end
+  end
+
+  class EmbedField
+    attr_reader :name
+
+    attr_reader :value
+
+    attr_reader :inline
+
+    def initialize(data)
+      @name, @value, @inline = data['name'], data['value'], data['inline']
+    end
+  end
+end

data/lib/mij-discord/data/emoji.rb

@@ -0,0 +1,62 @@
+# frozen_string_literal: true
+
+module MijDiscord::Data
+  class Emoji
+    include IDObject
+
+    attr_reader :name
+
+    attr_reader :server
+
+    attr_reader :roles
+
+    def initialize(data, bot, server)
+      @bot, @server = bot, server
+
+      @id = data['id'].to_i
+      @name = data['name']
+
+      @roles = []
+      if @server && (roles = data['roles'])
+        @roles = roles.map {|ro| @server.role(ro) }
+      end
+    end
+
+    def mention
+      "<:#{@name}:#{@id}>"
+    end
+
+    alias_method :to_s, :mention
+
+    def reaction
+      @id.zero? ? @name : "#{@name}:#{@id}"
+    end
+
+    def icon_url(format = :png)
+      MijDiscord::Core::API.emoji_icon_url(@id, format)
+    end
+
+    def inspect
+      %(<Emoji id=#{@id} name=#{@name}>)
+    end
+  end
+
+  class Reaction
+    attr_reader :id
+
+    attr_reader :name
+
+    attr_reader :count
+
+    attr_reader :me
+    alias_method :me?, :me
+
+    def initialize(data)
+      @me = !!data['me']
+      @count = data['count'] || 1
+
+      @id = data['emoji']['id']&.to_i
+      @name = data['emoji']['name']
+    end
+  end
+end

data/lib/mij-discord/data/invite.rb

@@ -0,0 +1,87 @@
+# frozen_string_literal: true
+
+module MijDiscord::Data
+  class InviteChannel
+    include IDObject
+
+    attr_reader :name
+
+    attr_reader :type
+
+    def initialize(data, bot)
+      @bot = bot
+
+      @id = data['id'].to_i
+      @name = data['name']
+      @type = data['type']
+    end
+  end
+
+  class InviteServer
+    include IDObject
+
+    attr_reader :name
+
+    attr_reader :splash_hash
+
+    def initialize(data, bot)
+      @bot = bot
+
+      @id = data['id'].to_i
+      @name = data['name']
+      @splash_hash = data['splash_hash']
+    end
+  end
+
+  class Invite
+    attr_reader :channel
+
+    attr_reader :server
+
+    attr_reader :code
+
+    attr_reader :max_uses
+    alias_method :uses, :max_uses
+
+    attr_reader :inviter
+    alias_method :user, :inviter
+
+    attr_reader :temporary
+    alias_method :temporary?, :temporary
+
+    attr_reader :revoked
+    alias_method :revoked?, :revoked
+
+    def initialize(data, bot)
+      @bot = bot
+
+      @channel = InviteChannel.new(data['channel'], bot)
+      @server = InviteServer.new(data['guild'], bot)
+
+      @code, @max_uses = data['code'], data['uses']
+      @temporary, @revoked = data['temporary'], data['revoked']
+
+      @inviter = data['inviter'] ? @bot.cache.put_user(data['inviter']) : nil
+    end
+
+    def ==(other)
+      @code == (other.respond_to?(:code) ? other.code : other)
+    end
+
+    alias_method :eql?, :==
+
+    def delete(reason = nil)
+      MijDiscord::Core::API::Invite.delete(@bot.token, @code, reason)
+    end
+
+    alias_method :revoke, :delete
+
+    def invite_url
+      "https://discord.gg/#{@code}"
+    end
+
+    def inspect
+      %(<Invite code=#{@code} uses=#{@max_uses} temporary=#{@temporary} revoked=#{@revoked}>)
+    end
+  end
+end

data/lib/mij-discord/data/member.rb

@@ -0,0 +1,174 @@
+# frozen_string_literal: true
+
+module MijDiscord::Data
+  class Member < DelegateClass(User)
+    include PermissionObject
+
+    attr_reader :bot
+
+    attr_reader :joined_at
+
+    attr_reader :nickname
+    alias_method :nick, :nickname
+
+    attr_reader :roles
+
+    attr_reader :server
+
+    def mute?
+      voice_state_attribute(:mute)
+    end
+
+    def deaf?
+      voice_state_attribute(:deaf)
+    end
+
+    def self_mute?
+      voice_state_attribute(:self_mute)
+    end
+
+    def self_deaf?
+      voice_state_attribute(:self_deaf)
+    end
+
+    def voice_channel
+      voice_state_attribute(:voice_channel)
+    end
+
+    def initialize(data, server, bot)
+      @bot = bot
+
+      @user = @bot.cache.put_user(data['user'])
+      super @user
+
+      raise ArgumentError, 'Cannot create member with no server' unless server || data['guild_id']
+      @server = server || @bot.servers(data['guild_id'])
+
+      @joined_at = Time.parse(data['joined_at']).utc if data['joined_at']
+
+      update_data(data)
+    end
+
+    def update_data(data)
+      @user.update_data(data['user'])
+
+      @nickname = data.fetch('nick', @nickname || '')
+
+      if (roles = data['roles'])
+        @roles = roles.map {|x| @server.role(x) }
+      end
+    end
+
+    def member?
+      true
+    end
+
+    def owner?
+      @server.owner == self
+    end
+
+    def role?(role)
+      role = role.to_id
+      @roles.any? {|x| x.id == role }
+    end
+
+    def set_roles(roles, reason = nil)
+      roles = roles.map(&:to_id)
+      MijDiscord::Core::API::Server.update_member(@bot.token, @server.id, @user.id, reason, roles: roles)
+    end
+
+    def modify_roles(add, remove, reason = nil)
+      add, remove = add.map(&:to_id), remove.map(&:to_id)
+      roles = (@roles.map(&:id) - remove + add).uniq
+      MijDiscord::Core::API::Server.update_member(@bot.tolen, @server.id, @user.id, reason, roles: roles)
+    end
+
+    def add_role(role, reason = nil)
+      if role.is_a?(Array)
+        modify_roles(role, [], reason)
+      else
+        role = role.to_id
+        MijDiscord::Core::API::Server.add_member_role(@bot.token, @server.id, @user.id, role, reason)
+      end
+    end
+
+    def remove_role(role, reason = nil)
+      if role.is_a?(Array)
+        modify_roles([], role, reason)
+      else
+        role = role.to_id
+        MijDiscord::Core::API::Server.remove_member_role(@bot.token, @server.id, @user.id, role, reason)
+      end
+    end
+
+    def highest_role
+      @roles.sort_by(&:position).last
+    end
+
+    def hoist_role
+      roles = @roles.select(&:hoist)
+      roles.sort_by(&:position).last
+    end
+
+    def color_role
+      roles = @roles.select {|x| x.color.to_i.nonzero? }
+      roles.sort_by(&:position).last
+    end
+
+    def color
+      color_role&.color
+    end
+
+    def set_nickname(nick, reason = nil)
+      nick ||= ''
+
+      if @user.current_bot?
+        MijDiscord::Core::API::User.change_own_nickname(@bot.token, @server.id, nick, reason)
+      else
+        MijDiscord::Core::API::Server.update_member(@bot.token, @server.id, @user.id, reason, nick: nick)
+      end
+    end
+
+    alias_method :nickname=, :set_nickname
+
+    def display_name
+      nickname.empty? ? username : nickname
+    end
+
+    def inspect
+      %(<Member user=#{@user.inspect} server=#{@server.id}>)
+    end
+
+    private
+
+    def voice_state_attribute(key)
+      @server.voice_states[@user.id]&.send(key)
+    end
+  end
+
+  class Recipient < DelegateClass(User)
+    attr_reader :joined_at
+
+    attr_reader :nickname
+    alias_method :nick, :nickname
+
+    attr_reader :roles
+
+    attr_reader :server
+
+    def initialize(user, channel, bot)
+      @bot, @channel, @user = bot, channel, user
+      raise ArgumentError, 'Recipient for public channel' unless channel.private?
+
+      super @user
+
+      @mute = @deaf = @self_mute = @self_deaf = false
+      @voice_channel, @server, @roles = nil, nil, []
+      @nickname, @joined_at = '', @channel.creation_time
+    end
+
+    def inspect
+      %(<Recipient user=#{@user.inspect} channel=#{@channel.inspect}>)
+    end
+  end
+end

data/lib/mij-discord/data/message.rb

@@ -0,0 +1,206 @@
+# frozen_string_literal: true
+
+module MijDiscord::Data
+  class Message
+    include IDObject
+
+    attr_reader :bot
+
+    attr_reader :content
+    alias_method :text, :content
+    alias_method :to_s, :content
+
+    attr_reader :author
+    alias_method :user, :author
+
+    attr_reader :channel
+
+    attr_reader :timestamp
+
+    attr_reader :edited_timestamp
+    alias_method :edit_timestamp, :edited_timestamp
+
+    attr_reader :user_mentions
+
+    attr_reader :role_mentions
+
+    attr_reader :mention_everyone
+
+    attr_reader :attachments
+
+    attr_reader :embeds
+
+    # attr_reader :reactions
+
+    attr_reader :tts
+    alias_method :tts?, :tts
+
+    # attr_reader :nonce
+
+    attr_reader :edited
+    alias_method :edited?, :edited
+
+    attr_reader :pinned
+    alias_method :pinned?, :pinned
+
+    attr_reader :webhook_id
+
+    def initialize(data, bot)
+      @bot = bot
+
+      @id = data['id'].to_i
+      @channel = @bot.channel(data['channel_id'])
+
+      # @nonce = data['nonce']
+      @webhook_id = data['webhook_id']&.to_i
+
+      if (author = data['author'])
+        if author['discriminator'] == '0000'
+          @author = @bot.cache.put_user(author)
+        elsif @channel.private?
+          @author = @channel.recipient
+        else
+          member = @channel.server.member(author['id'])
+          @author = member || @bot.user(author['id'])
+        end
+      end
+
+      update_data(data)
+    end
+
+    def update_data(data)
+      @content = data.fetch('content', @content)
+      @pinned = data.fetch('pinned', @pinned)
+      @tts = data.fetch('tts', @tts)
+
+      @timestamp = Time.parse(data['timestamp']).utc if data['timestamp']
+      @edited_timestamp = Time.parse(data['edited_timestamp']).utc if data['edited_timestamp']
+      @edited = !!@edited_timestamp
+
+      @mention_everyone = !!data['mention_everyone']
+
+      # @reactions = []
+      # if (reactions = data['reactions'])
+      #   reactions.each {|x| @reactions << Reaction.new(x) }
+      # end
+
+      @user_mentions = []
+      if (mentions = data['mentions'])
+        mentions.each {|x| @user_mentions << @bot.cache.put_user(x) }
+      end
+
+      @role_mentions = []
+      if @channel.text? && (mentions = data['mention_roles'])
+        mentions.each {|x| @role_mentions << @channel.server.role(x) }
+      end
+
+      @attachments = []
+      if (attachments = data['attachments'])
+        attachments.each {|x| @attachments << Attachment.new(x, self) }
+      end
+
+      @embeds = []
+      if (embeds = data['embeds'])
+        embeds.each {|x| @embeds << Embed.new(x) }
+      end
+    end
+
+    def reply(text: '', embed: nil, tts: false)
+      @channel.send_message(text: text, embed: embed, tts: tts)
+    end
+
+    def reply_file(file, caption: '', tts: false)
+      @channel.send_file(file, caption: caption, tts: tts)
+    end
+
+    def edit(text: '', embed: nil)
+      response = MijDiscord::Core::API::Channel.edit_message(@bot.token, @channel.id, @id,
+        text, [], embed&.to_h)
+      @channel.cache.put_message(JSON.parse(response), update: true)
+    end
+
+    def pin
+      MijDiscord::Core::API::Channel.pin_message(@bot.token, @channel.id, @id)
+      nil
+    end
+
+    def unpin
+      MijDiscord::Core::API::Channel.unpin_message(@bot.token, @channel.id, @id)
+      nil
+    end
+
+    def webhook?
+      !@webhook_id.nil?
+    end
+
+    # def my_reactions
+    #   @reactions.select(&:me)
+    # end
+
+    def create_reaction(reaction)
+      emoji = reaction.respond_to?(:reaction) ? reaction.reaction : reaction
+      MijDiscord::Core::API::Channel.create_reaction(@bot.token, @channel.id, @id, emoji)
+      nil
+    end
+
+    alias_method :add_reaction, :create_reaction
+
+    def reacted_with(reaction)
+      emoji = reaction.respond_to?(:reaction) ? reaction.reaction : reaction
+      response = MijDiscord::Core::API::Channel.get_reactions(@bot.token, @channel.id, @id, emoji)
+      JSON.parse(response).map {|x| @bot.cache.put_user(x) }
+    end
+
+    def delete_reaction(reaction, user: nil)
+      emoji = reaction.respond_to?(:reaction) ? reaction.reaction : reaction
+      if user.nil?
+        MijDiscord::Core::API::Channel.delete_own_reaction(@bot.token, @channel.id, @id, emoji)
+      else
+        MijDiscord::Core::API::Channel.delete_user_reaction(@bot.token, @channel.id, @id, emoji, user.to_id)
+      end
+      nil
+    end
+
+    def clear_reactions
+      MijDiscord::Core::API::Channel.delete_all_reactions(@bot.token, @channel.id, @id)
+      nil
+    end
+
+    def delete
+      MijDiscord::Core::API::Channel.delete_message(@bot.token, @channel.id, @id)
+      nil
+    end
+
+    def inspect
+      %(<Message id=#{@id} timestamp=#{@timestamp} content="#{@content}" author=#{@author.inspect} channel=#{@channel.inspect}>)
+    end
+  end
+
+  class Attachment
+    attr_reader :message
+
+    attr_reader :url
+
+    attr_reader :proxy_url
+
+    attr_reader :filename
+
+    attr_reader :size
+
+    attr_reader :width
+
+    attr_reader :height
+
+    def initialize(data, message)
+      @message = message
+
+      @url, @proxy_url = data['url'], data['proxy_url']
+      @filename, @size = data['filename'], data['size']
+      @width, @height = data['width'], data['height']
+    end
+
+    def image?
+      !@width.nil? && !@height.nil?
+    end
+  end
+end