package eztools

import (
	"database/sql"
	"encoding/xml"
)

const (
	strSQL     = "mysql"
	strCfgFile = "eztools"
)

// ConnectWtParam connects to the database using parameters.
func ConnectWtParam(user, pass, ip, database string) (db *sql.DB, err error) {
	if len(user) < 1 || len(database) < 1 || len(ip) < 1 || len(pass) < 1 {
		return nil, ErrInvalidInput
	}

	db, err = sql.Open(strSQL, user+":"+pass+"@"+ip+"/"+database)
	if err != nil {
		return
	}
	err = db.Ping()
	if err != nil {
		return nil, err
	}
	return
}

var cfg struct {
	Root    xml.Name `xml:"eztools"`
	StrUSER string   `xml:"user"`
	StrPASS string   `xml:"pass"`
	StrIP   string   `xml:"ip"`
	StrDB   string   `xml:"db"`
}

// ConnectWtCfg connects to the database using parameters from an xml file
// root element is named "root", elements include "ip", "db", "user" and "pass"
func ConnectWtCfg(file string) (*sql.DB, error) {
	if _, err := XMLsReadDefaultNoCreate("", file, &cfg); err != nil {
		return nil, err
	}
	return ConnectWtParam(cfg.StrUSER, cfg.StrPASS, cfg.StrIP, cfg.StrDB)
}

// ConnectWtPath connects to the database using path as config file
func ConnectWtPath(path string) (*sql.DB, error) {
	if _, err := XMLsReadDefaultNoCreate(path, "", &cfg); err != nil {
		return nil, err
	}
	return ConnectWtParam(cfg.StrUSER, cfg.StrPASS, cfg.StrIP, cfg.StrDB)
}

// Connect connects to the database using parameters from eztools.xml
// root element is named "root", elements include "ip", "db", "user" and "pass"
func Connect() (*sql.DB, error) {
	return ConnectWtCfg(strCfgFile)
}