// 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()
}