/*

 * Copyright (c) 1998, 2016, 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 jdk.javadoc.internal.doclets.formats.html;

import java.io.*;

import java.text.SimpleDateFormat;

import java.util.*;

import java.util.regex.Matcher;

import java.util.regex.Pattern;

import javax.lang.model.element.AnnotationMirror;

import javax.lang.model.element.AnnotationValue;

import javax.lang.model.element.Element;

import javax.lang.model.element.ExecutableElement;

import javax.lang.model.element.ModuleElement;

import javax.lang.model.element.Name;

import javax.lang.model.element.PackageElement;

import javax.lang.model.element.TypeElement;

import javax.lang.model.element.VariableElement;

import javax.lang.model.type.DeclaredType;

import javax.lang.model.type.TypeMirror;

import javax.lang.model.util.SimpleAnnotationValueVisitor9;

import javax.lang.model.util.SimpleElementVisitor9;

import javax.lang.model.util.SimpleTypeVisitor9;

import com.sun.source.doctree.AttributeTree;

import com.sun.source.doctree.AttributeTree.ValueKind;

import com.sun.source.doctree.CommentTree;

import com.sun.source.doctree.DocRootTree;

import com.sun.source.doctree.DocTree;

import com.sun.source.doctree.DocTree.Kind;

import com.sun.source.doctree.EndElementTree;

import com.sun.source.doctree.EntityTree;

import com.sun.source.doctree.ErroneousTree;

import com.sun.source.doctree.InheritDocTree;

import com.sun.source.doctree.IndexTree;

import com.sun.source.doctree.LinkTree;

import com.sun.source.doctree.LiteralTree;

import com.sun.source.doctree.SeeTree;

import com.sun.source.doctree.StartElementTree;

import com.sun.source.doctree.TextTree;

import com.sun.source.util.SimpleDocTreeVisitor;

import com.sun.tools.javac.util.DefinedBy;

import com.sun.tools.javac.util.DefinedBy.Api;

import jdk.javadoc.internal.doclets.formats.html.markup.Comment;

import jdk.javadoc.internal.doclets.formats.html.markup.ContentBuilder;

import jdk.javadoc.internal.doclets.formats.html.markup.DocType;

import jdk.javadoc.internal.doclets.formats.html.markup.HtmlAttr;

import jdk.javadoc.internal.doclets.formats.html.markup.HtmlConstants;

import jdk.javadoc.internal.doclets.formats.html.markup.HtmlDocWriter;

import jdk.javadoc.internal.doclets.formats.html.markup.HtmlDocument;

import jdk.javadoc.internal.doclets.formats.html.markup.HtmlStyle;

import jdk.javadoc.internal.doclets.formats.html.markup.HtmlTag;

import jdk.javadoc.internal.doclets.formats.html.markup.HtmlTree;

import jdk.javadoc.internal.doclets.formats.html.markup.RawHtml;

import jdk.javadoc.internal.doclets.formats.html.markup.StringContent;

import jdk.javadoc.internal.doclets.toolkit.AnnotationTypeWriter;

import jdk.javadoc.internal.doclets.toolkit.ClassWriter;

import jdk.javadoc.internal.doclets.toolkit.Configuration;

import jdk.javadoc.internal.doclets.toolkit.Content;

import jdk.javadoc.internal.doclets.toolkit.Messages;

import jdk.javadoc.internal.doclets.toolkit.PackageSummaryWriter;

import jdk.javadoc.internal.doclets.toolkit.Resources;

import jdk.javadoc.internal.doclets.toolkit.taglets.DocRootTaglet;

import jdk.javadoc.internal.doclets.toolkit.taglets.TagletWriter;

import jdk.javadoc.internal.doclets.toolkit.util.DocFile;

import jdk.javadoc.internal.doclets.toolkit.util.DocLink;

import jdk.javadoc.internal.doclets.toolkit.util.DocPath;

import jdk.javadoc.internal.doclets.toolkit.util.DocPaths;

import jdk.javadoc.internal.doclets.toolkit.util.DocletConstants;

import jdk.javadoc.internal.doclets.toolkit.util.ImplementedMethods;

import jdk.javadoc.internal.doclets.toolkit.util.Utils;

import jdk.javadoc.internal.doclets.toolkit.util.CommentHelper;

import static com.sun.source.doctree.AttributeTree.ValueKind.*;

import static com.sun.source.doctree.DocTree.Kind.*;

import static jdk.javadoc.internal.doclets.formats.html.markup.HtmlDocWriter.CONTENT_TYPE;

import static jdk.javadoc.internal.doclets.toolkit.util.CommentHelper.SPACER;

/**

 * Class for the Html Format Code Generation specific to JavaDoc.

 * This Class contains methods related to the Html Code Generation which

 * are used extensively while generating the entire documentation.

 *

 *  <p><b>This is NOT part of any supported API.

 *  If you write code that depends on this, you do so at your own risk.

 *  This code and its internal interfaces are subject to change or

 *  deletion without notice.</b>

 *

 * @author Atul M Dambalkar

 * @author Robert Field

 * @author Bhavesh Patel (Modified)

 */

public class HtmlDocletWriter extends HtmlDocWriter {

    /**

     * Relative path from the file getting generated to the destination

     * directory. For example, if the file getting generated is

     * "java/lang/Object.html", then the path to the root is "../..".

     * This string can be empty if the file getting generated is in

     * the destination directory.

     */

    public final DocPath pathToRoot;

    /**

     * Platform-independent path from the current or the

     * destination directory to the file getting generated.

     * Used when creating the file.

     */

    public final DocPath path;

    /**

     * Name of the file getting generated. If the file getting generated is

     * "java/lang/Object.html", then the filename is "Object.html".

     */

    public final DocPath filename;

    /**

     * The global configuration information for this run.

     */

    public final ConfigurationImpl configuration;

    protected final Utils utils;

    protected final Contents contents;

    protected final Messages messages;

    protected final Resources resources;

    /**

     * To check whether annotation heading is printed or not.

     */

    protected boolean printedAnnotationHeading = false;

    /**

     * To check whether annotation field heading is printed or not.

     */

    protected boolean printedAnnotationFieldHeading = false;

    /**

     * To check whether the repeated annotations is documented or not.

     */

    private boolean isAnnotationDocumented = false;

    /**

     * To check whether the container annotations is documented or not.

     */

    private boolean isContainerDocumented = false;

    HtmlTree fixedNavDiv = new HtmlTree(HtmlTag.DIV);

    final static Pattern IMPROPER_HTML_CHARS = Pattern.compile(".*[&<>].*");

    /**

     * Constructor to construct the HtmlStandardWriter object.

     *

     * @param path File to be generated.

     */

    public HtmlDocletWriter(ConfigurationImpl configuration, DocPath path) {

        super(configuration, path);

        this.configuration = configuration;

        this.contents = configuration.contents;

        this.messages = configuration.messages;

        this.resources = configuration.resources;

        this.utils = configuration.utils;

        this.path = path;

        this.pathToRoot = path.parent().invert();

        this.filename = path.basename();

    }

    /**

     * Replace {@docRoot} tag used in options that accept HTML text, such

     * as -header, -footer, -top and -bottom, and when converting a relative

     * HREF where commentTagsToString inserts a {@docRoot} where one was

     * missing.  (Also see DocRootTaglet for {@docRoot} tags in doc

     * comments.)

     * <p>

     * Replace {@docRoot} tag in htmlstr with the relative path to the

     * destination directory from the directory where the file is being

     * written, looping to handle all such tags in htmlstr.

     * <p>

     * For example, for "-d docs" and -header containing {@docRoot}, when

     * the HTML page for source file p/C1.java is being generated, the

     * {@docRoot} tag would be inserted into the header as "../",

     * the relative path from docs/p/ to docs/ (the document root).

     * <p>

     * Note: This doc comment was written with '@' representing '@'

     * to prevent the inline tag from being interpreted.

     */

    public String replaceDocRootDir(String htmlstr) {

        // Return if no inline tags exist

        int index = htmlstr.indexOf("{@");

        if (index < 0) {

            return htmlstr;

        }

        Matcher docrootMatcher = docrootPattern.matcher(htmlstr);

        if (!docrootMatcher.find()) {

            return htmlstr;

        }

        StringBuilder buf = new StringBuilder();

        int prevEnd = 0;

        do {

            int match = docrootMatcher.start();

            // append htmlstr up to start of next {@docroot}

            buf.append(htmlstr.substring(prevEnd, match));

            prevEnd = docrootMatcher.end();

            if (configuration.docrootparent.length() > 0 && htmlstr.startsWith("/..", prevEnd)) {

                // Insert the absolute link if {@docRoot} is followed by "/..".

                buf.append(configuration.docrootparent);

                prevEnd += 3;

            } else {

                // Insert relative path where {@docRoot} was located

                buf.append(pathToRoot.isEmpty() ? "." : pathToRoot.getPath());

            }

            // Append slash if next character is not a slash

            if (prevEnd < htmlstr.length() && htmlstr.charAt(prevEnd) != '/') {

                buf.append('/');

            }

        } while (docrootMatcher.find());

        buf.append(htmlstr.substring(prevEnd));

        return buf.toString();

    }

    //where:

        // Note: {@docRoot} is not case sensitive when passed in w/command line option:

        private static final Pattern docrootPattern =

                Pattern.compile(Pattern.quote("{@docroot}"), Pattern.CASE_INSENSITIVE);

    /**

     * Get the script to show or hide the All classes link.

     *

     * @param id id of the element to show or hide

     * @return a content tree for the script

     */

    public Content getAllClassesLinkScript(String id) {

        HtmlTree script = HtmlTree.SCRIPT();

        String scriptCode = "<!--\n" +

                "  allClassesLink = document.getElementById(\"" + id + "\");\n" +

                "  if(window==top) {\n" +

                "    allClassesLink.style.display = \"block\";\n" +

                "  }\n" +

                "  else {\n" +

                "    allClassesLink.style.display = \"none\";\n" +

                "  }\n" +

                "  //-->\n";

        Content scriptContent = new RawHtml(scriptCode.replace("\n", DocletConstants.NL));

        script.addContent(scriptContent);

        Content div = HtmlTree.DIV(script);

        Content div_noscript = HtmlTree.DIV(contents.noScriptMessage);

        Content noScript = HtmlTree.NOSCRIPT(div_noscript);

        div.addContent(noScript);

        return div;

    }

    /**

     * Add method information.

     *

     * @param method the method to be documented

     * @param dl the content tree to which the method information will be added

     */

    private void addMethodInfo(ExecutableElement method, Content dl) {

        TypeElement enclosing = utils.getEnclosingTypeElement(method);

        List<? extends TypeMirror> intfacs = enclosing.getInterfaces();

        ExecutableElement overriddenMethod = utils.overriddenMethod(method);

        // Check whether there is any implementation or overridden info to be

        // printed. If no overridden or implementation info needs to be

        // printed, do not print this section.

        if ((!intfacs.isEmpty()

                && new ImplementedMethods(method, this.configuration).build().isEmpty() == false)

                || overriddenMethod != null) {

            MethodWriterImpl.addImplementsInfo(this, method, dl);

            if (overriddenMethod != null) {

                MethodWriterImpl.addOverridden(this,

                        utils.overriddenType(method),

                        overriddenMethod,

                        dl);

            }

        }

    }

    /**

     * Adds the tags information.

     *

     * @param e the Element for which the tags will be generated

     * @param htmltree the documentation tree to which the tags will be added

     */

    protected void addTagsInfo(Element e, Content htmltree) {

        if (configuration.nocomment) {

            return;

        }

        Content dl = new HtmlTree(HtmlTag.DL);

        if (utils.isExecutableElement(e) && !utils.isConstructor(e)) {

            addMethodInfo((ExecutableElement)e, dl);

        }

        Content output = new ContentBuilder();

        TagletWriter.genTagOutput(configuration.tagletManager, e,

            configuration.tagletManager.getCustomTaglets(e),

                getTagletWriterInstance(false), output);

        dl.addContent(output);

        htmltree.addContent(dl);

    }

    /**

     * Check whether there are any tags for Serialization Overview

     * section to be printed.

     *

     * @param field the VariableElement object to check for tags.

     * @return true if there are tags to be printed else return false.

     */

    protected boolean hasSerializationOverviewTags(VariableElement field) {

        Content output = new ContentBuilder();

        TagletWriter.genTagOutput(configuration.tagletManager, field,

                configuration.tagletManager.getCustomTaglets(field),

                getTagletWriterInstance(false), output);

        return !output.isEmpty();

    }

    /**

     * Returns a TagletWriter that knows how to write HTML.

     *

     * @return a TagletWriter that knows how to write HTML.

     */

    public TagletWriter getTagletWriterInstance(boolean isFirstSentence) {

        return new TagletWriterImpl(this, isFirstSentence);

    }

    /**

     * Get Package link, with target frame.

     *

     * @param pkg The link will be to the "package-summary.html" page for this package

     * @param target name of the target frame

     * @param label tag for the link

     * @return a content for the target package link

     */

    public Content getTargetPackageLink(PackageElement pkg, String target,

            Content label) {

        return getHyperLink(pathString(pkg, DocPaths.PACKAGE_SUMMARY), label, "", target);

    }

    /**

     * Get Module Package link, with target frame.

     *

     * @param pkg the PackageElement

     * @param target name of the target frame

     * @param label tag for the link

     * @param mdle the module being documented

     * @return a content for the target module packages link

     */

    public Content getTargetModulePackageLink(PackageElement pkg, String target,

            Content label, ModuleElement mdle) {

        return getHyperLink(pathString(pkg, DocPaths.PACKAGE_SUMMARY),

                label, "", target);

    }

    /**

     * Get Module link, with target frame.

     *

     * @param target name of the target frame

     * @param label tag for the link

     * @param mdle the module being documented

     * @return a content for the target module link

     */

    public Content getTargetModuleLink(String target, Content label, ModuleElement mdle) {

        return getHyperLink(pathToRoot.resolve(

                DocPaths.moduleSummary(mdle)), label, "", target);

    }

    public void addClassesSummary(SortedSet<TypeElement> classes, String label,

            String tableSummary, List<String> tableHeader, Content summaryContentTree) {

        if (!classes.isEmpty()) {

            Content caption = getTableCaption(new RawHtml(label));

            Content table = (configuration.isOutputHtml5())

                    ? HtmlTree.TABLE(HtmlStyle.typeSummary, caption)

                    : HtmlTree.TABLE(HtmlStyle.typeSummary, tableSummary, caption);

            table.addContent(getSummaryTableHeader(tableHeader, "col"));

            Content tbody = new HtmlTree(HtmlTag.TBODY);

            boolean altColor = true;

            for (TypeElement te : classes) {

                if (!utils.isCoreClass(te) ||

                    !configuration.isGeneratedDoc(te)) {

                    continue;

                }

                Content classContent = getLink(new LinkInfoImpl(

                        configuration, LinkInfoImpl.Kind.PACKAGE, te));

                Content tdClass = HtmlTree.TD(HtmlStyle.colFirst, classContent);

                HtmlTree tr = HtmlTree.TR(tdClass);

                tr.addStyle(altColor ? HtmlStyle.altColor : HtmlStyle.rowColor);

                altColor = !altColor;

                HtmlTree tdClassDescription = new HtmlTree(HtmlTag.TD);

                tdClassDescription.addStyle(HtmlStyle.colLast);

                if (utils.isDeprecated(te)) {

                    tdClassDescription.addContent(contents.deprecatedLabel);

                    List<? extends DocTree> tags = utils.getDeprecatedTrees(te);

                    if (!tags.isEmpty()) {

                        addSummaryDeprecatedComment(te, tags.get(0), tdClassDescription);

                    }

                } else {

                    addSummaryComment(te, tdClassDescription);

                }

                tr.addContent(tdClassDescription);

                tbody.addContent(tr);

            }

            table.addContent(tbody);

            summaryContentTree.addContent(table);

        }

    }

    /**

     * Generates the HTML document tree and prints it out.

     *

     * @param metakeywords Array of String keywords for META tag. Each element

     *                     of the array is assigned to a separate META tag.

     *                     Pass in null for no array

     * @param includeScript true if printing windowtitle script

     *                      false for files that appear in the left-hand frames

     * @param body the body htmltree to be included in the document

     */

    public void printHtmlDocument(List<String> metakeywords, boolean includeScript,

            Content body) throws IOException {

        Content htmlDocType = configuration.isOutputHtml5()

                ? DocType.HTML5

                : DocType.TRANSITIONAL;

        Content htmlComment = new Comment(configuration.getText("doclet.New_Page"));

        Content head = new HtmlTree(HtmlTag.HEAD);

        head.addContent(getGeneratedBy(!configuration.notimestamp));

        head.addContent(getTitle());

        Content meta = HtmlTree.META("Content-Type", CONTENT_TYPE,

                (configuration.charset.length() > 0) ?

                        configuration.charset : HtmlConstants.HTML_DEFAULT_CHARSET);

        head.addContent(meta);

        if (!configuration.notimestamp) {

            SimpleDateFormat dateFormat = new SimpleDateFormat("yyyy-MM-dd");

            meta = HtmlTree.META(configuration.isOutputHtml5()

                    ? "dc.created"

                    : "date", dateFormat.format(new Date()));

            head.addContent(meta);

        }

        if (metakeywords != null) {

            for (String metakeyword : metakeywords) {

                meta = HtmlTree.META("keywords", metakeyword);

                head.addContent(meta);

            }

        }

        addStyleSheetProperties(head);

        addScriptProperties(head);

        Content htmlTree = HtmlTree.HTML(configuration.getLocale().getLanguage(),

                head, body);

        Content htmlDocument = new HtmlDocument(htmlDocType,

                htmlComment, htmlTree);

        write(htmlDocument);

    }

    /**

     * Get the window title.

     *

     * @param title the title string to construct the complete window title

     * @return the window title string

     */

    public String getWindowTitle(String title) {

        if (configuration.windowtitle.length() > 0) {

            title += " (" + configuration.windowtitle  + ")";

        }

        return title;

    }

    /**

     * Get user specified header and the footer.

     *