package token

import (
	"context"
	"database/sql"
	"encoding/json"
	"fmt"

	"git.faercol.me/faercol/polyculeconnect/polyculeconnect/internal/model"
	"github.com/google/uuid"
)

func strArrayToSlice(rawVal string) []string {
	var res []string
	if err := json.Unmarshal([]byte(rawVal), &res); err != nil {
		return nil
	}
	return res
}

func sliceToStrArray(rawVal []string) string {
	res, err := json.Marshal(rawVal)
	if err != nil {
		return "[]"
	}
	return string(res)
}

type TokenDB interface {
	AddRefreshToken(ctx context.Context, refreshToken *model.RefreshToken) error
	GetRefreshTokenByID(ctx context.Context, id uuid.UUID) (*model.RefreshToken, error)
}

type sqlTokenDB struct {
	db *sql.DB
}

func (db *sqlTokenDB) AddRefreshToken(ctx context.Context, refreshToken *model.RefreshToken) error {
	tx, err := db.db.BeginTx(ctx, nil)
	if err != nil {
		return fmt.Errorf("failed to start transaction: %w", err)
	}
	defer func() { _ = tx.Rollback() }()

	if _, err := tx.ExecContext(ctx, `INSERT INTO "refresh_token" ("id", "client_id", "user_id", "scopes", "auth_time") VALUES ($1, $2, $3, $4, $5)`, refreshToken.ID, refreshToken.ClientID, refreshToken.UserID, sliceToStrArray(refreshToken.Scopes), refreshToken.AuthTime); err != nil {
		return fmt.Errorf("failed to exec query: %w", err)
	}

	if err := tx.Commit(); err != nil {
		return fmt.Errorf("failed to commit transaction: %w", err)
	}
	return nil
}

func (db *sqlTokenDB) GetRefreshTokenByID(ctx context.Context, id uuid.UUID) (*model.RefreshToken, error) {
	row := db.db.QueryRowContext(ctx, `SELECT "id", "client_id", "user_id", "scopes", "auth_time" FROM "refresh_token" WHERE "id" = ?`, id)
	var res model.RefreshToken
	var strScopes string
	if err := row.Scan(&res.ID, &res.ClientID, &res.UserID, &strScopes, &res.AuthTime); err != nil {
		return nil, fmt.Errorf("failed to query DB: %w", err)
	}
	res.Scopes = strArrayToSlice(strScopes)
	return &res, nil
}

func New(db *sql.DB) TokenDB {
	return &sqlTokenDB{db: db}
}