huangguo_server/web/service/officialWebsiteser/partner.go

package officialwebsiteser

import (
	"time"

	"91porn-server/common/stderr"
	officialwebsitemod "91porn-server/models/v/officialWebsitemod"

	"go.mongodb.org/mongo-driver/bson/primitive"
)

// ─────────────────────────────────────────
// Service functions
// ─────────────────────────────────────────

func ListPartner(req *PartnerListReq) (int64, []officialwebsitemod.Partner, error) {
	filter, err := req.Query()
	if err != nil {
		return 0, nil, err
	}
	mod := &officialwebsitemod.Partner{}
	total, err := mod.Count(filter)
	if err != nil {
		return 0, nil, err
	}
	list, err := mod.FindMany(filter, req.FindOptions())
	if err != nil {
		return 0, nil, err
	}
	return total, list, nil
}

func CreatePartner(req *ModifyPartnerReq) (officialwebsitemod.Partner, error) {
	data := req.toMod()
	if err := data.Create(); err != nil {
		return officialwebsitemod.Partner{}, err
	}
	return data, nil
}

func UpdatePartner(req *ModifyPartnerReq) (resp CountResp, err error) {
	if req == nil || req.ID == "" {
		return
	}
	id, parseErr := primitive.ObjectIDFromHex(req.ID)
	if parseErr != nil || isNilObjectID(id) {
		err = stderr.ErrParamError
		return
	}
	mod := &officialwebsitemod.Partner{}
	if err = mod.FindOne(officialwebsitemod.M{"_id": id}); err != nil {
		return
	}
	var update officialwebsitemod.M
	if update, err = req.toM(); err != nil || update == nil {
		return
	}
	resp.Count, err = mod.Update(officialwebsitemod.M{"_id": id}, update)
	return
}

func DeletePartner(req *DeletePartnerReq) (data DeletePartnerResp, err error) {
	if req == nil {
		err = stderr.CodeEmptyData
		return
	}
	var id primitive.ObjectID
	if id, err = primitive.ObjectIDFromHex(req.ID); err != nil {
		return
	}
	if isNilObjectID(id) {
		err = stderr.ErrParamError
		return
	}
	mod := &officialwebsitemod.Partner{}
	data.Count, err = mod.Delete(officialwebsitemod.M{"_id": id})
	return
}

// ─────────────────────────────────────────
// Mutation (Modify) types
// ─────────────────────────────────────────

type ModifyPartnerReq struct {
	ID          string             `json:"id" form:"id"`
	Name        *string            `json:"name"`
	Description *string            `json:"description"`
	Logo        *string            `json:"logo"`
	Url         *string            `json:"url"`
	Sort        *int64             `json:"sort"`
	IsActive    *bool              `json:"isActive"`
}

// ─────────────────────────────────────────
// Delete types
// ─────────────────────────────────────────

type DeletePartnerReq struct {
	ID string `json:"id" form:"id"`
}

type DeletePartnerResp struct {
	Count int64 `json:"count"`
}

// ─────────────────────────────────────────
// toMod — ModifyPartnerReq → model struct
// ─────────────────────────────────────────

func (req *ModifyPartnerReq) toMod() officialwebsitemod.Partner {
	if req == nil {
		return officialwebsitemod.Partner{}
	}
	return officialwebsitemod.Partner{
		Name:        stringOrZero(req.Name),
		Description: stringOrZero(req.Description),
		Logo:        stringOrZero(req.Logo),
		Url:         stringOrZero(req.Url),
		SortModel: officialwebsitemod.SortModel{
			Sort:     int64OrZero(req.Sort),
			IsActive: boolOrZero(req.IsActive),
		},
	}
}

// ─────────────────────────────────────────
// toM — ModifyPartnerReq → officialwebsitemod.M
// ─────────────────────────────────────────

func (req *ModifyPartnerReq) toM() (officialwebsitemod.M, error) {
	if req == nil {
		return nil, stderr.CodeEmptyData
	}
	set := officialwebsitemod.M{}
	if req.Name != nil {
		set["name"] = *req.Name
	}
	if req.Description != nil {
		set["description"] = *req.Description
	}
	if req.Logo != nil {
		set["logo"] = *req.Logo
	}
	if req.Url != nil {
		set["url"] = *req.Url
	}
	if req.Sort != nil {
		set["sort"] = *req.Sort
	}
	if req.IsActive != nil {
		set["isActive"] = *req.IsActive
	}
	if len(set) == 0 {
		return nil, nil
	}
	set["updatedAt"] = time.Now().UTC()
	return officialwebsitemod.M{"$set": set}, nil
}

// ─────────────────────────────────────────
// List types
// ─────────────────────────────────────────

type PartnerListReq struct {
	PageReq
	ID          *string    `form:"id"`
	Name        *string    `form:"name"`
	Description *string    `form:"description"`
	Logo        *string    `form:"logo"`
	Url         *string    `form:"url"`
	Sort        *int64     `form:"sort"`
	IsActive    *bool      `form:"isActive"`
	CreatedAt   *time.Time `form:"createdAt"`
	UpdatedAt   *time.Time `form:"updatedAt"`
	DeletedAt   *time.Time `form:"deletedAt"`
}

func (r *PartnerListReq) Query() (officialwebsitemod.M, error) {
	filter := officialwebsitemod.M{}
	if err := parseObjectIDFilter(filter, "_id", r.ID); err != nil {
		return nil, err
	}
	if r.Name != nil {
		filter["name"] = *r.Name
	}
	if r.Description != nil {
		filter["description"] = *r.Description
	}
	if r.Logo != nil {
		filter["logo"] = *r.Logo
	}
	if r.Url != nil {
		filter["url"] = *r.Url
	}
	if r.Sort != nil {
		filter["sort"] = *r.Sort
	}
	if r.IsActive != nil {
		filter["isActive"] = *r.IsActive
	}
	if r.CreatedAt != nil {
		filter["createdAt"] = *r.CreatedAt
	}
	if r.UpdatedAt != nil {
		filter["updatedAt"] = *r.UpdatedAt
	}
	if r.DeletedAt != nil {
		filter["deletedAt"] = *r.DeletedAt
	}
	return filter, nil
}