/*
 * Copyright (c) 2003, 2023, 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 org.jdrupes.mdoclet.internal.doclets.formats.html;

import java.util.ArrayList;
import java.util.EnumSet;
import java.util.List;
import java.util.Set;

import javax.lang.model.element.AnnotationMirror;
import javax.lang.model.element.Element;
import javax.lang.model.element.TypeElement;
import javax.lang.model.element.TypeParameterElement;
import javax.lang.model.type.ArrayType;
import javax.lang.model.type.DeclaredType;
import javax.lang.model.type.TypeMirror;
import javax.lang.model.type.TypeVariable;
import javax.lang.model.type.WildcardType;
import javax.lang.model.util.SimpleTypeVisitor14;

import org.jdrupes.mdoclet.internal.doclets.formats.html.markup.ContentBuilder;
import org.jdrupes.mdoclet.internal.doclets.formats.html.markup.HtmlTree;
import org.jdrupes.mdoclet.internal.doclets.formats.html.markup.TagName;
import org.jdrupes.mdoclet.internal.doclets.formats.html.markup.Text;
import org.jdrupes.mdoclet.internal.doclets.toolkit.BaseConfiguration;
import org.jdrupes.mdoclet.internal.doclets.toolkit.Content;
import org.jdrupes.mdoclet.internal.doclets.toolkit.Resources;
import org.jdrupes.mdoclet.internal.doclets.toolkit.util.DocPath;
import org.jdrupes.mdoclet.internal.doclets.toolkit.util.DocPaths;
import org.jdrupes.mdoclet.internal.doclets.toolkit.util.Utils;
import org.jdrupes.mdoclet.internal.doclets.toolkit.util.Utils.ElementFlag;

/**
 * A factory that returns a link given the information about it.
 */
public class HtmlLinkFactory {

    private final HtmlDocletWriter m_writer;
    private final DocPaths docPaths;
    private final Utils utils;

    /**
     * Constructs a new link factory.
     *
     * @param writer the HTML doclet writer
     */
    public HtmlLinkFactory(HtmlDocletWriter writer) {
        m_writer = writer;
        docPaths = writer.configuration.docPaths;
        utils = writer.configuration.utils;
    }

    /**
     * {@return a new instance of a content object}
     */
    protected Content newContent() {
        return new ContentBuilder();
    }

    /**
     * Constructs a link from the given link information.
     *
     * @param linkInfo the information about the link.
     * @return the link.
     */
    public Content getLink(HtmlLinkInfo linkInfo) {
        if (linkInfo.getType() != null) {
            SimpleTypeVisitor14<Content, HtmlLinkInfo> linkVisitor
                = new SimpleTypeVisitor14<>() {

                    final Content link = newContent();

                    // handles primitives, no types and error types
                    @Override
                    protected Content defaultAction(TypeMirror type,
                            HtmlLinkInfo linkInfo) {
                        link.add(utils.getTypeName(type, false));
                        return link;
                    }

                    int currentDepth = 0;

                    @Override
                    public Content visitArray(ArrayType type,
                            HtmlLinkInfo linkInfo) {
                        // keep track of the dimension depth and replace the
                        // last dimension
                        // specifier with varargs, when the stack is fully
                        // unwound.
                        currentDepth++;
                        var componentType = type.getComponentType();
                        visit(componentType, linkInfo.forType(componentType));
                        currentDepth--;
                        if (utils.isAnnotated(type)) {
                            link.add(" ");
                            link.add(getTypeAnnotationLinks(linkInfo));
                        }
                        // use vararg if required
                        if (linkInfo.isVarArg() && currentDepth == 0) {
                            link.add("...");
                        } else {
                            link.add("[]");
                        }
                        return link;
                    }

                    @Override
                    public Content visitWildcard(WildcardType type,
                            HtmlLinkInfo linkInfo) {
                        link.add(getTypeAnnotationLinks(linkInfo));
                        link.add("?");
                        TypeMirror extendsBound = type.getExtendsBound();
                        if (extendsBound != null) {
                            link.add(" extends ");
                            link.add(getLink(
                                getBoundsLinkInfo(linkInfo, extendsBound)));
                        }
                        TypeMirror superBound = type.getSuperBound();
                        if (superBound != null) {
                            link.add(" super ");
                            link.add(getLink(
                                getBoundsLinkInfo(linkInfo, superBound)));
                        }
                        return link;
                    }

                    @Override
                    public Content visitTypeVariable(TypeVariable type,
                            HtmlLinkInfo linkInfo) {
                        link.add(getTypeAnnotationLinks(linkInfo));
                        TypeVariable typevariable = (utils.isArrayType(type))
                            ? (TypeVariable) utils.getComponentType(type)
                            : type;
                        Element owner
                            = typevariable.asElement().getEnclosingElement();
                        if (linkInfo.linkTypeParameters()
                            && utils.isTypeElement(owner)) {
                            linkInfo.setTypeElement((TypeElement) owner);
                            Content label = newContent();
                            label.add(utils.getTypeName(type, false));
                            linkInfo.label(label).skipPreview(true);
                            link.add(getClassLink(linkInfo));
                        } else {
                            // No need to link method type parameters.
                            link.add(utils.getTypeName(typevariable, false));
                        }

                        if (linkInfo.showTypeBounds()) {
                            linkInfo.showTypeBounds(false);
                            TypeParameterElement tpe
                                = ((TypeParameterElement) typevariable
                                    .asElement());
                            boolean more = false;
                            List<? extends TypeMirror> bounds
                                = utils.getBounds(tpe);
                            for (TypeMirror bound : bounds) {
                                // we get everything as extends java.lang.Object
                                // we suppress
                                // all of them except those that have multiple
                                // extends
                                if (bounds.size() == 1 &&
                                    utils.typeUtils.isSameType(bound,
                                        utils.getObjectType())
                                    &&
                                    !utils.isAnnotated(bound)) {
                                    continue;
                                }
                                link.add(more ? " & " : " extends ");
                                link.add(getLink(
                                    getBoundsLinkInfo(linkInfo, bound)));
                                more = true;
                            }
                        }
                        return link;
                    }

                    @Override
                    public Content visitDeclared(DeclaredType type,
                            HtmlLinkInfo linkInfo) {
                        TypeMirror enc = type.getEnclosingType();
                        if (enc instanceof DeclaredType dt
                            && utils.isGenericType(dt)) {
                            // If an enclosing type has type parameters render
                            // them as separate links as
                            // otherwise this information is lost. On the other
                            // hand, plain enclosing types
                            // are not linked separately as they are easy to
                            // reach from the nested type.
                            visitDeclared(dt, linkInfo.forType(dt));
                            link.add(".");
                        }
                        link.add(getTypeAnnotationLinks(linkInfo));
                        linkInfo.setTypeElement(utils.asTypeElement(type));
                        link.add(getClassLink(linkInfo));
                        if (linkInfo.showTypeParameters()) {
                            link.add(getTypeParameterLinks(linkInfo));
                        }
                        return link;
                    }
                };
            return linkVisitor.visit(linkInfo.getType(), linkInfo);
        } else if (linkInfo.getTypeElement() != null) {
            Content link = newContent();
            link.add(getClassLink(linkInfo));
            if (linkInfo.showTypeParameters()) {
                link.add(getTypeParameterLinks(linkInfo));
            }
            return link;
        } else {
            return null;
        }
    }

    /**
     * Returns a link to the given class.
     *
     * @param linkInfo the information about the link to construct
     * @return the link for the given class.
     */
    protected Content getClassLink(HtmlLinkInfo linkInfo) {
        BaseConfiguration configuration = m_writer.configuration;
        TypeElement typeElement = linkInfo.getTypeElement();
        // Create a tool tip if we are linking to a class or interface. Don't
        // create one if we are linking to a member.
        String title = "";
        String fragment = linkInfo.getFragment();
        boolean hasFragment = fragment != null && !fragment.isEmpty();
        if (!hasFragment) {
            boolean isTypeLink = linkInfo.getType() != null &&
                utils
                    .isTypeVariable(utils.getComponentType(linkInfo.getType()));
            title = getClassToolTip(typeElement, isTypeLink);
        }
        Content label = linkInfo.getClassLinkLabel(configuration);
        if (linkInfo
            .getContext() == HtmlLinkInfo.Kind.SHOW_TYPE_PARAMS_IN_LABEL) {
            // For this kind of link, type parameters are included in the link
            // label
            // (and obviously not added after the link).
            label.add(getTypeParameterLinks(linkInfo));
        }
        Set<ElementFlag> flags;
        Element previewTarget;
        boolean showPreview = !linkInfo.isSkipPreview();
        if (!hasFragment && showPreview) {
            flags = utils.elementFlags(typeElement);
            previewTarget = typeElement;
        } else if (linkInfo.getContext() == HtmlLinkInfo.Kind.SHOW_PREVIEW
            && linkInfo.getTargetMember() != null && showPreview) {
            flags = utils.elementFlags(linkInfo.getTargetMember());
            TypeElement enclosing
                = utils.getEnclosingTypeElement(linkInfo.getTargetMember());
            Set<ElementFlag> enclosingFlags = utils.elementFlags(enclosing);
            if (flags.contains(ElementFlag.PREVIEW)
                && enclosingFlags.contains(ElementFlag.PREVIEW)) {
                if (enclosing.equals(m_writer.getCurrentPageElement())) {
                    // skip the PREVIEW tag:
                    flags = EnumSet.copyOf(flags);
                    flags.remove(ElementFlag.PREVIEW);
                }
                previewTarget = enclosing;
            } else {
                previewTarget = linkInfo.getTargetMember();
            }
        } else {
            flags = EnumSet.noneOf(ElementFlag.class);
            previewTarget = null;
        }

        Content link = new ContentBuilder();
        if (utils.isIncluded(typeElement)) {
            if (configuration.isGeneratedDoc(typeElement)
                && !utils.hasHiddenTag(typeElement)) {
                DocPath filename = getPath(linkInfo);
                if (linkInfo.linkToSelf()
                    || typeElement != m_writer.getCurrentPageElement()) {
                    link.add(m_writer.links.createLink(
                        filename.fragment(linkInfo.getFragment()),
                        label,
                        linkInfo.getStyle(),
                        title));
                    if (flags.contains(ElementFlag.PREVIEW)) {
                        link.add(HtmlTree.SUP(m_writer.links.createLink(
                            filename.fragment(m_writer.htmlIds
                                .forPreviewSection(previewTarget).name()),
                            m_writer.contents.previewMark)));
                    }
                    return link;
                }
            }
        } else {
            Content crossLink = m_writer.getCrossClassLink(
                typeElement, linkInfo.getFragment(),
                label, linkInfo.getStyle(), true);
            if (crossLink != null) {
                link.add(crossLink);
                if (flags.contains(ElementFlag.PREVIEW)) {
                    link.add(HtmlTree.SUP(m_writer.getCrossClassLink(
                        typeElement,
                        m_writer.htmlIds.forPreviewSection(previewTarget)
                            .name(),
                        m_writer.contents.previewMark,
                        null, false)));
                }
                return link;
            }
        }
        // Can't link so just write label.
        link.add(label);
        if (flags.contains(ElementFlag.PREVIEW)) {
            link.add(HtmlTree.SUP(m_writer.contents.previewMark));
        }
        return link;
    }

    /**
     * Returns links to the type parameters.
     *
     * @param linkInfo the information about the link to construct
     * @return the links to the type parameters
     */
    protected Content getTypeParameterLinks(HtmlLinkInfo linkInfo) {
        Content links = newContent();
        List<TypeMirror> vars = new ArrayList<>();
        TypeMirror ctype = linkInfo.getType() != null
            ? utils.getComponentType(linkInfo.getType())
            : null;
        if (linkInfo.getExecutableElement() != null) {
            linkInfo.getExecutableElement().getTypeParameters()
                .forEach(t -> vars.add(t.asType()));
        } else if (linkInfo.getType() != null
            && utils.isDeclaredType(linkInfo.getType())) {
            vars.addAll(((DeclaredType) linkInfo.getType()).getTypeArguments());
        } else if (ctype != null && utils.isDeclaredType(ctype)) {
            vars.addAll(((DeclaredType) ctype).getTypeArguments());
        } else if (ctype == null && linkInfo.getTypeElement() != null) {
            linkInfo.getTypeElement().getTypeParameters()
                .forEach(t -> vars.add(t.asType()));
        } else {
            // Nothing to document.
            return links;
        }
        if (!vars.isEmpty()) {
            if (linkInfo.addLineBreakOpportunitiesInTypeParameters()) {
                links.add(new HtmlTree(TagName.WBR));
            }
            links.add("<");
            boolean many = false;
            for (TypeMirror t : vars) {
                if (many) {
                    links.add(",");
                    links.add(new HtmlTree(TagName.WBR));
                    if (linkInfo.addLineBreaksInTypeParameters()) {
                        links.add(Text.NL);
                    }
                }
                links.add(getLink(linkInfo.forType(t)));
                many = true;
            }
            links.add(">");
        }
        return links;
    }

    /**
     * Returns links to the type annotations.
     *
     * @param linkInfo the information about the link to construct
     * @return the links to the type annotations
     */
    public Content getTypeAnnotationLinks(HtmlLinkInfo linkInfo) {
        ContentBuilder links = new ContentBuilder();
        List<? extends AnnotationMirror> annotations;
        if (utils.isAnnotated(linkInfo.getType())) {
            annotations = linkInfo.getType().getAnnotationMirrors();
        } else if (utils.isTypeVariable(linkInfo.getType())
            && linkInfo.showTypeParameterAnnotations()) {
            Element element = utils.typeUtils.asElement(linkInfo.getType());
            annotations = element.getAnnotationMirrors();
        } else {
            return links;
        }

        if (annotations.isEmpty())
            return links;

        m_writer.getAnnotations(annotations, false)
            .forEach(a -> {
                links.add(a);
                links.add(" ");
            });

        return links;
    }

    /*
     * Returns a link info for a type bounds link.
     */
    private HtmlLinkInfo getBoundsLinkInfo(HtmlLinkInfo linkInfo,
            TypeMirror bound) {
        return linkInfo.forType(bound).skipPreview(false);
    }

    /**
     * Given a class, return the appropriate tool tip.
     *
     * @param typeElement the class to get the tool tip for.
     * @return the tool tip for the appropriate class.
     */
    private String getClassToolTip(TypeElement typeElement,
            boolean isTypeLink) {
        Resources resources = m_writer.configuration.getDocResources();
        if (isTypeLink) {
            return resources.getText("doclet.Href_Type_Param_Title",
                utils.getSimpleName(typeElement));
        } else if (utils.isPlainInterface(typeElement)) {
            return resources.getText("doclet.Href_Interface_Title",
                m_writer.getLocalizedPackageName(
                    utils.containingPackage(typeElement)));
        } else if (utils.isAnnotationInterface(typeElement)) {
            return resources.getText("doclet.Href_Annotation_Title",
                m_writer.getLocalizedPackageName(
                    utils.containingPackage(typeElement)));
        } else if (utils.isEnum(typeElement)) {
            return resources.getText("doclet.Href_Enum_Title",
                m_writer.getLocalizedPackageName(
                    utils.containingPackage(typeElement)));
        } else {
            return resources.getText("doclet.Href_Class_Title",
                m_writer.getLocalizedPackageName(
                    utils.containingPackage(typeElement)));
        }
    }

    /**
     * Return path to the given file name in the given package. So if the name
     * passed is "Object.html" and the name of the package is "java.lang", and
     * if the relative path is "../.." then returned string will be
     * "../../java/lang/Object.html"
     *
     * @param linkInfo the information about the link.
     */
    private DocPath getPath(HtmlLinkInfo linkInfo) {
        return m_writer.pathToRoot
            .resolve(docPaths.forClass(linkInfo.getTypeElement()));
    }
}