Examples with MimeTypeInfoManager com.helger.xml.util.mime.MimeTypeInfoManager used on opensource projects

Example 1 with MimeTypeInfoManager

use of com.helger.xml.util.mime.MimeTypeInfoManager in project ph-commons by phax.

the class MainReadSharedMimeInfo method main.

public static void main(final String[] args) throws MimeTypeParserException {
    LOGGER.info("Reading shared-mime-info/freedesktop.org.xml");
    final IMicroDocument aDoc = MicroReader.readMicroXML(new File("src/test/resources/shared-mime-info/freedesktop.org.xml.in"));
    if (aDoc == null)
        throw new IllegalStateException("Failed to read mime type info file!");
    final MimeTypeInfoManager aMgr = new MimeTypeInfoManager();
    for (final IMicroElement eSrcMimeType : aDoc.getDocumentElement().getAllChildElements(NS, "mime-type")) {
        final String sMIMEType = eSrcMimeType.getAttributeValue("type");
        final ICommonsOrderedSet<MimeTypeWithSource> aLocalNames = new CommonsLinkedHashSet<>();
        // Names
        aLocalNames.add(new MimeTypeWithSource(sMIMEType));
        for (final IMicroElement eSrcChild : eSrcMimeType.getAllChildElements(NS, "alias")) {
            final String sAlias = eSrcChild.getAttributeValue("type");
            aLocalNames.add(new MimeTypeWithSource(sAlias));
        }
        // Description
        String sComment = null;
        for (final IMicroElement eSrcChild : eSrcMimeType.getAllChildElements(NS, "comment")) if (!eSrcChild.hasAttribute("xml:lang")) {
            sComment = eSrcChild.getTextContentTrimmed();
            break;
        }
        // Sub class of
        final ICommonsOrderedSet<String> aSubClassOf = new CommonsLinkedHashSet<>();
        for (final IMicroElement eSrcChild : eSrcMimeType.getAllChildElements(NS, "sub-class-of")) {
            final String s = eSrcChild.getAttributeValue("type");
            aSubClassOf.add(s);
        }
        boolean bHasAnyGlob = false;
        final ICommonsOrderedSet<String> aGlobs = new CommonsLinkedHashSet<>();
        final ICommonsOrderedSet<ExtensionWithSource> aExts = new CommonsLinkedHashSet<>();
        for (final IMicroElement eSrcChild : eSrcMimeType.getAllChildElements(NS, "glob")) {
            final String sPattern = eSrcChild.getAttributeValue("pattern");
            if (RegExHelper.stringMatchesPattern("\\*\\.[0-9a-zA-Z]+", sPattern)) {
                final String sExt = sPattern.substring(2);
                aExts.add(new ExtensionWithSource(sExt));
            } else
                aGlobs.add(sPattern);
            bHasAnyGlob = true;
        }
        if (bHasAnyGlob) {
            // Append only if at least on filename pattern is present
            aMgr.registerMimeType(new MimeTypeInfo(aLocalNames, sComment, aSubClassOf, aGlobs, aExts, "shared-mime-info"));
        }
    }
    LOGGER.info("Read " + aMgr.getAllMimeTypeInfos().size() + " mime type infos");
    // Maps file extension to MIME type
    LOGGER.info("Reading shared-mime-info/fileext-mimetype-mapping-local.xml");
    final Map<String, String> FileExtMap = new HashMap<>();
    if (XMLMapHandler.readMap(new FileSystemResource("src/test/resources/shared-mime-info/fileext-mimetype-mapping-local.xml"), FileExtMap).isFailure())
        throw new InitializationException("Failed to init file extension to mimetype mapping file");
    LOGGER.info("Read " + FileExtMap.size() + " entries");
    // Check old data
    for (final Map.Entry<String, String> aEntry : CollectionHelper.getSortedByKey(FileExtMap).entrySet()) {
        final String sOldExt = aEntry.getKey();
        final String sOldMimeType = aEntry.getValue();
        final MimeType aOldMimeType = MimeTypeParser.parseMimeType(sOldMimeType);
        ICommonsList<MimeTypeInfo> aNew;
        // First check for Mime Type, as they are unique
        aNew = aMgr.getAllInfosOfMimeType(aOldMimeType);
        if (aNew != null) {
            // Mime type is present - check if extension is also present
            boolean bFound = false;
            for (final MimeTypeInfo aInfo : aNew) if (aInfo.containsExtension(sOldExt)) {
                bFound = true;
                break;
            }
            if (!bFound) {
                if (aNew.size() == 1) {
                    aMgr.addExtension(aNew.get(0), new ExtensionWithSource(sOldExt, "old"));
                    if (false)
                        LOGGER.info("Added extension '" + sOldExt + "' to " + sOldMimeType + "!");
                } else
                    LOGGER.error(sOldMimeType + ": '" + sOldExt + "' not found in " + aNew + "!");
            }
        } else {
            // no such mime type present - Check other direction: ext 2 mimetype
            aNew = aMgr.getAllInfosOfExtension(sOldExt);
            if (aNew != null) {
                // Found extension - check if MIME type matches that type
                boolean bFound = false;
                for (final MimeTypeInfo aInfo : aNew) if (aInfo.containsMimeType(sOldMimeType)) {
                    bFound = true;
                    break;
                }
                if (!bFound) {
                    if (aNew.size() == 1) {
                        aMgr.addMimeType(aNew.get(0), new MimeTypeWithSource(aOldMimeType, "old"));
                        if (false)
                            LOGGER.info("'" + sOldExt + "': " + sOldMimeType + " not found in " + aNew.get(0) + "!");
                    } else
                        LOGGER.error("'" + sOldExt + "': " + sOldMimeType + " not found in any of " + aNew + "!");
                }
            } else {
                // No such mapping from ext to mime type
                // Create a new entry
                aMgr.registerMimeType(new MimeTypeInfo(CollectionHelper.newOrderedSet(new MimeTypeWithSource(sOldMimeType)), null, new CommonsLinkedHashSet<>(), new CommonsLinkedHashSet<>(), CollectionHelper.newOrderedSet(new ExtensionWithSource(sOldExt)), "old"));
                if (false)
                    LOGGER.info("Creating new: " + sOldMimeType + " = '" + sOldExt + "'");
            }
        }
    }
    LOGGER.info("Finally having " + aMgr.getAllMimeTypeInfos().size() + " mime type infos");
    if (MicroWriter.writeToFile(aMgr.getAsDocument(), new File("src/main/resources/codelists/mime-type-info.xml")).isSuccess())
        LOGGER.info("done - run mvn license:format !!");
    else
        LOGGER.error("Error writing file");
}