added some missing CreatedAt() time.Time methods

disgoorg · Aug 16, 2022 · 7295dbe · 7295dbe
1 parent ead63ed
commit 7295dbe
Show file tree

Hide file tree

Showing 5 changed files with 45 additions and 3 deletions.
diff --git a/discord/attachment.go b/discord/attachment.go
@@ -1,6 +1,10 @@
 package discord
 
-import "github.com/disgoorg/snowflake/v2"
+import (
+	"time"
+
+	"github.com/disgoorg/snowflake/v2"
+)
 
 //Attachment is used for files sent in a Message
 type Attachment struct {
@@ -16,6 +20,10 @@ type Attachment struct {
 	Ephemeral   bool         `json:"ephemeral,omitempty"`
 }
 
+func (a Attachment) CreatedAt() time.Time {
+	return a.ID.Time()
+}
+
 type AttachmentUpdate interface {
 	attachmentUpdate()
 }

diff --git a/discord/auto_moderation.go b/discord/auto_moderation.go
@@ -1,6 +1,10 @@
 package discord
 
-import "github.com/disgoorg/snowflake/v2"
+import (
+	"time"
+
+	"github.com/disgoorg/snowflake/v2"
+)
 
 type AutoModerationEventType int
 
@@ -62,6 +66,10 @@ type AutoModerationRule struct {
 	ExemptChannels  []snowflake.ID                `json:"exempt_channels"`
 }
 
+func (r AutoModerationRule) CreatedAt() time.Time {
+	return r.ID.Time()
+}
+
 type AutoModerationRuleCreate struct {
 	Name            string                         `json:"name"`
 	EventType       AutoModerationEventType        `json:"event_type"`

diff --git a/discord/guild_scheduled_event.go b/discord/guild_scheduled_event.go
@@ -25,6 +25,10 @@ type GuildScheduledEvent struct {
 	UserCount          int                        `json:"user_count"`
 }
 
+func (e GuildScheduledEvent) CreatedAt() time.Time {
+	return e.ID.Time()
+}
+
 type GuildScheduledEventCreate struct {
 	ChannelID          snowflake.ID               `json:"channel_id,omitempty"`
 	EntityMetaData     *EntityMetaData            `json:"entity_metadata,omitempty"`

diff --git a/discord/integration.go b/discord/integration.go
@@ -2,6 +2,7 @@ package discord
 
 import (
 	"fmt"
+	"time"
 
 	"github.com/disgoorg/disgo/json"
 	"github.com/disgoorg/snowflake/v2"
@@ -38,6 +39,7 @@ type Integration interface {
 	json.Marshaler
 	Type() IntegrationType
 	ID() snowflake.ID
+	CreatedAt() time.Time
 }
 
 type UnmarshalIntegration struct {
@@ -121,6 +123,10 @@ func (i TwitchIntegration) ID() snowflake.ID {
 	return i.IntegrationID
 }
 
+func (i TwitchIntegration) CreatedAt() time.Time {
+	return i.IntegrationID.Time()
+}
+
 type YouTubeIntegration struct {
 	IntegrationID     snowflake.ID       `json:"id"`
 	Name              string             `json:"name"`
@@ -155,6 +161,10 @@ func (i YouTubeIntegration) ID() snowflake.ID {
 	return i.IntegrationID
 }
 
+func (i YouTubeIntegration) CreatedAt() time.Time {
+	return i.IntegrationID.Time()
+}
+
 type BotIntegration struct {
 	IntegrationID snowflake.ID           `json:"id"`
 	Name          string                 `json:"name"`
@@ -181,3 +191,7 @@ func (BotIntegration) Type() IntegrationType {
 func (i BotIntegration) ID() snowflake.ID {
 	return i.IntegrationID
 }
+
+func (i BotIntegration) CreatedAt() time.Time {
+	return i.IntegrationID.Time()
+}
diff --git a/discord/stage_instance.go b/discord/stage_instance.go
@@ -1,6 +1,10 @@
 package discord
 
-import "github.com/disgoorg/snowflake/v2"
+import (
+	"time"
+
+	"github.com/disgoorg/snowflake/v2"
+)
 
 type StagePrivacyLevel int
 
@@ -18,6 +22,10 @@ type StageInstance struct {
 	DiscoverableDisabled bool              `json:"discoverable_disabled"`
 }
 
+func (e StageInstance) CreatedAt() time.Time {
+	return e.ID.Time()
+}
+
 type StageInstanceCreate struct {
 	ChannelID             snowflake.ID      `json:"channel_id"`
 	Topic                 string            `json:"topic,omitempty"`