package datasource_processor

import (
	"gitee.com/kristas/booting-go/framework/common/util/ccmap"
	"gitee.com/kristas/booting-go/framework/core/bean"
	"gitee.com/kristas/booting-go/framework/data"
	"gitee.com/kristas/booting-go/framework/data/tunnel_client/ssh"
	"gitee.com/kristas/booting-go/framework/data/tunnel_client/tcp"
	"gitee.com/kristas/booting-go/framework/data/wrapper_manager"
	"gitee.com/kristas/booting-go/framework/logging"
	"sync"
)

var (
	tcpInit     sync.Once
	sshInit     sync.Once
	typeFactory = ccmap.NewConcurrentMap()
)

type DataBaseHandler func(d *DatasourceProcessor, configure data.Configure) data.DatasourceWrapper

func UseDataBaseHandler(typ string, handler DataBaseHandler) {
	typeFactory.Set(typ, handler)
}

type DatasourceProcessor struct {
	bean.Component
	Config    ssh.Config     `value:"ssh"`
	Log       logging.Logger `bean:"log"`
	tcpDialer *tcp.Dialer
	sshDialer *ssh.Dialer
}

func (d *DatasourceProcessor) Filter(bean bean.Bean) bool {
	_, ok := bean.(data.Configure)
	return ok
}

func (d *DatasourceProcessor) Do(bean bean.Bean) {
	configure := bean.(data.Configure)
	var wrapper data.DatasourceWrapper
	switch configure.Config().Tunnel {
	case "ssh":
		wrapper = d.buildSSHConnection(configure)
	case "tcp":
		wrapper = d.buildTCPConnection(configure)
	default:
		wrapper = d.buildTCPConnection(configure)
	}
	wrapper_manager.Save(wrapper)
}

func (d *DatasourceProcessor) buildTCPConnection(configure data.Configure) data.DatasourceWrapper {
	tcpInit.Do(func() {
		d.tcpDialer = tcp.NewDialer()
	})
	return doHandler(d, configure)
}

func (d *DatasourceProcessor) buildSSHConnection(configure data.Configure) data.DatasourceWrapper {
	sshInit.Do(func() {
		client := ssh.BuildSSHClient(d.Config)
		d.sshDialer = ssh.NewDialer(client)
	})
	return doHandler(d, configure)
}

func doHandler(d *DatasourceProcessor, configure data.Configure) data.DatasourceWrapper {
	handler, ok := typeFactory.Get(configure.Config().Type).(DataBaseHandler)
	if !ok {
		d.Log.Panic("datasource type not found")
	}
	return handler(d, configure)
}