/*

 * Copyright (c) 1998, 2013, Oracle and/or its affiliates. All rights reserved.

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

 *

 * This code is free software; you can redistribute it and/or modify it

 * under the terms of the GNU General Public License version 2 only, as

 * published by the Free Software Foundation.  Oracle designates this

 * particular file as subject to the "Classpath" exception as provided

 * by Oracle in the LICENSE file that accompanied this code.

 *

 * This code 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

 * version 2 for more details (a copy is included in the LICENSE file that

 * accompanied this code).

 *

 * You should have received a copy of the GNU General Public License version

 * 2 along with this work; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

 *

 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

 * or visit www.oracle.com if you need additional information or have any

 * questions.

 */

package build.tools.dtdbuilder;

import javax.swing.text.html.parser.*;

import java.io.DataOutputStream;

import java.io.File;

import java.io.FileInputStream;

import java.io.IOException;

import java.io.FileNotFoundException;

import java.io.BufferedInputStream;

import java.io.OutputStream;

import java.util.Hashtable;

import java.util.Vector;

import java.util.BitSet;

import java.util.StringTokenizer;

import java.util.Enumeration;

import java.util.Properties;

import java.util.zip.DeflaterOutputStream;

import java.util.zip.Deflater;

import java.net.URL;

/**

 * The representation of an SGML DTD. This is produced by the DTDParser.

 * The resulting DTD object describes a document syntax and is needed

 * to parse HTML documents using the Parser. It contains a list of

 * elements and their attributes as well as a list of entities defined

 * in the DTD.

 *

 * @see Element

 * @see AttributeList

 * @see ContentModel

 * @see DTDParser

 * @see Parser

 * @author Arthur van Hoff

 */

public

class DTDBuilder extends DTD {

    static PublicMapping mapping = null;

    // Hash from name to Integer

    private Hashtable<String, Integer> namesHash = new Hashtable<>();

    // Vector of all names

    private Vector<String> namesVector = new Vector<>();

    /**

     * Create a new DTD.

     */

    protected DTDBuilder(String name) {

        super(name);

    }

    /**

     * Save to a stream as a Java class. Instantiating this class will

     * reproduce a (virtually) identical DTD.

     */

    void save(DataOutputStream out, String className) throws IOException {

        out.writeInt(DTD.FILE_VERSION);

        buildNamesTable();

        int numNames = namesVector.size();

        out.writeShort((short) (namesVector.size()));

        for (int i = 0; i < namesVector.size(); i++) {

            String nm = namesVector.elementAt(i);

            out.writeUTF(nm);

        }

        saveEntities(out);

        out.writeShort((short) (elements.size()));

        for (Enumeration<Element> e = elements.elements() ; e.hasMoreElements() ; ) {

            saveElement(out, e.nextElement());

        }

        if (namesVector.size() != numNames) {

            System.err.println("!!! ERROR!  Names were added to the list!");

            Thread.dumpStack();

            System.exit(1);

        }

    }

    private void buildNamesTable() {

        for (Enumeration<Entity> e = entityHash.elements() ; e.hasMoreElements() ; ) {

            Entity ent = e.nextElement();

            // Do even if not isGeneral().  That way, exclusions and inclusions

            // will definitely have their element.

            getNameId(ent.getName());

        }

        for (Enumeration<Element> e = elements.elements() ; e.hasMoreElements() ; ) {

            Element el = e.nextElement();

            getNameId(el.getName());

            for (AttributeList atts = el.getAttributes() ; atts != null ; atts = atts.getNext()) {

                getNameId(atts.getName());

                if (atts.getValue() != null) {

                    getNameId(atts.getValue());

                }

                Enumeration<?> vals = atts.getValues();

                while (vals != null && vals.hasMoreElements()) {

                    String s = (String) vals.nextElement();

                    getNameId(s);

                }

            }

        }

    }

    //

    // The the id of a name from the list of names

    //

    private short getNameId(String name)  {

        Integer o = namesHash.get(name);

        if (o != null) {

            return (short) o.intValue();

        }

        int i = namesVector.size();

        namesVector.addElement(name);

        namesHash.put(name, new Integer(i));

        return (short) i;

    }

    /**

     * Save an entity to a stream.

     */

    void saveEntities(DataOutputStream out) throws IOException {

        int num = 0;

        for (Enumeration<Entity> e = entityHash.elements() ; e.hasMoreElements() ; ) {

            Entity ent = e.nextElement();

            if (ent.isGeneral()) {

                num++;

            }

        }

        out.writeShort((short) num);

        for (Enumeration<Entity> e = entityHash.elements() ; e.hasMoreElements() ; ) {

            Entity ent = e.nextElement();

            if (ent.isGeneral()) {

                out.writeShort(getNameId(ent.getName()));

                out.writeByte(ent.getType() & ~GENERAL);

                out.writeUTF(ent.getString());

            }

        }

    }

    /**

     * Save an element to a stream.

     */

    public void saveElement(DataOutputStream out, Element elem) throws IOException {

        out.writeShort(getNameId(elem.getName()));

        out.writeByte(elem.getType());

        byte flags = 0;

        if (elem.omitStart()) {

            flags |= 0x01;

        }

        if (elem.omitEnd()) {

            flags |= 0x02;

        }

        out.writeByte(flags);

        saveContentModel(out, elem.getContent());

        // Exclusions

        if (elem.exclusions == null) {

            out.writeShort(0);

        } else {

            short num = 0;

            for (int i = 0 ; i < elem.exclusions.size() ; i++) {

                if (elem.exclusions.get(i)) {

                    num++;

                }

            }

            out.writeShort(num);

            for (int i = 0 ; i < elem.exclusions.size() ; i++) {

                if (elem.exclusions.get(i)) {

                    out.writeShort(getNameId(getElement(i).getName()));

                }

            }

        }

        // Inclusions

        if (elem.inclusions == null) {

            out.writeShort(0);

        } else {

            short num = 0;

            for (int i = 0 ; i < elem.inclusions.size() ; i++) {

                if (elem.inclusions.get(i)) {

                    num++;

                }

            }

            out.writeShort(num);

            for (int i = 0 ; i < elem.inclusions.size() ; i++) {

                if (elem.inclusions.get(i)) {

                    out.writeShort(getNameId(getElement(i).getName()));

                }

            }

        }

        // Attributes

        {

            short numAtts = 0;

            for (AttributeList atts = elem.getAttributes() ; atts != null ; atts = atts.getNext()) {

                numAtts++;

            }

            out.writeByte(numAtts);

            for (AttributeList atts = elem.getAttributes() ; atts != null ; atts = atts.getNext()) {

                out.writeShort(getNameId(atts.getName()));

                out.writeByte(atts.getType());

                out.writeByte(atts.getModifier());

                if (atts.getValue() == null) {

                    out.writeShort(-1);

                } else {

                    out.writeShort(getNameId(atts.getValue()));

                }

                if (atts.values == null) {

                    out.writeShort(0);

                } else {

                    out.writeShort((short) atts.values.size());

                    for (int i = 0; i < atts.values.size(); i++) {

                        String s = (String) atts.values.elementAt(i);

                        out.writeShort(getNameId(s));

                    }

                }

            }

        }

    }

    /**

     * Save a content model to a stream. This does a

     * recursive decent of the entire model.

     */

    public void saveContentModel(DataOutputStream out, ContentModel model) throws IOException {

        if (model == null) {

            out.writeByte(0);

        } else if (model.content instanceof ContentModel) {

            out.writeByte(1);

            out.writeByte(model.type);

            saveContentModel(out, (ContentModel)model.content);

            saveContentModel(out, model.next);

        } else if (model.content instanceof Element) {

            out.writeByte(2);

            out.writeByte(model.type);

            out.writeShort(getNameId(((Element) model.content).getName()));

            saveContentModel(out, model.next);

        }

    }

    /**

     * Generate a class representing this DTD.

     */

    public static void main(String argv[]) {

        String dtd_home = System.getProperty("dtd_home") + File.separator;

        if (dtd_home == null) {

            System.err.println("Must set property 'dtd_home'");

            return;

        }

        DTDBuilder dtd = null;

        try {

            dtd = new DTDBuilder(argv[0]);

            mapping = new PublicMapping(dtd_home, "public.map");

            String path = mapping.get(argv[0]);

            new DTDParser().parse(new FileInputStream(path), dtd);

        } catch (IOException e) {

            System.err.println("Could not open DTD file "+argv[0]);

            e.printStackTrace(System.err);

            System.exit(1);

        }

        try {

            DataOutputStream str = new DataOutputStream(System.out);

            dtd.save(str, argv[0]);

            str.close();

        } catch (IOException ex) {

            ex.printStackTrace();

            System.exit(1);

        }

    }

}