enbas/internal/client/lists.go

package client

import (
	"bytes"
	"encoding/json"
	"fmt"
	"net/http"

	"codeflow.dananglin.me.uk/apollo/enbas/internal/model"
)

const (
	listPath string = "/api/v1/lists"
)

func (g *Client) GetAllLists() (model.Lists, error) {
	url := g.Authentication.Instance + listPath

	var lists []model.List

	if err := g.sendRequest(http.MethodGet, url, nil, &lists); err != nil {
		return nil, fmt.Errorf(
			"received an error after sending the request to get the list of lists; %w",
			err,
		)
	}

	return lists, nil
}

func (g *Client) GetList(listID string) (model.List, error) {
	url := g.Authentication.Instance + listPath + "/" + listID

	var list model.List

	if err := g.sendRequest(http.MethodGet, url, nil, &list); err != nil {
		return model.List{}, fmt.Errorf(
			"received an error after sending the request to get the list; %w",
			err,
		)
	}

	return list, nil
}

func (g *Client) CreateList(title string, repliesPolicy model.ListRepliesPolicy) (model.List, error) {
	form := struct {
		Title         string                  `json:"title"`
		RepliesPolicy model.ListRepliesPolicy `json:"replies_policy"`
	}{
		Title:         title,
		RepliesPolicy: repliesPolicy,
	}

	data, err := json.Marshal(form)
	if err != nil {
		return model.List{}, fmt.Errorf("unable to marshal the form; %w", err)
	}

	requestBody := bytes.NewBuffer(data)
	url := g.Authentication.Instance + "/api/v1/lists"

	var list model.List

	if err := g.sendRequest(http.MethodPost, url, requestBody, &list); err != nil {
		return model.List{}, fmt.Errorf(
			"received an error after sending the request to create the list; %w",
			err,
		)
	}

	return list, nil
}

func (g *Client) UpdateList(listToUpdate model.List) (model.List, error) {
	form := struct {
		Title         string                  `json:"title"`
		RepliesPolicy model.ListRepliesPolicy `json:"replies_policy"`
	}{
		Title:         listToUpdate.Title,
		RepliesPolicy: listToUpdate.RepliesPolicy,
	}

	data, err := json.Marshal(form)
	if err != nil {
		return model.List{}, fmt.Errorf("unable to marshal the form; %w", err)
	}

	requestBody := bytes.NewBuffer(data)
	url := g.Authentication.Instance + listPath + "/" + listToUpdate.ID

	var updatedList model.List

	if err := g.sendRequest(http.MethodPut, url, requestBody, &updatedList); err != nil {
		return model.List{}, fmt.Errorf(
			"received an error after sending the request to update the list; %w",
			err,
		)
	}

	return updatedList, nil
}

func (g *Client) DeleteList(listID string) error {
	url := g.Authentication.Instance + "/api/v1/lists/" + listID

	return g.sendRequest(http.MethodDelete, url, nil, nil)
}

func (g *Client) AddAccountsToList(listID string, accountIDs []string) error {
	form := struct {
		AccountIDs []string `json:"account_ids"`
	}{
		AccountIDs: accountIDs,
	}

	data, err := json.Marshal(form)
	if err != nil {
		return fmt.Errorf("unable to marshal the form; %w", err)
	}

	requestBody := bytes.NewBuffer(data)
	url := g.Authentication.Instance + listPath + "/" + listID + "/accounts"

	if err := g.sendRequest(http.MethodPost, url, requestBody, nil); err != nil {
		return fmt.Errorf(
			"received an error after sending the request to add the accounts to the list; %w",
			err,
		)
	}

	return nil
}

func (g *Client) RemoveAccountsFromList(listID string, accountIDs []string) error {
	form := struct {
		AccountIDs []string `json:"account_ids"`
	}{
		AccountIDs: accountIDs,
	}

	data, err := json.Marshal(form)
	if err != nil {
		return fmt.Errorf("unable to marshal the form; %w", err)
	}

	requestBody := bytes.NewBuffer(data)
	url := g.Authentication.Instance + listPath + "/" + listID + "/accounts"

	if err := g.sendRequest(http.MethodDelete, url, requestBody, nil); err != nil {
		return fmt.Errorf(
			"received an error after sending the request to remove the accounts from the list; %w",
			err,
		)
	}

	return nil
}

func (g *Client) GetAccountsFromList(listID string, limit int) ([]model.Account, error) {
	path := fmt.Sprintf("%s/%s/accounts?limit=%d", listPath, listID, limit)
	url := g.Authentication.Instance + path

	var accounts []model.Account

	if err := g.sendRequest(http.MethodGet, url, nil, &accounts); err != nil {
		return nil, fmt.Errorf(
			"received an error after sending the request to get the accounts from the list; %w",
			err,
		)
	}

	return accounts, nil
}