Initial commit (code only without large binaries)

EdgeUser/internal/web/actions/default/settings/identity/uploadEnterprise.go

@@ -0,0 +1,130 @@
+// Copyright 2022 Liuxiangchao iwind.liu@gmail.com. All rights reserved. Official site: https://goedge.cn .
+
+package identity
+
+import (
+	"github.com/TeaOSLab/EdgeCommon/pkg/langs/codes"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+	"github.com/TeaOSLab/EdgeCommon/pkg/userconfigs"
+	"github.com/TeaOSLab/EdgeUser/internal/utils/sizes"
+	"github.com/TeaOSLab/EdgeUser/internal/web/actions/actionutils"
+	"github.com/iwind/TeaGo/actions"
+	"regexp"
+	"strings"
+)
+
+type UploadEnterpriseAction struct {
+	actionutils.ParentAction
+}
+
+func (this *UploadEnterpriseAction) RunPost(params struct {
+	IdentityId int64
+
+	RealName  string
+	Number    string
+	FrontFile *actions.File
+
+	Must *actions.Must
+	CSRF *actionutils.CSRF
+}) {
+	defer this.CreateLogInfo(codes.UserIdentity_LogUpdateUserIdentityEnterprise)
+
+	// 检查IdentityId
+	var identityId = params.IdentityId
+	var oldFileIds = []int64{0, 0}
+	if identityId > 0 {
+		identityResp, err := this.RPC().UserIdentityRPC().FindEnabledUserIdentity(this.UserContext(), &pb.FindEnabledUserIdentityRequest{
+			UserIdentityId: identityId,
+		})
+		if err != nil {
+			this.ErrorPage(err)
+			return
+		}
+		if identityResp.UserIdentity == nil {
+			this.Fail("错误的表单数据，请刷新后重试")
+		}
+
+		var status = identityResp.UserIdentity.Status
+		if status != userconfigs.UserIdentityStatusNone && status != userconfigs.UserIdentityStatusRejected {
+			this.Fail("当前状态下无法修改")
+		}
+
+		oldFileIds = identityResp.UserIdentity.FileIds
+	}
+	var mustUpload = identityId <= 0
+
+	params.Must.
+		Field("realName", params.RealName).
+		Require("请输入企业全称").
+		Field("number", params.Number).
+		Require("请输入营业执照号码")
+
+	var number = regexp.MustCompile(`\s+`).ReplaceAllString(params.Number, "")
+	number = strings.ToUpper(number)
+
+	// front
+	var frontFile = params.FrontFile
+	if frontFile == nil {
+		if mustUpload {
+			this.Fail("请上传营业执照照片")
+		}
+	} else {
+		var frontContentType = strings.ToLower(frontFile.ContentType)
+		_, ok := allowedContentTypes[frontContentType]
+		if !ok {
+			this.Fail("请上传正确格式的图片")
+		}
+
+		// TODO 需要可以配置最大上传尺寸，包括界面上的提示
+		if frontFile.Size > 8*sizes.M {
+			this.Fail("要上传的图片（营业执照照片）不能超过8MiB")
+		}
+	}
+
+	// 上传到服务器
+
+	var frontFileId int64 = 0
+
+	if frontFile != nil {
+		fileId, ok, err := this.UploadFile(frontFile, "enterprise.license")
+		if err != nil {
+			this.ErrorPage(err)
+			return
+		}
+		if !ok {
+			this.Fail("上传失败，请刷新后重试")
+		}
+		frontFileId = fileId
+	}
+
+	if len(oldFileIds) == 1 {
+		if frontFileId == 0 {
+			frontFileId = oldFileIds[0]
+		}
+	}
+
+	var err error
+	if identityId > 0 {
+		_, err = this.RPC().UserIdentityRPC().UpdateUserIdentity(this.UserContext(), &pb.UpdateUserIdentityRequest{
+			UserIdentityId: identityId,
+			Type:           userconfigs.UserIdentityTypeEnterpriseLicense,
+			RealName:       params.RealName,
+			Number:         number,
+			FileIds:        []int64{frontFileId},
+		})
+	} else {
+		_, err = this.RPC().UserIdentityRPC().CreateUserIdentity(this.UserContext(), &pb.CreateUserIdentityRequest{
+			OrgType:  userconfigs.UserIdentityOrgTypeEnterprise,
+			Type:     userconfigs.UserIdentityTypeEnterpriseLicense,
+			RealName: params.RealName,
+			Number:   number,
+			FileIds:  []int64{frontFileId},
+		})
+	}
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+
+	this.Success()
+}