package application

import (
	"git.openprivacy.ca/openprivacy/libricochet-go/channels"
	"git.openprivacy.ca/openprivacy/libricochet-go/connection"
)

// ApplicationInstance is a  concrete instance of a ricochet application, encapsulating a connection
type ApplicationInstance struct {
	connection.AutoConnectionHandler
	Connection     *connection.Connection
	RemoteHostname string
}

// ApplicationInstanceFactory
type ApplicationInstanceFactory struct {
	handlerMap map[string]func(*ApplicationInstance) func() channels.Handler
}

// Init sets up an Application Factory
func (af *ApplicationInstanceFactory) Init() {
	af.handlerMap = make(map[string]func(*ApplicationInstance) func() channels.Handler)
}

// AddHandler defines a channel type -> handler construct function
func (af *ApplicationInstanceFactory) AddHandler(ctype string, chandler func(*ApplicationInstance) func() channels.Handler) {
	af.handlerMap[ctype] = chandler
}

func (af *ApplicationInstanceFactory) GetHandlers() []string {
	keys := make([]string, len(af.handlerMap))

	i := 0
	for k := range af.handlerMap {
		keys[i] = k
		i++
	}

	return keys
}

func (af *ApplicationInstanceFactory) GetHandler(ctype string) func(*ApplicationInstance) func() channels.Handler {
	return af.handlerMap[ctype]
}

// GetApplicationInstance builds a new application instance using a connection as a base.
func (af *ApplicationInstanceFactory) GetApplicationInstance(rc *connection.Connection) *ApplicationInstance {
	rai := new(ApplicationInstance)
	rai.Init()
	rai.RemoteHostname = rc.RemoteHostname
	rai.Connection = rc
	for t, h := range af.handlerMap {
		rai.RegisterChannelHandler(t, h(rai))
	}
	return rai
}