package jnpf.bizcommon.onlyoffice.service.impl;

import cn.hutool.json.JSONUtil;
import cn.hutool.jwt.JWTUtil;
import com.sun.net.httpserver.HttpServer;
import jnpf.bizcommon.onlyoffice.entity.OnlyOfficeProperties;
import jnpf.exception.DataException;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import java.net.InetSocketAddress;
import java.nio.charset.StandardCharsets;
import java.util.concurrent.atomic.AtomicReference;
import static org.junit.Assert.*;

public class OnlyOfficeComparisonRendererTest {
    private HttpServer server;
    private OnlyOfficeComparisonRenderer renderer;
    private final AtomicReference<String> request = new AtomicReference<>();
    private String response = "{\"end\":true,\"urls\":{\"Result.docx\":\"http://ds/cache/files/job/output/Result.docx?md5=signed\"}}";
    @Before public void setup() throws Exception {
        server = HttpServer.create(new InetSocketAddress("127.0.0.1", 0), 0);
        server.createContext("/docbuilder", exchange -> {
            request.set(new String(exchange.getRequestBody().readAllBytes(), StandardCharsets.UTF_8));
            byte[] bytes = response.getBytes(StandardCharsets.UTF_8);
            exchange.sendResponseHeaders(200, bytes.length);
            exchange.getResponseBody().write(bytes); exchange.close();
        });
        server.createContext("/cache/files/job/output/Result.docx", exchange -> {
            byte[] bytes = new byte[]{1, 2, 3};
            exchange.sendResponseHeaders(200, bytes.length);
            exchange.getResponseBody().write(bytes); exchange.close();
        });
        server.start();
        OnlyOfficeProperties properties = new OnlyOfficeProperties();
        properties.setDsInternalUrl("http://127.0.0.1:" + server.getAddress().getPort());
        properties.setFileFetchBaseUrl("http://gateway"); properties.setJwtSecret("test-secret");
        renderer = new OnlyOfficeComparisonRenderer(properties);
    }
    @After public void cleanup() { if (server != null) server.stop(0); }
    @Test public void signsFixedScriptAndArgumentsAndReturnsOnlyCachePath() {
        String path = renderer.generate("old-ticket-url", "new-ticket-url");
        cn.hutool.json.JSONObject body = JSONUtil.parseObj(request.get());
        assertEquals("http://gateway/api/biz/onlyoffice/compare/script.js", body.getStr("url"));
        assertEquals("old-ticket-url", body.getJSONObject("argument").getStr("originalUrl"));
        assertEquals("new-ticket-url", body.getJSONObject("argument").getStr("revisedUrl"));
        assertTrue(JWTUtil.verify(body.getStr("token"), "test-secret".getBytes(StandardCharsets.UTF_8)));
        assertEquals("/cache/files/job/output/Result.docx?md5=signed", path);
        assertArrayEquals(new byte[]{1, 2, 3}, renderer.readResult(path));
    }
    @Test public void rejectsIncompleteOrFailedGeneration() {
        for (String value : new String[]{"{\"end\":false}", "{\"error\":-3}", "{\"end\":true,\"urls\":{}}"}) {
            response = value;
            try { renderer.generate("old", "new"); fail(); } catch (DataException expected) { }
        }
    }
    @Test public void rejectsNonCacheAndTraversalResults() {
        for (String value : new String[]{"http://host/private/Result.docx", "/cache/files/../Result.docx", "/cache/files/%2e%2e/Result.docx"}) {
            try { OnlyOfficeComparisonRenderer.resultPath(value); fail(); } catch (DataException expected) { }
        }
    }
}