package jnpf.dmsService;

import jnpf.dmsEntity.permission.entity.DmsFilePermissionEntity;
import jnpf.dmsEntity.permission.enums.DmsResourceType;
import jnpf.dmsEntity.permission.model.DmsResourcePath;
import jnpf.dmsEntity.permission.vo.DmsCursorPageVO;
import jnpf.dmsEntity.permission.vo.DmsPermissionGrantVO;
import jnpf.dmsMapper.DmsFilePermissionMapper;
import jnpf.dmsMapper.DmsPermissionGroupMapper;
import jnpf.dmsMapper.DmsPermissionUsageMapper;
import jnpf.dmsPermission.DmsCurrentIdentity;
import jnpf.dmsPermission.DmsCursorCodec;
import jnpf.dmsPermission.DmsPermissionMatcher;
import jnpf.dmsPermission.DmsPermissionProperties;
import jnpf.dmsPermission.DmsResourceResolver;
import jnpf.dmsPermission.DmsUserDisplayNameResolver;
import org.junit.jupiter.api.Test;

import java.util.Arrays;
import java.util.Collections;
import java.util.Date;
import java.util.List;
import java.util.UUID;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertFalse;
import static org.junit.jupiter.api.Assertions.assertTrue;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

class DmsPermissionAdminQueryServiceTest {
    private static final String TENANT_ID = "tenant-1";
    private static final String USER_ID = "user-1";

    @Test
    void enrichesGrantPageUserNames() {
        Fixture fixture = new Fixture();
        DmsFilePermissionEntity permission = permission(UUID.randomUUID());
        when(fixture.permissionMapper.selectPage(TENANT_ID, null, null, null, null, null, 11))
                .thenReturn(Collections.singletonList(permission));
        when(fixture.grantService.toVO(permission)).thenReturn(toVO(permission));
        when(fixture.userDisplayNameResolver.resolve(any()))
                .thenReturn(Collections.singletonMap(USER_ID, "运营用户（yunying1）"));

        DmsCursorPageVO<DmsPermissionGrantVO> result = fixture.service.grants(
                null, null, null, null, 10);

        assertEquals(1, result.getList().size());
        assertEquals("运营用户（yunying1）", result.getList().get(0).getPrincipalName());
        verify(fixture.userDisplayNameResolver).resolve(any());
    }

    @Test
    void enrichesGrantDetailUserName() {
        Fixture fixture = new Fixture();
        DmsFilePermissionEntity permission = permission(UUID.randomUUID());
        when(fixture.permissionMapper.selectByTenantAndId(TENANT_ID, permission.getId()))
                .thenReturn(permission);
        when(fixture.grantService.toVO(permission)).thenReturn(toVO(permission));
        when(fixture.userDisplayNameResolver.resolve(any()))
                .thenReturn(Collections.singletonMap(USER_ID, "运营用户（yunying1）"));

        DmsPermissionGrantVO result = fixture.service.grant(permission.getId());

        assertEquals("运营用户（yunying1）", result.getPrincipalName());
        verify(fixture.userDisplayNameResolver).resolve(any());
    }

    @Test
    void enrichesDirectAndInheritedUserGrantsWithOneResolverCall() {
        Fixture fixture = new Fixture();
        UUID selectedResourceId = UUID.randomUUID();
        UUID ancestorResourceId = UUID.randomUUID();
        DmsResourcePath path = new DmsResourcePath();
        path.setActualResourceId(selectedResourceId);
        path.setActualResourceType("FOLDER");
        List<UUID> candidateIds = Arrays.asList(selectedResourceId, ancestorResourceId);
        DmsFilePermissionEntity direct = permission(selectedResourceId);
        DmsFilePermissionEntity inherited = permission(ancestorResourceId);
        when(fixture.resourceResolver.resolve(TENANT_ID, DmsResourceType.FOLDER, selectedResourceId))
                .thenReturn(path);
        when(fixture.resourceResolver.candidateResourceIds(path)).thenReturn(candidateIds);
        when(fixture.permissionMapper.selectForResourceIds(TENANT_ID, candidateIds, 1000))
                .thenReturn(Arrays.asList(direct, inherited));
        when(fixture.matcher.coversResource(any(), any())).thenReturn(true);
        when(fixture.grantService.toVO(any())).thenAnswer(invocation -> toVO(invocation.getArgument(0)));
        when(fixture.userDisplayNameResolver.resolve(any()))
                .thenReturn(Collections.singletonMap(USER_ID, "运营用户（yunying1）"));

        List<DmsPermissionGrantVO> result = fixture.service.resourceGrants(
                DmsResourceType.FOLDER, selectedResourceId);

        assertEquals(2, result.size());
        assertFalse(result.get(0).isInherited());
        assertTrue(result.get(1).isInherited());
        assertEquals("运营用户（yunying1）", result.get(0).getPrincipalName());
        assertEquals("运营用户（yunying1）", result.get(1).getPrincipalName());
        verify(fixture.userDisplayNameResolver).resolve(any());
    }

    private static DmsFilePermissionEntity permission(UUID resourceId) {
        DmsFilePermissionEntity result = new DmsFilePermissionEntity();
        result.setId(UUID.randomUUID());
        result.setResourceId(resourceId);
        result.setPrincipalType("USER");
        result.setPrincipalId(USER_ID);
        result.setValidFrom(new Date(System.currentTimeMillis() - 60_000));
        result.setUsedCount(0L);
        result.setCreatedAt(new Date());
        return result;
    }

    private static DmsPermissionGrantVO toVO(DmsFilePermissionEntity permission) {
        DmsPermissionGrantVO result = new DmsPermissionGrantVO();
        result.setId(permission.getId());
        result.setResourceId(permission.getResourceId());
        result.setPrincipalType(permission.getPrincipalType());
        result.setPrincipalId(permission.getPrincipalId());
        return result;
    }

    private static class Fixture {
        private final DmsCurrentIdentity currentIdentity = mock(DmsCurrentIdentity.class);
        private final DmsPermissionProperties properties = new DmsPermissionProperties();
        private final DmsCursorCodec cursorCodec = new DmsCursorCodec();
        private final DmsFilePermissionMapper permissionMapper = mock(DmsFilePermissionMapper.class);
        private final DmsPermissionGroupMapper permissionGroupMapper = mock(DmsPermissionGroupMapper.class);
        private final DmsPermissionUsageMapper usageMapper = mock(DmsPermissionUsageMapper.class);
        private final DmsPermissionGrantService grantService = mock(DmsPermissionGrantService.class);
        private final DmsResourceResolver resourceResolver = mock(DmsResourceResolver.class);
        private final DmsPermissionMatcher matcher = mock(DmsPermissionMatcher.class);
        private final DmsUserDisplayNameResolver userDisplayNameResolver = mock(DmsUserDisplayNameResolver.class);
        private final DmsPermissionAdminQueryService service = new DmsPermissionAdminQueryService(
                currentIdentity, properties, cursorCodec, permissionMapper, permissionGroupMapper, usageMapper,
                grantService, resourceResolver, matcher, userDisplayNameResolver);

        private Fixture() {
            when(currentIdentity.tenantId()).thenReturn(TENANT_ID);
        }
    }
}