package controllers

import (
	"server/models"

	"github.com/beego/beego/v2/client/orm"
	beego "github.com/beego/beego/v2/server/web"
)

// DashboardController 仪表盘控制器
type DashboardController struct {
	beego.Controller
}

// GetPlatformStats 获取平台统计数据（平台用户使用）
// @router /api/dashboard/platform-stats [get]
func (c *DashboardController) GetPlatformStats() {
	// 获取租户总数
	tenants, err := models.GetTenantList()
	tenantCount := int64(0)
	if err == nil {
		tenantCount = int64(len(tenants))
	}

	// 获取知识库总数及增长率（所有租户）
	knowledgeCount, currentMonthKnowledge, lastMonthKnowledge, knowledgeGrowthRate, _ := models.GetKnowledgeCountWithGrowth(0)

	// 获取用户总数（所有租户的用户，使用简单查询）
	o := orm.NewOrm()
	var userCount int64
	o.Raw("SELECT COUNT(*) FROM yz_users WHERE delete_time IS NULL").QueryRow(&userCount)

	// 获取员工总数（所有租户的员工）
	var employeeCount int64
	o.Raw("SELECT COUNT(*) FROM yz_tenant_employees WHERE delete_time IS NULL").QueryRow(&employeeCount)

	c.Data["json"] = map[string]interface{}{
		"code":    0,
		"message": "success",
		"data": map[string]interface{}{
			"tenantCount": tenantCount,
			"userCount":   userCount,
			"knowledgeCount": map[string]interface{}{
				"total":        knowledgeCount,
				"currentMonth": currentMonthKnowledge,
				"lastMonth":    lastMonthKnowledge,
				"growthRate":   knowledgeGrowthRate,
			},
			"employeeCount": employeeCount,
		},
	}
	c.ServeJSON()
}

// GetTenantStats 获取租户统计数据（租户员工使用）
// @router /api/dashboard/tenant-stats [get]
func (c *DashboardController) GetTenantStats() {
	// 获取租户ID（从JWT token中获取）
	tenantId := 0
	if tenantIdVal, ok := c.Ctx.Input.GetData("tenantId").(int); ok && tenantIdVal > 0 {
		if userType, ok := c.Ctx.Input.GetData("userType").(string); ok && userType == "employee" {
			tenantId = tenantIdVal
		}
	}

	if tenantId <= 0 {
		c.Data["json"] = map[string]interface{}{
			"code":    1,
			"message": "无法获取租户信息",
			"data":    nil,
		}
		c.ServeJSON()
		return
	}

	// 获取知识库数量及增长率
	knowledgeCount, currentMonthKnowledge, lastMonthKnowledge, knowledgeGrowthRate, _ := models.GetKnowledgeCountWithGrowth(tenantId)

	// 获取员工数量（简单查询）
	o := orm.NewOrm()
	var employeeCount int64
	o.Raw("SELECT COUNT(*) FROM yz_tenant_employees WHERE tenant_id = ? AND delete_time IS NULL", tenantId).QueryRow(&employeeCount)

	// 获取部门数量
	var departmentCount int64
	o.Raw("SELECT COUNT(*) FROM yz_tenant_departments WHERE tenant_id = ? AND delete_time IS NULL", tenantId).QueryRow(&departmentCount)

	// 获取职位数量
	var positionCount int64
	o.Raw("SELECT COUNT(*) FROM yz_tenant_positions WHERE tenant_id = ? AND delete_time IS NULL", tenantId).QueryRow(&positionCount)

	c.Data["json"] = map[string]interface{}{
		"code":    0,
		"message": "success",
		"data": map[string]interface{}{
			"knowledgeCount": map[string]interface{}{
				"total":        knowledgeCount,
				"currentMonth": currentMonthKnowledge,
				"lastMonth":    lastMonthKnowledge,
				"growthRate":   knowledgeGrowthRate,
			},
			"employeeCount":   employeeCount,
			"departmentCount": departmentCount,
			"positionCount":   positionCount,
		},
	}
	c.ServeJSON()
}