package com.kalessil.phpStorm.phpInspectionsEA.inspectors.strictOperators;

import com.intellij.codeInspection.ProblemHighlightType;
import com.intellij.codeInspection.ProblemsHolder;
import com.intellij.psi.PsiElement;
import com.intellij.psi.PsiElementVisitor;
import com.jetbrains.php.PhpIndex;
import com.jetbrains.php.lang.psi.elements.*;
import com.kalessil.phpStorm.phpInspectionsEA.openApi.BasePhpElementVisitor;
import com.kalessil.phpStorm.phpInspectionsEA.openApi.BasePhpInspection;
import org.jetbrains.annotations.NotNull;

import java.util.Collection;

public class StrictArgumentsInspector extends BasePhpInspection {
    private static final String strProblemDescriptionArgumentTypeMismatch = "Type of argument (%t1%) doesn't match type of parameter (%t2%).";

    @NotNull
    public String getShortName() {
        return "StrictArgumentsInspection";
    }

    @Override
    public PsiElementVisitor buildVisitor(@NotNull final ProblemsHolder holder, boolean isOnTheFly) {
        return new BasePhpElementVisitor() {

            public void visitPhpMethodReference(MethodReference reference) {
                PhpIndex objIndex = PhpIndex.getInstance(holder.getProject());

                Collection<? extends PhpNamedElement> methods = objIndex.getBySignature(reference.getSignature());
                if (methods.size() != 1) {
                    return;
                }

                PhpNamedElement method = methods.iterator().next();
                if (!(method instanceof Method)) {
                    return;
                }

                Parameter[] params = ((Method) method).getParameters();
                PsiElement[] args = reference.getParameters();

                inspectArguments(args, params);
            }

            public void visitPhpFunctionCall(FunctionReference reference) {
                PhpIndex objIndex = PhpIndex.getInstance(holder.getProject());

                Collection<Function> funcs = objIndex.getFunctionsByName(reference.getName());
                if (funcs.size() != 1) {
                    return;
                }

                Parameter[] params = funcs.iterator().next().getParameters();
                PsiElement[] args = reference.getParameters();

                inspectArguments(args, params);
            }

            private void inspectArguments(final PsiElement[] args, final Parameter[] params) {
                final int length = Math.min(args.length, params.length);
                for (int i = 0; i < length; ++i) {
                    PhpExpressionTypes argType = new PhpExpressionTypes(args[i], holder);
                    final PhpExpressionTypes paramType = new PhpExpressionTypes(params[i].getType().toString(), holder);

                    if (!paramType.isMixed()
                            && !argType.equals(paramType)
                            && !argType.instanceOf(paramType)) {
                        final String strWarning = strProblemDescriptionArgumentTypeMismatch
                                .replace("%t1%", argType.toString())
                                .replace("%t2%", paramType.toString());
                        holder.registerProblem(args[i], strWarning, ProblemHighlightType.WEAK_WARNING);
                    }
                }
            }
        };
    }
}

@@ -2670,24 +2670,236 @@
 , holder);%0D%0A
+                    if (argType.isMixed() && (args%5Bi%5D instanceof Variable)) %7B%0D%0A                        argType = new PhpExpressionTypes(((Variable) args%5Bi%5D).getType().toString(), holder);%0D%0A                    %7D%0D%0A
             

resolve variables if possible