package db

import (
	"context"
	"errors"
	"github.com/jackc/pgx/v4/pgxpool"
	"go.uber.org/zap"
	"os"
)

type Client struct {
	pool *pgxpool.Pool
}

func NewClient(l *zap.Logger) *Client {
	l.Info("Setting up database connection")

	dbPool := setupPool()
	testConnection(dbPool)

	l.Info("Database successfully connected")

	return &Client{pool: dbPool}
}

func setupPool() *pgxpool.Pool {
	dbUrl := os.Getenv("DATABASE_URL")
	if dbUrl == "" {
		panic(errors.New("DATABASE_URL is not set"))
	}
	dbPool, err := pgxpool.Connect(context.Background(), dbUrl)
	if err != nil {
		panic(err)
	}

	return dbPool
}

func testConnection(dbPool *pgxpool.Pool) {
	var greeting string
	err := dbPool.QueryRow(context.Background(), "select 'Hello, world!'").Scan(&greeting)
	if err != nil {
		panic(err)
	}
}

func (c *Client) GetConn(ctx context.Context) *pgxpool.Conn {
	conn, _ := c.pool.Acquire(ctx)
	return conn
}