app.go

package spine

import (
	"strings"

	"github.com/NARUBROWN/spine/core"
	"github.com/NARUBROWN/spine/internal/bootstrap"
	"github.com/NARUBROWN/spine/internal/event/consumer"
	"github.com/NARUBROWN/spine/internal/router"
	"github.com/NARUBROWN/spine/internal/ws"
	"github.com/NARUBROWN/spine/pkg/boot"
)

type App interface {
	// 생성자 선언
	Constructor(constructors ...any)
	// 라우트 선언
	Route(method string, path string, handler any, opts ...router.RouteOption)
	// 인터셉터 선언
	Interceptor(interceptors ...core.Interceptor)
	// HTTP Transport 확장 (Echo 등)
	Transport(fn func(any))
	// 독립 실행되는 Custom Transport 등록
	RegisterTransport(t core.CustomTransport)
	// 실행
	Run(opts boot.Options) error
	// 이벤트 소비자 레지스트리 반환
	Consumers() *consumer.Registry
	// 웹 소켓 레지스트리 반환
	WebSocket() *ws.Registry
}

type app struct {
	constructors      []any
	routes            []router.RouteSpec
	interceptors      []core.Interceptor
	transportHooks    []func(any)
	customTransports  []core.CustomTransport
	consumerRegistry  *consumer.Registry
	websocketRegistry *ws.Registry
}

func New() App {
	return &app{}
}

func (a *app) Constructor(constructors ...any) {
	a.constructors = append(a.constructors, constructors...)
}

func (a *app) Route(method string, path string, handler any, opts ...router.RouteOption) {
	// HTTP 메서드를 대문자로 변환해 라우팅 시 대소문자 불일치 문제를 방지합니다.
	method = strings.ToUpper(strings.TrimSpace(method))

	spec := router.RouteSpec{
		Method:  method,
		Path:    path,
		Handler: handler,
	}

	for _, opt := range opts {
		opt(&spec)
	}

	a.routes = append(a.routes, spec)
}

func (a *app) Interceptor(interceptors ...core.Interceptor) {
	a.interceptors = append(a.interceptors, interceptors...)
}

func (a *app) Transport(fn func(any)) {
	a.transportHooks = append(a.transportHooks, fn)
}

func (a *app) RegisterTransport(t core.CustomTransport) {
	a.customTransports = append(a.customTransports, t)
}

func (a *app) Run(opts boot.Options) error {
	internalConfig := bootstrap.Config{
		Address:                opts.Address,
		Constructors:           a.constructors,
		Routes:                 a.routes,
		Interceptors:           a.interceptors,
		TransportHooks:         a.transportHooks,
		CustomTransports:       a.customTransports,
		EnableGracefulShutdown: opts.EnableGracefulShutdown,
		ShutdownTimeout:        opts.ShutdownTimeout,
		Kafka:                  opts.Kafka,
		RabbitMQ:               opts.RabbitMQ,
		ConsumerRegistry:       a.consumerRegistry,
		WebSocketRegistry:      a.websocketRegistry,
		HTTP:                   opts.HTTP,
	}

	return bootstrap.Run(internalConfig)
}

func (a *app) Consumers() *consumer.Registry {
	if a.consumerRegistry == nil {
		a.consumerRegistry = consumer.NewRegistry()
	}
	return a.consumerRegistry
}

func (a *app) WebSocket() *ws.Registry {
	if a.websocketRegistry == nil {
		a.websocketRegistry = ws.NewRegistry()
	}
	return a.websocketRegistry
}