package datasource

import (
	"errors"
	"gitee.com/kristas/booting-go/framework/core/statement/types"
)

type DataSourceImpl struct {
	types.Component
	Drivers []DataSourceDriver `wire:""`
	db      interface{}
	name    string
}

func (d *DataSourceImpl) Bean() string {
	return d.name
}

func NewDataSource(name ...string) DataSource {
	d := &DataSourceImpl{}
	if len(name) > 0 {
		d.name = name[0]
	} else {
		d.name = PropertiesPath
	}
	return d
}

func (d *DataSourceImpl) Build(properties *DataSourceProperties) (DataSource, error) {
	found := false
	for _, driver := range d.Drivers {
		if driver.Type() == properties.Type && driver.Driver() == properties.Driver {
			found = true
			conn, err := driver.Open(properties)
			if err != nil {
				return nil, err
			}
			d.db = conn
		}
	}
	if !found {
		return nil, errors.New(d.Bean() + " not found matched driver!")
	}
	return d, nil
}

func (d *DataSourceImpl) DB() interface{} {
	return d.db
}