TriTih
/
DIP
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Refactor phases 4.1

This commit is contained in:
trifonovt 2026-03-18 15:46:08 +01:00
parent f337af56b5
commit 1ba8cccb62
1 changed files with 31 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
src/main/java/at/procon/dip/classification/detector/BasicMimeAndExtensionDocumentTypeDetector.java
View File

@ -26,9 +26,21 @@ public class BasicMimeAndExtensionDocumentTypeDetector implements DocumentTypeDe
@Override @Override
public DetectionResult detect(SourceDescriptor sourceDescriptor) { public DetectionResult detect(SourceDescriptor sourceDescriptor) {
String normalizedMediaType = DocumentImportSupport.normalizeMediaType(sourceDescriptor.mediaType()); String normalizedMediaType = DocumentImportSupport.normalizeMediaType(sourceDescriptor.mediaType());
if (sourceDescriptor.sourceType() == at.procon.dip.domain.document.SourceType.TED_PACKAGE) {
Map<String, String> attributes = new HashMap<>();
attributes.put("sourceType", sourceDescriptor.sourceType().name());
if (StringUtils.hasText(sourceDescriptor.fileName())) {
attributes.put("fileName", sourceDescriptor.fileName());
}
return new DetectionResult(DocumentType.TED_PACKAGE, DocumentFamily.PROCUREMENT,
normalizedMediaType != null ? normalizedMediaType : "application/gzip", null, attributes);
}
DocumentType hintedType = detectByHint(sourceDescriptor);
String extension = DocumentImportSupport.extensionOf(sourceDescriptor.fileName()); String extension = DocumentImportSupport.extensionOf(sourceDescriptor.fileName());
DocumentType documentType = detectByMediaType(normalizedMediaType); DocumentType documentType = hintedType != null ? hintedType : detectByMediaType(normalizedMediaType);
if (documentType == DocumentType.UNKNOWN) { if (documentType == DocumentType.UNKNOWN) {
documentType = detectByExtension(extension); documentType = detectByExtension(extension);
} }
@ -46,10 +58,28 @@ public class BasicMimeAndExtensionDocumentTypeDetector implements DocumentTypeDe
if (StringUtils.hasText(sourceDescriptor.fileName())) { if (StringUtils.hasText(sourceDescriptor.fileName())) {
attributes.put("fileName", sourceDescriptor.fileName()); attributes.put("fileName", sourceDescriptor.fileName());
} }
if (hintedType != null) {
attributes.put("documentTypeHint", hintedType.name());
}
return new DetectionResult(documentType, family, normalizedMediaType, languageCode, attributes); return new DetectionResult(documentType, family, normalizedMediaType, languageCode, attributes);
} }
private DocumentType detectByHint(SourceDescriptor sourceDescriptor) {
if (sourceDescriptor.attributes() == null) {
return null;
}
String hint = sourceDescriptor.attributes().get("documentTypeHint");
if (!StringUtils.hasText(hint)) {
return null;
}
try {
return DocumentType.valueOf(hint.trim().toUpperCase(Locale.ROOT));
} catch (IllegalArgumentException ignored) {
return null;
}
}
private DocumentType detectByMediaType(String mediaType) { private DocumentType detectByMediaType(String mediaType) {
if (!StringUtils.hasText(mediaType)) { if (!StringUtils.hasText(mediaType)) {
return DocumentType.UNKNOWN; return DocumentType.UNKNOWN;