package database

import (
	"database/sql"
	"fmt"
	"github.com/joho/godotenv"
	_ "github.com/lib/pq"
	"log"
	"os"
)

func InitDatabase() (*sql.DB, error) {
	err := godotenv.Load(".env")
	if err != nil {
		log.Fatal("Error loading .env file")
	}
	password := os.Getenv("PG_PASSWORD")
	host := os.Getenv("PG_HOST")
	connStr := fmt.Sprintf("user=postgres host=%s dbname=relay password=%s sslmode=disable", host, password)
	db, err := sql.Open("postgres", connStr)
	if err != nil {
		log.Fatal(err)
	}

	return db, nil

}

func GetUsers(db *sql.DB) ([]string, error) {
	query := `SELECT username FROM accounts;`

	rows, err := db.Query(query)
	if err != nil {
		log.Fatal(err)
	}

	var users []string
	for rows.Next() {
		var user string
		if err := rows.Scan(&user); err != nil {
			return nil, err
		}
		users = append(users, user)
	}
	return users, err
}