package queries

import (
	"backend/typedefs"
	"fmt"
	"time"

	"github.com/georgysavva/scany/v2/pgxscan"
)

// STORAGE

func StorageNewId() (string, error) {
	var newId string
	err := Db.Dbpool.QueryRow(Db.Ctx, "INSERT INTO public.storageindex(filename)VALUES (null) RETURNING id;").Scan(&newId)
	if err != nil {
		return "", err
	}
	if newId == "" {
		return "", fmt.Errorf("failed to create new id")
	}
	return newId, nil

}

func StorageGet_min() ([]struct {
	Id  string
	Ext string
}, error) {
	var items_db []struct {
		Id  string
		Ext string
	}

	err := pgxscan.Select(Db.Ctx, Db.Dbpool, &items_db, "SELECT id, ext FROM public.storageindex")
	if err != nil {
		return nil, fmt.Errorf("failed to get items: %w", err)
	}
	return items_db, nil
}

func StorageUpdate(id, filename, ext string, from_admin bool, expires_at *time.Time) error {
	_, err := Db.Dbpool.Exec(Db.Ctx, "UPDATE public.storageindex set filename= $2, ext = $3, from_admin = $4, expires_at = $5 where id = $1", id, filename, ext, from_admin, expires_at)
	if err != nil {
		return fmt.Errorf("failed to update storage index: %w", err)
	}
	return nil
}

func StorageDelete(id string) error {
	_, err := Db.Dbpool.Exec(Db.Ctx, "DELETE FROM public.storageindex WHERE id = $1", id)
	if err != nil {
		return fmt.Errorf("failed to delete storage index: %w", err)
	}
	return nil

}

func StorageDeleteUnused() error {
	_, err := Db.Dbpool.Exec(Db.Ctx, "DELETE FROM public.storageindex WHERE filename IS NULL")
	if err != nil {
		return fmt.Errorf("failed to delete storage index: %w", err)
	}
	return nil
}

func StorageCheckToken(token string) error {
	var tk []string
	err := pgxscan.Select(Db.Ctx, Db.Dbpool, &tk, "SELECT token FROM public.temp_tokens WHERE token = $1 and elapse > $2", token, time.Now())
	if err != nil {
		return fmt.Errorf("failed to check token: %w", err)
	}
	if tk == nil {
		return fmt.Errorf("invalid or expired token")
	}
	return nil
}

func StorageCleanExpiredTokens(token string) error {
	_, err := Db.Dbpool.Exec(Db.Ctx, "DELETE FROM public.temp_tokens WHERE elapse < $1", time.Now())
	if err != nil {
		return err
	}
	return nil
}

func StorageGetExpiredFiles() ([]typedefs.Storageindex, error) {

	var items_db []typedefs.Storageindex
	err := pgxscan.Select(Db.Ctx, Db.Dbpool, &items_db, "SELECT * FROM public.storageindex WHERE expires_at is not null and expires_at < NOW()")
	if err != nil {
		return nil, fmt.Errorf("failed to get expired items: %w", err)
	}
	return items_db, nil
}