package jnpf.dmsService;

import jnpf.dmsEntity.storage.*;
import jnpf.dmsMapper.DmsRevisionEditMapper;
import jnpf.exception.DataException;
import jnpf.util.JsonUtil;
import org.junit.jupiter.api.Test;
import java.util.*;
import static org.junit.jupiter.api.Assertions.*;
import static org.mockito.Mockito.*;

public class DmsRevisionEditServiceTest {
    @Test public void nonEditableSourcesReceiveFreshVerifiableReceiptsWithoutOldUploadTokens() {
        for (String extension : List.of("txt", "pdf", "doc", "docx")) {
            DmsRevisionEditMapper mapper = mock(DmsRevisionEditMapper.class);
            DmsRevisionEditSource source = source();
            source.setFiles(JsonUtil.getObjectToString(List.of(Map.of("fileId", "original", "name", "source." + extension,
                    "uploadReceipt", "expired-upload", "laiyuan_muban_id", "template"))));
            when(mapper.sources("tenant", "owner", "source", "model")).thenReturn(List.of(source));
            Map<String, String> tickets = new HashMap<>();
            DmsRevisionEditService service = new DmsRevisionEditService(mapper, tickets::get,
                    (json, ttl) -> { tickets.put("receipt", json); return "receipt"; });
            Map<?, ?> file = (Map<?, ?>)((List<?>)service.prepare(request(), "tenant", "owner").get(0).get("files")).get(0);
            assertEquals("receipt", file.get("revisionEditReceipt"));
            assertFalse(file.containsKey("uploadReceipt"));
            Map<?, ?> verified = (Map<?, ?>) service.verify(request()).get("file");
            assertEquals("source." + extension, verified.get("name"));
            assertEquals("template", verified.get("laiyuan_muban_id"));
            source.setFiles(JsonUtil.getObjectToString(List.of(Map.of("fileId", "original", "name", "source." + extension,
                    "uploadReceipt", "expired-upload"))));
            Map<?, ?> ordinary = (Map<?, ?>)((List<?>)service.prepare(request(), "tenant", "owner").get(0).get("files")).get(0);
            assertFalse(ordinary.containsKey("laiyuan_muban_id"));
            assertFalse(((Map<?, ?>)service.verify(request()).get("file")).containsKey("laiyuan_muban_id"));
            when(mapper.sources("tenant", "owner", "source", "model")).thenReturn(List.of());
            assertThrows(DataException.class, () -> service.verify(request()));
        }
    }
    @Test public void unchangedSourceReturnsCompleteTrustedJsonButEditedCopyDoesNot() throws Exception {
        DmsRevisionEditMapper mapper = mock(DmsRevisionEditMapper.class);
        DmsRevisionEditSource source = source();
        String original = "{\"fileId\":\"original\",\"name\":\"source.txt\",\"uploadReceipt\":\"historical-owner-token\",\"fileVersionId\":null}";
        source.setFiles("[" + original + "]");
        when(mapper.sources("tenant", "owner", "source", "model")).thenReturn(List.of(source));
        Map<String,Object> grant = proof();
        grant.put("file", Map.of("fileId", "original", "name", "source.txt"));
        grant.put("sourceFileJson", "forged");
        DmsRevisionEditService service = new DmsRevisionEditService(mapper,
                token -> JsonUtil.getObjectToString(grant), (json, ttl) -> "issued");
        Map<String,Object> verified = service.verify(request());
        com.fasterxml.jackson.databind.ObjectMapper json = new com.fasterxml.jackson.databind.ObjectMapper();
        assertEquals(json.readTree(original), json.readTree((String)verified.get("sourceFileJson")));
        assertFalse(((Map<?,?>)verified.get("file")).containsKey("laiyuan_muban_id"));
        grant.put("file", Map.of("fileId", "copy", "name", "source.docx"));
        assertFalse(service.verify(request()).containsKey("sourceFileJson"));
    }

    private static void assertThrows(Class<? extends Throwable> type, Runnable action) {
        try { action.run(); } catch (Throwable error) { if (type.isInstance(error)) return; throw error; }
        fail("Expected " + type.getSimpleName());
    }

    private DmsRevisionEditRequest request() {
        DmsRevisionEditRequest r = new DmsRevisionEditRequest();
        r.setTenantId("tenant"); r.setOwnerId("owner"); r.setModelId("model");
        r.setSourceItemId("source"); r.setField("wenjian"); r.setParentField(""); r.setReceipt("receipt"); return r;
    }
    private DmsRevisionEditSource source() {
        DmsRevisionEditSource s = new DmsRevisionEditSource();
        s.setVersionId("version"); s.setField("wenjian"); s.setParentField("");
        s.setFiles("[{\"fileId\":\"original\",\"name\":\"source.docx\"}]"); return s;
    }
    private Map<String,Object> proof() {
        Map<String,Object> p = new HashMap<>();
        p.put("purpose", "dms-revision-edit"); p.put("tenantId", "tenant"); p.put("ownerId", "owner");
        p.put("modelId", "model"); p.put("sourceItemId", "source"); p.put("field", "wenjian"); p.put("parentField", "");
        p.put("versionId", "version"); p.put("originFileId", "original"); p.put("expiresAt", System.currentTimeMillis()+60000);
        p.put("file", Map.of("fileId", "copy", "name", "source.docx")); return p;
    }
    @Test public void receiptRevalidatesCanonicalSourceAndRejectsTampering() {
        DmsRevisionEditMapper mapper = mock(DmsRevisionEditMapper.class);
        when(mapper.sources("tenant","owner","source","model")).thenReturn(List.of(source()));
        Map<String,Object> current = proof();
        DmsRevisionEditService service = new DmsRevisionEditService(mapper,
                token -> JsonUtil.getObjectToString(current), (json, ttl) -> "issued");
        assertEquals("copy", ((Map<?,?>)service.verify(request()).get("file")).get("fileId"));
        for (String field : List.of("purpose", "tenantId", "ownerId", "modelId", "field", "parentField", "versionId", "originFileId")) {
            Object original = current.put(field, "foreign");
            assertThrows(DataException.class, () -> service.verify(request())); current.put(field, original);
        }
        current.put("expiresAt", 1); assertThrows(DataException.class, () -> service.verify(request()));
    }
    @Test public void sourceRemovedAfterIssuanceIsRejected() {
        DmsRevisionEditMapper mapper = mock(DmsRevisionEditMapper.class);
        DmsRevisionEditService service = new DmsRevisionEditService(mapper,
                token -> JsonUtil.getObjectToString(proof()), (json, ttl) -> "issued");
        assertThrows(DataException.class, () -> service.verify(request()));
    }
    @Test public void prepareUsesOnlyServerFilesAndBindsChildVersion() {
        DmsRevisionEditMapper mapper = mock(DmsRevisionEditMapper.class);
        DmsRevisionEditSource child = source(); child.setParentField("jilu_mingxi"); child.setField("jilu_wenjian");
        when(mapper.sources("tenant","owner","source","model")).thenReturn(List.of(child));
        DmsRevisionEditService service = new DmsRevisionEditService(mapper, token -> null, (json, ttl) -> {
            Map<?,?> proof = JsonUtil.getJsonToBean(json, Map.class);
            assertEquals("jilu_mingxi", proof.get("parentField")); assertEquals("version", proof.get("versionId"));
            assertEquals("original", proof.get("originFileId")); assertEquals(14400L, ttl.longValue()); return "issued";
        });
        List<Map<String,Object>> rows = service.prepare(request(), "tenant", "owner");
        assertEquals("issued", ((Map<?,?>)((List<?>)rows.get(0).get("files")).get(0)).get("revisionEditReceipt"));
    }
}