payments/internal/models/connector_id.go

package models

import (
	"database/sql/driver"
	"encoding/base64"
	"errors"
	"fmt"

	"github.com/gibson042/canonicaljson-go"
	"github.com/google/uuid"
)

// TODO(polo): change reference to uuid for temporal purpose
type ConnectorID struct {
	Reference uuid.UUID
	Provider  string
}

func (cid *ConnectorID) String() string {
	if cid == nil || cid.Reference == uuid.Nil {
		return ""
	}

	data, err := canonicaljson.Marshal(cid)
	if err != nil {
		panic(err)
	}

	return base64.URLEncoding.WithPadding(base64.NoPadding).EncodeToString(data)
}

func ConnectorIDFromString(value string) (ConnectorID, error) {
	data, err := base64.URLEncoding.WithPadding(base64.NoPadding).DecodeString(value)
	if err != nil {
		return ConnectorID{}, err
	}
	ret := ConnectorID{}
	err = canonicaljson.Unmarshal(data, &ret)
	if err != nil {
		return ConnectorID{}, err
	}

	return ret, nil
}

func MustConnectorIDFromString(value string) ConnectorID {
	id, err := ConnectorIDFromString(value)
	if err != nil {
		panic(err)
	}
	return id
}

func (cid ConnectorID) Value() (driver.Value, error) {
	return cid.String(), nil
}

func (cid *ConnectorID) Scan(value interface{}) error {
	if value == nil {
		return errors.New("connector id is nil")
	}

	if s, err := driver.String.ConvertValue(value); err == nil {

		if v, ok := s.(string); ok {

			id, err := ConnectorIDFromString(v)
			if err != nil {
				return fmt.Errorf("failed to parse connector id %s: %v", v, err)
			}

			*cid = id
			return nil
		}
	}

	return fmt.Errorf("failed to scan connector id: %v", value)
}