package executor

import (
	"flag"
	"fmt"

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

type FollowExecutor struct {
	*flag.FlagSet

	topLevelFlags TopLevelFlags
	resourceType  string
	accountName   string
	showReposts   bool
	notify        bool
	unfollow      bool
}

func NewFollowExecutor(tlf TopLevelFlags, name, summary string, unfollow bool) *FollowExecutor {
	command := FollowExecutor{
		FlagSet:       flag.NewFlagSet(name, flag.ExitOnError),
		unfollow:      unfollow,
		topLevelFlags: tlf,
	}

	command.StringVar(&command.resourceType, flagType, "", "specify the type of resource to follow")
	command.StringVar(&command.accountName, flagAccountName, "", "specify the account name in full (username@domain)")
	command.BoolVar(&command.showReposts, flagShowReposts, true, "show reposts from the account you want to follow")
	command.BoolVar(&command.notify, flagNotify, false, "get notifications when the account you want to follow posts a status")

	command.Usage = commandUsageFunc(name, summary, command.FlagSet)

	return &command
}

func (c *FollowExecutor) Execute() error {
	funcMap := map[string]func(*client.Client) error{
		resourceAccount: c.followAccount,
	}

	doFunc, ok := funcMap[c.resourceType]
	if !ok {
		return UnsupportedTypeError{resourceType: c.resourceType}
	}

	gtsClient, err := client.NewClientFromConfig(c.topLevelFlags.ConfigDir)
	if err != nil {
		return fmt.Errorf("unable to create the GoToSocial client; %w", err)
	}

	return doFunc(gtsClient)
}

func (c *FollowExecutor) followAccount(gtsClient *client.Client) error {
	accountID, err := getAccountID(gtsClient, false, c.accountName, c.topLevelFlags.ConfigDir)
	if err != nil {
		return fmt.Errorf("received an error while getting the account ID; %w", err)
	}

	if c.unfollow {
		return c.unfollowAccount(gtsClient, accountID)
	}

	form := client.FollowAccountForm{
		AccountID:   accountID,
		ShowReposts: c.showReposts,
		Notify:      c.notify,
	}

	if err := gtsClient.FollowAccount(form); err != nil {
		return fmt.Errorf("unable to follow the account; %w", err)
	}

	fmt.Println("The follow request was sent successfully.")

	return nil
}

func (c *FollowExecutor) unfollowAccount(gtsClient *client.Client, accountID string) error {
	if err := gtsClient.UnfollowAccount(accountID); err != nil {
		return fmt.Errorf("unable to unfollow the account; %w", err)
	}

	fmt.Println("Successfully unfollowed the account.")

	return nil
}