From 786d62fc6964799942f5723a621fc6ff680fe8bb Mon Sep 17 00:00:00 2001 From: Dan Anglin Date: Thu, 15 Aug 2024 19:58:19 +0100 Subject: [PATCH] checkpoint: add ability to mute and unmute statuses --- internal/client/statuses.go | 46 ++++++++++++++++++++++++++++++++-- internal/executor/executors.go | 4 +++ internal/executor/mute.go | 15 +++++++++++ internal/executor/unmute.go | 15 +++++++++++ internal/printer/status.go | 1 + schema/enbas_cli_schema.json | 2 ++ 6 files changed, 81 insertions(+), 2 deletions(-) diff --git a/internal/client/statuses.go b/internal/client/statuses.go index 6e1a59e..6689d88 100644 --- a/internal/client/statuses.go +++ b/internal/client/statuses.go @@ -240,7 +240,7 @@ func (g *Client) ReblogStatus(statusID string) error { if err := g.sendRequest(params); err != nil { return fmt.Errorf( - "received an error after sending the request to reblog the status; %w", + "received an error after sending the request to reblog the status: %w", err, ) } @@ -261,7 +261,49 @@ func (g *Client) UnreblogStatus(statusID string) error { if err := g.sendRequest(params); err != nil { return fmt.Errorf( - "received an error after sending the request to un-reblog the status; %w", + "received an error after sending the request to un-reblog the status: %w", + err, + ) + } + + return nil +} + +func (g *Client) MuteStatus(statusID string) error { + url := g.Authentication.Instance + baseStatusesPath + "/" + statusID + "/mute" + + params := requestParameters{ + httpMethod: http.MethodPost, + url: url, + requestBody: nil, + contentType: "", + output: nil, + } + + if err := g.sendRequest(params); err != nil { + return fmt.Errorf( + "received an error after sending the request to mute the status: %w", + err, + ) + } + + return nil +} + +func (g *Client) UnmuteStatus(statusID string) error { + url := g.Authentication.Instance + baseStatusesPath + "/" + statusID + "/unmute" + + params := requestParameters{ + httpMethod: http.MethodPost, + url: url, + requestBody: nil, + contentType: "", + output: nil, + } + + if err := g.sendRequest(params); err != nil { + return fmt.Errorf( + "received an error after sending the request to unmute the status: %w", err, ) } diff --git a/internal/executor/executors.go b/internal/executor/executors.go index 4ce5cfc..5b6b01a 100644 --- a/internal/executor/executors.go +++ b/internal/executor/executors.go @@ -326,6 +326,7 @@ type MuteExecutor struct { accountName internalFlag.StringSliceValue muteDuration internalFlag.TimeDurationValue muteNotifications bool + statusID string resourceType string } @@ -346,6 +347,7 @@ func NewMuteExecutor( exe.Var(&exe.accountName, "account-name", "The name of the account") exe.Var(&exe.muteDuration, "mute-duration", "Specify how long the mute should last for. To mute indefinitely, set this to 0s") exe.BoolVar(&exe.muteNotifications, "mute-notifications", false, "Set to true to mute notifications as well as posts") + exe.StringVar(&exe.statusID, "status-id", "", "The ID of the status") exe.StringVar(&exe.resourceType, "type", "", "The type of resource you want to action on (e.g. account, status)") return &exe @@ -570,6 +572,7 @@ type UnmuteExecutor struct { printer *printer.Printer config *config.Config accountName internalFlag.StringSliceValue + statusID string resourceType string } @@ -587,6 +590,7 @@ func NewUnmuteExecutor( exe.Usage = usage.ExecutorUsageFunc("unmute", "Umutes a specific resource (e.g. an account)", exe.FlagSet) exe.Var(&exe.accountName, "account-name", "The name of the account") + exe.StringVar(&exe.statusID, "status-id", "", "The ID of the status") exe.StringVar(&exe.resourceType, "type", "", "The type of resource you want to action on (e.g. account, status)") return &exe diff --git a/internal/executor/mute.go b/internal/executor/mute.go index 2416f76..1696e45 100644 --- a/internal/executor/mute.go +++ b/internal/executor/mute.go @@ -9,6 +9,7 @@ import ( func (m *MuteExecutor) Execute() error { funcMap := map[string]func(*client.Client) error{ resourceAccount: m.muteAccount, + resourceStatus: m.muteStatus, } doFunc, ok := funcMap[m.resourceType] @@ -43,3 +44,17 @@ func (m *MuteExecutor) muteAccount(gtsClient *client.Client) error { return nil } + +func (m *MuteExecutor) muteStatus(gtsClient *client.Client) error { + if m.statusID == "" { + return FlagNotSetError{flagText: flagStatusID} + } + + if err := gtsClient.MuteStatus(m.statusID); err != nil { + return fmt.Errorf("unable to mute the status: %w", err) + } + + m.printer.PrintSuccess("Successfully muted the status.") + + return nil +} diff --git a/internal/executor/unmute.go b/internal/executor/unmute.go index ae19720..487451d 100644 --- a/internal/executor/unmute.go +++ b/internal/executor/unmute.go @@ -9,6 +9,7 @@ import ( func (m *UnmuteExecutor) Execute() error { funcMap := map[string]func(*client.Client) error{ resourceAccount: m.unmuteAccount, + resourceStatus: m.unmuteStatus, } doFunc, ok := funcMap[m.resourceType] @@ -38,3 +39,17 @@ func (m *UnmuteExecutor) unmuteAccount(gtsClient *client.Client) error { return nil } + +func (m *UnmuteExecutor) unmuteStatus(gtsClient *client.Client) error { + if m.statusID == "" { + return FlagNotSetError{flagText: flagStatusID} + } + + if err := gtsClient.UnmuteStatus(m.statusID); err != nil { + return fmt.Errorf("unable to unmute the status: %w", err) + } + + m.printer.PrintSuccess("Successfully unmuted the status.") + + return nil +} diff --git a/internal/printer/status.go b/internal/printer/status.go index 9b847aa..d083e21 100644 --- a/internal/printer/status.go +++ b/internal/printer/status.go @@ -66,6 +66,7 @@ func (p Printer) PrintStatus(status model.Status, userAccountID string) { builder.WriteString("\n" + p.fieldFormat("Boosted: ") + strconv.FormatBool(status.Reblogged)) builder.WriteString("\n" + p.fieldFormat("Liked: ") + strconv.FormatBool(status.Favourited)) builder.WriteString("\n" + p.fieldFormat("Bookmarked: ") + strconv.FormatBool(status.Bookmarked)) + builder.WriteString("\n" + p.fieldFormat("Muted: ") + strconv.FormatBool(status.Muted)) // Status visibility builder.WriteString("\n\n" + p.headerFormat("VISIBILITY:")) diff --git a/schema/enbas_cli_schema.json b/schema/enbas_cli_schema.json index bbed96d..ad94996 100644 --- a/schema/enbas_cli_schema.json +++ b/schema/enbas_cli_schema.json @@ -316,6 +316,7 @@ { "flag": "account-name" }, { "flag": "mute-duration" }, { "flag": "mute-notifications", "default": "false" }, + { "flag": "status-id", "fieldName": "statusID", "default": "" }, { "flag": "type", "fieldName": "resourceType", "default": "" } ], "summary": "Mutes a specific resource (e.g. an account)", @@ -408,6 +409,7 @@ "additionalFields": [], "flags": [ { "flag": "account-name" }, + { "flag": "status-id", "fieldName": "statusID", "default": "" }, { "flag": "type", "fieldName": "resourceType", "default": "" } ], "summary": "Umutes a specific resource (e.g. an account)",