package db

import (
	"fmt"
	"os"
	"os/user"

	"github.com/jmoiron/sqlx"

	_ "github.com/lib/pq" // load Postgresql driver here, since we have no access to main
)

var db *sqlx.DB

// Connect attempts to establish a database connection
func Connect() error {

	if db != nil {
		return nil
	}

	var username = "nobody"

	u, err := user.Current()
	if err == nil {
		username = u.Username
	}

	dsn := fmt.Sprintf("postgresql://%s@localhost:26257/cycore?sslmode=disable", username)
	if os.Getenv("DSN") != "" {
		dsn = os.Getenv("DSN")
	}

	db, err = sqlx.Open("postgres", dsn)

	return err
}

// Get returns a database connection handle for the database
func Get() *sqlx.DB {
	if err := Connect(); err != nil {
		panic("no database connection: " + err.Error())
	}
	return db
}