Dan Anglin
ccdd8b6530
Add a new internal printer package for printing resources to the screen or pager. With the new printer in place, most of the settings such as the pager command, colour theme, whether or not colour output is disabled, etc are defined in one place which saves us the trouble of passing an increasing number of parameters to an increasing number of Display methods throughout the code base. The old Displayer interface and associated Display methods in the model package are removed as this is now handled by the printer. The format functions in the utilities package has essentially been rewritten as methods to the Printer type. Additional changes: - All indentation when displaying information about resources (e.g. statuses, instance, accounts) are removed. - The application's build information now has colour output.
295 lines
7.7 KiB
Go
295 lines
7.7 KiB
Go
// SPDX-FileCopyrightText: 2024 Dan Anglin <d.n.i.anglin@gmail.com>
|
|
//
|
|
// SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
package executor
|
|
|
|
import (
|
|
"errors"
|
|
"flag"
|
|
"fmt"
|
|
|
|
"codeflow.dananglin.me.uk/apollo/enbas/internal/client"
|
|
"codeflow.dananglin.me.uk/apollo/enbas/internal/printer"
|
|
)
|
|
|
|
type AddExecutor struct {
|
|
*flag.FlagSet
|
|
|
|
printer *printer.Printer
|
|
configDir string
|
|
resourceType string
|
|
toResourceType string
|
|
listID string
|
|
statusID string
|
|
pollID string
|
|
choices MultiIntFlagValue
|
|
accountNames MultiStringFlagValue
|
|
content string
|
|
}
|
|
|
|
func NewAddExecutor(printer *printer.Printer, configDir, name, summary string) *AddExecutor {
|
|
emptyArr := make([]string, 0, 3)
|
|
|
|
addExe := AddExecutor{
|
|
FlagSet: flag.NewFlagSet(name, flag.ExitOnError),
|
|
|
|
printer: printer,
|
|
configDir: configDir,
|
|
accountNames: MultiStringFlagValue(emptyArr),
|
|
}
|
|
|
|
addExe.StringVar(&addExe.resourceType, flagType, "", "Specify the resource type to add (e.g. account, note)")
|
|
addExe.StringVar(&addExe.toResourceType, flagTo, "", "Specify the target resource type to add to (e.g. list, account, etc)")
|
|
addExe.StringVar(&addExe.listID, flagListID, "", "The ID of the list")
|
|
addExe.StringVar(&addExe.statusID, flagStatusID, "", "The ID of the status")
|
|
addExe.StringVar(&addExe.content, flagContent, "", "The content of the resource")
|
|
addExe.StringVar(&addExe.pollID, flagPollID, "", "The ID of the poll")
|
|
addExe.Var(&addExe.accountNames, flagAccountName, "The name of the account")
|
|
addExe.Var(&addExe.choices, flagChoose, "Specify your choice ")
|
|
|
|
addExe.Usage = commandUsageFunc(name, summary, addExe.FlagSet)
|
|
|
|
return &addExe
|
|
}
|
|
|
|
func (a *AddExecutor) Execute() error {
|
|
if a.toResourceType == "" {
|
|
return FlagNotSetError{flagText: flagTo}
|
|
}
|
|
|
|
funcMap := map[string]func(*client.Client) error{
|
|
resourceList: a.addToList,
|
|
resourceAccount: a.addToAccount,
|
|
resourceBookmarks: a.addToBookmarks,
|
|
resourceStatus: a.addToStatus,
|
|
resourcePoll: a.addToPoll,
|
|
}
|
|
|
|
doFunc, ok := funcMap[a.toResourceType]
|
|
if !ok {
|
|
return UnsupportedTypeError{resourceType: a.toResourceType}
|
|
}
|
|
|
|
gtsClient, err := client.NewClientFromConfig(a.configDir)
|
|
if err != nil {
|
|
return fmt.Errorf("unable to create the GoToSocial client: %w", err)
|
|
}
|
|
|
|
return doFunc(gtsClient)
|
|
}
|
|
|
|
func (a *AddExecutor) addToList(gtsClient *client.Client) error {
|
|
funcMap := map[string]func(*client.Client) error{
|
|
resourceAccount: a.addAccountsToList,
|
|
}
|
|
|
|
doFunc, ok := funcMap[a.resourceType]
|
|
if !ok {
|
|
return UnsupportedAddOperationError{
|
|
ResourceType: a.resourceType,
|
|
AddToResourceType: a.toResourceType,
|
|
}
|
|
}
|
|
|
|
return doFunc(gtsClient)
|
|
}
|
|
|
|
func (a *AddExecutor) addAccountsToList(gtsClient *client.Client) error {
|
|
if a.listID == "" {
|
|
return FlagNotSetError{flagText: flagListID}
|
|
}
|
|
|
|
if len(a.accountNames) == 0 {
|
|
return NoAccountSpecifiedError{}
|
|
}
|
|
|
|
accountIDs := make([]string, len(a.accountNames))
|
|
|
|
for ind := range a.accountNames {
|
|
accountID, err := getTheirAccountID(gtsClient, a.accountNames[ind])
|
|
if err != nil {
|
|
return fmt.Errorf("unable to get the account ID for %s: %w", a.accountNames[ind], err)
|
|
}
|
|
|
|
relationship, err := gtsClient.GetAccountRelationship(accountID)
|
|
if err != nil {
|
|
return fmt.Errorf("unable to get your relationship to %s: %w", a.accountNames[ind], err)
|
|
}
|
|
|
|
if !relationship.Following {
|
|
return NotFollowingError{Account: a.accountNames[ind]}
|
|
}
|
|
|
|
accountIDs[ind] = accountID
|
|
}
|
|
|
|
if err := gtsClient.AddAccountsToList(a.listID, accountIDs); err != nil {
|
|
return fmt.Errorf("unable to add the accounts to the list: %w", err)
|
|
}
|
|
|
|
a.printer.PrintSuccess("Successfully added the account(s) to the list.")
|
|
|
|
return nil
|
|
}
|
|
|
|
func (a *AddExecutor) addToAccount(gtsClient *client.Client) error {
|
|
funcMap := map[string]func(*client.Client) error{
|
|
resourceNote: a.addNoteToAccount,
|
|
}
|
|
|
|
doFunc, ok := funcMap[a.resourceType]
|
|
if !ok {
|
|
return UnsupportedAddOperationError{
|
|
ResourceType: a.resourceType,
|
|
AddToResourceType: a.toResourceType,
|
|
}
|
|
}
|
|
|
|
return doFunc(gtsClient)
|
|
}
|
|
|
|
func (a *AddExecutor) addNoteToAccount(gtsClient *client.Client) error {
|
|
if len(a.accountNames) != 1 {
|
|
return fmt.Errorf("unexpected number of accounts specified: want 1, got %d", len(a.accountNames))
|
|
}
|
|
|
|
accountID, err := getAccountID(gtsClient, false, a.accountNames[0], a.configDir)
|
|
if err != nil {
|
|
return fmt.Errorf("received an error while getting the account ID: %w", err)
|
|
}
|
|
|
|
if a.content == "" {
|
|
return EmptyContentError{
|
|
ResourceType: resourceNote,
|
|
Hint: "please use --" + flagContent,
|
|
}
|
|
}
|
|
|
|
if err := gtsClient.SetPrivateNote(accountID, a.content); err != nil {
|
|
return fmt.Errorf("unable to add the private note to the account: %w", err)
|
|
}
|
|
|
|
a.printer.PrintSuccess("Successfully added the private note to the account.")
|
|
|
|
return nil
|
|
}
|
|
|
|
func (a *AddExecutor) addToBookmarks(gtsClient *client.Client) error {
|
|
funcMap := map[string]func(*client.Client) error{
|
|
resourceStatus: a.addStatusToBookmarks,
|
|
}
|
|
|
|
doFunc, ok := funcMap[a.resourceType]
|
|
if !ok {
|
|
return UnsupportedAddOperationError{
|
|
ResourceType: a.resourceType,
|
|
AddToResourceType: a.toResourceType,
|
|
}
|
|
}
|
|
|
|
return doFunc(gtsClient)
|
|
}
|
|
|
|
func (a *AddExecutor) addStatusToBookmarks(gtsClient *client.Client) error {
|
|
if a.statusID == "" {
|
|
return FlagNotSetError{flagText: flagStatusID}
|
|
}
|
|
|
|
if err := gtsClient.AddStatusToBookmarks(a.statusID); err != nil {
|
|
return fmt.Errorf("unable to add the status to your bookmarks: %w", err)
|
|
}
|
|
|
|
a.printer.PrintSuccess("Successfully added the status to your bookmarks.")
|
|
|
|
return nil
|
|
}
|
|
|
|
func (a *AddExecutor) addToStatus(gtsClient *client.Client) error {
|
|
if a.statusID == "" {
|
|
return FlagNotSetError{flagText: flagStatusID}
|
|
}
|
|
|
|
funcMap := map[string]func(*client.Client) error{
|
|
resourceStar: a.addStarToStatus,
|
|
resourceLike: a.addStarToStatus,
|
|
resourceBoost: a.addBoostToStatus,
|
|
}
|
|
|
|
doFunc, ok := funcMap[a.resourceType]
|
|
if !ok {
|
|
return UnsupportedAddOperationError{
|
|
ResourceType: a.resourceType,
|
|
AddToResourceType: a.toResourceType,
|
|
}
|
|
}
|
|
|
|
return doFunc(gtsClient)
|
|
}
|
|
|
|
func (a *AddExecutor) addStarToStatus(gtsClient *client.Client) error {
|
|
if err := gtsClient.LikeStatus(a.statusID); err != nil {
|
|
return fmt.Errorf("unable to add the %s to the status: %w", a.resourceType, err)
|
|
}
|
|
|
|
a.printer.PrintSuccess("Successfully added a " + a.resourceType + " to the status.")
|
|
|
|
return nil
|
|
}
|
|
|
|
func (a *AddExecutor) addBoostToStatus(gtsClient *client.Client) error {
|
|
if err := gtsClient.ReblogStatus(a.statusID); err != nil {
|
|
return fmt.Errorf("unable to add the boost to the status: %w", err)
|
|
}
|
|
|
|
a.printer.PrintSuccess("Successfully added the boost to the status.")
|
|
|
|
return nil
|
|
}
|
|
|
|
func (a *AddExecutor) addToPoll(gtsClient *client.Client) error {
|
|
if a.pollID == "" {
|
|
return FlagNotSetError{flagText: flagPollID}
|
|
}
|
|
|
|
funcMap := map[string]func(*client.Client) error{
|
|
resourceVote: a.addVoteToPoll,
|
|
}
|
|
|
|
doFunc, ok := funcMap[a.resourceType]
|
|
if !ok {
|
|
return UnsupportedAddOperationError{
|
|
ResourceType: a.resourceType,
|
|
AddToResourceType: a.toResourceType,
|
|
}
|
|
}
|
|
|
|
return doFunc(gtsClient)
|
|
}
|
|
|
|
func (a *AddExecutor) addVoteToPoll(gtsClient *client.Client) error {
|
|
if len(a.choices) == 0 {
|
|
return errors.New("please use --" + flagChoose + " to make a choice in this poll")
|
|
}
|
|
|
|
poll, err := gtsClient.GetPoll(a.pollID)
|
|
if err != nil {
|
|
return fmt.Errorf("unable to retrieve the poll: %w", err)
|
|
}
|
|
|
|
if poll.Expired {
|
|
return PollClosedError{}
|
|
}
|
|
|
|
if !poll.Multiple && len(a.choices) > 1 {
|
|
return MultipleChoiceError{}
|
|
}
|
|
|
|
if err := gtsClient.VoteInPoll(a.pollID, []int(a.choices)); err != nil {
|
|
return fmt.Errorf("unable to add your vote(s) to the poll: %w", err)
|
|
}
|
|
|
|
a.printer.PrintSuccess("Successfully added your vote(s) to the poll.")
|
|
|
|
return nil
|
|
}
|