package database

import (
	"context"
	"fmt"
	"time"
)

func AddKey(ctx context.Context, exec Executor, keyName string, key []byte) (int, error) {
	row := exec.QueryRowContext(ctx, `
INSERT INTO keys (kms_key_name, encrypted_key)
VALUES ($1, $2)
RETURNING id;`, keyName, key)

	var keyId int
	err := row.Scan(&keyId)
	if err != nil {
		return 0, fmt.Errorf("unable to scan key id: %w", err)
	}

	return keyId, nil
}

func GetKey(ctx context.Context, exec Executor, keyId int) (*Key, error) {
	row := exec.QueryRowContext(ctx, `
SELECT id, kms_key_name, encrypted_key, created_at
FROM keys
WHERE id = $1;`, keyId)

	key := &Key{}
	err := row.Scan(&key.Id, &key.Name, &key.Key, &key.Created)
	if err != nil {
		return nil, fmt.Errorf("unable to scan key: %w", err)
	}

	return key, nil
}

type Key struct {
	Id      int
	Name    string
	Key     []byte
	Created time.Time
}