2024-05-23 18:06:49 +01:00
|
|
|
package executor
|
2024-05-21 21:06:18 +01:00
|
|
|
|
|
|
|
import (
|
|
|
|
"fmt"
|
|
|
|
|
|
|
|
"codeflow.dananglin.me.uk/apollo/enbas/internal/client"
|
2024-08-13 14:53:26 +01:00
|
|
|
internalFlag "codeflow.dananglin.me.uk/apollo/enbas/internal/flag"
|
2024-05-21 21:06:18 +01:00
|
|
|
"codeflow.dananglin.me.uk/apollo/enbas/internal/model"
|
|
|
|
)
|
|
|
|
|
2024-08-13 14:53:26 +01:00
|
|
|
func getAccountID(
|
|
|
|
gtsClient *client.Client,
|
|
|
|
myAccount bool,
|
|
|
|
accountNames internalFlag.StringSliceValue,
|
|
|
|
) (string, error) {
|
2024-08-13 23:41:12 +01:00
|
|
|
account, err := getAccount(gtsClient, myAccount, accountNames)
|
2024-08-13 14:53:26 +01:00
|
|
|
if err != nil {
|
|
|
|
return "", fmt.Errorf("unable to get the account information: %w", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
return account.ID, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func getAccount(
|
|
|
|
gtsClient *client.Client,
|
|
|
|
myAccount bool,
|
|
|
|
accountNames internalFlag.StringSliceValue,
|
|
|
|
) (model.Account, error) {
|
2024-05-21 21:06:18 +01:00
|
|
|
var (
|
2024-08-13 14:53:26 +01:00
|
|
|
account model.Account
|
|
|
|
err error
|
2024-05-21 21:06:18 +01:00
|
|
|
)
|
|
|
|
|
|
|
|
switch {
|
|
|
|
case myAccount:
|
2024-08-13 23:41:12 +01:00
|
|
|
account, err = getMyAccount(gtsClient)
|
2024-05-21 21:06:18 +01:00
|
|
|
if err != nil {
|
2024-08-13 14:53:26 +01:00
|
|
|
return account, fmt.Errorf("unable to get your account ID: %w", err)
|
2024-05-21 21:06:18 +01:00
|
|
|
}
|
2024-08-13 14:53:26 +01:00
|
|
|
case !accountNames.Empty():
|
|
|
|
account, err = getOtherAccount(gtsClient, accountNames)
|
2024-05-21 21:06:18 +01:00
|
|
|
if err != nil {
|
2024-08-13 14:53:26 +01:00
|
|
|
return account, fmt.Errorf("unable to get the account ID: %w", err)
|
2024-05-21 21:06:18 +01:00
|
|
|
}
|
|
|
|
default:
|
2024-08-13 14:53:26 +01:00
|
|
|
return account, NoAccountSpecifiedError{}
|
2024-05-21 21:06:18 +01:00
|
|
|
}
|
|
|
|
|
2024-08-13 14:53:26 +01:00
|
|
|
return account, nil
|
2024-05-21 21:06:18 +01:00
|
|
|
}
|
|
|
|
|
2024-08-13 23:41:12 +01:00
|
|
|
func getMyAccount(gtsClient *client.Client) (model.Account, error) {
|
|
|
|
account, err := gtsClient.VerifyCredentials()
|
2024-05-21 21:06:18 +01:00
|
|
|
if err != nil {
|
2024-06-02 11:35:43 +01:00
|
|
|
return model.Account{}, fmt.Errorf("unable to retrieve your account: %w", err)
|
2024-05-21 21:06:18 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
return account, nil
|
|
|
|
}
|
|
|
|
|
2024-08-13 14:53:26 +01:00
|
|
|
func getOtherAccount(gtsClient *client.Client, accountNames internalFlag.StringSliceValue) (model.Account, error) {
|
|
|
|
expectedNumAccountNames := 1
|
|
|
|
if !accountNames.ExpectedLength(expectedNumAccountNames) {
|
|
|
|
return model.Account{}, fmt.Errorf(
|
|
|
|
"received an unexpected number of account names: want %d",
|
|
|
|
expectedNumAccountNames,
|
|
|
|
)
|
|
|
|
}
|
|
|
|
|
|
|
|
account, err := gtsClient.GetAccount(accountNames[0])
|
2024-05-21 21:06:18 +01:00
|
|
|
if err != nil {
|
2024-06-02 11:35:43 +01:00
|
|
|
return model.Account{}, fmt.Errorf("unable to retrieve the account details: %w", err)
|
2024-05-21 21:06:18 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
return account, nil
|
|
|
|
}
|
2024-08-13 14:53:26 +01:00
|
|
|
|
|
|
|
func getOtherAccounts(gtsClient *client.Client, accountNames internalFlag.StringSliceValue) ([]model.Account, error) {
|
|
|
|
numAccountNames := len(accountNames)
|
|
|
|
accounts := make([]model.Account, numAccountNames)
|
|
|
|
|
2024-08-17 21:56:54 +01:00
|
|
|
for ind := range numAccountNames {
|
2024-08-13 14:53:26 +01:00
|
|
|
var err error
|
|
|
|
|
|
|
|
accounts[ind], err = gtsClient.GetAccount(accountNames[ind])
|
|
|
|
if err != nil {
|
|
|
|
return nil, fmt.Errorf("unable to retrieve the account information for %s: %w", accountNames[ind], err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return accounts, nil
|
|
|
|
}
|