package app

import (
	"context"
	"net/http"
	"time"

	"git.118im.com/im_team/im_workbench/internal/accounts"
	"git.118im.com/im_team/im_workbench/internal/auth"
	"git.118im.com/im_team/im_workbench/internal/config"
	"git.118im.com/im_team/im_workbench/internal/finance"
	"git.118im.com/im_team/im_workbench/internal/httpapi"
	"git.118im.com/im_team/im_workbench/internal/mall"
	"git.118im.com/im_team/im_workbench/internal/portal"
	"git.118im.com/im_team/im_workbench/internal/store"
	"git.118im.com/im_team/im_workbench/internal/tasks"
	"github.com/prometheus/client_golang/prometheus/promhttp"
	"github.com/redis/go-redis/v9"
)

type App struct {
	store *store.Store
	redis *redis.Client
	api   *httpapi.API
}

func New(cfg *config.Config) (*App, error) {
	st, err := store.Open(cfg.MySQL)
	if err != nil {
		return nil, err
	}
	redisClient := redis.NewClient(&redis.Options{
		Addr: cfg.Redis.Addr, Username: cfg.Redis.Username, Password: cfg.Redis.Password, DB: cfg.Redis.DB,
	})
	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()
	if err := redisClient.Ping(ctx).Err(); err != nil {
		_ = redisClient.Close()
		_ = st.Close()
		return nil, err
	}
	authenticator := auth.New(cfg.Auth, redisClient)
	accountService := accounts.New(st, redisClient, cfg.Auth)
	portalService := portal.New(st)
	taskService := tasks.New(st)
	mallService := mall.New(st)
	financeService := finance.New(st, redisClient, cfg.Auth)
	return &App{store: st, redis: redisClient, api: httpapi.New(authenticator, accountService, portalService, taskService, mallService, financeService)}, nil
}

func (a *App) Handler() http.Handler {
	mux := http.NewServeMux()
	mux.HandleFunc("/healthz", a.api.Healthz)
	mux.Handle("/metrics", promhttp.Handler())
	a.api.Register(mux)
	return mux
}

func (a *App) Close() {
	if a.redis != nil {
		_ = a.redis.Close()
	}
	if a.store != nil {
		_ = a.store.Close()
	}
}