package model

import (
	"database/sql/driver"
	"encoding/base32"
	"fmt"
	"io"

	"github.com/google/uuid"
)

// Crockford's low-ambiguity base32 alphabet
var base32Encoding = base32.
	NewEncoding("0123456789abcdefghjkmnpqrstvwxyz").
	WithPadding(base32.NoPadding)

// An RID is a unique resource ID.
type RID struct {
	uuid uuid.UUID
}

// Creates a new resource ID.
func NewRID() RID {
	uuid, err := uuid.NewV7()
	if err != nil {
		panic(err)
	}
	return RID{
		uuid: uuid,
	}
}

// Returns the UUID representation of this RID.
func (rid *RID) UUID() uuid.UUID {
	return rid.uuid
}

// Returns the string representation of this RID.
func (rid *RID) String() string {
	return base32Encoding.EncodeToString(rid.uuid[:])
}

func (rid RID) MarshalGQL(w io.Writer) {
	w.Write(fmt.Appendf(nil, `"%s"`, rid.String()))
}

func (rid *RID) unmarshalString(s string) error {
	bytes, err := base32Encoding.DecodeString(s)
	if err != nil {
		return err
	}
	rid.uuid, err = uuid.FromBytes(bytes)
	if err != nil {
		return err
	}
	return nil
}

func (rid *RID) UnmarshalGQL(v any) error {
	switch v := v.(type) {
	case string:
		return rid.unmarshalString(v)
	default:
		return fmt.Errorf("%T is not a valid RID", v)
	}
}

// database/sql.Scanner
func (rid *RID) Scan(src any) error {
	switch src := src.(type) {
	case string:
		return rid.unmarshalString(src)
	default:
		var uuid uuid.UUID
		err := uuid.Scan(src)
		if err != nil {
			return err
		}
		rid.uuid = uuid
		return nil
	}
}

// database/sql/driver.Valuer
func (rid RID) Value() (driver.Value, error) {
	return rid.UUID().Value()
}