package jnpf.dmsService;
|
|
import jnpf.base.UserInfo;
|
import jnpf.dmsEntity.permission.entity.DmsFileVersionEntity;
|
import jnpf.dmsEntity.permission.enums.DmsPermissionAction;
|
import jnpf.dmsEntity.permission.enums.DmsResourceType;
|
import jnpf.dmsEntity.preview.form.DmsFilePreviewTicketRequest;
|
import jnpf.dmsEntity.preview.model.DmsFilePreviewTicket;
|
import jnpf.dmsEntity.preview.vo.DmsFilePreviewTicketVO;
|
import jnpf.dmsEntity.storage.entity.DmsFileObjectEntity;
|
import jnpf.dmsMapper.DmsFileObjectMapper;
|
import jnpf.dmsMapper.DmsFileVersionMapper;
|
import jnpf.dmsPermission.DmsCurrentIdentity;
|
import jnpf.dmsPermission.DmsFileControlPolicy;
|
import jnpf.dmsPermission.DmsPermissionError;
|
import jnpf.dmsPermission.DmsPermissionException;
|
import jnpf.dmsPermission.DmsPermissionExecutionContext;
|
import jnpf.dmsPermission.DmsPermissionGuard;
|
import jnpf.dmsPermission.DmsPermissionOperation;
|
import jnpf.dmsService.pdf.DmsPersistentPdfSources;
|
import jnpf.dmsService.pdf.DmsPdfGenerationTaskService;
|
import lombok.RequiredArgsConstructor;
|
import org.springframework.stereotype.Service;
|
|
import java.util.Date;
|
import java.util.UUID;
|
|
@Service
|
@RequiredArgsConstructor
|
public class DmsFilePreviewService {
|
static final long TICKET_TTL_SECONDS = 5L * 60L;
|
|
private final DmsCurrentIdentity currentIdentity;
|
private final DmsPermissionGuard permissionGuard;
|
private final DmsFileVersionMapper fileVersionMapper;
|
private final DmsFileObjectMapper fileObjectMapper;
|
private final DmsFilePreviewTicketIssuer ticketIssuer;
|
private final DmsFileOperationLogService operationLogService;
|
private final DmsWatermarkPreviewResolver watermarkPreviewResolver;
|
private final DmsPreviewPreparationClient previewPreparationClient;
|
private final DmsPdfGenerationTaskService pdfGenerationTaskService;
|
|
public DmsFilePreviewTicketVO createTicket(DmsFilePreviewTicketRequest request) {
|
request.normalized();
|
UserInfo user = currentIdentity.requireUser();
|
String tenantId = currentIdentity.tenantId(user);
|
DmsFileVersionEntity resolvedVersion = resolveFileVersion(request, tenantId);
|
UUID fileVersionId = resolvedVersion.getId();
|
long expiresAtMillis = System.currentTimeMillis() + TICKET_TTL_SECONDS * 1000L;
|
|
PreviewSource source = DmsFileControlPolicy.requiresPermissionCheck(
|
resolvedVersion.getShifouShoukong())
|
? loadPreviewSourceWithPermission(request, tenantId, fileVersionId, expiresAtMillis)
|
: loadPreviewSource(tenantId, fileVersionId);
|
|
DmsFilePreviewTicket payload = previewPreparationClient.prepare(
|
payload(user, tenantId, source, expiresAtMillis, request));
|
String accessTicket = ticketIssuer.issue(payload, TICKET_TTL_SECONDS);
|
DmsFilePreviewTicketVO result = new DmsFilePreviewTicketVO();
|
result.setFileVersionId(fileVersionId);
|
result.setFileName(payload.getFileName());
|
result.setMimeType(payload.getMimeType());
|
result.setSizeBytes(payload.getSizeBytes());
|
result.setAccessTicket(accessTicket);
|
result.setExpiresAt(new Date(expiresAtMillis));
|
operationLogService.recordView(user, tenantId, source.version.getDocumentId(),
|
source.version.getId(), source.version.getFileName(), request.getOperationId());
|
return result;
|
}
|
|
private DmsFileVersionEntity resolveFileVersion(DmsFilePreviewTicketRequest request,
|
String tenantId) {
|
if (request.getResourceType() == DmsResourceType.DOCUMENT) {
|
DmsFileVersionEntity current = fileVersionMapper.selectCurrentEffective(tenantId,
|
request.getResourceId());
|
if (current == null) {
|
throw DmsPermissionException.of(DmsPermissionError.RESOURCE_NOT_FOUND,
|
"文件没有现行有效版本");
|
}
|
return current;
|
}
|
if (request.getResourceType() == DmsResourceType.FILE_VERSION) {
|
DmsFileVersionEntity version = fileVersionMapper.selectReadableByTenantAndId(tenantId,
|
request.getResourceId());
|
if (version == null) {
|
throw DmsPermissionException.of(DmsPermissionError.RESOURCE_NOT_FOUND,
|
"文件版本不存在");
|
}
|
return version;
|
}
|
throw DmsPermissionException.of(DmsPermissionError.INVALID_REQUEST,
|
"仅支持预览 DOCUMENT 或 FILE_VERSION");
|
}
|
|
private PreviewSource loadPreviewSourceWithPermission(DmsFilePreviewTicketRequest request,
|
String tenantId, UUID fileVersionId,
|
long expiresAtMillis) {
|
DmsPermissionExecutionContext context = new DmsPermissionExecutionContext();
|
context.setResourceType(DmsResourceType.FILE_VERSION);
|
context.setResourceId(fileVersionId);
|
context.setActionCode(DmsPermissionAction.READ);
|
context.setOperationId(request.getOperationId());
|
context.setSessionExpiresAt(new Date(expiresAtMillis));
|
return permissionGuard.execute(context, new DmsPermissionOperation<PreviewSource>() {
|
@Override
|
public PreviewSource execute() {
|
return loadPreviewSource(tenantId, fileVersionId);
|
}
|
|
@Override
|
public PreviewSource replay() {
|
return loadPreviewSource(tenantId, fileVersionId);
|
}
|
});
|
}
|
|
private PreviewSource loadPreviewSource(String tenantId, UUID fileVersionId) {
|
DmsFileVersionEntity version = fileVersionMapper.selectReadableByTenantAndId(
|
tenantId, fileVersionId);
|
DmsFileObjectEntity object = fileObjectMapper.selectByTenantAndFileVersionId(tenantId,
|
fileVersionId);
|
if (version == null || object == null) {
|
throw DmsPermissionException.of(DmsPermissionError.RESOURCE_NOT_FOUND,
|
"文件版本或存储对象不存在");
|
}
|
return new PreviewSource(version, object);
|
}
|
|
private DmsFilePreviewTicket payload(UserInfo user, String tenantId, PreviewSource source,
|
long expiresAtMillis, DmsFilePreviewTicketRequest request) {
|
DmsFilePreviewTicket payload = new DmsFilePreviewTicket();
|
payload.setScope(DmsFilePreviewTicket.SCOPE);
|
payload.setTenantId(tenantId);
|
payload.setUserId(user.getUserId().trim());
|
payload.setFileVersionId(source.version.getId());
|
payload.setHistoricalAccess(source.version.getLifecycleStatus() != null
|
&& !"EFFECTIVE".equals(source.version.getLifecycleStatus()));
|
payload.setFileName(source.version.getFileName());
|
payload.setMimeType(source.version.getMimeType());
|
payload.setSizeBytes(source.version.getSizeBytes());
|
payload.setStoragePlatform(source.object.getStoragePlatform());
|
payload.setBucketName(source.object.getBucketName());
|
payload.setBasePath(source.object.getBasePath());
|
payload.setObjectPath(source.object.getObjectPath());
|
payload.setObjectName(source.object.getObjectName());
|
payload.setPdfSource(DmsPersistentPdfSources.valid(source.version, source.object));
|
if (payload.getPdfSource() == null) {
|
pdfGenerationTaskService.enqueueMissingEligible(tenantId, source.version.getId(),
|
source.version.getFileName(), source.version.getChecksumSha256());
|
}
|
payload.setPreviewFormat(request.getPreviewFormat());
|
if ("download".equals(request.getWatermarkType())) {
|
payload.setWatermarks(watermarkPreviewResolver.resolveDownload(user, tenantId, source.version, request));
|
} else if ("print".equals(request.getWatermarkType())) {
|
payload.setWatermarks(watermarkPreviewResolver.resolvePrint(user, tenantId, source.version, request));
|
} else {
|
payload.setWatermarks(watermarkPreviewResolver.resolve(user, tenantId, source.version, request));
|
}
|
payload.setExpiresAt(expiresAtMillis);
|
return payload;
|
}
|
|
private static final class PreviewSource {
|
private final DmsFileVersionEntity version;
|
private final DmsFileObjectEntity object;
|
|
private PreviewSource(DmsFileVersionEntity version, DmsFileObjectEntity object) {
|
this.version = version;
|
this.object = object;
|
}
|
}
|
}
|
