new

internal/infrastructure/http/handlers/certification_handler.go

@@ -215,6 +215,86 @@ func (h *CertificationHandler) ApplyContract(c *gin.Context) {
 	h.response.Success(c, result, "合同申请成功")
 }
 
+// RecognizeBusinessLicense OCR识别营业执照
+// @Summary OCR识别营业执照
+// @Description 上传营业执照图片进行OCR识别，自动填充企业信息
+// @Tags 认证管理
+// @Accept multipart/form-data
+// @Produce json
+// @Security Bearer
+// @Param image formData file true "营业执照图片文件"
+// @Success 200 {object} responses.BusinessLicenseResult "营业执照识别成功"
+// @Failure 400 {object} map[string]interface{} "请求参数错误"
+// @Failure 401 {object} map[string]interface{} "未认证"
+// @Failure 500 {object} map[string]interface{} "服务器内部错误"
+// @Router /api/v1/certifications/ocr/business-license [post]
+func (h *CertificationHandler) RecognizeBusinessLicense(c *gin.Context) {
+	userID := h.getCurrentUserID(c)
+	if userID == "" {
+		h.response.Unauthorized(c, "用户未登录")
+		return
+	}
+
+	// 获取上传的文件
+	file, err := c.FormFile("image")
+	if err != nil {
+		h.logger.Error("获取上传文件失败", zap.Error(err), zap.String("user_id", userID))
+		h.response.BadRequest(c, "请选择要上传的营业执照图片")
+		return
+	}
+
+	// 验证文件类型
+	allowedTypes := map[string]bool{
+		"image/jpeg": true,
+		"image/jpg":  true,
+		"image/png":  true,
+		"image/webp": true,
+	}
+	if !allowedTypes[file.Header.Get("Content-Type")] {
+		h.response.BadRequest(c, "只支持JPG、PNG、WEBP格式的图片")
+		return
+	}
+
+	// 验证文件大小（限制为5MB）
+	if file.Size > 5*1024*1024 {
+		h.response.BadRequest(c, "图片大小不能超过5MB")
+		return
+	}
+
+	// 打开文件
+	src, err := file.Open()
+	if err != nil {
+		h.logger.Error("打开上传文件失败", zap.Error(err), zap.String("user_id", userID))
+		h.response.BadRequest(c, "文件读取失败")
+		return
+	}
+	defer src.Close()
+
+	// 读取文件内容
+	imageBytes, err := io.ReadAll(src)
+	if err != nil {
+		h.logger.Error("读取文件内容失败", zap.Error(err), zap.String("user_id", userID))
+		h.response.BadRequest(c, "文件读取失败")
+		return
+	}
+
+	// 调用OCR服务识别营业执照
+	result, err := h.appService.RecognizeBusinessLicense(c.Request.Context(), imageBytes)
+	if err != nil {
+		h.logger.Error("营业执照OCR识别失败", zap.Error(err), zap.String("user_id", userID))
+		h.response.BadRequest(c, "营业执照识别失败："+err.Error())
+		return
+	}
+
+	h.logger.Info("营业执照OCR识别成功",
+		zap.String("user_id", userID),
+		zap.String("company_name", result.CompanyName),
+		zap.Float64("confidence", result.Confidence),
+	)
+
+	h.response.Success(c, result, "营业执照识别成功")
+}
+
 // ListCertifications 获取认证列表（管理员）
 // @Summary 获取认证列表
 // @Description 管理员获取认证申请列表