Function completeWPSFileUpload

api/controller/wps.go:945–1076

completeWPSFileUpload 完成WPS文件上传回调

(c *gin.Context)

// completeWPSFileUpload 完成WPS文件上传回调
func completeWPSFileUpload(c *gin.Context) (any, error) {
	// 获取文件信息
	file, wpsErr := getWPSFileCommon(c, true)
	if wpsErr != nil {
		return nil, wpsErr
	}

	// 验证文件是否存在
	if file == nil {
		return nil, weboffice.ErrFileNotExists
	}

	// 解析请求体
	var req UploadCompleteRequest
	if err := c.ShouldBindJSON(&req); err != nil {
		return nil, weboffice.ErrInvalidArguments.WithMessage("请求参数错误")
	}

	// 验证 send_back_params 中的 upload_token
	if req.SendBackParams == nil {
		return nil, weboffice.ErrInvalidArguments.WithMessage("缺少 send_back_params")
	}

	uploadToken, exists := req.SendBackParams["upload_token"]
	if !exists || uploadToken == "" {
		return nil, weboffice.ErrInvalidArguments.WithMessage("缺少 upload_token")
	}

	// 从 Redis 验证上传许可
	redisKey := fmt.Sprintf("wps_upload_token_%s", uploadToken)
	uploadTokenRaw, err := common.RedisGet(redisKey)
	if err != nil {
		return nil, weboffice.ErrPermissionDenied.WithMessage("上传许可无效或已过期")
	}

	if uploadTokenRaw == "" {
		return nil, weboffice.ErrPermissionDenied.WithMessage("上传许可不存在")
	}

	// 解析原始数据验证匹配
	ctx, err := weboffice.ParseContext(c.Request, true)
	if err != nil {
		return nil, weboffice.ErrInvalidArguments.WithMessage(err.Error())
	}
	expectedPrefix := fmt.Sprintf("wps_upload_%d_%d_", file.ID, ctx.UserModel().UserID)
	if !strings.HasPrefix(uploadTokenRaw, expectedPrefix) {
		return nil, weboffice.ErrPermissionDenied.WithMessage("上传许可不匹配")
	}

	// 解析请求信息
	rawNameValue, _ := req.Request["name"]
	name, ok := rawNameValue.(string)
	if !ok {
		return nil, weboffice.ErrInvalidArguments.WithMessage("缺少文档名称")
	}
	name = strings.TrimSpace(path.Base(name))
	if name == "" || name == "." || name == "/" {
		return nil, weboffice.ErrInvalidArguments.WithMessage("缺少文档名称")