package controllers

import (
	"database/sql"
	"fmt"
	"regexp"
	"strings"

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

	"panel/app/models"
	"panel/app/services"
	"panel/pkg/tools"
)

type MenuItem struct {
	Name  string `json:"name"`
	Title string `json:"title"`
	Icon  string `json:"icon"`
	Jump  string `json:"jump"`
}

type InfoController struct {
	plugin  services.Plugin
	setting services.Setting
}

func NewInfoController() *InfoController {
	return &InfoController{
		plugin:  services.NewPluginImpl(),
		setting: services.NewSettingImpl(),
	}
}

// Name 获取面板名称
func (c *InfoController) Name(ctx http.Context) http.Response {
	var setting models.Setting
	err := facades.Orm().Query().Where("key", "name").First(&setting)
	if err != nil {
		facades.Log().Error("[面板][InfoController] 查询面板名称失败 ", err)
		return Error(ctx, http.StatusInternalServerError, "系统内部错误")
	}

	return Success(ctx, http.Json{
		"name": setting.Value,
	})
}

// HomePlugins 获取首页插件
func (c *InfoController) HomePlugins(ctx http.Context) http.Response {
	var plugins []models.Plugin
	err := facades.Orm().Query().Where("show", 1).Find(&plugins)
	if err != nil {
		facades.Log().Error("[面板][InfoController] 查询首页插件失败 ", err)
		return Error(ctx, http.StatusInternalServerError, "系统内部错误")
	}

	type pluginsData struct {
		models.Plugin
		Name string `json:"name"`
	}

	var pluginsJson []pluginsData
	for _, plugin := range plugins {
		pluginsJson = append(pluginsJson, pluginsData{
			Plugin: plugin,
			Name:   services.NewPluginImpl().GetBySlug(plugin.Slug).Name,
		})
	}

	return Success(ctx, pluginsJson)
}

// NowMonitor 获取当前监控信息
func (c *InfoController) NowMonitor(ctx http.Context) http.Response {
	return Success(ctx, tools.GetMonitoringInfo())
}

// SystemInfo 获取系统信息
func (c *InfoController) SystemInfo(ctx http.Context) http.Response {
	monitorInfo := tools.GetMonitoringInfo()

	return Success(ctx, http.Json{
		"os_name":       monitorInfo.Host.Platform + " " + monitorInfo.Host.PlatformVersion,
		"uptime":        fmt.Sprintf("%.2f", float64(monitorInfo.Host.Uptime)/86400),
		"panel_version": facades.Config().GetString("panel.version"),
	})
}

// CountInfo 获取面板统计信息
func (c *InfoController) CountInfo(ctx http.Context) http.Response {
	var websiteCount int64
	err := facades.Orm().Query().Model(models.Website{}).Count(&websiteCount)
	if err != nil {
		websiteCount = -1
	}

	var mysql models.Plugin
	mysqlInstalled := true
	err = facades.Orm().Query().Where("slug like ?", "mysql%").FirstOrFail(&mysql)
	if err != nil {
		mysqlInstalled = false
	}
	var postgresql models.Plugin
	postgresqlInstalled := true
	err = facades.Orm().Query().Where("slug like ?", "postgresql%").FirstOrFail(&postgresql)
	if err != nil {
		postgresqlInstalled = false
	}
	var databaseCount int64
	if mysqlInstalled {
		status := tools.Exec("systemctl status mysqld | grep Active | grep -v grep | awk '{print $2}'")
		if status == "active" {
			rootPassword := c.setting.Get(models.SettingKeyMysqlRootPassword)
			type database struct {
				Name string `json:"name"`
			}

			db, err := sql.Open("mysql", "root:"+rootPassword+"@unix(/tmp/mysql.sock)/")
			if err != nil {
				facades.Log().With(map[string]any{
					"error": err.Error(),
				}).Error("[面板][InfoController] 获取数据库列表失败")
				databaseCount = -1
			} else {
				defer db.Close()
				rows, err := db.Query("SHOW DATABASES")
				if err != nil {
					facades.Log().With(map[string]any{
						"error": err.Error(),
					}).Error("[面板][InfoController] 获取数据库列表失败")
					databaseCount = -1
				}
				defer rows.Close()

				var databases []database
				for rows.Next() {
					var d database
					err := rows.Scan(&d.Name)
					if err != nil {
						continue
					}

					databases = append(databases, d)
				}
				databaseCount = int64(len(databases))
			}
		}
	}
	if postgresqlInstalled {
		status := tools.Exec("systemctl status postgresql | grep Active | grep -v grep | awk '{print $2}'")
		if status == "active" {
			raw := tools.Exec(`echo "\l" | su - postgres -c "psql"`)
			databases := strings.Split(raw, "\n")
			databases = databases[3 : len(databases)-1]
			databaseCount = int64(len(databases))
		}
	}

	var ftpCount int64
	var ftpPlugin = c.plugin.GetInstalledBySlug("pureftpd")
	if ftpPlugin.ID != 0 {
		listRaw := tools.Exec("pure-pw list")
		if len(listRaw) != 0 {
			listArr := strings.Split(listRaw, "\n")
			ftpCount = int64(len(listArr))
		}
	}

	var cronCount int64
	err = facades.Orm().Query().Model(models.Cron{}).Count(&cronCount)
	if err != nil {
		cronCount = -1
	}

	return Success(ctx, http.Json{
		"website":  websiteCount,
		"database": databaseCount,
		"ftp":      ftpCount,
		"cron":     cronCount,
	})
}

// InstalledDbAndPhp 获取已安装的数据库和 PHP 版本
func (c *InfoController) InstalledDbAndPhp(ctx http.Context) http.Response {
	var php []models.Plugin
	err := facades.Orm().Query().Where("slug like ?", "php%").Find(&php)
	if err != nil {
		return Error(ctx, http.StatusInternalServerError, "系统内部错误")
	}

	var mysql models.Plugin
	mysqlInstalled := true
	err = facades.Orm().Query().Where("slug like ?", "mysql%").FirstOrFail(&mysql)
	if err != nil {
		mysqlInstalled = false
	}

	var postgresql models.Plugin
	postgresqlInstalled := true
	err = facades.Orm().Query().Where("slug like ?", "postgresql%").FirstOrFail(&postgresql)
	if err != nil {
		postgresqlInstalled = false
	}

	type data struct {
		Label string `json:"label"`
		Value string `json:"value"`
	}
	var phpData []data
	var dbData []data
	phpData = append(phpData, data{Value: "0", Label: "不使用"})
	dbData = append(dbData, data{Value: "0", Label: "不使用"})
	for _, p := range php {
		match := regexp.MustCompile(`php(\d+)`).FindStringSubmatch(p.Slug)
		if len(match) == 0 {
			continue
		}

		phpData = append(phpData, data{Value: strings.ReplaceAll(p.Slug, "php", ""), Label: c.plugin.GetBySlug(p.Slug).Name})
	}

	if mysqlInstalled {
		dbData = append(dbData, data{Value: "mysql", Label: "MySQL"})
	}
	if postgresqlInstalled {
		dbData = append(dbData, data{Value: "postgresql", Label: "PostgreSQL"})
	}

	return Success(ctx, http.Json{
		"php": phpData,
		"db":  dbData,
	})
}

// CheckUpdate 检查面板更新
func (c *InfoController) CheckUpdate(ctx http.Context) http.Response {
	version := facades.Config().GetString("panel.version")
	remote, err := tools.GetLatestPanelVersion()
	if err != nil {
		return Error(ctx, http.StatusInternalServerError, "获取最新版本失败")
	}

	if tools.VersionCompare(version, remote.Version, ">=") {
		return Success(ctx, http.Json{
			"update":  false,
			"version": remote.Version,
			"name":    remote.Name,
			"body":    remote.Body,
			"date":    remote.Date,
		})
	}

	return Success(ctx, http.Json{
		"update":  true,
		"version": remote.Version,
		"name":    remote.Name,
		"body":    remote.Body,
		"date":    remote.Date,
	})
}

// Update 更新面板
func (c *InfoController) Update(ctx http.Context) http.Response {
	var task models.Task
	err := facades.Orm().Query().Where("status", models.TaskStatusRunning).OrWhere("status", models.TaskStatusWaiting).FirstOrFail(&task)
	if err == nil {
		return Error(ctx, http.StatusInternalServerError, "当前有任务正在执行，禁止更新")
	}

	panel, err := tools.GetLatestPanelVersion()
	if err != nil {
		facades.Log().With(map[string]any{
			"error": err.Error(),
		}).Error("[面板][InfoController] 获取最新版本失败")
		return Error(ctx, http.StatusInternalServerError, "获取最新版本失败")
	}

	err = tools.UpdatePanel(panel)
	if err != nil {
		facades.Log().With(map[string]any{
			"error": err.Error(),
		}).Error("[面板][InfoController] 更新面板失败")
		return Error(ctx, http.StatusInternalServerError, "更新失败: "+err.Error())
	}

	tools.RestartPanel()
	return Success(ctx, nil)
}

// Restart 重启面板
func (c *InfoController) Restart(ctx http.Context) http.Response {
	var task models.Task
	err := facades.Orm().Query().Where("status", models.TaskStatusRunning).OrWhere("status", models.TaskStatusWaiting).FirstOrFail(&task)
	if err == nil {
		return Error(ctx, http.StatusInternalServerError, "当前有任务正在执行，禁止重启")
	}

	tools.RestartPanel()
	return Success(ctx, nil)
}