Codebase list jd-gui / beb5dc6 services / src / main / java / org / jd / gui / service / indexer / XmlBasedFileIndexerProvider.java
beb5dc6

Tree @beb5dc6 (Download .tar.gz)

XmlBasedFileIndexerProvider.java @beb5dc6 — raw · history · blame 

  1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
/*
 * Copyright (c) 2008-2019 Emmanuel Dupuy.
 * This project is distributed under the GPLv3 license.
 * This is a Copyleft license that gives the user the right to use,
 * copy and modify the code freely for non-commercial purposes.
 */

package org.jd.gui.service.indexer;

import org.jd.gui.api.API;
import org.jd.gui.api.model.Container;
import org.jd.gui.api.model.Indexes;
import org.jd.gui.util.exception.ExceptionUtil;

import javax.xml.stream.XMLInputFactory;
import javax.xml.stream.XMLStreamConstants;
import javax.xml.stream.XMLStreamException;
import javax.xml.stream.XMLStreamReader;
import java.util.Collection;
import java.util.HashSet;
import java.util.Map;

public class XmlBasedFileIndexerProvider extends AbstractIndexerProvider {
    protected XMLInputFactory factory;

    public XmlBasedFileIndexerProvider() {
        factory = XMLInputFactory.newInstance();
        factory.setProperty(XMLInputFactory.SUPPORT_DTD, false);
    }

    @Override public String[] getSelectors() { return appendSelectors("*:file:*.xsl", "*:file:*.xslt", "*:file:*.xsd"); }

    @Override
    @SuppressWarnings("unchecked")
    public void index(API api, Container.Entry entry, Indexes indexes) {
        HashSet<String> stringSet = new HashSet<>();
        XMLStreamReader reader = null;

        try {
            reader = factory.createXMLStreamReader(entry.getInputStream());

            stringSet.add(reader.getVersion());
            stringSet.add(reader.getEncoding());
            stringSet.add(reader.getCharacterEncodingScheme());

            while (reader.hasNext()) {
                switch (reader.next()) {
                    case XMLStreamConstants.START_ELEMENT:
                        stringSet.add(reader.getLocalName());
                        for (int i = reader.getAttributeCount() - 1; i >= 0; i--) {
                            stringSet.add(reader.getAttributeLocalName(i));
                            stringSet.add(reader.getAttributeValue(i));
                        }
                        for (int i = reader.getNamespaceCount() - 1; i >= 0; i--) {
                            stringSet.add(reader.getNamespacePrefix(i));
                            stringSet.add(reader.getNamespaceURI(i));
                        }
                        break;
                    case XMLStreamConstants.PROCESSING_INSTRUCTION:
                        stringSet.add(reader.getPITarget());
                        stringSet.add(reader.getPIData());
                        break;
                    case XMLStreamConstants.START_DOCUMENT:
                        stringSet.add(reader.getVersion());
                        stringSet.add(reader.getEncoding());
                        stringSet.add(reader.getCharacterEncodingScheme());
                        break;
                    case XMLStreamConstants.ENTITY_REFERENCE:
                        stringSet.add(reader.getLocalName());
                        stringSet.add(reader.getText());
                        break;
                    case XMLStreamConstants.ATTRIBUTE:
                        stringSet.add(reader.getPrefix());
                        stringSet.add(reader.getNamespaceURI());
                        stringSet.add(reader.getLocalName());
                        stringSet.add(reader.getText());
                        break;
                    case XMLStreamConstants.COMMENT:
                    case XMLStreamConstants.DTD:
                    case XMLStreamConstants.CDATA:
                    case XMLStreamConstants.CHARACTERS:
                        stringSet.add(reader.getText().trim());
                        break;
                    case XMLStreamConstants.NAMESPACE:
                        for (int i = reader.getNamespaceCount() - 1; i >= 0; i--) {
                            stringSet.add(reader.getNamespacePrefix(i));
                            stringSet.add(reader.getNamespaceURI(i));
                        }
                        break;
                }
            }
        } catch (Exception e) {
            assert ExceptionUtil.printStackTrace(e);
        } finally {
            if (reader != null) {
                try {
                    reader.close();
                } catch (XMLStreamException e) {
                    assert ExceptionUtil.printStackTrace(e);
                }
            }
        }

        Map<String, Collection> stringIndex = indexes.getIndex("strings");

        for (String string : stringSet) {
            if ((string != null) && !string.isEmpty()) {
                stringIndex.get(string).add(entry);
            }
        }
    }
}