package db

import (
	"context"
	"fmt"

	"github.com/jmoiron/sqlx"
)

type Ownership struct {
	ID uint

	Player    string
	Trusted   string
	Assistant string
}

func (o Ownership) Query(data *InsertId[string]) {
	data.query = fmt.Sprintf(`
		INSERT INTO ownership (%s, player, trusted, assistant)
		VALUES ($1, $2, $3, $4)
		RETURNING id`, data.fieldName)
}

func (o Ownership) Insert(tx *sqlx.Tx, data *InsertId[string]) error {
	if data.query == "" {
		return ErrNoQuery
	}

	args := []any{data.id, o.Player, o.Trusted, o.Assistant}

	err := tx.QueryRowx(data.query, args...).Scan(&o.ID)
	if err != nil {
		return err
	}

	return nil
}

func (o Ownership) InsertCtx(ctx context.Context, tx *sqlx.Tx, data *InsertId[string]) error {
	if data.query == "" {
		return ErrNoQuery
	}

	args := []any{data.id, o.Player, o.Trusted, o.Assistant}

	err := tx.QueryRowxContext(ctx, data.query, args...).Scan(&o.ID)
	if err != nil {
		return err
	}

	return nil
}

type OwnershipString struct {
	ID uint

	Key   string
	Value int
}

func (o OwnershipString) Query(data *InsertId[string]) {
	data.query = fmt.Sprintf(`
		INSERT INTO ownership_string (%s, key_, value)
		VALUES ($1, $2, $3)
		RETURNING id`, data.fieldName)
}

func (o OwnershipString) Insert(tx *sqlx.Tx, data *InsertId[string]) error {
	if data.query == "" {
		return ErrNoQuery
	}

	args := []any{data.id, o.Key, o.Value}

	err := tx.QueryRowx(data.query, args...).Scan(&o.ID)
	if err != nil {
		return err
	}

	return nil
}

func (o OwnershipString) InsertCtx(ctx context.Context, tx *sqlx.Tx, data *InsertId[string]) error {
	if data.query == "" {
		return ErrNoQuery
	}

	args := []any{data.id, o.Key, o.Value}

	err := tx.QueryRowxContext(ctx, data.query, args...).Scan(&o.ID)
	if err != nil {
		return err
	}

	return nil
}