apollo/enbas
1
0
Fork 0
Code Issues 4 Pull requests Projects Activity Actions

checkpoint(final): remove statuses from bookmarks

Browse source
This commit is contained in:
Dan Anglin 2024-06-03 05:26:51 +01:00
parent c0a56f0b7a
commit 31e90b31e2
Signed by: dananglin
GPG key ID: 0C1D44CFBEE68638
1 changed files with 39 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

45
internal/executor/remove.go
View file

@ -18,6 +18,7 @@ type RemoveExecutor struct {
resourceType string resourceType string
fromResourceType string fromResourceType string
listID string listID string
statusID string
accountNames AccountNames accountNames AccountNames
} }
@ -33,6 +34,7 @@ func NewRemoveExecutor(tlf TopLevelFlags, name, summary string) *RemoveExecutor
removeExe.StringVar(&removeExe.resourceType, flagType, "", "specify the resource type to remove (e.g. account, note)") removeExe.StringVar(&removeExe.resourceType, flagType, "", "specify the resource type to remove (e.g. account, note)")
removeExe.StringVar(&removeExe.fromResourceType, flagFrom, "", "specify the resource type to remove from (e.g. list, account, etc)") removeExe.StringVar(&removeExe.fromResourceType, flagFrom, "", "specify the resource type to remove from (e.g. list, account, etc)")
removeExe.StringVar(&removeExe.listID, flagListID, "", "the ID of the list to remove from") removeExe.StringVar(&removeExe.listID, flagListID, "", "the ID of the list to remove from")
removeExe.StringVar(&removeExe.statusID, flagStatusID, "", "the ID of the status")
removeExe.Var(&removeExe.accountNames, flagAccountName, "the name of the account to remove from the resource") removeExe.Var(&removeExe.accountNames, flagAccountName, "the name of the account to remove from the resource")
removeExe.Usage = commandUsageFunc(name, summary, removeExe.FlagSet) removeExe.Usage = commandUsageFunc(name, summary, removeExe.FlagSet)
@ -46,8 +48,9 @@ func (r *RemoveExecutor) Execute() error {
} }
funcMap := map[string]func(*client.Client) error{ funcMap := map[string]func(*client.Client) error{
resourceList: r.removeFromList, resourceList: r.removeFromList,
resourceAccount: r.removeFromAccount, resourceAccount: r.removeFromAccount,
resourceBookmarks: r.removeFromBookmarks,
} }
doFunc, ok := funcMap[r.fromResourceType] doFunc, ok := funcMap[r.fromResourceType]
@ -90,13 +93,13 @@ func (r *RemoveExecutor) removeAccountsFromList(gtsClient *client.Client) error
accountIDs := make([]string, len(r.accountNames)) accountIDs := make([]string, len(r.accountNames))
for i := range r.accountNames { for ind := range r.accountNames {
accountID, err := getTheirAccountID(gtsClient, r.accountNames[i]) accountID, err := getTheirAccountID(gtsClient, r.accountNames[ind])
if err != nil { if err != nil {
return fmt.Errorf("unable to get the account ID for %s, %w", r.accountNames[i], err) return fmt.Errorf("unable to get the account ID for %s, %w", r.accountNames[ind], err)
} }
accountIDs[i] = accountID accountIDs[ind] = accountID
} }
if err := gtsClient.RemoveAccountsFromList(r.listID, accountIDs); err != nil { if err := gtsClient.RemoveAccountsFromList(r.listID, accountIDs); err != nil {
@ -142,3 +145,33 @@ func (r *RemoveExecutor) removeNoteFromAccount(gtsClient *client.Client) error {
return nil return nil
} }
func (r *RemoveExecutor) removeFromBookmarks(gtsClient *client.Client) error {
funcMap := map[string]func(*client.Client) error{
resourceStatus: r.removeStatusFromBookmarks,
}
doFunc, ok := funcMap[r.resourceType]
if !ok {
return UnsupportedRemoveOperationError{
ResourceType: r.resourceType,
RemoveFromResourceType: r.fromResourceType,
}
}
return doFunc(gtsClient)
}
func (r *RemoveExecutor) removeStatusFromBookmarks(gtsClient *client.Client) error {
if r.statusID == "" {
return FlagNotSetError{flagText: flagStatusID}
}
if err := gtsClient.RemoveStatusFromBookmarks(r.statusID); err != nil {
return fmt.Errorf("unable to remove the status from your bookmarks: %w", err)
}
fmt.Println("Successfully removed the status from your bookmarks.")
return nil
}