package db

import (
	"database/sql"
)

var Engine *DB

type DB struct {
	DatabaseManager
	LogFunc func(log Log)
}

func (d *DB) SetLogger(f func(log Log)) *DB {
	d.LogFunc = f
	return d
}

func Open(config map[string]DBConfig) *DB {
	var configP = make(map[string]*DBConfig)

	for name := range config {
		c := config[name]
		configP[name] = &c
	}

	db := DB{
		DatabaseManager: DatabaseManager{
			Configs:     configP,
			Connections: make(map[string]*Connection),
		},
	}

	db.Connection("default")

	Engine = &db

	return Engine
}

func (d *DB) AddConfig(name string, config *DBConfig) *DB {
	Engine.Configs[name] = config
	return d
}

func (d *DB) GetConfigs() map[string]*DBConfig {
	return Engine.Configs
}

func (*DB) Raw(connectionName ...string) *sql.DB {
	if len(connectionName) > 0 {
		c := Engine.Connection(connectionName[0])
		return (*c).GetDB()
	} else {
		c := Engine.Connection("default")
		return (*c).GetDB()
	}
}