client_store.go

package pg

import (
	"context"
	"encoding/json"
	"fmt"
	"log"
	"os"

	"github.com/go-oauth2/oauth2/v4"
	"github.com/go-oauth2/oauth2/v4/models"

	pgAdapter "github.com/vgarvardt/go-pg-adapter"
)

// ClientStore PostgreSQL client store
type ClientStore struct {
	adapter   pgAdapter.Adapter
	tableName string
	logger    Logger

	initTableDisabled bool
}

// ClientStoreItem data item
type ClientStoreItem struct {
	ID     string `db:"id"`
	Secret string `db:"secret"`
	Domain string `db:"domain"`
	Data   []byte `db:"data"`
}

// NewClientStore creates PostgreSQL store instance
func NewClientStore(adapter pgAdapter.Adapter, options ...ClientStoreOption) (*ClientStore, error) {
	store := &ClientStore{
		adapter:   adapter,
		tableName: "oauth2_clients",
		logger:    log.New(os.Stderr, "[OAUTH2-PG-ERROR]", log.LstdFlags),
	}

	for _, o := range options {
		o(store)
	}

	var err error
	if !store.initTableDisabled {
		err = store.initTable()
	}

	if err != nil {
		return store, err
	}

	return store, err
}

func (s *ClientStore) initTable() error {
	return s.adapter.Exec(context.Background(), fmt.Sprintf(`
CREATE TABLE IF NOT EXISTS %[1]s (
	"id"     TEXT  NOT NULL,
	"secret" TEXT  NOT NULL,
	"domain" TEXT  NOT NULL,
	"data"   JSONB NOT NULL,
	CONSTRAINT %[1]s_pkey PRIMARY KEY (id)
);
`, s.tableName))
}

func (s *ClientStore) toClientInfo(data []byte) (oauth2.ClientInfo, error) {
	var cm models.Client
	err := json.Unmarshal(data, &cm)
	return &cm, err
}

// GetByID retrieves and returns client information by id
func (s *ClientStore) GetByID(ctx context.Context, id string) (oauth2.ClientInfo, error) {
	if id == "" {
		return nil, nil
	}

	var item ClientStoreItem
	if err := s.adapter.SelectOne(ctx, &item, fmt.Sprintf(`SELECT "id", "secret", "domain", "data" FROM "%s" WHERE "id" = $1`, s.tableName), id); err != nil {
		return nil, err
	}

	return s.toClientInfo(item.Data)
}

// Create creates and stores the new client information
func (s *ClientStore) Create(info oauth2.ClientInfo) error {
	data, err := json.Marshal(info)
	if err != nil {
		return err
	}

	return s.adapter.Exec(
		context.Background(),
		fmt.Sprintf(`INSERT INTO %s ("id", "secret", "domain", "data") VALUES ($1, $2, $3, $4)`, s.tableName),
		info.GetID(),
		info.GetSecret(),
		info.GetDomain(),
		data,
	)
}