package ssh

import (
	"context"
	"errors"
	"sync"

	"github.com/Neur0toxine/sshpoke/internal/config"
	"github.com/Neur0toxine/sshpoke/internal/model"
	"github.com/Neur0toxine/sshpoke/internal/server/driver/iface"
	"github.com/Neur0toxine/sshpoke/internal/server/driver/util"
	"github.com/Neur0toxine/sshpoke/internal/server/proto/sshtun"
)

type SSH struct {
	ctx      context.Context
	name     string
	params   Params
	sessions map[string]conn
	wg       sync.WaitGroup
}

type conn struct {
	container model.Container
	tun       *sshtun.Tunnel
}

func New(ctx context.Context, name string, params config.DriverParams) (iface.Driver, error) {
	drv := &SSH{ctx: ctx, name: name, sessions: make(map[string]conn)}
	if err := util.UnmarshalParams(params, &drv.params); err != nil {
		return nil, err
	}
	return drv, nil
}

func (d *SSH) Handle(event model.Event) error {
	// TODO: Implement event handling & connections management.
	return errors.New(d.name + " server handler is not implemented yet")
}

func (d *SSH) Driver() config.DriverType {
	return config.DriverSSH
}

func (d *SSH) WaitForShutdown() {
	d.wg.Wait()
}