/*

 * Copyright (c) 1999, 2015, 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 com.sun.tools.doclets.internal.toolkit.util;

import java.io.*;

import java.lang.annotation.Documented;

import java.lang.annotation.ElementType;

import java.lang.annotation.Target;

import java.text.Collator;

import java.util.*;

import javax.tools.JavaFileManager.Location;

import com.sun.javadoc.*;

import com.sun.tools.doclets.internal.toolkit.*;

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

/**

 * Utilities Class for Doclets.

 *

 *  <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 Jamie Ho

 */

public class Utils {

    /**

     * Return array of class members whose documentation is to be generated.

     * If the member is deprecated do not include such a member in the

     * returned array.

     *

     * @param  members             Array of members to choose from.

     * @return ProgramElementDoc[] Array of eligible members for whom

     *                             documentation is getting generated.

     */

    public ProgramElementDoc[] excludeDeprecatedMembers(

        ProgramElementDoc[] members) {

        return

            toProgramElementDocArray(excludeDeprecatedMembersAsList(members));

    }

    /**

     * Return array of class members whose documentation is to be generated.

     * If the member is deprecated do not include such a member in the

     * returned array.

     *

     * @param  members    Array of members to choose from.

     * @return List       List of eligible members for whom

     *                    documentation is getting generated.

     */

    public List<ProgramElementDoc> excludeDeprecatedMembersAsList(

        ProgramElementDoc[] members) {

        List<ProgramElementDoc> list = new ArrayList<>();

        for (ProgramElementDoc member : members) {

            if (member.tags("deprecated").length == 0) {

                list.add(member);

            }

        }

        Collections.sort(list);

        return list;

    }

    /**

     * Return the list of ProgramElementDoc objects as Array.

     */

    public ProgramElementDoc[] toProgramElementDocArray(List<ProgramElementDoc> list) {

        ProgramElementDoc[] pgmarr = new ProgramElementDoc[list.size()];

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

            pgmarr[i] = list.get(i);

        }

        return pgmarr;

    }

    /**

     * Return true if a non-public member found in the given array.

     *

     * @param  members Array of members to look into.

     * @return boolean True if non-public member found, false otherwise.

     */

    public boolean nonPublicMemberFound(ProgramElementDoc[] members) {

        for (ProgramElementDoc member : members) {

            if (!member.isPublic()) {

                return true;

            }

        }

        return false;

    }

    /**

     * Search for the given method in the given class.

     *

     * @param  cd        Class to search into.

     * @param  method    Method to be searched.

     * @return MethodDoc Method found, null otherwise.

     */

    public MethodDoc findMethod(ClassDoc cd, MethodDoc method) {

        MethodDoc[] methods = cd.methods();

        for (MethodDoc m : methods) {

            if (executableMembersEqual(method, m)) {

                return m;

            }

        }

        return null;

    }

    /**

     * @param member1 the first method to compare.

     * @param member2 the second method to compare.

     * @return true if member1 overrides/hides or is overriden/hidden by member2.

     */

    public boolean executableMembersEqual(ExecutableMemberDoc member1,

            ExecutableMemberDoc member2) {

        if (! (member1 instanceof MethodDoc && member2 instanceof MethodDoc))

            return false;

        MethodDoc method1 = (MethodDoc) member1;

        MethodDoc method2 = (MethodDoc) member2;

        if (method1.isStatic() && method2.isStatic()) {

            Parameter[] targetParams = method1.parameters();

            Parameter[] currentParams;

            if (method1.name().equals(method2.name()) &&

                   (currentParams = method2.parameters()).length ==

                targetParams.length) {

                int j;

                for (j = 0; j < targetParams.length; j++) {

                    if (! (targetParams[j].typeName().equals(

                              currentParams[j].typeName()) ||

                                   currentParams[j].type() instanceof TypeVariable ||

                                   targetParams[j].type() instanceof TypeVariable)) {

                        break;

                    }

                }

                if (j == targetParams.length) {

                    return true;

                }

            }

            return false;

        } else {

                return method1.overrides(method2) ||

                method2.overrides(method1) ||

                                member1 == member2;

        }

    }

    /**

     * According to

     * <cite>The Java&trade; Language Specification</cite>,

     * all the outer classes and static inner classes are core classes.

     */

    public boolean isCoreClass(ClassDoc cd) {

        return cd.containingClass() == null || cd.isStatic();

    }

    public boolean matches(ProgramElementDoc doc1,

            ProgramElementDoc doc2) {

        if (doc1 instanceof ExecutableMemberDoc &&

            doc2 instanceof ExecutableMemberDoc) {

            ExecutableMemberDoc ed1 = (ExecutableMemberDoc)doc1;

            ExecutableMemberDoc ed2 = (ExecutableMemberDoc)doc2;

            return executableMembersEqual(ed1, ed2);

        } else {

            return doc1.name().equals(doc2.name());

        }

    }

    /**

     * Copy the given directory contents from the source package directory

     * to the generated documentation directory. For example for a package

     * java.lang this method find out the source location of the package using

     * {@link SourcePath} and if given directory is found in the source

     * directory structure, copy the entire directory, to the generated

     * documentation hierarchy.

     *

     * @param configuration The configuration of the current doclet.

     * @param path The relative path to the directory to be copied.

     * @param dir The original directory name to copy from.

     * @param overwrite Overwrite files if true.

     */

    public void copyDocFiles(Configuration configuration, PackageDoc pd) {

        Location locn = configuration.getLocationForPackage(pd);

        copyDocFiles(configuration, locn, DocPath.forPackage(pd).resolve(DocPaths.DOC_FILES));

    }

    public void copyDocFiles(Configuration configuration, Location locn, DocPath dir) {

        try {

            boolean first = true;

            for (DocFile f : DocFile.list(configuration, locn, dir)) {

                if (!f.isDirectory()) {

                    continue;

                }

                DocFile srcdir = f;

                DocFile destdir = DocFile.createFileForOutput(configuration, dir);

                if (srcdir.isSameFile(destdir)) {

                    continue;

                }

                for (DocFile srcfile: srcdir.list()) {

                    DocFile destfile = destdir.resolve(srcfile.getName());

                    if (srcfile.isFile()) {

                        if (destfile.exists() && !first) {

                            configuration.message.warning((SourcePosition) null,

                                    "doclet.Copy_Overwrite_warning",

                                    srcfile.getPath(), destdir.getPath());

                        } else {

                            configuration.message.notice(

                                    "doclet.Copying_File_0_To_Dir_1",

                                    srcfile.getPath(), destdir.getPath());

                            destfile.copyFile(srcfile);

                        }

                    } else if (srcfile.isDirectory()) {

                        if (configuration.copydocfilesubdirs

                                && !configuration.shouldExcludeDocFileDir(srcfile.getName())) {

                            copyDocFiles(configuration, locn, dir.resolve(srcfile.getName()));

                        }

                    }

                }

                first = false;

            }

        } catch (SecurityException | IOException exc) {

            throw new DocletAbortException(exc);

        }

    }

    /**

     * Returns a TypeComparator object suitable for sorting Types.

     * @return a TypeComparator objectt

     */

    public Comparator<Type> makeTypeComparator() {

        return new TypeComparator();

    }

    /**

     * We want the list of types in alphabetical order.  However, types are not

     * comparable.  We need a comparator for now.

     */

    private static class TypeComparator implements Comparator<Type> {

        public int compare(Type type1, Type type2) {

            return compareStrings(type1.qualifiedTypeName(), type2.qualifiedTypeName());

        }

    }

    /**

     * For the class return all implemented interfaces including the

     * superinterfaces of the implementing interfaces, also iterate over for

     * all the superclasses. For interface return all the extended interfaces

     * as well as superinterfaces for those extended interfaces.

     *

     * @param  type       type whose implemented or

     *                    super interfaces are sought.

     * @param  configuration the current configuration of the doclet.

     * @param  sort if true, return list of interfaces sorted alphabetically.

     * @return List of all the required interfaces.

     */

    public List<Type> getAllInterfaces(Type type,

            Configuration configuration, boolean sort) {

        Map<ClassDoc,Type> results = sort ?

                new TreeMap<ClassDoc,Type>() :

                new LinkedHashMap<ClassDoc,Type>();

        Type[] interfaceTypes = null;

        Type superType = null;

        if (type instanceof ParameterizedType) {

            interfaceTypes = ((ParameterizedType) type).interfaceTypes();

            superType = ((ParameterizedType) type).superclassType();

        } else if (type instanceof ClassDoc) {

            interfaceTypes = ((ClassDoc) type).interfaceTypes();

            superType = ((ClassDoc) type).superclassType();

        } else {

            interfaceTypes = type.asClassDoc().interfaceTypes();

            superType = type.asClassDoc().superclassType();

        }

        for (Type interfaceType : interfaceTypes) {

            ClassDoc interfaceClassDoc = interfaceType.asClassDoc();

            if (!(interfaceClassDoc.isPublic() ||

                  (configuration == null ||

                   isLinkable(interfaceClassDoc, configuration)))) {

                continue;

            }

            results.put(interfaceClassDoc, interfaceType);

            for (Type t : getAllInterfaces(interfaceType, configuration, sort)) {

                results.put(t.asClassDoc(), t);

            }

        }

        if (superType == null)

            return new ArrayList<>(results.values());

        //Try walking the tree.

        addAllInterfaceTypes(results,

            superType,

            interfaceTypesOf(superType),

            false, configuration);

        List<Type> resultsList = new ArrayList<>(results.values());

        if (sort) {

                Collections.sort(resultsList, new TypeComparator());

        }

        return resultsList;

    }

    private Type[] interfaceTypesOf(Type type) {

        if (type instanceof AnnotatedType)

            type = ((AnnotatedType)type).underlyingType();

        return type instanceof ClassDoc ?

                ((ClassDoc)type).interfaceTypes() :

                ((ParameterizedType)type).interfaceTypes();

    }

    public List<Type> getAllInterfaces(Type type, Configuration configuration) {

        return getAllInterfaces(type, configuration, true);

    }

    private void findAllInterfaceTypes(Map<ClassDoc,Type> results, ClassDoc c, boolean raw,

            Configuration configuration) {

        Type superType = c.superclassType();

        if (superType == null)

            return;

        addAllInterfaceTypes(results, superType,

                interfaceTypesOf(superType),

                raw, configuration);

    }

    private void findAllInterfaceTypes(Map<ClassDoc,Type> results, ParameterizedType p,

            Configuration configuration) {

        Type superType = p.superclassType();

        if (superType == null)

            return;

        addAllInterfaceTypes(results, superType,

                interfaceTypesOf(superType),

                false, configuration);

    }

    private void addAllInterfaceTypes(Map<ClassDoc,Type> results, Type type,

            Type[] interfaceTypes, boolean raw,

            Configuration configuration) {

        for (Type interfaceType : interfaceTypes) {

            ClassDoc interfaceClassDoc = interfaceType.asClassDoc();

            if (!(interfaceClassDoc.isPublic() ||

                  (configuration != null &&

                   isLinkable(interfaceClassDoc, configuration)))) {

                continue;

            }

            if (raw)

                interfaceType = interfaceType.asClassDoc();

            results.put(interfaceClassDoc, interfaceType);

            List<Type> superInterfaces = getAllInterfaces(interfaceType, configuration);

            for (Type superInterface : superInterfaces) {

                results.put(superInterface.asClassDoc(), superInterface);

            }

        }

        if (type instanceof AnnotatedType)

            type = ((AnnotatedType)type).underlyingType();

        if (type instanceof ParameterizedType)

            findAllInterfaceTypes(results, (ParameterizedType) type, configuration);

        else if (((ClassDoc) type).typeParameters().length == 0)

            findAllInterfaceTypes(results, (ClassDoc) type, raw, configuration);

        else

            findAllInterfaceTypes(results, (ClassDoc) type, true, configuration);

    }

    /**

     * Enclose in quotes, used for paths and filenames that contains spaces

     */

    public String quote(String filepath) {

        return ("\"" + filepath + "\"");

    }

    /**

     * Given a package, return its name.

     * @param packageDoc the package to check.

     * @return the name of the given package.

     */

    public String getPackageName(PackageDoc packageDoc) {

        return packageDoc == null || packageDoc.name().length() == 0 ?

            DocletConstants.DEFAULT_PACKAGE_NAME : packageDoc.name();

    }

    /**

     * Given a package, return its file name without the extension.

     * @param packageDoc the package to check.

     * @return the file name of the given package.

     */

    public String getPackageFileHeadName(PackageDoc packageDoc) {

        return packageDoc == null || packageDoc.name().length() == 0 ?

            DocletConstants.DEFAULT_PACKAGE_FILE_NAME : packageDoc.name();

    }

    /**

     * Given a string, replace all occurrences of 'newStr' with 'oldStr'.

     * @param originalStr the string to modify.

     * @param oldStr the string to replace.

     * @param newStr the string to insert in place of the old string.

     */

    public String replaceText(String originalStr, String oldStr,

            String newStr) {

        if (oldStr == null || newStr == null || oldStr.equals(newStr)) {

            return originalStr;

        }

        return originalStr.replace(oldStr, newStr);

    }

    /**

     * Given an annotation, return true if it should be documented and false

     * otherwise.

     *

     * @param annotationDoc the annotation to check.

     *

     * @return true return true if it should be documented and false otherwise.

     */

    public boolean isDocumentedAnnotation(AnnotationTypeDoc annotationDoc) {

        for (AnnotationDesc anno : annotationDoc.annotations()) {

            if (anno.annotationType().qualifiedName().equals(

                    Documented.class.getName())) {

                return true;

            }

        }

        return false;

    }

    private boolean isDeclarationTarget(AnnotationDesc targetAnno) {

        // The error recovery steps here are analogous to TypeAnnotations

        if (elems == null

            || elems.length != 1

            || !"value".equals(elems[0].element().name())

            || !(elems[0].value().value() instanceof AnnotationValue[]))

            return true;    // error recovery

        for (AnnotationValue aValue : (AnnotationValue[])elems[0].value().value()) {

            Object value = aValue.value();

            if (!(value instanceof FieldDoc))

                return true; // error recovery

            FieldDoc eValue = (FieldDoc) value;

            if (isJava5DeclarationElementType(eValue)) {

                return true;

            }

        }

        return false;

    }

    /**

     * Returns true if the {@code annotationDoc} is to be treated

     * as a declaration annotation, when targeting the

     * {@code elemType} element type.

     *

     * @param annotationDoc the annotationDoc to check

     * @param elemType  the targeted elemType

     * @return true if annotationDoc is a declaration annotation

     */

    public boolean isDeclarationAnnotation(AnnotationTypeDoc annotationDoc,

            boolean isJava5DeclarationLocation) {

        if (!isJava5DeclarationLocation)

            return false;

        AnnotationDesc[] annotationDescList = annotationDoc.annotations();

        // Annotations with no target are treated as declaration as well

        if (annotationDescList.length==0)

            return true;

        for (AnnotationDesc anno : annotationDescList) {

            if (anno.annotationType().qualifiedName().equals(

                    Target.class.getName())) {

                if (isDeclarationTarget(anno))

                    return true;

            }

        }

        return false;

    }

    /**

     * Return true if this class is linkable and false if we can't link to the

     * desired class.

     * <br>

     * <b>NOTE:</b>  You can only link to external classes if they are public or

     * protected.

     *

     * @param classDoc the class to check.

     * @param configuration the current configuration of the doclet.

     *

     * @return true if this class is linkable and false if we can't link to the

     * desired class.

     */