package models

import (
	"database/sql"
	"fmt"
	"log"

	_ "github.com/lib/pq"
)

var DB *sql.DB

func InitDB() {
	var err error

	// Example DSN format for PostgreSQL:
	// "postgres://username:password@host:port/dbname?sslmode=disable"
	dsn := "postgres://mannpatel:Admin@localhost:5432/poll_database?sslmode=disable"

	DB, err = sql.Open("postgres", dsn)
	if err != nil {
		log.Fatalf("Failed to connect to DB: %v", err)
	}

	err = DB.Ping()
	if err != nil {
		log.Fatalf("Failed to ping DB: %v", err)
	}

	fmt.Println("Database connection successful")
}