/*

 * Copyright (c) 1997, 2017, 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.util.*;

import java.util.stream.Collectors;

import javax.lang.model.element.Element;

import javax.lang.model.element.ExecutableElement;

import javax.lang.model.element.Modifier;

import javax.lang.model.element.TypeElement;

import javax.lang.model.element.TypeParameterElement;

import javax.lang.model.type.TypeMirror;

import com.sun.source.doctree.DocTree;

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.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.StringContent;

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

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

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

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

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

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

import static javax.lang.model.element.Modifier.*;

/**

 * The base class for member writers.

 *

 *  <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 Robert Field

 * @author Atul M Dambalkar

 * @author Jamie Ho (Re-write)

 * @author Bhavesh Patel (Modified)

 */

public abstract class AbstractMemberWriter {

    protected final HtmlConfiguration configuration;

    protected final Utils utils;

    protected final SubWriterHolderWriter writer;

    protected final Contents contents;

    protected final Resources resources;

    protected final TypeElement typeElement;

    protected Map<String, Integer> typeMap = new LinkedHashMap<>();

    protected Set<MethodTypes> methodTypes = EnumSet.noneOf(MethodTypes.class);

    private int methodTypesOr = 0;

    public final boolean nodepr;

    protected boolean printedSummaryHeader = false;

    public AbstractMemberWriter(SubWriterHolderWriter writer, TypeElement typeElement) {

        this.configuration = writer.configuration;

        this.writer = writer;

        this.nodepr = configuration.nodeprecated;

        this.typeElement = typeElement;

        this.utils = configuration.utils;

        this.contents = configuration.contents;

        this.resources = configuration.resources;

    }

    public AbstractMemberWriter(SubWriterHolderWriter writer) {

        this(writer, null);

    }

    /*** abstracts ***/

    /**

     * Add the summary label for the member.

     *

     * @param memberTree the content tree to which the label will be added

     */

    public abstract void addSummaryLabel(Content memberTree);

    /**

     * Get the summary for the member summary table.

     *

     * @return a string for the table summary

     */

    public abstract String getTableSummary();

    /**

     * Get the caption for the member summary table.

     *

     * @return a string for the table caption

     */

    public abstract Content getCaption();

    /**

     * Get the summary table header for the member.

     *

     * @param member the member to be documented

     * @return the summary table header

     */

    public abstract List<String> getSummaryTableHeader(Element member);

    /**

     * Add inherited summary label for the member.

     *

     * @param typeElement the TypeElement to which to link to

     * @param inheritedTree the content tree to which the inherited summary label will be added

     */

    public abstract void addInheritedSummaryLabel(TypeElement typeElement, Content inheritedTree);

    /**

     * Add the anchor for the summary section of the member.

     *

     * @param typeElement the TypeElement to be documented

     * @param memberTree the content tree to which the summary anchor will be added

     */

    public abstract void addSummaryAnchor(TypeElement typeElement, Content memberTree);

    /**

     * Add the anchor for the inherited summary section of the member.

     *

     * @param typeElement the TypeElement to be documented

     * @param inheritedTree the content tree to which the inherited summary anchor will be added

     */

    public abstract void addInheritedSummaryAnchor(TypeElement typeElement, Content inheritedTree);

    /**

     * Add the summary type for the member.

     *

     * @param member the member to be documented

     * @param tdSummaryType the content tree to which the type will be added

     */

    protected abstract void addSummaryType(Element member, Content tdSummaryType);

    /**

     * Add the summary link for the member.

     *

     * @param typeElement the TypeElement to be documented

     * @param member the member to be documented

     * @param tdSummary the content tree to which the link will be added

     */

    protected void addSummaryLink(TypeElement typeElement, Element member, Content tdSummary) {

        addSummaryLink(LinkInfoImpl.Kind.MEMBER, typeElement, member, tdSummary);

    }

    /**

     * Add the summary link for the member.

     *

     * @param context the id of the context where the link will be printed

     * @param typeElement the TypeElement to be documented

     * @param member the member to be documented

     * @param tdSummary the content tree to which the summary link will be added

     */

    protected abstract void addSummaryLink(LinkInfoImpl.Kind context,

            TypeElement typeElement, Element member, Content tdSummary);

    /**

     * Add the inherited summary link for the member.

     *

     * @param typeElement the TypeElement to be documented

     * @param member the member to be documented

     * @param linksTree the content tree to which the inherited summary link will be added

     */

    protected abstract void addInheritedSummaryLink(TypeElement typeElement,

            Element member, Content linksTree);

    /**

     * Get the deprecated link.

     *

     * @param member the member being linked to

     * @return a content tree representing the link

     */

    protected abstract Content getDeprecatedLink(Element member);

    /**

     * Get the navigation summary link.

     *

     * @param typeElement the TypeElement to be documented

     * @param link true if its a link else the label to be printed

     * @return a content tree for the navigation summary link.

     */

    protected abstract Content getNavSummaryLink(TypeElement typeElement, boolean link);

    /**

     * Add the navigation detail link.

     *

     * @param link true if its a link else the label to be printed

     * @param liNav the content tree to which the navigation detail link will be added

     */

    protected abstract void addNavDetailLink(boolean link, Content liNav);

    /**

     * Add the member name to the content tree.

     *

     * @param name the member name to be added to the content tree.

     * @param htmltree the content tree to which the name will be added.

     */

    protected void addName(String name, Content htmltree) {

        htmltree.addContent(name);

    }

    /**

     * Add the modifier for the member. The modifiers are ordered as specified

     * by <em>The Java Language Specification</em>.

     *

     * @param member the member for which teh modifier will be added.

     * @param htmltree the content tree to which the modifier information will be added.

     */

    protected void addModifiers(Element member, Content htmltree) {

        Set<Modifier> set = new TreeSet<>(member.getModifiers());

        // remove the ones we really don't need

        set.remove(NATIVE);

        set.remove(SYNCHRONIZED);

        set.remove(STRICTFP);

        // According to JLS, we should not be showing public modifier for

        // interface methods.

        if ((utils.isField(member) || utils.isMethod(member))

            && ((writer instanceof ClassWriterImpl

                 && utils.isInterface(((ClassWriterImpl) writer).getTypeElement())  ||

                 writer instanceof AnnotationTypeWriterImpl) )) {

            // Remove the implicit abstract and public modifiers

            if (utils.isMethod(member) &&

                (utils.isInterface(member.getEnclosingElement()) ||

                 utils.isAnnotationType(member.getEnclosingElement()))) {

                set.remove(ABSTRACT);

                set.remove(PUBLIC);

            }

            if (!utils.isMethod(member)) {

                set.remove(PUBLIC);

            }

        }

        if (!set.isEmpty()) {

            String mods = set.stream().map(Modifier::toString).collect(Collectors.joining(" "));

            htmltree.addContent(mods);

            htmltree.addContent(Contents.SPACE);

        }

    }

    protected CharSequence makeSpace(int len) {

        if (len <= 0) {

            return "";

        }

        StringBuilder sb = new StringBuilder(len);

        for (int i = 0; i < len; i++) {

            sb.append(' ');

        }

        return sb;

    }

    /**

     * Add the modifier and type for the member in the member summary.

     *

     * @param member the member to add the type for

     * @param type the type to add

     * @param tdSummaryType the content tree to which the modified and type will be added

     */

    protected void addModifierAndType(Element member, TypeMirror type,

            Content tdSummaryType) {

        HtmlTree code = new HtmlTree(HtmlTag.CODE);

        addModifier(member, code);

        if (type == null) {

            code.addContent(utils.isClass(member) ? "class" : "interface");

            code.addContent(Contents.SPACE);

        } else {

            List<? extends TypeParameterElement> list = utils.isExecutableElement(member)

                    ? ((ExecutableElement)member).getTypeParameters()

                    : null;

            if (list != null && !list.isEmpty()) {

                Content typeParameters = ((AbstractExecutableMemberWriter) this)

                        .getTypeParameters((ExecutableElement)member);

                    code.addContent(typeParameters);

                //Code to avoid ugly wrapping in member summary table.

                if (typeParameters.charCount() > 10) {

                    code.addContent(new HtmlTree(HtmlTag.BR));

                } else {

                    code.addContent(Contents.SPACE);

                }

                code.addContent(

                        writer.getLink(new LinkInfoImpl(configuration,

                        LinkInfoImpl.Kind.SUMMARY_RETURN_TYPE, type)));

            } else {

                code.addContent(

                        writer.getLink(new LinkInfoImpl(configuration,

                        LinkInfoImpl.Kind.SUMMARY_RETURN_TYPE, type)));

            }

        }

        tdSummaryType.addContent(code);

    }

    /**

     * Add the modifier for the member.

     *

     * @param member the member to add the type for

     * @param code the content tree to which the modified will be added

     */

    private void addModifier(Element member, Content code) {

        if (utils.isProtected(member)) {

            code.addContent("protected ");

        } else if (utils.isPrivate(member)) {

            code.addContent("private ");

        } else if (!utils.isPublic(member)) { // Package private

            code.addContent(configuration.getText("doclet.Package_private"));

            code.addContent(" ");

        }

        boolean isAnnotatedTypeElement = utils.isAnnotationType(member.getEnclosingElement());

        if (!isAnnotatedTypeElement && utils.isMethod(member)) {

            if (!utils.isInterface(member.getEnclosingElement()) && utils.isAbstract(member)) {

                code.addContent("abstract ");

            }

            if (utils.isDefault(member)) {

                code.addContent("default ");

            }

        }

        if (utils.isStatic(member)) {

            code.addContent("static ");

        }

    }

    /**

     * Add the deprecated information for the given member.

     *

     * @param member the member being documented.

     * @param contentTree the content tree to which the deprecated information will be added.

     */

    protected void addDeprecatedInfo(Element member, Content contentTree) {

        Content output = (new DeprecatedTaglet()).getTagletOutput(member,

            writer.getTagletWriterInstance(false));

        if (!output.isEmpty()) {

            Content deprecatedContent = output;

            Content div = HtmlTree.DIV(HtmlStyle.block, deprecatedContent);

            contentTree.addContent(div);

        }

    }

    /**

     * Add the comment for the given member.

     *

     * @param member the member being documented.

     * @param htmltree the content tree to which the comment will be added.

     */

    protected void addComment(Element member, Content htmltree) {

        if (!utils.getFullBody(member).isEmpty()) {

            writer.addInlineComment(member, htmltree);

        }

    }

    protected String name(Element member) {

        return utils.getSimpleName(member);

    }

    /**

     * Get the header for the section.

     *

     * @param member the member being documented.

     * @return a header content for the section.

     */

    protected Content getHead(Element member) {

        Content memberContent = new StringContent(name(member));

        Content heading = HtmlTree.HEADING(HtmlConstants.MEMBER_HEADING, memberContent);

        return heading;

    }

    /**

    * Return true if the given <code>ProgramElement</code> is inherited

    * by the class that is being documented.

    *

    * @param ped The <code>ProgramElement</code> being checked.

    * return true if the <code>ProgramElement</code> is being inherited and

    * false otherwise.

     *@return true if inherited

    */

    protected boolean isInherited(Element ped){

        return (!utils.isPrivate(ped) &&

                (!utils.isPackagePrivate(ped) ||

                    ped.getEnclosingElement().equals(ped.getEnclosingElement())));

    }

    /**

     * Add use information to the documentation tree.

     *

     * @param mems list of program elements for which the use information will be added

     * @param heading the section heading

     * @param tableSummary the summary for the use table

     * @param contentTree the content tree to which the use information will be added

     */

    protected void addUseInfo(List<? extends Element> mems,

            Content heading, String tableSummary, Content contentTree) {

        if (mems == null || mems.isEmpty()) {

            return;

        }

        List<? extends Element> members = mems;

        boolean printedUseTableHeader = false;

        if (members.size() > 0) {

            Content caption = writer.getTableCaption(heading);

            Content table = (configuration.isOutputHtml5())

                    ? HtmlTree.TABLE(HtmlStyle.useSummary, caption)

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

            Content tbody = new HtmlTree(HtmlTag.TBODY);

            boolean altColor = true;

            for (Element element : members) {

                TypeElement te = utils.getEnclosingTypeElement(element);

                if (!printedUseTableHeader) {

                    table.addContent(writer.getSummaryTableHeader(

                            this.getSummaryTableHeader(element), "col"));

                    printedUseTableHeader = true;

                }

                HtmlTree tr = new HtmlTree(HtmlTag.TR);

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

                altColor = !altColor;

                HtmlTree tdFirst = new HtmlTree(HtmlTag.TD);

                tdFirst.addStyle(HtmlStyle.colFirst);

                writer.addSummaryType(this, element, tdFirst);

                tr.addContent(tdFirst);

                HtmlTree thType = new HtmlTree(HtmlTag.TH);

                thType.addStyle(HtmlStyle.colSecond);

                thType.addAttr(HtmlAttr.SCOPE, "row");

                if (te != null

                        && !utils.isConstructor(element)

                        && !utils.isClass(element)

                        && !utils.isInterface(element)

                        && !utils.isAnnotationType(element)) {

                    HtmlTree name = new HtmlTree(HtmlTag.SPAN);

                    name.addStyle(HtmlStyle.typeNameLabel);

                    name.addContent(name(te) + ".");

                    thType.addContent(name);

                }

                addSummaryLink(utils.isClass(element) || utils.isInterface(element)

                        ? LinkInfoImpl.Kind.CLASS_USE

                        : LinkInfoImpl.Kind.MEMBER,

                        te, element, thType);

                tr.addContent(thType);

                HtmlTree tdDesc = new HtmlTree(HtmlTag.TD);

                tdDesc.addStyle(HtmlStyle.colLast);

                writer.addSummaryLinkComment(this, element, tdDesc);

                tr.addContent(tdDesc);

                tbody.addContent(tr);

            }

            table.addContent(tbody);

            contentTree.addContent(table);

        }

    }

    /**

     * Add the navigation detail link.

     *

     * @param members the members to be linked

     * @param liNav the content tree to which the navigation detail link will be added

     */

    protected void addNavDetailLink(SortedSet<Element> members, Content liNav) {

        addNavDetailLink(!members.isEmpty(), liNav);

    }

    /**

     * Add the navigation summary link.

     *

     * @param members members to be linked

     * @param visibleMemberMap the visible inherited members map

     * @param liNav the content tree to which the navigation summary link will be added

     */

    protected void addNavSummaryLink(SortedSet<? extends Element> members,