package middleware

import (
	"strings"

	"github.com/goravel/framework/contracts/http"

	"panel/internal/services"
)

// MustInstall 确保已安装插件
func MustInstall() http.Middleware {
	return func(ctx http.Context) {
		path := ctx.Request().Path()
		var slug string
		if strings.HasPrefix(path, "/api/panel/website") {
			slug = "openresty"
		} else {
			pathArr := strings.Split(path, "/")
			if len(pathArr) < 4 {
				ctx.Request().AbortWithStatusJson(http.StatusForbidden, http.Json{
					"message": "插件不存在",
				})
				return
			}
			slug = pathArr[3]
		}

		plugin := services.NewPluginImpl().GetBySlug(slug)
		installedPlugin := services.NewPluginImpl().GetInstalledBySlug(slug)
		installedPlugins, err := services.NewPluginImpl().AllInstalled()
		if err != nil {
			ctx.Request().AbortWithStatusJson(http.StatusInternalServerError, http.Json{
				"message": "系统内部错误",
			})
			return
		}

		if installedPlugin.Slug != plugin.Slug {
			ctx.Request().AbortWithStatusJson(http.StatusForbidden, http.Json{
				"message": "插件 " + slug + " 未安装",
			})
			return
		}

		pluginsMap := make(map[string]bool)

		for _, p := range installedPlugins {
			pluginsMap[p.Slug] = true
		}

		for _, require := range plugin.Requires {
			_, requireFound := pluginsMap[require]
			if !requireFound {
				ctx.Request().AbortWithStatusJson(http.StatusForbidden, http.Json{
					"message": "插件 " + slug + " 需要依赖 " + require + " 插件",
				})
				return
			}
		}

		for _, exclude := range plugin.Excludes {
			_, excludeFound := pluginsMap[exclude]
			if excludeFound {
				ctx.Request().AbortWithStatusJson(http.StatusForbidden, http.Json{
					"message": "插件 " + slug + " 不兼容 " + exclude + " 插件",
				})
				return
			}
		}

		ctx.Request().Next()
	}
}