package base

import (
	"context"

	"github.com/Neur0toxine/sshpoke/internal/logger"
	"github.com/Neur0toxine/sshpoke/pkg/dto"
	"go.uber.org/zap"
)

type Base struct {
	ctx           context.Context
	name          string
	log           *zap.SugaredLogger
	eventStatusCb EventStatusCallback
}

func New(ctx context.Context, name string) Base {
	return Base{
		ctx:  ctx,
		name: name,
		log:  logger.Default.With(zap.String("serverName", name)).Sugar(),
	}
}

func (b *Base) SetEventStatusCallback(callback EventStatusCallback) {
	b.eventStatusCb = callback
}

func (b *Base) PushEventStatus(status dto.EventStatus) {
	if b.eventStatusCb != nil {
		b.eventStatusCb(status)
	}
}

func (b *Base) Context() context.Context {
	return b.ctx
}

func (b *Base) Name() string {
	return b.name
}

func (b *Base) Log() *zap.SugaredLogger {
	return b.log
}