92 lines
2.5 KiB
Go
92 lines
2.5 KiB
Go
|
// SPDX-FileCopyrightText: 2024 Dan Anglin <d.n.i.anglin@gmail.com>
|
||
|
//
|
||
|
// SPDX-License-Identifier: GPL-3.0-or-later
|
||
|
|
||
|
package executor
|
||
|
|
||
|
import (
|
||
|
"flag"
|
||
|
"fmt"
|
||
|
|
||
|
"codeflow.dananglin.me.uk/apollo/enbas/internal/client"
|
||
|
)
|
||
|
|
||
|
type AcceptOrRejectExecutor struct {
|
||
|
*flag.FlagSet
|
||
|
|
||
|
topLevelFlags TopLevelFlags
|
||
|
resourceType string
|
||
|
accountName string
|
||
|
action string
|
||
|
}
|
||
|
|
||
|
func NewAcceptOrRejectExecutor(tlf TopLevelFlags, name, summary string) *AcceptOrRejectExecutor {
|
||
|
acceptExe := AcceptOrRejectExecutor{
|
||
|
FlagSet: flag.NewFlagSet(name, flag.ExitOnError),
|
||
|
|
||
|
topLevelFlags: tlf,
|
||
|
action: name,
|
||
|
}
|
||
|
|
||
|
acceptExe.StringVar(&acceptExe.resourceType, flagType, "", "Specify the type of resource to accept or reject")
|
||
|
acceptExe.StringVar(&acceptExe.accountName, flagAccountName, "", "Specify the account name in full (username@domain)")
|
||
|
|
||
|
acceptExe.Usage = commandUsageFunc(name, summary, acceptExe.FlagSet)
|
||
|
|
||
|
return &acceptExe
|
||
|
}
|
||
|
|
||
|
func (a *AcceptOrRejectExecutor) Execute() error {
|
||
|
funcMap := map[string]func(*client.Client) error{
|
||
|
resourceFollowRequest: a.acceptOrRejectFollowRequest,
|
||
|
}
|
||
|
|
||
|
doFunc, ok := funcMap[a.resourceType]
|
||
|
if !ok {
|
||
|
return UnsupportedTypeError{resourceType: a.resourceType}
|
||
|
}
|
||
|
|
||
|
gtsClient, err := client.NewClientFromConfig(a.topLevelFlags.ConfigDir)
|
||
|
if err != nil {
|
||
|
return fmt.Errorf("unable to create the GoToSocial client: %w", err)
|
||
|
}
|
||
|
|
||
|
return doFunc(gtsClient)
|
||
|
}
|
||
|
|
||
|
func (a *AcceptOrRejectExecutor) acceptOrRejectFollowRequest(gtsClient *client.Client) error {
|
||
|
accountID, err := getAccountID(gtsClient, false, a.accountName, a.topLevelFlags.ConfigDir)
|
||
|
if err != nil {
|
||
|
return fmt.Errorf("received an error while getting the account ID: %w", err)
|
||
|
}
|
||
|
|
||
|
switch a.action {
|
||
|
case "accept":
|
||
|
return a.acceptFollowRequest(gtsClient, accountID)
|
||
|
case "reject":
|
||
|
return a.rejectFollowRequest(gtsClient, accountID)
|
||
|
default:
|
||
|
return nil
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (a *AcceptOrRejectExecutor) acceptFollowRequest(gtsClient *client.Client, accountID string) error {
|
||
|
if err := gtsClient.AcceptFollowRequest(accountID); err != nil {
|
||
|
return fmt.Errorf("unable to accept the follow request: %w", err)
|
||
|
}
|
||
|
|
||
|
fmt.Println("Successfully accepted the follow request.")
|
||
|
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func (a *AcceptOrRejectExecutor) rejectFollowRequest(gtsClient *client.Client, accountID string) error {
|
||
|
if err := gtsClient.RejectFollowRequest(accountID); err != nil {
|
||
|
return fmt.Errorf("unable to reject the follow request: %w", err)
|
||
|
}
|
||
|
|
||
|
fmt.Println("Successfully rejected the follow request.")
|
||
|
|
||
|
return nil
|
||
|
}
|