経費精算シナリオ（既存ワークスペース）

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.ai";
    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