费用报销场景（已有工作空间）

package com.docflow;

import com.google.gson.*;
import okhttp3.*;
import java.io.*;
import java.text.SimpleDateFormat;
import java.util.*;

/**
 * DocFlow 费用报销场景示例（已完成配置版）
 *
 * 适用于工作空间、文件类别、审核规则库已配置完毕的场景。
 * 流程：
 *   1. 上传待处理文件
 *   2. 轮询获取抽取结果（分类识别 + 字段抽取）
 *   3. 提交审核任务并获取审核结论
 *
 * 依赖：okhttp3、gson（见 pom.xml）
 */
public class ExpenseReimbursementConfigured {

    // ============================================================
    // 配置项 — 请替换为您的实际值
    // ============================================================
    private static final String APP_ID       = "your-app-id";
    private static final String SECRET_CODE  = "your-secret-code";
    private static final String WORKSPACE_ID = "your-workspace-id";
    private static final String REPO_ID      = "your-repo-id";

    private static final String BASE_URL  = "https://docflow.textin.com";
    private static final String FILES_DIR = "../sample_files/expense_reimbursement";

    private static final OkHttpClient HTTP      = new OkHttpClient.Builder()
            .callTimeout(java.time.Duration.ofSeconds(60)).build();
    private static final Gson         GSON      = new Gson();
    private static final MediaType    JSON_TYPE = MediaType.get("application/json; charset=utf-8");

    // ============================================================
    // 工具辅助方法
    // ============================================================

    private static Headers authHeaders() {
        return new Headers.Builder()
                .add("x-ti-app-id", APP_ID)
                .add("x-ti-secret-code", SECRET_CODE)
                .build();
    }

    private static JsonObject checkResponse(String body, String action) {
        JsonObject obj = JsonParser.parseString(body).getAsJsonObject();
        if (obj.get("code").getAsInt() != 200)
            throw new RuntimeException(action + " 失败: " + body);
        return obj;
    }

    private static String mimeType(String fileName) {
        String ext = fileName.substring(fileName.lastIndexOf('.')).toLowerCase();
        switch (ext) {
            case ".png":  return "image/png";
            case ".jpg":
            case ".jpeg": return "image/jpeg";
            case ".xls":  return "application/vnd.ms-excel";
            case ".xlsx": return "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet";
            case ".pdf":  return "application/pdf";
            default:      return "application/octet-stream";
        }
    }

    // ============================================================
    // 步骤 1：上传待处理文件
    // ============================================================

    public static String uploadFile(String workspaceId, String filePath) throws IOException {
        File file = new File(filePath);
        HttpUrl url = HttpUrl.parse(BASE_URL + "/api/app-api/sip/platform/v2/file/upload")
                .newBuilder().addQueryParameter("workspace_id", workspaceId).build();
        MultipartBody body = new MultipartBody.Builder().setType(MultipartBody.FORM)
                .addFormDataPart("file", file.getName(),
                        RequestBody.create(file, MediaType.get(mimeType(file.getName()))))
                .build();
        Request req = new Request.Builder().url(url).headers(authHeaders()).post(body).build();
        try (Response resp = HTTP.newCall(req).execute()) {
            String batchNumber = checkResponse(resp.body().string(), "上传文件")
                    .getAsJsonObject("result").get("batch_number").getAsString();
            System.out.println("[步骤1] 文件上传成功  name=" + file.getName()
                    + "  batch_number=" + batchNumber);
            return batchNumber;
        }
    }

    // ============================================================
    // 步骤 2：轮询获取抽取结果
    // ============================================================

    public static JsonObject waitForResult(String workspaceId, String batchNumber,
            int timeoutSec, int intervalSec) throws IOException, InterruptedException {
        HttpUrl url = HttpUrl.parse(BASE_URL + "/api/app-api/sip/platform/v2/file/fetch")
                .newBuilder().addQueryParameter("workspace_id", workspaceId)
                .addQueryParameter("batch_number", batchNumber).build();
        long deadline = System.currentTimeMillis() + (long) timeoutSec * 1000;
        System.out.print("[步骤2] 等待处理结果（batch_number=" + batchNumber + "）");
        while (System.currentTimeMillis() < deadline) {
            Request req = new Request.Builder().url(url).headers(authHeaders()).get().build();
            try (Response resp = HTTP.newCall(req).execute()) {
                JsonObject data = checkResponse(resp.body().string(), "获取处理结果");
                JsonArray files = data.getAsJsonObject("result").getAsJsonArray("files");
                if (files != null && files.size() > 0) {
                    JsonObject file = files.get(0).getAsJsonObject();
                    int status = file.get("recognition_status").getAsInt();
                    if (status == 1) { System.out.println(" 完成"); return file; }
                    if (status == 2) throw new RuntimeException("文件处理失败");
                }
            }
            System.out.print(".");
            Thread.sleep((long) intervalSec * 1000);
        }
        throw new RuntimeException("等待超时");
    }

    // ============================================================
    // 步骤 3：提交审核任务
    // ============================================================

    public static String submitReviewTask(String workspaceId, String name,
            String repoId, List<String> extractTaskIds) throws IOException {
        JsonObject payload = new JsonObject();
        payload.addProperty("workspace_id", workspaceId);
        payload.addProperty("name", name);
        payload.addProperty("repo_id", repoId);
        JsonArray ids = new JsonArray();
        extractTaskIds.forEach(ids::add);
        payload.add("extract_task_ids", ids);
        Request req = new Request.Builder()
                .url(BASE_URL + "/api/app-api/sip/platform/v2/review/task/submit")
                .headers(authHeaders())
                .post(RequestBody.create(GSON.toJson(payload), JSON_TYPE)).build();
        try (Response resp = HTTP.newCall(req).execute()) {
            String taskId = checkResponse(resp.body().string(), "提交审核任务")
                    .getAsJsonObject("result").get("task_id").getAsString();
            System.out.println("[步骤3] 审核任务提交成功  task_id=" + taskId);
            return taskId;
        }
    }

    // ============================================================
    // 步骤 4：轮询获取审核结果
    // ============================================================

    public static JsonObject waitForReview(String workspaceId, String taskId,
            int timeoutSec, int intervalSec) throws IOException, InterruptedException {
        JsonObject payload = new JsonObject();
        payload.addProperty("workspace_id", workspaceId);
        payload.addProperty("task_id", taskId);
        long deadline = System.currentTimeMillis() + (long) timeoutSec * 1000;
        System.out.print("[步骤3] 等待审核结果（task_id=" + taskId + "）");
        while (System.currentTimeMillis() < deadline) {
            Request req = new Request.Builder()
                    .url(BASE_URL + "/api/app-api/sip/platform/v2/review/task/result")
                    .headers(authHeaders())
                    .post(RequestBody.create(GSON.toJson(payload), JSON_TYPE)).build();
            try (Response resp = HTTP.newCall(req).execute()) {
                JsonObject result = checkResponse(resp.body().string(), "获取审核结果")
                        .getAsJsonObject("result");
                int status = result.get("status").getAsInt();
                if (status == 1 || status == 2 || status == 4 || status == 7) {
                    System.out.println(" 完成");
                    return result;
                }
            }
            System.out.print(".");
            Thread.sleep((long) intervalSec * 1000);
        }
        throw new RuntimeException("等待审核结果超时");
    }

    // ============================================================
    // 主流程
    // ============================================================

    public static void main(String[] args) throws Exception {
        System.out.println("=".repeat(60));
        System.out.println("  DocFlow 费用报销场景示例（已完成配置版）");
        System.out.println("=".repeat(60));
        System.out.println("工作空间: " + WORKSPACE_ID);
        System.out.println("规则库:   " + REPO_ID + "\n");

        // 步骤 1：上传文件
        System.out.println("开始上传待处理文件...");
        String[] filePaths = {
            FILES_DIR + "/sample_expense_form.xls",
            FILES_DIR + "/sample_hotel_receipt.png",
            FILES_DIR + "/sample_payment_record.pdf"
        };
        List<String> batchNumbers = new ArrayList<>();
        for (String path : filePaths) batchNumbers.add(uploadFile(WORKSPACE_ID, path));

        // 步骤 2：获取抽取结果
        System.out.println("\n开始获取处理结果...");
        List<JsonObject> rawResults = new ArrayList<>();
        for (String bn : batchNumbers) {
            rawResults.add(waitForResult(WORKSPACE_ID, bn, 120, 3));
        }

        // 步骤 3：提交审核任务
        System.out.println("\n开始审核...");
        String taskName = "费用报销审核_"
                + new SimpleDateFormat("yyyyMMdd_HHmmss").format(new Date());
        List<String> extractTaskIds = new ArrayList<>();
        for (JsonObject r : rawResults) {
            if (r.has("task_id")) extractTaskIds.add(r.get("task_id").getAsString());
        }
        String reviewTaskId = submitReviewTask(WORKSPACE_ID, taskName, REPO_ID, extractTaskIds);

        // 步骤 4：获取审核结果
        JsonObject reviewResult = waitForReview(WORKSPACE_ID, reviewTaskId, 300, 5);
        System.out.println("\n审核完成，状态码: " + reviewResult.get("status").getAsInt());
    }
}

private static final String APP_ID       = "your-app-id";
private static final String SECRET_CODE  = "your-secret-code";
private static final String WORKSPACE_ID = "your-workspace-id";
private static final String REPO_ID      = "your-repo-id";

cd examples/java
mvn clean package -q
java -jar target/expense-reimbursement-1.0.0.jar