cas_tt_cloud_backend/app/api/enterprise/controller/manage/enterprise.go

package manage

import (
	"SciencesServer/app/api/enterprise/model"
	"SciencesServer/app/basic/controller"
	model2 "SciencesServer/app/common/model"
	"SciencesServer/app/session"
	"errors"
)

// Enterprise 企业管理信息
type Enterprise struct {
	*session.Enterprise
	tenantID uint64
}

type EnterpriseHandle func(session *session.Enterprise, tenantID uint64) *Enterprise

type (
	// EnterpriseInfo 企业信息
	EnterpriseInfo struct {
		ID string `json:"id"`
		*model.ManageCooperateEnterpriseInfo
		Papers  []string `json:"papers"`
		Patents []string `json:"patents"`
	}
	// EnterpriseParams 企业参数信息
	EnterpriseParams struct {
		ID                           uint64
		Mode                         int
		Name, Contact, ContactMobile string
		Papers, Patents              []uint64
		Content                      string
	}
)

// List 列表信息
func (c *Enterprise) List(name string, mode, page, pageSize int) (*controller.ReturnPages, error) {
	mManageCooperateEnterprise := model.NewManageCooperateEnterprise()

	where := []*model2.ModelWhere{
		model2.NewWhere("mode", mode),
		model2.NewWhere("tenant_id", c.tenantID),
		model2.NewWhere("uid", c.UID),
	}
	if name != "" {
		where = append(where, model2.NewWhereLike("name", name))
	}
	var count int64

	out, err := mManageCooperateEnterprise.Enterprise(page, pageSize, &count, where...)

	if err != nil {
		return nil, err
	}
	list := make([]*EnterpriseInfo, 0)

	for _, v := range out {
		list = append(list, &EnterpriseInfo{
			ID:                            v.GetEncodeID(),
			ManageCooperateEnterpriseInfo: v,
			Papers:                        v.GetPaperAttribute(),
			Patents:                       v.GetPatentAttribute(),
		})
	}
	return &controller.ReturnPages{Data: list, Count: count}, nil
}

// Form 数据操作
func (c *Enterprise) Form(params *EnterpriseParams) error {
	mManageCooperateEnterprise := model.NewManageCooperateEnterprise()

	if params.ID > 0 {
		mManageCooperateEnterprise.ID = params.ID

		isExist, err := model2.FirstField(mManageCooperateEnterprise.ManageCooperateEnterprise, []string{"id", "uid", "created_at"})

		if err != nil {
			return err
		} else if !isExist {
			return errors.New("操作错误，企业信息不存在或已被删除")
		} else if mManageCooperateEnterprise.UID != c.UID {
			return errors.New("无权限操作")
		}
	}
	mManageCooperateEnterprise.Name = params.Name
	mManageCooperateEnterprise.Mode = model2.ManageCooperateEnterpriseMode(params.Mode)
	// 已合作信息
	if mManageCooperateEnterprise.Mode == model2.ManageCooperateEnterpriseModeForCooperateAlready {
		mManageCooperateEnterprise.SetPaperAttribute(params.Papers)
		mManageCooperateEnterprise.SetPatentAttribute(params.Patents)
		// 想合作信息
	} else if mManageCooperateEnterprise.Mode == model2.ManageCooperateEnterpriseModeForCooperateToWant {
		mManageCooperateEnterprise.Contact = params.Contact
		mManageCooperateEnterprise.ContactMobile = params.ContactMobile
		mManageCooperateEnterprise.Content = params.Content
	}
	if mManageCooperateEnterprise.ID > 0 {
		return model2.Updates(mManageCooperateEnterprise.ManageCooperateEnterprise, mManageCooperateEnterprise.ManageCooperateEnterprise)
	}
	mManageCooperateEnterprise.TenantID = c.tenantID
	mManageCooperateEnterprise.UID = c.UID
	return model2.Create(mManageCooperateEnterprise.ManageCooperateEnterprise)
}

// Delete 删除操作
func (c *Enterprise) Delete(id uint64) error {
	mManageCooperateEnterprise := model.NewManageCooperateEnterprise()

	mManageCooperateEnterprise.ID = id

	isExist, err := model2.FirstField(mManageCooperateEnterprise.ManageCooperateEnterprise, []string{"id", "uid", "created_at"})

	if err != nil {
		return err
	} else if !isExist {
		return errors.New("操作错误，企业信息不存在或已被删除")
	} else if mManageCooperateEnterprise.UID != c.UID {
		return errors.New("无权限操作")
	}
	return model2.Delete(mManageCooperateEnterprise.ManageCooperateEnterprise)
}

func NewEnterprise() EnterpriseHandle {
	return func(session *session.Enterprise, tenantID uint64) *Enterprise {
		return &Enterprise{Enterprise: session, tenantID: tenantID}
	}
}