/*

 * Copyright (c) 1999, 2019, 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.javac.comp;

import java.util.*;

import java.util.function.Supplier;

import javax.tools.JavaFileManager;

import com.sun.tools.javac.code.*;

import com.sun.tools.javac.code.Attribute.Compound;

import com.sun.tools.javac.code.Directive.ExportsDirective;

import com.sun.tools.javac.code.Directive.RequiresDirective;

import com.sun.tools.javac.code.Source.Feature;

import com.sun.tools.javac.comp.Annotate.AnnotationTypeMetadata;

import com.sun.tools.javac.jvm.*;

import com.sun.tools.javac.resources.CompilerProperties.Errors;

import com.sun.tools.javac.resources.CompilerProperties.Fragments;

import com.sun.tools.javac.resources.CompilerProperties.Warnings;

import com.sun.tools.javac.tree.*;

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

import com.sun.tools.javac.util.JCDiagnostic.DiagnosticFlag;

import com.sun.tools.javac.util.JCDiagnostic.DiagnosticPosition;

import com.sun.tools.javac.util.JCDiagnostic.Error;

import com.sun.tools.javac.util.JCDiagnostic.Fragment;

import com.sun.tools.javac.util.JCDiagnostic.Warning;

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

import com.sun.tools.javac.code.Lint;

import com.sun.tools.javac.code.Lint.LintCategory;

import com.sun.tools.javac.code.Scope.WriteableScope;

import com.sun.tools.javac.code.Type.*;

import com.sun.tools.javac.code.Symbol.*;

import com.sun.tools.javac.comp.DeferredAttr.DeferredAttrContext;

import com.sun.tools.javac.comp.Infer.FreeTypeListener;

import com.sun.tools.javac.tree.JCTree.*;

import static com.sun.tools.javac.code.Flags.*;

import static com.sun.tools.javac.code.Flags.ANNOTATION;

import static com.sun.tools.javac.code.Flags.SYNCHRONIZED;

import static com.sun.tools.javac.code.Kinds.*;

import static com.sun.tools.javac.code.Kinds.Kind.*;

import static com.sun.tools.javac.code.Scope.LookupKind.NON_RECURSIVE;

import static com.sun.tools.javac.code.TypeTag.*;

import static com.sun.tools.javac.code.TypeTag.WILDCARD;

import static com.sun.tools.javac.tree.JCTree.Tag.*;

/** Type checking helper class for the attribution phase.

 *

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

 */

public class Check {

    protected static final Context.Key<Check> checkKey = new Context.Key<>();

    private final Names names;

    private final Log log;

    private final Resolve rs;

    private final Symtab syms;

    private final Enter enter;

    private final DeferredAttr deferredAttr;

    private final Infer infer;

    private final Types types;

    private final TypeAnnotations typeAnnotations;

    private final JCDiagnostic.Factory diags;

    private final JavaFileManager fileManager;

    private final Source source;

    private final Target target;

    private final Profile profile;

    private final boolean warnOnAnyAccessToMembers;

    // The set of lint options currently in effect. It is initialized

    // from the context, and then is set/reset as needed by Attr as it

    // visits all the various parts of the trees during attribution.

    private Lint lint;

    // The method being analyzed in Attr - it is set/reset as needed by

    // Attr as it visits new method declarations.

    private MethodSymbol method;

    public static Check instance(Context context) {

        Check instance = context.get(checkKey);

        if (instance == null)

            instance = new Check(context);

        return instance;

    }

    protected Check(Context context) {

        context.put(checkKey, this);

        names = Names.instance(context);

        dfltTargetMeta = new Name[] { names.PACKAGE, names.TYPE,

            names.FIELD, names.METHOD, names.CONSTRUCTOR,

            names.ANNOTATION_TYPE, names.LOCAL_VARIABLE, names.PARAMETER};

        log = Log.instance(context);

        rs = Resolve.instance(context);

        syms = Symtab.instance(context);

        enter = Enter.instance(context);

        deferredAttr = DeferredAttr.instance(context);

        infer = Infer.instance(context);

        types = Types.instance(context);

        typeAnnotations = TypeAnnotations.instance(context);

        diags = JCDiagnostic.Factory.instance(context);

        Options options = Options.instance(context);

        lint = Lint.instance(context);

        fileManager = context.get(JavaFileManager.class);

        source = Source.instance(context);

        target = Target.instance(context);

        warnOnAnyAccessToMembers = options.isSet("warnOnAccessToMembers");

        Target target = Target.instance(context);

        syntheticNameChar = target.syntheticNameChar();

        profile = Profile.instance(context);

        boolean verboseDeprecated = lint.isEnabled(LintCategory.DEPRECATION);

        boolean verboseRemoval = lint.isEnabled(LintCategory.REMOVAL);

        boolean verboseUnchecked = lint.isEnabled(LintCategory.UNCHECKED);

        boolean enforceMandatoryWarnings = true;

        deprecationHandler = new MandatoryWarningHandler(log, verboseDeprecated,

                enforceMandatoryWarnings, "deprecated", LintCategory.DEPRECATION);

        removalHandler = new MandatoryWarningHandler(log, verboseRemoval,

                enforceMandatoryWarnings, "removal", LintCategory.REMOVAL);

        uncheckedHandler = new MandatoryWarningHandler(log, verboseUnchecked,

                enforceMandatoryWarnings, "unchecked", LintCategory.UNCHECKED);

        sunApiHandler = new MandatoryWarningHandler(log, false,

                enforceMandatoryWarnings, "sunapi", null);

        deferredLintHandler = DeferredLintHandler.instance(context);

    }

    /** Character for synthetic names

     */

    char syntheticNameChar;

    /** A table mapping flat names of all compiled classes for each module in this run

     *  to their symbols; maintained from outside.

     */

    private Map<Pair<ModuleSymbol, Name>,ClassSymbol> compiled = new HashMap<>();

    /** A handler for messages about deprecated usage.

     */

    private MandatoryWarningHandler deprecationHandler;

    /** A handler for messages about deprecated-for-removal usage.

     */

    private MandatoryWarningHandler removalHandler;

    /** A handler for messages about unchecked or unsafe usage.

     */

    private MandatoryWarningHandler uncheckedHandler;

    /** A handler for messages about using proprietary API.

     */

    private MandatoryWarningHandler sunApiHandler;

    /** A handler for deferred lint warnings.

     */

    private DeferredLintHandler deferredLintHandler;

/* *************************************************************************

 * Errors and Warnings

 **************************************************************************/

    Lint setLint(Lint newLint) {

        Lint prev = lint;

        lint = newLint;

        return prev;

    }

    MethodSymbol setMethod(MethodSymbol newMethod) {

        MethodSymbol prev = method;

        method = newMethod;

        return prev;

    }

    /** Warn about deprecated symbol.

     *  @param pos        Position to be used for error reporting.

     *  @param sym        The deprecated symbol.

     */

    void warnDeprecated(DiagnosticPosition pos, Symbol sym) {

        if (sym.isDeprecatedForRemoval()) {

            if (!lint.isSuppressed(LintCategory.REMOVAL)) {

                if (sym.kind == MDL) {

                    removalHandler.report(pos, Warnings.HasBeenDeprecatedForRemovalModule(sym));

                } else {

                    removalHandler.report(pos, Warnings.HasBeenDeprecatedForRemoval(sym, sym.location()));

                }

            }

        } else if (!lint.isSuppressed(LintCategory.DEPRECATION)) {

            if (sym.kind == MDL) {

                deprecationHandler.report(pos, Warnings.HasBeenDeprecatedModule(sym));

            } else {

                deprecationHandler.report(pos, Warnings.HasBeenDeprecated(sym, sym.location()));

            }

        }

    }

    /** Warn about unchecked operation.

     *  @param pos        Position to be used for error reporting.

     *  @param msg        A string describing the problem.

     */

    public void warnUnchecked(DiagnosticPosition pos, Warning warnKey) {

        if (!lint.isSuppressed(LintCategory.UNCHECKED))

            uncheckedHandler.report(pos, warnKey);

    }

    /** Warn about unsafe vararg method decl.

     *  @param pos        Position to be used for error reporting.

     */

    void warnUnsafeVararg(DiagnosticPosition pos, Warning warnKey) {

        if (lint.isEnabled(LintCategory.VARARGS))

            log.warning(LintCategory.VARARGS, pos, warnKey);

    }

    public void warnStatic(DiagnosticPosition pos, Warning warnKey) {

        if (lint.isEnabled(LintCategory.STATIC))

            log.warning(LintCategory.STATIC, pos, warnKey);

    }

    /** Warn about division by integer constant zero.

     *  @param pos        Position to be used for error reporting.

     */

    void warnDivZero(DiagnosticPosition pos) {

        if (lint.isEnabled(LintCategory.DIVZERO))

            log.warning(LintCategory.DIVZERO, pos, Warnings.DivZero);

    }

    /**

     * Report any deferred diagnostics.

     */

    public void reportDeferredDiagnostics() {

        deprecationHandler.reportDeferredDiagnostic();

        removalHandler.reportDeferredDiagnostic();

        uncheckedHandler.reportDeferredDiagnostic();

        sunApiHandler.reportDeferredDiagnostic();

    }

    /** Report a failure to complete a class.

     *  @param pos        Position to be used for error reporting.

     *  @param ex         The failure to report.

     */

    public Type completionError(DiagnosticPosition pos, CompletionFailure ex) {

        log.error(JCDiagnostic.DiagnosticFlag.NON_DEFERRABLE, pos, Errors.CantAccess(ex.sym, ex.getDetailValue()));

        return syms.errType;

    }

    /** Report an error that wrong type tag was found.

     *  @param pos        Position to be used for error reporting.

     *  @param required   An internationalized string describing the type tag

     *                    required.

     *  @param found      The type that was found.

     */

    Type typeTagError(DiagnosticPosition pos, JCDiagnostic required, Object found) {

        // this error used to be raised by the parser,

        // but has been delayed to this point:

        if (found instanceof Type && ((Type)found).hasTag(VOID)) {

            log.error(pos, Errors.IllegalStartOfType);

            return syms.errType;

        }

        log.error(pos, Errors.TypeFoundReq(found, required));

        return types.createErrorType(found instanceof Type ? (Type)found : syms.errType);

    }

    /** Report an error that symbol cannot be referenced before super

     *  has been called.

     *  @param pos        Position to be used for error reporting.

     *  @param sym        The referenced symbol.

     */

    void earlyRefError(DiagnosticPosition pos, Symbol sym) {

        log.error(pos, Errors.CantRefBeforeCtorCalled(sym));

    }

    /** Report duplicate declaration error.

     */

    void duplicateError(DiagnosticPosition pos, Symbol sym) {

        if (!sym.type.isErroneous()) {

            Symbol location = sym.location();

            if (location.kind == MTH &&

                    ((MethodSymbol)location).isStaticOrInstanceInit()) {

                log.error(pos,

                          Errors.AlreadyDefinedInClinit(kindName(sym),

                                                        sym,

                                                        kindName(sym.location()),

                                                        kindName(sym.location().enclClass()),

                                                        sym.location().enclClass()));

            } else {

                log.error(pos,

                          Errors.AlreadyDefined(kindName(sym),

                                                sym,

                                                kindName(sym.location()),

                                                sym.location()));

            }

        }

    }

    /** Report array/varargs duplicate declaration

     */

    void varargsDuplicateError(DiagnosticPosition pos, Symbol sym1, Symbol sym2) {

        if (!sym1.type.isErroneous() && !sym2.type.isErroneous()) {

            log.error(pos, Errors.ArrayAndVarargs(sym1, sym2, sym2.location()));

        }

    }

/* ************************************************************************

 * duplicate declaration checking

 *************************************************************************/

    /** Check that variable does not hide variable with same name in

     *  immediately enclosing local scope.

     *  @param pos           Position for error reporting.

     *  @param v             The symbol.

     *  @param s             The scope.

     */

    void checkTransparentVar(DiagnosticPosition pos, VarSymbol v, Scope s) {

        for (Symbol sym : s.getSymbolsByName(v.name)) {

            if (sym.owner != v.owner) break;

            if (sym.kind == VAR &&

                sym.owner.kind.matches(KindSelector.VAL_MTH) &&

                v.name != names.error) {

                duplicateError(pos, sym);

                return;

            }

        }

    }

    /** Check that a class or interface does not hide a class or

     *  interface with same name in immediately enclosing local scope.

     *  @param pos           Position for error reporting.

     *  @param c             The symbol.

     *  @param s             The scope.

     */

    void checkTransparentClass(DiagnosticPosition pos, ClassSymbol c, Scope s) {

        for (Symbol sym : s.getSymbolsByName(c.name)) {

            if (sym.owner != c.owner) break;

            if (sym.kind == TYP && !sym.type.hasTag(TYPEVAR) &&

                sym.owner.kind.matches(KindSelector.VAL_MTH) &&

                c.name != names.error) {

                duplicateError(pos, sym);

                return;

            }

        }

    }

    /** Check that class does not have the same name as one of

     *  its enclosing classes, or as a class defined in its enclosing scope.

     *  return true if class is unique in its enclosing scope.

     *  @param pos           Position for error reporting.

     *  @param name          The class name.

     *  @param s             The enclosing scope.

     */

    boolean checkUniqueClassName(DiagnosticPosition pos, Name name, Scope s) {

        for (Symbol sym : s.getSymbolsByName(name, NON_RECURSIVE)) {

            if (sym.kind == TYP && sym.name != names.error) {

                duplicateError(pos, sym);

                return false;

            }

        }

        for (Symbol sym = s.owner; sym != null; sym = sym.owner) {

            if (sym.kind == TYP && sym.name == name && sym.name != names.error) {

                duplicateError(pos, sym);

                return true;

            }

        }

        return true;

    }

/* *************************************************************************

 * Class name generation

 **************************************************************************/

    private Map<Pair<Name, Name>, Integer> localClassNameIndexes = new HashMap<>();

    /** Return name of local class.

     *  This is of the form   {@code <enclClass> $ n <classname> }

     *  where

     *    enclClass is the flat name of the enclosing class,

     *    classname is the simple name of the local class

     */

    public Name localClassName(ClassSymbol c) {

        Name enclFlatname = c.owner.enclClass().flatname;

        String enclFlatnameStr = enclFlatname.toString();

        Pair<Name, Name> key = new Pair<>(enclFlatname, c.name);

        Integer index = localClassNameIndexes.get(key);

        for (int i = (index == null) ? 1 : index; ; i++) {

            Name flatname = names.fromString(enclFlatnameStr

                    + syntheticNameChar + i + c.name);

            if (getCompiled(c.packge().modle, flatname) == null) {

                localClassNameIndexes.put(key, i + 1);

                return flatname;

            }

        }

    }

    public void clearLocalClassNameIndexes(ClassSymbol c) {

        if (c.owner != null && c.owner.kind != NIL) {

            localClassNameIndexes.remove(new Pair<>(

                    c.owner.enclClass().flatname, c.name));

        }

    }

    public void newRound() {

        compiled.clear();

        localClassNameIndexes.clear();

    }

    public void putCompiled(ClassSymbol csym) {

        compiled.put(Pair.of(csym.packge().modle, csym.flatname), csym);

    }

    public ClassSymbol getCompiled(ClassSymbol csym) {

        return compiled.get(Pair.of(csym.packge().modle, csym.flatname));

    }

    public ClassSymbol getCompiled(ModuleSymbol msym, Name flatname) {

        return compiled.get(Pair.of(msym, flatname));

    }

    public void removeCompiled(ClassSymbol csym) {

        compiled.remove(Pair.of(csym.packge().modle, csym.flatname));

    }

/* *************************************************************************

 * Type Checking

 **************************************************************************/

    /**

     * A check context is an object that can be used to perform compatibility

     * checks - depending on the check context, meaning of 'compatibility' might

     * vary significantly.

     */

    public interface CheckContext {

        /**

         * Is type 'found' compatible with type 'req' in given context

         */

        boolean compatible(Type found, Type req, Warner warn);

        /**

         * Report a check error

         */

        void report(DiagnosticPosition pos, JCDiagnostic details);

        /**

         * Obtain a warner for this check context

         */

        public Warner checkWarner(DiagnosticPosition pos, Type found, Type req);

        public InferenceContext inferenceContext();

        public DeferredAttr.DeferredAttrContext deferredAttrContext();

    }

    /**

     * This class represent a check context that is nested within another check

     * context - useful to check sub-expressions. The default behavior simply

     * redirects all method calls to the enclosing check context leveraging

     * the forwarding pattern.

     */

    static class NestedCheckContext implements CheckContext {

        CheckContext enclosingContext;

        NestedCheckContext(CheckContext enclosingContext) {

            this.enclosingContext = enclosingContext;

        }

        public boolean compatible(Type found, Type req, Warner warn) {

            return enclosingContext.compatible(found, req, warn);

        }