package extension

import (
	"context"

	"github.com/go-sonic/sonic/model/dto"
	"github.com/go-sonic/sonic/model/param"
	"github.com/go-sonic/sonic/model/property"
	"github.com/go-sonic/sonic/model/vo"
	"github.com/go-sonic/sonic/service"
	"github.com/go-sonic/sonic/template"
)

type menuExtension struct {
	MenuService   service.MenuService
	OptionService service.OptionService
	Template      *template.Template
}

func RegisterMenuFunc(template *template.Template, menuService service.MenuService, optionService service.OptionService) {
	m := &menuExtension{
		Template:      template,
		MenuService:   menuService,
		OptionService: optionService,
	}
	m.addListMenuFunc()
	m.addListMenuAsTree()
	m.addListTeams()
	m.addListMenuByTeam()
	m.addListMenuAsTreeByTeam()
	m.addGetMenuCount()
}

func (m *menuExtension) addListMenuFunc() {
	listMenu := func() ([]*dto.Menu, error) {
		ctx := context.Background()
		listTeam := m.OptionService.GetOrByDefault(ctx, property.DefaultMenuTeam)
		menus, err := m.MenuService.ListByTeam(ctx, listTeam.(string), &param.Sort{
			Fields: []string{"priority,desc"},
		})
		if err != nil {
			return nil, err
		}
		menuDTOs := m.MenuService.ConvertToDTOs(ctx, menus)
		return menuDTOs, nil
	}
	m.Template.AddFunc("listMenu", listMenu)
}

func (m *menuExtension) addListMenuAsTree() {
	listMenuAsTree := func() ([]*vo.Menu, error) {
		ctx := context.Background()
		listTeam := m.OptionService.GetOrByDefault(ctx, property.DefaultMenuTeam)
		menus, err := m.MenuService.ListAsTreeByTeam(ctx, listTeam.(string), &param.Sort{Fields: []string{"priority,desc"}})
		return menus, err
	}
	m.Template.AddFunc("listMenuAsTree", listMenuAsTree)
}

func (m *menuExtension) addListTeams() {
	listMenuTeams := func() ([]string, error) {
		ctx := context.Background()
		teams, err := m.MenuService.ListTeams(ctx)
		return teams, err
	}
	m.Template.AddFunc("listMenuTeams", listMenuTeams)
}

func (m *menuExtension) addListMenuByTeam() {
	listMenuByTeam := func(team string) ([]*dto.Menu, error) {
		ctx := context.Background()
		menus, err := m.MenuService.ListByTeam(ctx, team, &param.Sort{
			Fields: []string{"priority,desc"},
		})
		if err != nil {
			return nil, err
		}
		menuDTOs := m.MenuService.ConvertToDTOs(ctx, menus)
		return menuDTOs, nil
	}
	m.Template.AddFunc("listMenuByTeam", listMenuByTeam)
}

func (m *menuExtension) addListMenuAsTreeByTeam() {
	listMenuAsTreeByTeam := func(team string) ([]*vo.Menu, error) {
		ctx := context.Background()
		menus, err := m.MenuService.ListAsTreeByTeam(ctx, team, &param.Sort{
			Fields: []string{"priority,desc"},
		})
		return menus, err
	}
	m.Template.AddFunc("listMenuAsTreeByTeam", listMenuAsTreeByTeam)
}

func (m *menuExtension) addGetMenuCount() {
	getMenuCount := func() (int64, error) {
		ctx := context.Background()
		return m.MenuService.GetMenuCount(ctx)
	}
	m.Template.AddFunc("getMenuCount", getMenuCount)
}