stack/components/fctl/cmd/wallets/internal/subject.go

package internal

import (
	"strings"

	formance "github.com/formancehq/formance-sdk-go/v2"
	"github.com/formancehq/formance-sdk-go/v2/pkg/models/shared"
	"github.com/pkg/errors"
	"github.com/spf13/cobra"
)

func ParseSubject(subject string, cmd *cobra.Command, client *formance.Formance) (*shared.Subject, error) {
	var err error
	switch {
	case strings.HasPrefix(subject, "wallet="):
		walletDefinition := strings.TrimPrefix(subject, "wallet=")
		parts := strings.SplitN(walletDefinition, "/", 2)
		balance := "main"
		if len(parts) > 1 {
			balance = parts[1]
		}

		var walletID string
		switch {
		case strings.HasPrefix(walletDefinition, "id:"):
			walletID = strings.TrimPrefix(parts[0], "id:")
		case strings.HasPrefix(walletDefinition, "name:"):
			walletID, err = DiscoverWalletIDFromName(cmd, client, strings.TrimPrefix(parts[0], "name:"))
			if err != nil {
				return nil, err
			}
		default:
			return nil, errors.New("malformed wallet source definition")
		}
		subject := shared.CreateSubjectWallet(shared.WalletSubject{
			Identifier: walletID,
			Balance:    &balance,
		})
		return &subject, nil
	case strings.HasPrefix(subject, "account="):
		subject := shared.CreateSubjectAccount(shared.LedgerAccountSubject{
			Identifier: strings.TrimPrefix(subject, "account="),
		})
		return &subject, nil
	default:
		return nil, errors.New("malformed source definition")
	}
}