feat: add bookmark support #21

Manually merged
dananglin merged 1 commit from bookmarks into main 2024-06-03 06:55:31 +01:00
5 changed files with 140 additions and 9 deletions

View file

@ -62,3 +62,51 @@ func (g *Client) CreateStatus(form CreateStatusForm) (model.Status, error) {
return status, nil
}
func (g *Client) GetBookmarks(limit int) (model.StatusList, error) {
path := fmt.Sprintf("/api/v1/bookmarks?limit=%d", limit)
url := g.Authentication.Instance + path
bookmarks := model.StatusList{
Type: model.StatusListBookMarks,
Name: "BOOKMARKS",
Statuses: nil,
}
if err := g.sendRequest(http.MethodGet, url, nil, &bookmarks.Statuses); err != nil {
return bookmarks, fmt.Errorf(
"received an error after sending the request to get the bookmarks: %w",
err,
)
}
return bookmarks, nil
}
func (g *Client) AddStatusToBookmarks(statusID string) error {
path := fmt.Sprintf("/api/v1/statuses/%s/bookmark", statusID)
url := g.Authentication.Instance + path
if err := g.sendRequest(http.MethodPost, url, nil, nil); err != nil {
return fmt.Errorf(
"received an error after sending the request to add the status to the list of bookmarks: %w",
err,
)
}
return nil
}
func (g *Client) RemoveStatusFromBookmarks(statusID string) error {
path := fmt.Sprintf("/api/v1/statuses/%s/unbookmark", statusID)
url := g.Authentication.Instance + path
if err := g.sendRequest(http.MethodPost, url, nil, nil); err != nil {
return fmt.Errorf(
"received an error after sending the request to remove the status from the list of bookmarks: %w",
err,
)
}
return nil
}

View file

@ -18,6 +18,7 @@ type AddExecutor struct {
resourceType string
toResourceType string
listID string
statusID string
accountNames AccountNames
content string
}
@ -34,7 +35,8 @@ func NewAddExecutor(tlf TopLevelFlags, name, summary string) *AddExecutor {
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 to add to")
addExe.Var(&addExe.accountNames, flagAccountName, "the name of the account to add to the resource")
addExe.StringVar(&addExe.statusID, flagStatusID, "", "the ID of the status")
addExe.Var(&addExe.accountNames, flagAccountName, "the name of the account")
addExe.StringVar(&addExe.content, flagContent, "", "the content of the note")
addExe.Usage = commandUsageFunc(name, summary, addExe.FlagSet)
@ -48,8 +50,9 @@ func (a *AddExecutor) Execute() error {
}
funcMap := map[string]func(*client.Client) error{
resourceList: a.addToList,
resourceAccount: a.addToAccount,
resourceList: a.addToList,
resourceAccount: a.addToAccount,
resourceBookmarks: a.addToBookmarks,
}
doFunc, ok := funcMap[a.toResourceType]
@ -151,3 +154,33 @@ func (a *AddExecutor) addNoteToAccount(gtsClient *client.Client) error {
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)
}
fmt.Println("Successfully added the status to your bookmarks.")
return nil
}

View file

@ -44,4 +44,5 @@ const (
resourceNote = "note"
resourceStatus = "status"
resourceTimeline = "timeline"
resourceBookmarks = "bookmarks"
)

View file

@ -18,6 +18,7 @@ type RemoveExecutor struct {
resourceType string
fromResourceType string
listID string
statusID string
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.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.statusID, flagStatusID, "", "the ID of the status")
removeExe.Var(&removeExe.accountNames, flagAccountName, "the name of the account to remove from the resource")
removeExe.Usage = commandUsageFunc(name, summary, removeExe.FlagSet)
@ -46,8 +48,9 @@ func (r *RemoveExecutor) Execute() error {
}
funcMap := map[string]func(*client.Client) error{
resourceList: r.removeFromList,
resourceAccount: r.removeFromAccount,
resourceList: r.removeFromList,
resourceAccount: r.removeFromAccount,
resourceBookmarks: r.removeFromBookmarks,
}
doFunc, ok := funcMap[r.fromResourceType]
@ -90,13 +93,13 @@ func (r *RemoveExecutor) removeAccountsFromList(gtsClient *client.Client) error
accountIDs := make([]string, len(r.accountNames))
for i := range r.accountNames {
accountID, err := getTheirAccountID(gtsClient, r.accountNames[i])
for ind := range r.accountNames {
accountID, err := getTheirAccountID(gtsClient, r.accountNames[ind])
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 {
@ -142,3 +145,33 @@ func (r *RemoveExecutor) removeNoteFromAccount(gtsClient *client.Client) error {
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
}

View file

@ -66,6 +66,7 @@ func (c *ShowExecutor) Execute() error {
resourceFollowers: c.showFollowers,
resourceFollowing: c.showFollowing,
resourceBlocked: c.showBlocked,
resourceBookmarks: c.showBookmarks,
}
doFunc, ok := funcMap[c.resourceType]
@ -313,3 +314,18 @@ func (c *ShowExecutor) showBlocked(gtsClient *client.Client) error {
return nil
}
func (c *ShowExecutor) showBookmarks(gtsClient *client.Client) error {
bookmarks, err := gtsClient.GetBookmarks(c.limit)
if err != nil {
return fmt.Errorf("unable to retrieve the list of bookmarks: %w", err)
}
if len(bookmarks.Statuses) > 0 {
utilities.Display(bookmarks, *c.topLevelFlags.NoColor)
} else {
fmt.Println("You have no bookmarks.")
}
return nil
}