discorb 0.0.1

data/lib/discorb/interaction.rb

@@ -0,0 +1,333 @@
+# frozen_string_literal: true
+
+module Discorb
+  #
+  # Represents a user interaction with the bot.
+  #
+  class Interaction < DiscordModel
+    # @return [Discorb::Snowflake] The ID of the interaction.
+    attr_reader :id
+    # @return [Discorb::Snowflake] The ID of the application that created the interaction.
+    attr_reader :application_id
+    # @return [Symbol] The type of interaction.
+    attr_reader :type
+    # @return [Discorb::Member] The member that created the interaction.
+    attr_reader :member
+    # @return [Discorb::User] The user that created the interaction.
+    attr_reader :user
+    # @return [Integer] The type of interaction.
+    # @note This is always `1` for now.
+    attr_reader :version
+    # @return [String] The token for the interaction.
+    attr_reader :token
+
+    # @!attribute [r] guild
+    #   @macro client_cache
+    #   @return [Discorb::Guild] The guild the interaction took place in.
+    # @!attribute [r] channel
+    #   @macro client_cache
+    #   @return [Discorb::Channel] The channel the interaction took place in.
+    # @!attribute [r] target
+    #   @return [Discorb::User, Discorb::Member] The user or member the interaction took place with.
+
+    @interaction_type = nil
+    @interaction_name = nil
+
+    # @!visibility private
+    def initialize(client, data)
+      @client = client
+      @id = Snowflake.new(data[:id])
+      @application_id = Snowflake.new(data[:application_id])
+      @type = self.class.interaction_name
+      @type_id = self.class.interaction_type
+      @guild_id = data[:guild_id] && Snowflake.new(data[:guild_id])
+      @channel_id = data[:channel_id] && Snowflake.new(data[:channel_id])
+      @member = guild.members[data[:member][:id]] || Member.new(@client, @guild_id, data[:member][:user], data[:member]) if data[:member]
+      @user = @client.users[data[:user][:id]] || User.new(@client, data[:user]) if data[:user]
+      @token = data[:token]
+      @version = data[:version]
+      @defered = false
+      @responded = false
+      _set_data(data[:data])
+    end
+
+    def guild
+      @client.guilds[@guild_id]
+    end
+
+    def channel
+      @client.channels[@channel_id]
+    end
+
+    def target
+      @member || @user
+    end
+
+    alias fired_by target
+
+    def inspect
+      "#<#{self.class} id=#{@id}>"
+    end
+
+    class << self
+      # @!visibility private
+      attr_reader :interaction_type, :interaction_name, :event_name
+
+      # @!visibility private
+      def make_interaction(client, data)
+        descendants.each do |klass|
+          return klass.make_interaction(client, data) if !klass.interaction_type.nil? && klass.interaction_type == data[:type]
+        end
+        client.log.warn("Unknown interaction type #{data[:type]}, initialized Interaction")
+        Interaction.new(client, data)
+      end
+
+      # @!visibility private
+      def descendants
+        ObjectSpace.each_object(Class).select { |klass| klass < self }
+      end
+    end
+
+    #
+    # A module for response with source.
+    #
+    module SourceResponse
+      #
+      # Response with `DEFERRED_CHANNEL_MESSAGE_WITH_SOURCE`(`5`).
+      #
+      # @param [Boolean] hide Whether to hide the response (ephemeral).
+      #
+      def defer_source(hide: false)
+        Async do
+          @client.internet.post("/interactions/#{@id}/#{@token}/callback", {
+            type: 5,
+            data: {
+              flags: (hide ? 1 << 6 : 0),
+            },
+          }).wait
+          @defered = true
+        end
+      end
+
+      #
+      # Response with `CHANNEL_MESSAGE_WITH_SOURCE`(`4`).
+      #
+      # @param [String] content The content of the response.
+      # @param [Boolean] tts Whether to send the message as text-to-speech.
+      # @param [Discorb::Embed] embed The embed to send.
+      # @param [Array<Discorb::Embed>] embeds The embeds to send. (max: 10)
+      # @param [Discorb::AllowedMentions] allowed_mentions The allowed mentions to send.
+      # @param [Array<Discorb::Components>, Array<Array<Discorb::Components>>] components The components to send.
+      # @param [Boolean] hide Whether to hide the response (ephemeral).
+      #
+      def post(content, tts: false, embed: nil, embeds: nil, allowed_mentions: nil, components: nil, hide: false)
+        payload = {}
+        payload[:content] = content if content
+        payload[:tts] = tts
+        tmp_embed = if embed
+            [embed]
+          elsif embeds
+            embeds
+          end
+        payload[:embeds] = tmp_embed.map(&:to_hash) if tmp_embed
+        payload[:allowed_mentions] = allowed_mentions ? allowed_mentions.to_hash(@client.allowed_mentions) : @client.allowed_mentions.to_hash
+        if components
+          tmp_components = []
+          tmp_row = []
+          components.each do |c|
+            case c
+            when Array
+              tmp_components << tmp_row
+              tmp_row = []
+              tmp_components << c
+            when SelectMenu
+              tmp_components << tmp_row
+              tmp_row = []
+              tmp_components << [c]
+            else
+              tmp_row << c
+            end
+          end
+          tmp_components << tmp_row
+          payload[:components] = tmp_components.filter { |c| c.length.positive? }.map { |c| { type: 1, components: c.map(&:to_hash) } }
+        end
+        payload[:flags] = (hide ? 1 << 6 : 0)
+        if @responded
+          @client.internet.post("/webhooks/#{@id}/#{@token}", { type: 4, data: payload }).wait
+        elsif @defered
+          @client.internet.post("/interactions/#{@id}/#{@token}/@original/edit", { type: 4, data: payload }).wait
+        else
+          @client.internet.post("/interactions/#{@id}/#{@token}/callback", { type: 4, data: payload }).wait
+        end
+        @responded = true
+      end
+    end
+
+    #
+    # A module for response with update.
+    #
+    module UpdateResponse
+      #
+      # Response with `DEFERRED_UPDATE_MESSAGE`(`6`).
+      #
+      # @param [Boolean] hide Whether to hide the response (ephemeral).
+      #
+      def defer_update(hide: false)
+        Async do
+          @client.internet.post("/interactions/#{@id}/#{@token}/callback", {
+            type: 7,
+            data: {
+              flags: (hide ? 1 << 6 : 0),
+            },
+          }).wait
+        end
+      end
+
+      #
+      # Response with `UPDATE_MESSAGE`(`7`).
+      #
+      # @param [String] content The content of the response.
+      # @param [Boolean] tts Whether to send the message as text-to-speech.
+      # @param [Discorb::Embed] embed The embed to send.
+      # @param [Array<Discorb::Embed>] embeds The embeds to send. (max: 10)
+      # @param [Discorb::AllowedMentions] allowed_mentions The allowed mentions to send.
+      # @param [Array<Discorb::Components>, Array<Array<Discorb::Components>>] components The components to send.
+      # @param [Boolean] hide Whether to hide the response (ephemeral).
+      #
+      def edit(content, tts: false, embed: nil, embeds: nil, allowed_mentions: nil, components: nil, hide: false)
+        payload = {}
+        payload[:content] = content if content
+        payload[:tts] = tts
+        tmp_embed = if embed
+            [embed]
+          elsif embeds
+            embeds
+          end
+        payload[:embeds] = tmp_embed.map(&:to_hash) if tmp_embed
+        payload[:allowed_mentions] = allowed_mentions ? allowed_mentions.to_hash(@client.allowed_mentions) : @client.allowed_mentions.to_hash
+        if components
+          tmp_components = []
+          tmp_row = []
+          components.each do |c|
+            case c
+            when Array
+              tmp_components << tmp_row
+              tmp_row = []
+              tmp_components << c
+            when SelectMenu
+              tmp_components << tmp_row
+              tmp_row = []
+              tmp_components << [c]
+            else
+              tmp_row << c
+            end
+          end
+          tmp_components << tmp_row
+          payload[:components] = tmp_components.filter { |c| c.length.positive? }.map { |c| { type: 1, components: c.map(&:to_hash) } }
+        end
+        payload[:flags] = (hide ? 1 << 6 : 0)
+        @client.internet.post("/interactions/#{@id}/#{@token}/callback", { type: 6, data: payload }).wait
+      end
+    end
+
+    private
+
+    def _set_data(*)
+      nil
+    end
+  end
+
+  #
+  # Represents a slash command interaction.
+  # @todo Implement this.
+  #
+  class SlashCommandInteraction < Interaction
+    @interaction_type = 2
+    @interaction_name = :slash_command
+
+    def _set_data(data)
+      p data
+    end
+  end
+
+  #
+  # Represents a message component interaction.
+  # @abstract
+  #
+  class MessageComponentInteraction < Interaction
+    include Interaction::SourceResponse
+    include Interaction::UpdateResponse
+    # @return [String] The content of the response.
+    attr_reader :custom_id
+
+    @interaction_type = 3
+    @interaction_name = :message_component
+
+    # @!visibility private
+    def initialize(client, data)
+      super
+      @message = Message.new(@client, data[:message].merge({ member: data[:member] }))
+    end
+
+    class << self
+      # @!visibility private
+      attr_reader :component_type
+
+      # @!visibility private
+      def make_interaction(client, data)
+        nested_classes.each do |klass|
+          return klass.new(client, data) if !klass.component_type.nil? && klass.component_type == data[:type]
+        end
+        client.log.warn("Unknown component type #{data[:component_type]}, initialized Interaction")
+        MessageComponentInteraction.new(client, data)
+      end
+
+      # @!visibility private
+      def nested_classes
+        constants.select { |c| const_get(c).is_a? Class }.map { |c| const_get(c) }
+      end
+    end
+
+    #
+    # Represents a button interaction.
+    #
+    class Button < MessageComponentInteraction
+      @component_type = 2
+      @event_name = :button_click
+      # @return [String] The custom id of the button.
+      attr_reader :custom_id
+
+      private
+
+      def _set_data(data)
+        @custom_id = data[:custom_id]
+      end
+    end
+
+    #
+    # Represents a select menu interaction.
+    #
+    class SelectMenu < MessageComponentInteraction
+      @component_type = 3
+      @event_name = :select_menu_select
+      # @return [String] The custom id of the select menu.
+      attr_reader :custom_id
+      # @return [Array<String>] The selected options.
+      attr_reader :values
+
+      # @!attribute [r] value
+      #   @return [String] The first selected value.
+
+      def value
+        @values[0]
+      end
+
+      private
+
+      def _set_data(data)
+        @custom_id = data[:custom_id]
+        @values = data[:values]
+      end
+    end
+  end
+end

data/lib/discorb/internet.rb

@@ -0,0 +1,285 @@
+# frozen_string_literal: true
+
+require "net/https"
+
+module Discorb
+  #
+  # A class to handle internet requests.
+  # This class is internal use only.
+  #
+  class Internet
+    @nil_body = nil
+
+    # @!visibility private
+    def initialize(client)
+      @client = client
+    end
+
+    #
+    # Execute a GET request.
+    # @macro async
+    #
+    # @param [String] path The path to the resource.
+    # @param [Hash] headers The headers to send with the request.
+    # @param [String] audit_log_reason The audit log reason to send with the request.
+    # @param [Hash] kwargs The keyword arguments.
+    #
+    # @return [Array[Net::HTTPResponse, Hash]] The response and as JSON.
+    # @return [Array[Net::HTTPResponse, nil]] The response was 204.
+    #
+    # @raise [Discorb::HTTPError] The request was failed.
+    #
+    def get(path, headers: nil, audit_log_reason: nil, **kwargs)
+      Async do |task|
+        resp = http.get(get_path(path), get_headers(headers, "", audit_log_reason), **kwargs)
+        rd = resp.body
+        data = if rd.nil? || rd.empty?
+            nil
+          else
+            JSON.parse(rd, symbolize_names: true)
+          end
+        test_error(if resp.code == "429"
+          @client.log.warn "Ratelimit exceeded for #{path}, trying again in #{data[:retry_after]} seconds."
+          task.sleep(data[:retry_after])
+          get(path, headers: headers, audit_log_reason: audit_log_reason, **kwargs).wait
+        else
+          [resp, data]
+        end)
+      end
+    end
+
+    #
+    # Execute a POST request.
+    # @macro async
+    #
+    # @param [String] path The path to the resource.
+    # @param [String, Hash] body The body of the request.
+    # @param [Hash] headers The headers to send with the request.
+    # @param [String] audit_log_reason The audit log reason to send with the request.
+    # @param [Hash] kwargs The keyword arguments.
+    #
+    # @return [Array[Net::HTTPResponse, Hash]] The response and as JSON.
+    # @return [Array[Net::HTTPResponse, nil]] The response was 204.
+    #
+    # @raise [Discorb::HTTPError] The request was failed.
+    #
+    def post(path, body = "", headers: nil, audit_log_reason: nil, **kwargs)
+      Async do |task|
+        resp = http.post(get_path(path), get_body(body), get_headers(headers, body, audit_log_reason), **kwargs)
+        rd = resp.body
+        data = if rd.nil? || rd.empty?
+            nil
+          else
+            JSON.parse(rd, symbolize_names: true)
+          end
+        test_error(if resp.code == "429"
+          task.sleep(data[:retry_after])
+          post(path, body, headers: headers, audit_log_reason: audit_log_reason, **kwargs).wait
+        else
+          [resp, data]
+        end)
+      end
+    end
+
+    #
+    # Execute a PATCH request.
+    # @macro async
+    #
+    # @param [String] path The path to the resource.
+    # @param [String, Hash] body The body of the request.
+    # @param [Hash] headers The headers to send with the request.
+    # @param [String] audit_log_reason The audit log reason to send with the request.
+    # @param [Hash] kwargs The keyword arguments.
+    #
+    # @return [Array[Net::HTTPResponse, Hash]] The response and as JSON.
+    # @return [Array[Net::HTTPResponse, nil]] The response was 204.
+    #
+    # @raise [Discorb::HTTPError] The request was failed.
+    #
+    def patch(path, body = "", headers: nil, audit_log_reason: nil, **kwargs)
+      Async do |task|
+        resp = http.patch(get_path(path), get_body(body), get_headers(headers, body, audit_log_reason), **kwargs)
+        rd = resp.body
+        data = if rd.nil? || rd.empty?
+            nil
+          else
+            JSON.parse(rd, symbolize_names: true)
+          end
+        test_error(if resp.code == "429"
+          task.sleep(data[:retry_after])
+          patch(path, body, headers: headers, audit_log_reason: audit_log_reason, **kwargs).wait
+        else
+          [resp, data]
+        end)
+      end
+    end
+
+    #
+    # Execute a PUT request.
+    # @macro async
+    #
+    # @param [String] path The path to the resource.
+    # @param [String, Hash] body The body of the request.
+    # @param [Hash] headers The headers to send with the request.
+    # @param [String] audit_log_reason The audit log reason to send with the request.
+    # @param [Hash] kwargs The keyword arguments.
+    #
+    # @return [Array[Net::HTTPResponse, Hash]] The response and as JSON.
+    # @return [Array[Net::HTTPResponse, nil]] The response was 204.
+    #
+    # @raise [Discorb::HTTPError] The request was failed.
+    #
+    def put(path, body = "", headers: nil, audit_log_reason: nil, **kwargs)
+      Async do |task|
+        resp = http.put(get_path(path), get_body(body), get_headers(headers, body, audit_log_reason), **kwargs)
+        rd = resp.body
+        data = if rd.nil? || rd.empty?
+            nil
+          else
+            JSON.parse(rd, symbolize_names: true)
+          end
+        test_error(if resp.code == "429"
+          task.sleep(data[:retry_after])
+          put(path, body, headers: headers, audit_log_reason: audit_log_reason, **kwargs).wait
+        else
+          [resp, data]
+        end)
+      end
+    end
+
+    #
+    # Execute a DELETE request.
+    # @macro async
+    #
+    # @param [String] path The path to the resource.
+    # @param [Hash] headers The headers to send with the request.
+    # @param [String] audit_log_reason The audit log reason to send with the request.
+    # @param [Hash] kwargs The keyword arguments.
+    #
+    # @return [Array[Net::HTTPResponse, Hash]] The response and as JSON.
+    # @return [Array[Net::HTTPResponse, nil]] The response was 204.
+    #
+    # @raise [Discorb::HTTPError] The request was failed.
+    #
+    def delete(path, headers: nil, audit_log_reason: nil, **kwargs)
+      Async do |task|
+        resp = http.delete(get_path(path), get_headers(headers, "", audit_log_reason))
+        rd = resp.body
+        data = if rd.nil? || rd.empty?
+            nil
+          else
+            JSON.parse(rd, symbolize_names: true)
+          end
+        test_error(if resp.code == "429"
+          task.sleep(data[:retry_after])
+          delete(path, headers: headers, audit_log_reason: audit_log_reason, **kwargs).wait
+        else
+          [resp, data]
+        end)
+      end
+    end
+
+    def inspect
+      "#<#{self.class} client=#{@client}>"
+    end
+
+    #
+    # A helper method to send multipart/form-data requests.
+    #
+    # @param [Hash] payload The payload to send.
+    # @param [Array<Discorb::File>] files The files to send.
+    #
+    # @return [Array[String, String]] The boundary and body.
+    #
+    def self.multipart(payload, files)
+      boundary = "DiscorbBySevenC7CMultipartFormData#{Time.now.to_f}"
+      str_payloads = [<<~HTTP]
+        Content-Disposition: form-data; name="payload_json"
+        Content-Type: application/json
+
+        #{payload.to_json}
+      HTTP
+      files.each do |single_file|
+        str_payloads << <<~HTTP
+          Content-Disposition: form-data; name="file"; filename="#{single_file.filename}"
+          Content-Type: #{single_file.content_type}
+
+          #{single_file.io.read}
+        HTTP
+      end
+      [boundary, "--#{boundary}\n#{str_payloads.join("\n--#{boundary}\n")}\n--#{boundary}--"]
+    end
+
+    private
+
+    def test_error(ary)
+      resp, data = *ary
+      case resp.code
+      when "400"
+        raise BadRequestError.new(resp, data)
+      when "403"
+        raise ForbiddenError.new(resp, data)
+      when "404"
+        raise NotFoundError.new(resp, data)
+      else
+        [resp, data]
+      end
+    end
+
+    def get_headers(headers, body = "", audit_log_reason = nil)
+      ret = if body.nil? || body.empty?
+          { "User-Agent" => USER_AGENT, "authorization" => "Bot #{@client.token}" }
+        else
+          { "User-Agent" => USER_AGENT, "authorization" => "Bot #{@client.token}",
+            "content-type" => "application/json" }
+        end
+      ret.merge(headers) if !headers.nil? && headers.length.positive?
+      ret["X-Audit-Log-Reason"] = audit_log_reason unless audit_log_reason.nil?
+      ret
+    end
+
+    def get_body(body)
+      if body.nil?
+        ""
+      elsif body.is_a?(String)
+        body
+      else
+        recr_utf8(body).to_json
+      end
+    end
+
+    def get_path(path)
+      full_path = if path.start_with?("https://")
+          path
+        else
+          API_BASE_URL + path
+        end
+      URI(full_path).path
+    end
+
+    def http
+      https = Net::HTTP.new("discord.com", 443)
+      https.use_ssl = true
+      https
+    end
+
+    def recr_utf8(data)
+      case data
+      when Hash
+        data.each do |k, v|
+          data[k] = recr_utf8(v)
+        end
+        data
+      when Array
+        data.each_index do |i|
+          data[i] = recr_utf8(data[i])
+        end
+        data
+      when String
+        data.dup.force_encoding(Encoding::UTF_8)
+      else
+        data
+      end
+    end
+  end
+end