package gas

import (
	"time"

	"git.apinb.com/bsm-sdk/core/errcode"
	"git.apinb.com/bsm-sdk/core/infra"
	"git.apinb.com/heqiapp/platforms/backend/api/internal/impl"
	"git.apinb.com/heqiapp/platforms/backend/api/internal/logic/common"
	"git.apinb.com/heqiapp/platforms/backend/api/internal/models"
	"github.com/gin-gonic/gin"
)

var gasStaffRoles = map[string]bool{"installer": true, "delivery": true, "operations": true}
var gasWorkStatuses = map[string]bool{"on_duty": true, "off_duty": true}

func ListStaff(ctx *gin.Context) {
	station, ok := currentGas(ctx)
	if !ok {
		return
	}
	page, size := common.PageSize(ctx)
	var list []models.StaffAccount
	var total int64
	query := common.ApplyKeywordFilter(ctx,
		common.ActiveRecords(impl.DBService.Model(&models.StaffAccount{})).Where("gas_basic_id = ?", station.ID),
		&models.StaffAccount{})
	if roleCode := ctx.Query("role_code"); roleCode != "" {
		if !gasStaffRoles[roleCode] {
			infra.Response.Error(ctx, errcode.ErrInvalidArgument)
			return
		}
		query = query.Where("role_code = ?", roleCode)
	}
	if err := query.Count(&total).Error; err != nil {
		infra.Response.Error(ctx, err)
		return
	}
	if err := query.Order("created_at desc").Offset((page - 1) * size).Limit(size).Find(&list).Error; err != nil {
		infra.Response.Error(ctx, err)
		return
	}
	respondList(ctx, list, total)
}

func GetStaff(ctx *gin.Context) {
	station, ok := currentGas(ctx)
	if !ok {
		return
	}
	respondScopedRecord(ctx, common.ActiveRecords(impl.DBService).
		Where("identity = ? AND gas_basic_id = ?", ctx.Param("identity"), station.ID), &models.StaffAccount{})
}

type staffRequest struct {
	Username              string `json:"username"`
	Password              string `json:"password"`
	Name                  string `json:"name" binding:"required,max=64"`
	Phone                 string `json:"phone" binding:"max=32"`
	Avatar                string `json:"avatar" binding:"max=512"`
	RoleCode              string `json:"role_code" binding:"required"`
	DeliveryBasicIdentity string `json:"delivery_basic_identity"`
	WorkStatus            string `json:"work_status" binding:"required"`
}

func CreateStaff(ctx *gin.Context) {
	station, ok := currentGas(ctx)
	if !ok {
		return
	}
	var request staffRequest
	if err := ctx.ShouldBindJSON(&request); err != nil || request.Username == "" ||
		!common.IsValidAccountPassword(request.Password) || !gasStaffRoles[request.RoleCode] || !gasWorkStatuses[request.WorkStatus] {
		infra.Response.Error(ctx, errcode.ErrInvalidArgument)
		return
	}
	var deliveryID uint64
	if request.DeliveryBasicIdentity != "" {
		delivery, ok := requireDelivery(ctx, request.DeliveryBasicIdentity, station.ID)
		if !ok {
			return
		}
		deliveryID = delivery.ID
	}
	hash, err := common.PasswordHash(request.Password)
	if err != nil {
		infra.Response.Error(ctx, err)
		return
	}
	staff := models.StaffAccount{
		Entity: common.NewEntity(common.StatusEnable), Username: request.Username, PasswordHash: hash,
		Name: request.Name, Phone: request.Phone, Avatar: request.Avatar, RoleCode: request.RoleCode,
		GasBasicID: station.ID, DeliveryBasicID: deliveryID, WorkStatus: request.WorkStatus,
	}
	if err := impl.DBService.Create(&staff).Error; err != nil {
		infra.Response.Error(ctx, err)
		return
	}
	common.RespondCreatedResource(ctx, staff)
}

func UpdateStaff(ctx *gin.Context) {
	station, ok := currentGas(ctx)
	if !ok {
		return
	}
	if _, ok := requireStaff(ctx, ctx.Param("identity"), station.ID); !ok {
		return
	}
	var request staffRequest
	if err := ctx.ShouldBindJSON(&request); err != nil || !gasStaffRoles[request.RoleCode] || !gasWorkStatuses[request.WorkStatus] {
		infra.Response.Error(ctx, errcode.ErrInvalidArgument)
		return
	}
	var deliveryID uint64
	if request.DeliveryBasicIdentity != "" {
		delivery, ok := requireDelivery(ctx, request.DeliveryBasicIdentity, station.ID)
		if !ok {
			return
		}
		deliveryID = delivery.ID
	}
	common.UpdateAllowedByIdentity(ctx, &models.StaffAccount{}, gin.H{
		"name": request.Name, "phone": request.Phone, "avatar": request.Avatar, "role_code": request.RoleCode,
		"delivery_basic_id": deliveryID, "work_status": request.WorkStatus,
	}, []string{"name", "phone", "avatar", "role_code", "delivery_basic_id", "work_status"})
}

func UpdateStaffStatus(ctx *gin.Context) {
	station, ok := currentGas(ctx)
	if !ok {
		return
	}
	if _, ok := requireStaff(ctx, ctx.Param("identity"), station.ID); !ok {
		return
	}
	common.UpdateRecordStatus(ctx, &models.StaffAccount{})
}

func ArchiveStaff(ctx *gin.Context) {
	station, ok := currentGas(ctx)
	if !ok {
		return
	}
	staff, ok := requireStaff(ctx, ctx.Param("identity"), station.ID)
	if !ok {
		return
	}
	var blocking int64
	if err := impl.DBService.Model(&models.GasorderBasic{}).
		Where("staff_account_id = ? AND order_status NOT IN ?", staff.ID, []int{common.StatusCompleted, common.StatusCancelled}).
		Count(&blocking).Error; err != nil || blocking > 0 {
		infra.Response.Error(ctx, errcode.ErrInvalidArgument)
		return
	}
	if err := impl.DBService.Model(&staff).Update("status", common.StatusArchived).Error; err != nil {
		infra.Response.Error(ctx, err)
		return
	}
	infra.Response.Success(ctx, gin.H{"archived": true})
}

func ListCredential(ctx *gin.Context) {
	station, ok := currentGas(ctx)
	if !ok {
		return
	}
	page, size := common.PageSize(ctx)
	var list []models.StaffCredential
	var total int64
	query := common.ActiveRecords(impl.DBService.Model(&models.StaffCredential{})).
		Joins("JOIN staff_account ON staff_account.id = staff_credential.staff_account_id").
		Where("staff_account.gas_basic_id = ?", station.ID)
	if err := query.Count(&total).Error; err != nil {
		infra.Response.Error(ctx, err)
		return
	}
	if err := query.Order("staff_credential.created_at desc").Offset((page - 1) * size).Limit(size).Find(&list).Error; err != nil {
		infra.Response.Error(ctx, err)
		return
	}
	respondList(ctx, list, total)
}

func GetCredential(ctx *gin.Context) {
	station, ok := currentGas(ctx)
	if !ok {
		return
	}
	query := common.ActiveRecords(impl.DBService.Model(&models.StaffCredential{})).
		Joins("JOIN staff_account ON staff_account.id = staff_credential.staff_account_id").
		Where("staff_credential.identity = ? AND staff_account.gas_basic_id = ?", ctx.Param("identity"), station.ID)
	respondScopedRecord(ctx, query, &models.StaffCredential{})
}

type credentialRequest struct {
	StaffAccountIdentity string     `json:"staff_account_identity" binding:"required"`
	CredentialType       string     `json:"credential_type" binding:"required,max=64"`
	CredentialNo         string     `json:"credential_no" binding:"max=128"`
	ExpiredAt            *time.Time `json:"expired_at"`
}

func CreateCredential(ctx *gin.Context) {
	station, ok := currentGas(ctx)
	if !ok {
		return
	}
	var request credentialRequest
	if err := ctx.ShouldBindJSON(&request); err != nil {
		infra.Response.Error(ctx, errcode.ErrInvalidArgument)
		return
	}
	staff, ok := requireStaff(ctx, request.StaffAccountIdentity, station.ID)
	if !ok {
		return
	}
	credential := models.StaffCredential{
		Entity: common.NewEntity(common.StatusEnable), StaffAccountID: staff.ID,
		CredentialType: request.CredentialType, CredentialNo: request.CredentialNo, ExpiredAt: request.ExpiredAt,
	}
	if err := impl.DBService.Create(&credential).Error; err != nil {
		infra.Response.Error(ctx, err)
		return
	}
	common.RespondCreatedResource(ctx, credential)
}

func UpdateCredential(ctx *gin.Context) {
	station, ok := currentGas(ctx)
	if !ok {
		return
	}
	var existing models.StaffCredential
	if err := common.ActiveRecords(impl.DBService.Model(&models.StaffCredential{})).
		Select("staff_credential.*").
		Joins("JOIN staff_account ON staff_account.id = staff_credential.staff_account_id").
		Where("staff_credential.identity = ? AND staff_account.gas_basic_id = ?", ctx.Param("identity"), station.ID).
		First(&existing).Error; err != nil {
		infra.Response.Error(ctx, errcode.ErrInvalidArgument)
		return
	}
	var request credentialRequest
	if err := ctx.ShouldBindJSON(&request); err != nil {
		infra.Response.Error(ctx, errcode.ErrInvalidArgument)
		return
	}
	staff, ok := requireStaff(ctx, request.StaffAccountIdentity, station.ID)
	if !ok {
		return
	}
	common.UpdateAllowedByIdentity(ctx, &models.StaffCredential{}, gin.H{
		"staff_account_id": staff.ID, "credential_type": request.CredentialType,
		"credential_no": request.CredentialNo, "expired_at": request.ExpiredAt,
	}, []string{"staff_account_id", "credential_type", "credential_no", "expired_at"})
}

func UpdateCredentialStatus(ctx *gin.Context) {
	archiveCredentialWithStatus(ctx, 0, false)
}

func ArchiveCredential(ctx *gin.Context) {
	archiveCredentialWithStatus(ctx, common.StatusArchived, true)
}

func archiveCredentialWithStatus(ctx *gin.Context, status int, fixed bool) {
	station, ok := currentGas(ctx)
	if !ok {
		return
	}
	var count int64
	if err := impl.DBService.Model(&models.StaffCredential{}).
		Joins("JOIN staff_account ON staff_account.id = staff_credential.staff_account_id").
		Where("staff_credential.identity = ? AND staff_account.gas_basic_id = ?", ctx.Param("identity"), station.ID).
		Count(&count).Error; err != nil || count != 1 {
		infra.Response.Error(ctx, errcode.ErrInvalidArgument)
		return
	}
	if fixed {
		if err := impl.DBService.Model(&models.StaffCredential{}).Where("identity = ?", ctx.Param("identity")).Update("status", status).Error; err != nil {
			infra.Response.Error(ctx, err)
			return
		}
		infra.Response.Success(ctx, gin.H{"archived": true})
		return
	}
	common.UpdateRecordStatus(ctx, &models.StaffCredential{})
}