channellistmaker.listmaker.XmlTvDtdResolver.java Source code

Introduction

Here is the source code for channellistmaker.listmaker.XmlTvDtdResolver.java

Source

/*
 * Copyright (C) 2016 normal
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package channellistmaker.listmaker;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.lang.invoke.MethodHandles;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.xml.sax.EntityResolver;
import org.xml.sax.InputSource;
import org.xml.sax.SAXException;

/**
 * xmltv.dtd??
 *
 * @author normal
 */
public class XmlTvDtdResolver implements EntityResolver {

    private static final Log LOG;

    static {
        final Class<?> myClass = MethodHandles.lookup().lookupClass();
        LOG = LogFactory.getLog(myClass);
    }
    private static final File DTD_FILE = new File("./dtd/xmltv.dtd");

    /**
     * xmltv.dtd?????
     * dtd??????????DocumentBuilder???
     *
     * @author normal
     */
    public XmlTvDtdResolver() {
    }

    private static final String DTD_NAME = "xmltv.dtd";

    /**
     * ???????xmltv.dtd????????DocumentBuilder?????????xmltv.dtd???
     * ????????null?
     */
    @Override
    public InputSource resolveEntity(String publicId, String systemId) throws SAXException, IOException {
        if (this.DTD_FILE == null) {
            LOG.warn("DTD???????");
            return null;
        } else {
            LOG.trace("DTD????");
        }
        if ((publicId != null && publicId.contains(DTD_NAME))
                || (systemId != null && systemId.contains(DTD_NAME))) {
            LOG.trace("??????");
            InputSource source = new InputSource(new FileInputStream(this.DTD_FILE));
            source.setPublicId(publicId);
            source.setSystemId(systemId);
            return source;
        } else {
            LOG.trace("?????" + DTD_NAME
                    + " ?????????");
            return null;
        }
    }
}