package jnpf.base.service;

import jnpf.base.ActionResult;
import jnpf.base.UserInfo;
import jnpf.base.model.flow.DataModel;
import jnpf.base.model.uploadedit.*;
import jnpf.dms.DmsRevisionEditApi;
import jnpf.exception.DataException;
import jnpf.util.JsonUtil;
import jnpf.util.TicketUtil;
import jnpf.util.UserProvider;
import org.junit.Test;
import org.mockito.MockedStatic;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.data.redis.core.ValueOperations;
import java.util.*;
import static org.junit.Assert.*;
import static org.mockito.Mockito.*;

public class DmsAuditTemporaryEditServiceTest {
    @org.junit.BeforeClass public static void initializeStandaloneContext() {
        jnpf.util.context.SpringContext.setApplicationContext(mock(org.springframework.context.ApplicationContext.class));
    }
    @Test public void txtAndPdfSourcesCanSaveButCannotStartOnlineEditing() {
        when(access.isDmsAuditTemporary(any())).thenReturn(true);
        try(MockedStatic<UserProvider> u=user(); MockedStatic<TicketUtil> tickets=mockStatic(TicketUtil.class)) {
            for (String extension : List.of("txt", "pdf")) {
                proof.put("file", Map.of("fileId", "original", "name", "source." + extension,
                        "laiyuan_muban_id", "template"));
                DataModel model = new DataModel(); model.setVisualId("model");
                Map<String,Object> data = new HashMap<>();
                data.put("xiuding_shenqing_zixiang_id", "source"); data.put("yuan_wenjian_banben_id", "version");
                data.put("wenjian", List.of(Map.of("fileId", "original", "name", "tampered.docx",
                        "uploadReceipt", "expired-upload", "revisionEditReceipt", "receipt")));
                model.setDataNewMap(data);
                service.prepareSave(model);
                Map<?,?> saved = (Map<?,?>)((List<?>)data.get("wenjian")).get(0);
                assertEquals("source." + extension, saved.get("name"));
                assertEquals("template", saved.get("laiyuan_muban_id"));
                assertFalse(saved.containsKey("uploadReceipt"));
                assertFalse(saved.containsKey("revisionEditReceipt"));
                assertThrows(DataException.class, () -> service.begin(command()));
                assertTrue(cache.isEmpty());
            }
        }
    }
    @Test public void unchangedHistoricalFilesReachNativeSaveWithFullSourceAndWithoutClientTemplate() throws Exception {
        when(access.isDmsAuditTemporary(any())).thenReturn(true);
        try(MockedStatic<UserProvider> u=user(); MockedStatic<TicketUtil> tickets=mockStatic(TicketUtil.class)) {
            for (boolean asText : List.of(false, true)) for (String template : List.of("", "templateA")) {
                String sourceJson = "{\"fileId\":\"original\",\"name\":\"source.txt\",\"uploadReceipt\":\"historical-owner-token\",\"fileVersionId\":null}";
                sourceJson = template.isEmpty() ? sourceJson : sourceJson.substring(0, sourceJson.length()-1) + ",\"laiyuan_muban_id\":\"" + template + "\"}";
                final String expectedJson = sourceJson;
                proof.put("purpose", "dms-revision-edit"); proof.put("originFileId", "original");
                proof.put("sourceFileJson", sourceJson);
                Map<String,Object> submitted = new LinkedHashMap<>(Map.of("fileId", "original",
                        "name", "forged.txt", "laiyuan_muban_id", "forged-template", "revisionEditReceipt", "receipt"));
                DataModel model = new DataModel(); model.setVisualId("model");
                Map<String,Object> child = new HashMap<>(); child.put("yuan_wenjian_banben_id", "version");
                child.put("jilu_wenjian", asText ? JsonUtil.getObjectToString(List.of(submitted)) : List.of(submitted));
                Map<String,Object> data = new HashMap<>(); data.put("xiuding_shenqing_zixiang_id", "source");
                data.put("yuan_wenjian_banben_id", "version"); data.put("wenjian", List.of(submitted));
                data.put("jilu_mingxi", List.of(child)); model.setDataNewMap(data);
                // Run the real native create entry before the existing persistence boundary.
                UploadEditDraftService drafts = mock(UploadEditDraftService.class);
                doAnswer(call -> { service.prepareSave(call.getArgument(0)); return null; }).when(drafts).prepareRevisionSave(model);
                jnpf.base.util.FlowFormDataUtil forms = spy(new jnpf.base.util.FlowFormDataUtil(
                        null, null, null, null, null, null, drafts, mock(UploadFilePermissionService.class)));
                doAnswer(call -> {
                    com.fasterxml.jackson.databind.ObjectMapper json = new com.fasterxml.jackson.databind.ObjectMapper();
                    assertEquals(json.readTree("[" + expectedJson + "]"), json.readTree(JsonUtil.getObjectToString(data.get("wenjian"))));
                    Object saved = child.get("jilu_wenjian");
                    assertEquals(json.readTree("[" + expectedJson + "]"), json.readTree(saved instanceof String ? (String)saved : JsonUtil.getObjectToString(saved)));
                    return new HashMap<>();
                }).when(forms).createDataList(model);
                forms.create(model);
                Map<String,Object> saved = jnpf.base.util.UploadEditStateMachine.files(data.get("wenjian")).get(0);
                // Subsequent save uses persisted provenance, never treats historical receipt as a new upload.
                jnpf.base.util.UploadTemplateSource.normalize(List.of(new LinkedHashMap<>(saved)), List.of(saved), "tenant", "owner");
                tickets.verify(() -> TicketUtil.parseTicket("historical-owner-token"), never());
            }
        }
    }

    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 final UploadEditAccess access = mock(UploadEditAccess.class);
    private final DmsRevisionEditApi api = mock(DmsRevisionEditApi.class);
    private final StringRedisTemplate redis = mock(StringRedisTemplate.class);
    private final ValueOperations<String,String> values = mock(ValueOperations.class);
    private final Map<String,String> cache = new HashMap<>();
    private final DmsAuditTemporaryEditService service = new DmsAuditTemporaryEditService(access, api, redis);
    private final Map<String,Object> proof = new HashMap<>();
    private final UserInfo owner = new UserInfo();
    public DmsAuditTemporaryEditServiceTest() {
        owner.setUserId("owner"); owner.setTenantId("tenant");
        proof.put("file", Map.of("fileId","original","name","source.docx"));
        proof.put("expiresAt",System.currentTimeMillis()+60000); proof.put("sourceItemId","source"); proof.put("versionId","version");
        ActionResult<Map<String,Object>> verified = mock(ActionResult.class);
        when(verified.getCode()).thenReturn(200);
        when(verified.getData()).thenReturn(proof);
        when(api.verify(any())).thenReturn(verified);
        when(redis.opsForValue()).thenReturn(values);
        when(values.get(anyString())).thenAnswer(call -> cache.get(call.getArgument(0)));
        doAnswer(call -> {cache.put(call.getArgument(0),call.getArgument(1));return null;})
                .when(values).set(anyString(),anyString(),anyLong(),any());
        when(redis.delete(anyString())).thenAnswer(call -> cache.remove(call.getArgument(0)) != null);
    }
    private MockedStatic<UserProvider> user() {
        MockedStatic<UserProvider> u = mockStatic(UserProvider.class);
        u.when(UserProvider::getUser).thenReturn(owner); u.when(UserProvider::getLoginUserId).thenReturn("owner"); return u;
    }
    private UploadEditCommand command() {
        UploadEditContext context = new UploadEditContext(); context.setModelId("model"); context.setField("wenjian");
        UploadEditCommand c = new UploadEditCommand();c.setContext(context);c.setFileId("original");
        c.setFiles(List.of(Map.of("fileId","original","revisionEditReceipt","receipt")));return c;
    }
    @Test public void beginRejectsSavedRecordsForeignFilesAndExpiredProof() {
        try(MockedStatic<UserProvider> u=user(); MockedStatic<TicketUtil> tickets=mockStatic(TicketUtil.class)) {
            UploadEditCommand c=command();c.getContext().setRecordId("saved");assertThrows(DataException.class,()->service.begin(c));
            c.getContext().setRecordId(null);c.setFileId("foreign");assertThrows(DataException.class,()->service.begin(c));
            c.setFileId("original");proof.put("expiresAt",1);assertThrows(DataException.class,()->service.begin(c));
            assertTrue(cache.isEmpty());
        }
    }
    @Test public void callbackCannotResurrectCleanedLeaseAndCannotWriteOutsideLease() {
        try(MockedStatic<UserProvider> u=user(); MockedStatic<TicketUtil> tickets=mockStatic(TicketUtil.class)) {
            String id=(String)service.begin(command()).get("leaseId");
            assertTrue(service.find("tenant","owner",id).isRevisionTemporary());
            assertThrows(DataException.class,()->service.find("tenant","other",id));
            UploadEditSaveCommand c=new UploadEditSaveCommand();c.setTenantId("tenant");c.setOwnerId("owner");c.setLeaseId(id);
            c.setOperationId("operation");c.setFile(Map.of("fileId","foreign","name","source.docx"));
            assertThrows(DataException.class,()->service.save(c));
            service.cleanup("tenant","owner",id);assertFalse(service.save(c));
        }
    }
    @Test public void editedCopyKeepsAbsoluteExpiryAndCasFailureDoesNotAdopt() {
        try(MockedStatic<UserProvider> u=user();MockedStatic<TicketUtil> tickets=mockStatic(TicketUtil.class)) {
            proof.put("file", Map.of("fileId", "original", "name", "source.docx",
                    "contentSavedAt", 100L, "laiyuan_muban_id", "templateA"));
            String id=(String)service.begin(command()).get("leaseId");
            UploadEditSaveCommand c=new UploadEditSaveCommand();c.setTenantId("tenant");c.setOwnerId("owner");c.setLeaseId(id);
            c.setOperationId("operation");c.setFile(Map.of("fileId","dir,"+id+",copy.docx","name","source.docx", "contentSavedAt", 200L));
            tickets.when(()->TicketUtil.createTicket(anyString(),anyLong())).thenAnswer(call->{
                Map<?,?> p=JsonUtil.getJsonToBean(call.getArgument(0),Map.class);
                Map<?,?> saved = (Map<?,?>) p.get("file");
                assertEquals(200L, ((Number) saved.get("contentSavedAt")).longValue());
                assertEquals("templateA", saved.get("laiyuan_muban_id"));
                assertEquals(((Number)proof.get("expiresAt")).longValue(),((Number)p.get("expiresAt")).longValue());return "next";
            });
            when(redis.execute(any(org.springframework.data.redis.core.script.RedisScript.class),anyList(),any(),any())).thenReturn(0L);
            assertFalse(service.save(c));tickets.verify(()->TicketUtil.deleteTicket("next"));
            assertEquals("original",service.find("tenant","owner",id).getFile().get("fileId"));
            when(redis.execute(any(org.springframework.data.redis.core.script.RedisScript.class),anyList(),any(),any()))
                    .thenAnswer(call -> {
                        List<String> keys=call.getArgument(1); String old=call.getArgument(2), updated=call.getArgument(3);
                        if (!Objects.equals(cache.get(keys.get(0)),old)) return 0L;
                        cache.put(keys.get(0),updated); return 1L;
                    });
            assertTrue(service.save(c));
            assertEquals("dir,"+id+",copy.docx",service.result(id).get("source") instanceof Map<?,?>
                    ? ((Map<?,?>)service.result(id).get("source")).get("fileId") : null);
            assertEquals(200L, ((Number) service.find("tenant", "owner", id).getFile().get("contentSavedAt")).longValue());
            clearInvocations(redis);
            assertTrue(service.save(c));
            verify(redis,never()).execute(any(org.springframework.data.redis.core.script.RedisScript.class),anyList(),any(),any());
        }
    }
    @Test public void nativeCreateBindsSourceVersionAndFileThenStripsReceipt() {
        try(MockedStatic<UserProvider> u=user(); MockedStatic<TicketUtil> tickets=mockStatic(TicketUtil.class)) {
            proof.put("file", Map.of("fileId","original","name","source.docx","laiyuan_muban_id","templateA"));
            DataModel model=new DataModel();model.setVisualId("model");
            Map<String,Object> data=new HashMap<>();data.put("xiuding_shenqing_zixiang_id","source");data.put("yuan_wenjian_banben_id","wrong");
            data.put("wenjian",List.of(Map.of("fileId","original","revisionEditReceipt","receipt")));model.setDataNewMap(data);
            assertThrows(DataException.class,()->service.prepareSave(model));
            data.put("yuan_wenjian_banben_id","version");data.put("xiuding_shenqing_zixiang_id","wrong");
            assertThrows(DataException.class,()->service.prepareSave(model));
            data.put("xiuding_shenqing_zixiang_id","source");service.prepareSave(model);
            Map<?,?> saved=(Map<?,?>)((List<?>)data.get("wenjian")).get(0);
            assertFalse(saved.containsKey("revisionEditReceipt"));assertEquals("source.docx",saved.get("name"));
            assertEquals("templateA", saved.get("laiyuan_muban_id"));
        }
    }
    @Test public void newAuditUploadAndTemplateUseSameCopyAndNativeSaveLifecycle() {
        when(access.isDmsAuditTemporary(any())).thenReturn(true);
        try(MockedStatic<UserProvider> u=user(); MockedStatic<TicketUtil> tickets=mockStatic(TicketUtil.class)) {
            Map<String,String> grants=new HashMap<>();
            tickets.when(()->TicketUtil.createTicket(anyString(),anyLong())).thenAnswer(call->{
                grants.put("upload-grant",call.getArgument(0));return "upload-grant";
            });
            tickets.when(()->TicketUtil.parseTicket(anyString())).thenAnswer(call->grants.get(call.getArgument(0)));
            Map<String,Object> result=service.beginUpload(command(),Map.of("fileId","original","name","source.docx","laiyuan_muban_id","templateA"));
            assertEquals(true,result.get("revisionTemporary"));
            String id=(String)result.get("leaseId");assertTrue(service.find("tenant","owner",id).isRevisionTemporary());
            DataModel model=new DataModel();model.setVisualId("model");
            Map<String,Object> data=new HashMap<>();data.put("wenjian",result.get("files"));model.setDataNewMap(data);
            service.prepareSave(model);
            assertFalse(((Map<?,?>)((List<?>)data.get("wenjian")).get(0)).containsKey("revisionEditReceipt"));
            assertEquals("templateA", ((Map<?,?>)((List<?>)data.get("wenjian")).get(0)).get("laiyuan_muban_id"));
            verifyNoInteractions(api);
        }
    }

}