package db

import (
	"context"
	"os"

	"github.com/flowy-live/llink/internal/utils/flog"

	"github.com/jackc/pgx/v5/pgxpool"
)

var db *pgxpool.Pool

func Cleanup() {
	db.Close()
}

func Pool() *pgxpool.Pool {
	return db
}

func Init() {
	connString := os.Getenv("LLINK_POSTGRES_CONNECTION_URL")
	if connString == "" {
		flog.Error("must provide LLINK_POSTGRES_CONNECTION_URL in env")
		os.Exit(1)
	}

	dbpool, err := pgxpool.New(context.Background(), connString)
	if err != nil {
		flog.Error("unable to create connection pool", "error", err)
		os.Exit(1)
	}

	var greeting string
	err = dbpool.QueryRow(context.Background(), "select 'Hello, world!'").Scan(&greeting)
	if err != nil {
		flog.Error("queryRow failed", "error", err)
		os.Exit(1)
	}

	flog.Info("successfully connected to database", "greeting", greeting)

	db = dbpool
}