discord_rda 0.1.3 → 0.2.0

data/lib/discord_rda/entity/support.rb

@@ -291,4 +291,291 @@ module DiscordRDA
       @user ||= User.new(@raw_data['user']) if @raw_data['user']
     end
   end
+
+  class Integration < Entity
+    attribute :name, type: :string
+    attribute :type, type: :string
+    attribute :enabled, type: :boolean, default: false
+    attribute :syncing, type: :boolean, default: false
+    attribute :role_id, type: :snowflake
+    attribute :enable_emoticons, type: :boolean, default: false
+    attribute :expire_behavior, type: :integer
+    attribute :expire_grace_period, type: :integer
+    attribute :subscriber_count, type: :integer
+    attribute :revoked, type: :boolean, default: false
+    attribute :scopes, type: :array, default: []
+    attribute :guild_id, type: :snowflake
+    attribute :application_id, type: :snowflake
+
+    def user
+      @raw_data['user'] ? User.new(@raw_data['user']) : nil
+    end
+
+    def account
+      @raw_data['account'] || {}
+    end
+
+    def synced_at
+      @raw_data['synced_at'] ? Time.parse(@raw_data['synced_at']) : nil
+    end
+
+    def application
+      @raw_data['application'] ? Application.new(@raw_data['application']) : nil
+    end
+  end
+
+  class Presence < Entity
+    attribute :guild_id, type: :snowflake
+    attribute :status, type: :string
+
+    def user
+      @raw_data['user'] ? User.new(@raw_data['user']) : nil
+    end
+
+    def user_id
+      user&.id
+    end
+
+    def activities
+      @raw_data['activities'] || []
+    end
+
+    def client_status
+      @raw_data['client_status'] || {}
+    end
+
+    def desktop_status
+      client_status['desktop']
+    end
+
+    def mobile_status
+      client_status['mobile']
+    end
+
+    def web_status
+      client_status['web']
+    end
+
+    def online?
+      status == 'online'
+    end
+
+    def idle?
+      status == 'idle'
+    end
+
+    def dnd?
+      status == 'dnd'
+    end
+
+    def offline?
+      status == 'offline'
+    end
+  end
+
+  class VoiceState < Entity
+    attribute :guild_id, type: :snowflake
+    attribute :channel_id, type: :snowflake
+    attribute :user_id, type: :snowflake
+    attribute :session_id, type: :string
+    attribute :deaf, type: :boolean, default: false
+    attribute :mute, type: :boolean, default: false
+    attribute :self_deaf, type: :boolean, default: false
+    attribute :self_mute, type: :boolean, default: false
+    attribute :self_stream, type: :boolean, default: false
+    attribute :self_video, type: :boolean, default: false
+    attribute :suppress, type: :boolean, default: false
+
+    def member
+      return nil unless @raw_data['member']
+
+      Member.new(@raw_data['member'].merge('guild_id' => @raw_data['guild_id']))
+    end
+
+    def request_to_speak_timestamp
+      @raw_data['request_to_speak_timestamp'] ? Time.parse(@raw_data['request_to_speak_timestamp']) : nil
+    end
+
+    def connected?
+      !channel_id.nil?
+    end
+  end
+
+  class VoiceServer < Entity
+    attribute :token, type: :string
+    attribute :guild_id, type: :snowflake
+    attribute :endpoint, type: :string
+  end
+
+  class Webhook < Entity
+    TYPES = {
+      incoming: 1,
+      channel_follower: 2,
+      application: 3
+    }.freeze
+
+    attribute :type, type: :integer
+    attribute :guild_id, type: :snowflake
+    attribute :channel_id, type: :snowflake
+    attribute :name, type: :string
+    attribute :avatar, type: :string
+    attribute :token, type: :string
+    attribute :application_id, type: :snowflake
+    attribute :url, type: :string
+
+    def user
+      @raw_data['user'] ? User.new(@raw_data['user']) : nil
+    end
+
+    def source_guild
+      @raw_data['source_guild'] ? Guild.new(@raw_data['source_guild']) : nil
+    end
+
+    def source_channel
+      @raw_data['source_channel'] ? Channel.new(@raw_data['source_channel']) : nil
+    end
+
+    def incoming?
+      type == 1
+    end
+
+    def channel_follower?
+      type == 2
+    end
+
+    def application?
+      type == 3
+    end
+  end
+
+  class StageInstance < Entity
+    attribute :guild_id, type: :snowflake
+    attribute :channel_id, type: :snowflake
+    attribute :topic, type: :string
+    attribute :privacy_level, type: :integer
+    attribute :discoverable_disabled, type: :boolean, default: false
+    attribute :guild_scheduled_event_id, type: :snowflake
+
+    def guild_only?
+      privacy_level == 2
+    end
+  end
+
+  class Entitlement < Entity
+    attribute :sku_id, type: :snowflake
+    attribute :application_id, type: :snowflake
+    attribute :user_id, type: :snowflake
+    attribute :guild_id, type: :snowflake
+    attribute :type, type: :integer
+    attribute :deleted, type: :boolean, default: false
+    attribute :consumed, type: :boolean, default: false
+
+    def starts_at
+      @raw_data['starts_at'] ? Time.parse(@raw_data['starts_at']) : nil
+    end
+
+    def ends_at
+      @raw_data['ends_at'] ? Time.parse(@raw_data['ends_at']) : nil
+    end
+
+    def active?(at: Time.now.utc)
+      return false if deleted
+      return false if starts_at && at < starts_at
+      return false if ends_at && at > ends_at
+
+      true
+    end
+
+    def owner_id
+      user_id || guild_id
+    end
+
+    def owner_type
+      user_id ? :user : :guild
+    end
+  end
+
+  class Application < Entity
+    attribute :name, type: :string
+    attribute :icon, type: :string
+    attribute :description, type: :string
+    attribute :rpc_origins, type: :array, default: []
+    attribute :bot_public, type: :boolean, default: true
+    attribute :bot_require_code_grant, type: :boolean, default: false
+    attribute :terms_of_service_url, type: :string
+    attribute :privacy_policy_url, type: :string
+    attribute :summary, type: :string
+    attribute :verify_key, type: :string
+    attribute :cover_image, type: :string
+    attribute :flags, type: :integer, default: 0
+    attribute :approximate_guild_count, type: :integer
+    attribute :redirect_uris, type: :array, default: []
+    attribute :interactions_endpoint_url, type: :string
+    attribute :role_connections_verification_url, type: :string
+    attribute :tags, type: :array, default: []
+    attribute :custom_install_url, type: :string
+    attribute :install_params, type: :hash
+
+    def owner
+      @owner ||= User.new(@raw_data['owner']) if @raw_data['owner']
+    end
+
+    def team
+      @team ||= Team.new(@raw_data['team']) if @raw_data['team']
+    end
+  end
+
+  class Team < Entity
+    attribute :icon, type: :string
+    attribute :name, type: :string
+    attribute :owner_user_id, type: :snowflake
+
+    def members
+      (@raw_data['members'] || []).map do |member|
+        user = member['user'] ? User.new(member['user']) : nil
+        member.merge('user' => user)
+      end
+    end
+  end
+
+  class AuditLogEntry < Entity
+    attribute :target_id, type: :string
+    attribute :changes, type: :array, default: []
+    attribute :user_id, type: :snowflake
+    attribute :action_type, type: :integer
+    attribute :options, type: :hash
+    attribute :reason, type: :string
+
+    def user(users_index = nil)
+      return users_index[user_id.to_s] if users_index && user_id
+
+      @user ||= User.new(@raw_data['user']) if @raw_data['user']
+    end
+  end
+
+  class AuditLog
+    attr_reader :entries, :users, :webhooks, :integrations, :threads
+
+    def initialize(data = {})
+      @entries = (data['audit_log_entries'] || []).map { |entry| AuditLogEntry.new(entry) }
+      @users = (data['users'] || []).map { |user| User.new(user) }
+      @webhooks = data['webhooks'] || []
+      @integrations = data['integrations'] || []
+      @threads = (data['threads'] || []).map { |thread| Channel.new(thread) }
+      @application_commands = data['application_commands'] || []
+      @auto_moderation_rules = data['auto_moderation_rules'] || []
+    end
+
+    def users_index
+      @users_index ||= @users.each_with_object({}) { |user, index| index[user.id.to_s] = user }
+    end
+
+    def application_commands
+      @application_commands
+    end
+
+    def auto_moderation_rules
+      @auto_moderation_rules
+    end
+  end
 end

data/lib/discord_rda/event/base.rb

@@ -23,10 +23,9 @@ module DiscordRDA
     # @param shard_id [Integer] Shard ID
     def initialize(type, data, shard_id: 0)
       @type = type.to_s
-      @data = data.freeze
+      @data = data.each_with_object({}) { |(key, value), hash| hash[key.to_s] = value }.freeze
       @shard_id = shard_id
       @timestamp = Time.now.utc.freeze
-      freeze
     end
 
     # Get creation time from data if available
@@ -152,7 +151,7 @@ module DiscordRDA
       'ENTITLEMENT_CREATE' => 'EntitlementCreateEvent',
       'ENTITLEMENT_UPDATE' => 'EntitlementUpdateEvent',
       'ENTITLEMENT_DELETE' => 'EntitlementDeleteEvent'
-    }.freeze
+    }
 
     class << self
       # Create an event from Gateway data
@@ -161,10 +160,10 @@ module DiscordRDA
       # @param shard_id [Integer] Shard ID
       # @return [Event] Event instance
       def create(event_type, data, shard_id = 0)
-        class_name = EVENT_CLASSES[event_type]
+        class_name = event_classes[event_type.to_s]
 
-        if class_name && Event.const_defined?(class_name)
-          Event.const_get(class_name).new(data, shard_id: shard_id)
+        if class_name && DiscordRDA.const_defined?(class_name, false)
+          DiscordRDA.const_get(class_name, false).new(data, shard_id: shard_id)
         else
           Event.new(event_type, data, shard_id: shard_id)
         end
@@ -175,7 +174,13 @@ module DiscordRDA
       # @param klass [Class] Event class
       # @return [void]
       def register(event_type, klass)
-        EVENT_CLASSES[event_type] = klass.name
+        event_classes[event_type.to_s] = klass.name.split('::').last
+      end
+
+      private
+
+      def event_classes
+        @event_classes ||= EVENT_CLASSES.dup
       end
     end
   end
@@ -983,4 +988,250 @@ module DiscordRDA
       @data['removed_member_ids'] || []
     end
   end
+
+  class GuildIntegrationsUpdateEvent < Event
+    def initialize(data, shard_id:)
+      super('GUILD_INTEGRATIONS_UPDATE', data, shard_id: shard_id)
+    end
+
+    def guild_id
+      @data['guild_id'] ? Snowflake.new(@data['guild_id']) : nil
+    end
+  end
+
+  class PresenceUpdateEvent < Event
+    def initialize(data, shard_id:)
+      super('PRESENCE_UPDATE', data, shard_id: shard_id)
+    end
+
+    def presence
+      Presence.new(@data)
+    end
+
+    def user
+      presence.user
+    end
+
+    def guild_id
+      presence.guild_id
+    end
+
+    def status
+      presence.status
+    end
+
+    def activities
+      presence.activities
+    end
+
+    def client_status
+      presence.client_status
+    end
+  end
+
+  class TypingStartEvent < Event
+    def initialize(data, shard_id:)
+      super('TYPING_START', data, shard_id: shard_id)
+    end
+
+    def channel_id
+      @data['channel_id'] ? Snowflake.new(@data['channel_id']) : nil
+    end
+
+    def guild_id
+      @data['guild_id'] ? Snowflake.new(@data['guild_id']) : nil
+    end
+
+    def user_id
+      @data['user_id'] ? Snowflake.new(@data['user_id']) : nil
+    end
+
+    def member
+      return nil unless @data['member']
+
+      Member.new(@data['member'].merge('guild_id' => @data['guild_id']))
+    end
+
+    def started_at
+      Time.at(@data['timestamp'].to_i).utc if @data['timestamp']
+    end
+
+    def guild?
+      !guild_id.nil?
+    end
+  end
+
+  class UserUpdateEvent < Event
+    def initialize(data, shard_id:)
+      super('USER_UPDATE', data, shard_id: shard_id)
+    end
+
+    def user
+      User.new(@data)
+    end
+  end
+
+  class VoiceStateUpdateEvent < Event
+    def initialize(data, shard_id:)
+      super('VOICE_STATE_UPDATE', data, shard_id: shard_id)
+    end
+
+    def voice_state
+      VoiceState.new(@data)
+    end
+
+    def guild_id
+      voice_state.guild_id
+    end
+
+    def channel_id
+      voice_state.channel_id
+    end
+
+    def user_id
+      voice_state.user_id
+    end
+
+    def member
+      voice_state.member
+    end
+
+    def session_id
+      voice_state.session_id
+    end
+  end
+
+  class VoiceServerUpdateEvent < Event
+    def initialize(data, shard_id:)
+      super('VOICE_SERVER_UPDATE', data, shard_id: shard_id)
+    end
+
+    def server
+      VoiceServer.new(@data)
+    end
+
+    def guild_id
+      server.guild_id
+    end
+
+    def token
+      server.token
+    end
+
+    def endpoint
+      server.endpoint
+    end
+  end
+
+  class WebhooksUpdateEvent < Event
+    def initialize(data, shard_id:)
+      super('WEBHOOKS_UPDATE', data, shard_id: shard_id)
+    end
+
+    def guild_id
+      @data['guild_id'] ? Snowflake.new(@data['guild_id']) : nil
+    end
+
+    def channel_id
+      @data['channel_id'] ? Snowflake.new(@data['channel_id']) : nil
+    end
+  end
+
+  class StageInstanceCreateEvent < Event
+    def initialize(data, shard_id:)
+      super('STAGE_INSTANCE_CREATE', data, shard_id: shard_id)
+    end
+
+    def stage_instance
+      StageInstance.new(@data)
+    end
+
+    def guild_id
+      stage_instance.guild_id
+    end
+
+    def channel_id
+      stage_instance.channel_id
+    end
+  end
+
+  class StageInstanceUpdateEvent < Event
+    def initialize(data, shard_id:)
+      super('STAGE_INSTANCE_UPDATE', data, shard_id: shard_id)
+    end
+
+    def stage_instance
+      StageInstance.new(@data)
+    end
+
+    def guild_id
+      stage_instance.guild_id
+    end
+
+    def channel_id
+      stage_instance.channel_id
+    end
+  end
+
+  class StageInstanceDeleteEvent < Event
+    def initialize(data, shard_id:)
+      super('STAGE_INSTANCE_DELETE', data, shard_id: shard_id)
+    end
+
+    def stage_instance
+      StageInstance.new(@data)
+    end
+
+    def guild_id
+      stage_instance.guild_id
+    end
+
+    def channel_id
+      stage_instance.channel_id
+    end
+  end
+
+  class GuildAuditLogEntryCreateEvent < Event
+    def initialize(data, shard_id:)
+      super('GUILD_AUDIT_LOG_ENTRY_CREATE', data, shard_id: shard_id)
+    end
+
+    def entry
+      AuditLogEntry.new(@data)
+    end
+
+    def guild_id
+      @data['guild_id'] ? Snowflake.new(@data['guild_id']) : nil
+    end
+  end
+
+  class EntitlementCreateEvent < Event
+    def initialize(data, shard_id:)
+      super('ENTITLEMENT_CREATE', data, shard_id: shard_id)
+    end
+
+    def entitlement
+      Entitlement.new(@data)
+    end
+  end
+
+  class EntitlementUpdateEvent < Event
+    def initialize(data, shard_id:)
+      super('ENTITLEMENT_UPDATE', data, shard_id: shard_id)
+    end
+
+    def entitlement
+      Entitlement.new(@data)
+    end
+  end
+
+  class EntitlementDeleteEvent < Event
+    def initialize(data, shard_id:)
+      super('ENTITLEMENT_DELETE', data, shard_id: shard_id)
+    end
+
+    def entitlement
+      Entitlement.new(@data)
+    end
+  end
 end

data/lib/discord_rda/interactions/application_command.rb

@@ -5,6 +5,8 @@ module DiscordRDA
   # Supports Chat Input Commands (slash), User Commands, and Message Commands.
   #
   class ApplicationCommand < Entity
+    EMPTY_EXECUTION_POLICY = {}.freeze
+
     # Command types
     TYPES = {
       chat_input: 1,      # Slash commands
@@ -116,6 +118,10 @@ module DiscordRDA
       @handler
     end
 
+    def execution_policy
+      @execution_policy || EMPTY_EXECUTION_POLICY
+    end
+
     # Set the handler block
     # @param block [Proc] Handler block
     def handler=(block)
@@ -204,9 +210,10 @@ module DiscordRDA
 
   # Builder for creating application commands with DSL
   class CommandBuilder
-    def initialize(name, description = nil)
+    def initialize(name, description = nil, type: 1)
       @name = name
       @description = description || ''
+      @type = type
       @options = []
       @name_localizations = {}
       @description_localizations = {}
@@ -214,6 +221,15 @@ module DiscordRDA
       @dm_permission = true
       @nsfw = false
       @handler = nil
+      @execution_policy = {}
+    end
+
+    # Set command type
+    # @param type [Integer, Symbol] Discord command type
+    # @return [self]
+    def type(type)
+      @type = type.is_a?(Symbol) ? ApplicationCommand::TYPES.fetch(type) : type
+      self
     end
 
     # Set localized name
@@ -346,8 +362,7 @@ module DiscordRDA
     # @param description [String] Subcommand description
     # @yield [CommandBuilder] Block for building subcommand options
     def subcommand(name, description, &block)
-      builder = CommandBuilder.new(name, description)
-      builder.instance_variable_set(:@type, 1) # sub_command type
+      builder = CommandBuilder.new(name, description, type: 1)
       block.call(builder) if block
       @options << builder.to_h
       self
@@ -358,8 +373,7 @@ module DiscordRDA
     # @param description [String] Group description
     # @yield [CommandBuilder] Block for building subcommands in this group
     def group(name, description, &block)
-      builder = CommandBuilder.new(name, description)
-      builder.instance_variable_set(:@type, 2) # sub_command_group type
+      builder = CommandBuilder.new(name, description, type: 2)
       block.call(builder) if block
       @options << builder.to_h
       self
@@ -397,6 +411,27 @@ module DiscordRDA
       self
     end
 
+    def timeout(seconds)
+      @execution_policy[:timeout_seconds] = seconds.to_f
+      self
+    end
+
+    def max_concurrency(value)
+      @execution_policy[:max_concurrency] = value.to_i
+      self
+    end
+
+    def circuit_breaker(failures:, cooldown:)
+      @execution_policy[:failure_threshold] = failures.to_i
+      @execution_policy[:cooldown_seconds] = cooldown.to_f
+      self
+    end
+
+    def execution_policy(**policy)
+      @execution_policy.merge!(policy)
+      self
+    end
+
     # Convert to hash for API
     # @return [Hash] Command hash
     def to_h
@@ -408,6 +443,7 @@ module DiscordRDA
         options: @options.empty? ? nil : @options,
         default_member_permissions: @default_member_permissions,
         dm_permission: @dm_permission,
+        type: @type,
         nsfw: @nsfw
       }.compact
     end
@@ -415,8 +451,10 @@ module DiscordRDA
     # Build and return ApplicationCommand
     # @return [ApplicationCommand] Command instance
     def build
-      cmd = ApplicationCommand.new(to_h)
+      cmd = ApplicationCommand.new(to_h).dup
       cmd.handler = @handler
+      cmd.instance_variable_set(:@execution_policy, @execution_policy.dup)
+      cmd.freeze
       cmd
     end