package usecases

import (
	"cargo-erp-backend/internal/domain"
	"cargo-erp-backend/internal/handlers/dto"
	"cargo-erp-backend/pkg/datatable"
	"cargo-erp-backend/pkg/helpers"
	"time"
)

type ProvinceUsecaseInterface interface {
	GetAll(page int, limit int) ([]domain.Province, int64, error)
	GetList(dto.DataTableRequest) (dto.DataTableResponse, error)
	GetByID(id string) (domain.Province, error)
	Create(name string, createdBy string) (domain.Province, error)
	Update(id string, name string, updatedBy string) (domain.Province, error)
	Delete(id string) error
}

type ProvinceUsecase struct {
	Helper   helpers.HelperInterface
	AuthUser domain.User
}

func NewProvinceUsecase(helper helpers.HelperInterface, authUser domain.User) ProvinceUsecaseInterface {
	return &ProvinceUsecase{Helper: helper, AuthUser: authUser}
}

func (u *ProvinceUsecase) GetAll(page int, limit int) ([]domain.Province, int64, error) {
	var provinces []domain.Province
	var total int64

	db := u.Helper.GetDB("slave")
	if u.AuthUser.UsersRole_Relation.RoleKey != "SPM" {
		db.Where(db.Where("merchant_id IS NULL").Or("merchant_id=?", u.AuthUser.MerchantID))
	}
	db.Model(&domain.Province{}).Count(&total)

	offset := (page - 1) * limit
	if err := db.Offset(offset).Limit(limit).Find(&provinces).Error; err != nil {
		return nil, 0, err
	}

	return provinces, total, nil
}

func (u *ProvinceUsecase) GetList(req dto.DataTableRequest) (dto.DataTableResponse, error) {

	var response dto.DataTableResponse
	db := u.Helper.GetDB("slave")
	tx := db.Table("province").Joins("LEFT JOIN merchant ON province.merchant_id=merchant.id")
	dt := datatable.NewDatatable(tx, req)
	coldef := make([]dto.DataTableColDef, 0)
	coldef = append(coldef, dto.DataTableColDef{
		Field:     "province.name",
		Alias:     "name",
		Formatter: nil,
	})
	coldef = append(coldef, dto.DataTableColDef{
		Field:     "merchant.id",
		Alias:     "merchant_id",
		Formatter: nil,
	})
	coldef = append(coldef, dto.DataTableColDef{
		Field:     "province.id",
		Alias:     "id",
		Formatter: nil,
	})

	response = dt.Render(coldef)

	return response, nil
}

func (u *ProvinceUsecase) GetByID(id string) (domain.Province, error) {
	var province domain.Province

	db := u.Helper.GetDB("slave")

	if err := db.Where("id = ?", id).First(&province).Error; err != nil {
		return domain.Province{}, err
	}

	return province, nil
}

func (u *ProvinceUsecase) Create(name string, createdBy string) (domain.Province, error) {
	now := time.Now()
	province := domain.Province{
		Name:      &name,
		CreatedBy: &createdBy,
		CreatedOn: &now,
		UpdatedBy: &createdBy,
		UpdatedOn: &now,
	}

	db := u.Helper.GetDB("master")

	if err := db.Create(&province).Error; err != nil {
		return domain.Province{}, err
	}

	return province, nil
}

func (u *ProvinceUsecase) Update(id string, name string, updatedBy string) (domain.Province, error) {
	var province domain.Province

	db := u.Helper.GetDB("master")
	if u.AuthUser.UsersRole_Relation.RoleKey != "SPM" {
		db.Where("merchant_id=?", u.AuthUser.MerchantID)
	}
	if err := db.Where("id = ?", id).First(&province).Error; err != nil {
		return domain.Province{}, err
	}

	now := time.Now()
	province.Name = &name
	province.UpdatedBy = &updatedBy
	province.UpdatedOn = &now

	if err := db.Save(&province).Error; err != nil {
		return domain.Province{}, err
	}

	return province, nil
}

func (u *ProvinceUsecase) Delete(id string) error {
	db := u.Helper.GetDB("master")
	if u.AuthUser.UsersRole_Relation.RoleKey != "SPM" {
		db.Where("merchant_id=?", u.AuthUser.MerchantID)
	}
	if err := db.Delete(&domain.Province{}, "id = ?", id).Error; err != nil {
		return err
	}

	return nil
}