kpt-pasture/module/backend/enum_options.go

package backend

import (
	"context"
	"fmt"
	"kpt-pasture/model"
	"net/http"

	"gitee.com/xuyiping_admin/pkg/xerr"

	pasturePb "gitee.com/xuyiping_admin/go_proto/proto/go/backend/cow"
)

func (s *StoreEntry) BarnTypeOptions(ctx context.Context) (*pasturePb.ConfigOptionsListResponse, error) {
	userModel, err := s.GetUserModel(ctx)
	if err != nil {
		return nil, xerr.WithStack(err)
	}

	excludeTypes := make([]pasturePb.PenType_Kind, 0)
	if userModel.AppPasture.Category == pasturePb.PastureCategory_Beef {
		excludeTypes = append(excludeTypes, pasturePb.PenType_Dry_Milking, pasturePb.PenType_Peripartum)
	}

	return &pasturePb.ConfigOptionsListResponse{
		Code: http.StatusOK,
		Msg:  "ok",
		Data: s.BarnTypeEnumList(userModel, excludeTypes, ""),
	}, nil
}

func (s *StoreEntry) BarnListOptions(ctx context.Context, penType int, isAll string) (*pasturePb.ConfigOptionsListResponse, error) {
	userModel, err := s.GetUserModel(ctx)
	if err != nil {
		return nil, xerr.WithStack(err)
	}

	penList := make([]*model.Pen, 0)
	pref := s.DB.Table(new(model.Pen).TableName()).
		Where("pasture_id = ?", userModel.AppPasture.Id).
		Where("is_delete = ?", pasturePb.IsShow_Ok)
	if penType != -1 {
		pref.Where("pen_type = ?", penType)
	}
	if err = pref.Find(&penList).Error; err != nil {
		return nil, err
	}

	excludeTypes := make([]pasturePb.PenType_Kind, 0)
	if userModel.AppPasture.Category == pasturePb.PastureCategory_Beef {
		excludeTypes = append(excludeTypes, pasturePb.PenType_Dry_Milking, pasturePb.PenType_Peripartum)
	}

	barnTypeList := s.BarnTypeEnumList(userModel, excludeTypes, "")
	return &pasturePb.ConfigOptionsListResponse{
		Code: http.StatusOK,
		Msg:  "ok",
		Data: model.PenSlice(penList).ToPB2(barnTypeList, isAll),
	}, nil
}

func (s *StoreEntry) DiseaseTypeOptions(ctx context.Context, isChildren string) (*pasturePb.ConfigOptionsListResponse, error) {
	userModel, err := s.GetUserModel(ctx)
	if err != nil {
		return nil, xerr.WithStack(err)
	}
	diseaseTypeList := make([]*model.ConfigDiseaseType, 0)
	if err = s.DB.Table(new(model.ConfigDiseaseType).TableName()).
		Where("is_show =? ", pasturePb.IsShow_Ok).
		Order("kind ASC").
		Find(&diseaseTypeList).Error; err != nil {
		return nil, xerr.WithStack(err)
	}

	diseaseList := make([]*model.Disease, 0)
	if isChildren == model.IsAllYes {
		if err = s.DB.Model(new(model.Disease)).
			Where("pasture_id =? ", userModel.AppPasture.Id).
			Where("is_show =? ", pasturePb.IsShow_Ok).
			Find(&diseaseList).Error; err != nil {
			return nil, xerr.WithStack(err)
		}
	}

	return &pasturePb.ConfigOptionsListResponse{
		Code: http.StatusOK,
		Msg:  "ok",
		Data: model.ConfigDiseaseTypeSlice(diseaseTypeList).ToPB2(userModel, diseaseList),
	}, nil
}

func (s *StoreEntry) DiseaseOptions(ctx context.Context) (*pasturePb.ConfigOptionsListResponse, error) {
	userModel, err := s.GetUserModel(ctx)
	if err != nil {
		return nil, xerr.WithStack(err)
	}

	diseaseList := make([]*model.Disease, 0)
	if err = s.DB.Model(new(model.Disease)).
		Where("is_show =? ", pasturePb.IsShow_Ok).
		Where("pasture_id =? ", userModel.AppPasture.Id).
		Find(&diseaseList).Error; err != nil {
		return nil, xerr.WithStack(err)
	}
	return &pasturePb.ConfigOptionsListResponse{
		Code: http.StatusOK,
		Msg:  "ok",
		Data: model.DiseaseSlice(diseaseList).ToPB2(),
	}, nil
}

func (s *StoreEntry) PrescriptionOptions(ctx context.Context) (*pasturePb.ConfigOptionsListResponse, error) {
	userModel, err := s.GetUserModel(ctx)
	if err != nil {
		return nil, xerr.WithStack(err)
	}

	prescriptionList := make([]*model.Prescription, 0)
	if err = s.DB.Model(new(model.Prescription)).
		Where("pasture_id = ? ", userModel.AppPasture.Id).
		Where("is_show = ? ", pasturePb.IsShow_Ok).
		Find(&prescriptionList).Error; err != nil {
		return nil, xerr.WithStack(err)
	}
	return &pasturePb.ConfigOptionsListResponse{
		Code: http.StatusOK,
		Msg:  "ok",
		Data: model.PrescriptionSlice(prescriptionList).ToPB2(),
	}, nil
}

func (s *StoreEntry) BreedStatusOptions(ctx context.Context) (*pasturePb.ConfigOptionsListResponse, error) {
	userModel, err := s.GetUserModel(ctx)
	if err != nil {
		return nil, xerr.WithStack(err)
	}
	return &pasturePb.ConfigOptionsListResponse{
		Code: http.StatusOK,
		Msg:  "ok",
		Data: s.BreedStatusEnumList(userModel, ""),
	}, nil
}
func (s *StoreEntry) CowKindOptions(ctx context.Context) (*pasturePb.ConfigOptionsListResponse, error) {
	userModel, err := s.GetUserModel(ctx)
	if err != nil {
		return nil, xerr.WithStack(err)
	}
	return &pasturePb.ConfigOptionsListResponse{
		Code: http.StatusOK,
		Msg:  "ok",
		Data: s.CowKindEnumList(userModel, ""),
	}, nil
}

func (s *StoreEntry) CowSourceOptions(ctx context.Context) (*pasturePb.ConfigOptionsListResponse, error) {
	userModel, err := s.GetUserModel(ctx)
	if err != nil {
		return nil, xerr.WithStack(err)
	}

	return &pasturePb.ConfigOptionsListResponse{
		Code: http.StatusOK,
		Msg:  "ok",
		Data: s.CowSourceEnumList(userModel, ""),
	}, nil
}

func (s *StoreEntry) CowTypeOptions(ctx context.Context, optionName, isAll string) (*pasturePb.ConfigOptionsListResponse, error) {
	userModel, err := s.GetUserModel(ctx)
	if err != nil {
		return nil, xerr.WithStack(err)
	}
	return &pasturePb.ConfigOptionsListResponse{
		Code: http.StatusOK,
		Msg:  "ok",
		Data: s.CowTypeEnumList(userModel, optionName, isAll),
	}, nil
}

func (s *StoreEntry) CowTransferPenReasonOptions(ctx context.Context) (*pasturePb.ConfigOptionsListResponse, error) {
	userModel, err := s.GetUserModel(ctx)
	if err != nil {
		return nil, xerr.WithStack(err)
	}
	return &pasturePb.ConfigOptionsListResponse{
		Code: http.StatusOK,
		Msg:  "ok",
		Data: s.TransferPenEnumList(userModel, ""),
	}, nil
}

// SystemUserOptions 系统用户下拉框
func (s *StoreEntry) SystemUserOptions(ctx context.Context, depName string) (*pasturePb.ConfigOptionsListResponse, error) {
	userModel, err := s.GetUserModel(ctx)
	if err != nil {
		return nil, xerr.WithStack(err)
	}

	systemDepth := &model.SystemDept{}
	if err = s.DB.Model(new(model.SystemDept)).
		Where("name = ?", depName).
		Where("pasture_id = ?", userModel.AppPasture.Id).
		First(systemDepth).Error; err != nil {
		return nil, xerr.WithStack(err)
	}

	if systemDepth.IsDelete == pasturePb.IsShow_No {
		return nil, xerr.Custom("部门已经删除")
	}

	if systemDepth.IsShow == pasturePb.IsShow_No {
		return nil, xerr.Custom("部门已经禁用")
	}

	systemUserDepthRoleList := make([]*model.SystemUserDepthRole, 0)
	if err = s.DB.Model(new(model.SystemUserDepthRole)).
		Where("FIND_IN_SET(?,depth_ids) > 0", systemDepth.Id).
		Where("pasture_id = ?", userModel.AppPasture.Id).
		Find(&systemUserDepthRoleList).Error; err != nil {
		return nil, xerr.WithStack(err)
	}

	if len(systemUserDepthRoleList) <= 0 {
		return &pasturePb.ConfigOptionsListResponse{
			Code: http.StatusOK,
			Msg:  "ok",
			Data: make([]*pasturePb.ConfigOptionsList, 0),
		}, nil
	}

	userIds := make([]int64, 0)
	for _, v := range systemUserDepthRoleList {
		userIds = append(userIds, v.UserId)
	}

	systemUserList := make([]*model.SystemUser, 0)
	if err = s.DB.Table(new(model.SystemUser).TableName()).
		Where("id IN ?", userIds).
		Where("is_delete = ?", pasturePb.IsShow_Ok).
		Where("is_show = ? ", pasturePb.IsShow_Ok).
		Find(&systemUserList).Error; err != nil {
		return nil, xerr.WithStack(err)
	}

	return &pasturePb.ConfigOptionsListResponse{
		Code: http.StatusOK,
		Msg:  "ok",
		Data: model.SystemUserSlice(systemUserList).ToPB2(),
	}, nil
}

func (s *StoreEntry) BullOptions(ctx context.Context) (*pasturePb.BullOptionsListResponse, error) {
	userModel, err := s.GetUserModel(ctx)
	if err != nil {
		return nil, xerr.WithStack(err)
	}

	return &pasturePb.BullOptionsListResponse{
		Code: http.StatusOK,
		Msg:  "ok",
		Data: s.BullNumberEnumList(userModel, ""),
	}, nil
}

func (s *StoreEntry) SystemBaseConfigOptions(ctx context.Context, optionsName, isAll string) (*pasturePb.ConfigOptionsListResponse, error) {
	userModel, err := s.GetUserModel(ctx)
	if err != nil {
		return nil, xerr.WithStack(err)
	}

	if optionsName == "" {
		return nil, xerr.New("optionsName is empty")
	}
	getConfigFuncMap := map[string]func(userModel *model.UserModel, isAll string) []*pasturePb.ConfigOptionsList{
		"childNumber":                s.ChildNumberEnumList,
		"calvingLevel":               s.CalvingLevelEnumList,
		"dystociaReason":             s.DystociaReasonEnumList,
		"drugCategory":               s.DrugCategoryEnumList,
		"drugUsage":                  s.DrugUsageEnumList,
		"unit":                       s.UnitEnumList,
		"pregnantCheckResult":        s.PregnantCheckResultEnumList,
		"pregnantCheckMethod":        s.PregnantCheckMethodEnumList,
		"exposeEstrusType":           s.ExposeEstrusTypeEnumList,
		"frozenSemenType":            s.FrozenSemenTypeEnumList,
		"week":                       s.WeekEnumList,
		"month":                      s.MonthEnumList,
		"sameTimeCowType":            s.SameTimeCowTypeEnumList,
		"sameTimeType":               s.SameTimeTypeEnumList,
		"immunizationCowType":        s.ImmunizationCowTypeEnumList,
		"workOrderFrequency":         s.WorkOrderFrequencyEnumList,
		"workOrderSubUnit":           s.WorkOrderSubUnitEnumList,
		"workOrderPriority":          s.WorkOrderPriorityEnumList,
		"workOrderCategory":          s.WorkOrderCategoryEnumList,
		"immunizationConditions":     s.ImmunizationConditionsEnumList,
		"abortionReasons":            s.AbortionReasonsEnumList,
		"healthStatus":               s.HealthStatusEnumList,
		"calendarType":               s.CalendarTypeEnumList,
		"calvingAnalysisMethod":      s.CalvingAnalysisMethodEnumList,
		"lact":                       s.LactEnumList,
		"diseaseAnalysisMethod":      s.DiseaseAnalysisMethodEnumList,
		"singleFactorAnalysisMethod": s.SingleFactorAnalysisMethodEnumList,
		"lactIntervalSymbol":         s.LactIntervalSymbolEnumList,
		"multiFactorAnalysisMethod":  s.MultiFactorAnalysisMethodEnumList,
		"saleCowAnalysisMethod":      s.SaleCowAnalysisMethodEnumList,
		"neckRingIsBind":             s.NeckRingIsBindEnumList,
		"outType":                    s.OutTypeEnumList,
		"auditStatus":                s.AuditStatusEnumList,
		"pregnantCheckName":          s.PregnantCheckNameEnumList,
		"unMatingReasons":            s.UnMatingReasonsEnumList,
		"evenType":                   s.EventTypeEnumList,
		"outReason":                  s.CowOutReasonEnumList,
		"deathReason":                s.DeathReasonEnumList,
		"categoryKind":               s.EventCategoryEnumList,
		"indicatorsDetails":          s.IndicatorsDetailsList,
		"purposeKind":                s.CowPurposeList,
		"forbiddenMatingReasons":     s.ForbiddenMatingReasonsEnumList,
		"cowOutReason":               s.CowOutReasonEnumList,
		"deathDestination":           s.CowDeathDestinationList,
		"warningHealthLevel":         s.WarningHealthLevel,
		"behavior":                   s.Behavior,
		"matingWindowPeriod":         s.MatingWindowPeriodEnumList,
		"neckRingError":              s.NeckRingErrorEnumList,
		"indicatorType":              s.IndicatorsCategoryEnumList,
	}

	getConfigFunc, ok := getConfigFuncMap[optionsName]
	if !ok {
		return nil, fmt.Errorf("invalid optionsName: %s", optionsName)
	}

	configOptions := getConfigFunc(userModel, isAll)
	if configOptions == nil {
		return nil, fmt.Errorf("failed to retrieve configOptions for %s", optionsName)
	}

	return &pasturePb.ConfigOptionsListResponse{
		Code: http.StatusOK,
		Msg:  "ok",
		Data: configOptions,
	}, nil
}