/*
 * (c) Copyright 2019 Palantir Technologies Inc. All rights reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.palantir.baseline.errorprone;

import com.google.auto.service.AutoService;
import com.google.errorprone.BugPattern;
import com.google.errorprone.VisitorState;
import com.google.errorprone.bugpatterns.BugChecker;
import com.google.errorprone.fixes.SuggestedFix;
import com.google.errorprone.matchers.Description;
import com.google.errorprone.matchers.Matcher;
import com.google.errorprone.matchers.Matchers;
import com.sun.source.tree.ExpressionTree;
import com.sun.source.tree.MethodInvocationTree;

@AutoService(BugChecker.class)
@BugPattern(
        name = "JavaTimeSystemDefaultTimeZone",
        link = "https://github.com/palantir/gradle-baseline#baseline-error-prone-checks",
        linkType = BugPattern.LinkType.CUSTOM,
        severity = BugPattern.SeverityLevel.ERROR,
        summary = "The system default time zone should not be used, since the behavior is system dependent. "
                + "Instead, UTC should always be used.")
public final class JavaTimeSystemDefaultTimeZone extends BugChecker implements BugChecker.MethodInvocationTreeMatcher {

    private static final Matcher<ExpressionTree> CLOCK_SYSTEM_DEFAULT_ZONE_MATCHER = Matchers.staticMethod()
            .onClass("java.time.Clock")
            .named("systemDefaultZone")
            .withParameters();
    private static final Matcher<ExpressionTree> ZONE_ID_SYSTEM_DEFAULT_MATCHER = Matchers.staticMethod()
            .onClass("java.time.ZoneId")
            .named("systemDefault")
            .withParameters();

    @Override
    public Description matchMethodInvocation(MethodInvocationTree tree, VisitorState state) {
        if (CLOCK_SYSTEM_DEFAULT_ZONE_MATCHER.matches(tree, state)) {
            return buildDescription(tree)
                    .addFix(SuggestedFix.builder()
                            .replace(tree, "Clock.systemUTC()")
                            .addImport("java.time.Clock")
                            .build())
                    .build();
        }

        if (ZONE_ID_SYSTEM_DEFAULT_MATCHER.matches(tree, state)) {
            return buildDescription(tree)
                    .addFix(SuggestedFix.builder()
                            .replace(tree, "ZoneOffset.UTC")
                            .addImport("java.time.ZoneOffset")
                            .build())
                    .build();
        }

        return Description.NO_MATCH;
    }
}

@@ -902,16 +902,67 @@
 tedFix;%0A
+import com.google.errorprone.fixes.SuggestedFixes;%0A
 import c
@@ -1190,16 +1190,45 @@
 ionTree;
+%0Aimport java.time.ZoneOffset;
 %0A%0A@AutoS
@@ -2488,63 +2488,41 @@
 dFix
-.builder()%0A                            .replace
+es.renameMethodInvocation
 (tree, %22
 Cloc
@@ -2521,14 +2521,8 @@
 e, %22
-Clock.
 syst
@@ -2530,106 +2530,16 @@
 mUTC
-()%22)%0A                            .addImport(%22java.time.Clock%22)%0A                            .build(
+%22, state
 ))%0A 
@@ -2637,32 +2637,95 @@
 tree, state)) %7B%0A
+            SuggestedFix.Builder fix = SuggestedFix.builder();%0A
             retu
@@ -2778,38 +2778,28 @@
 .addFix(
-SuggestedFix.builder()
+fix.replace(
 %0A       
@@ -2823,101 +2823,96 @@
     
-.replace(tree, %22ZoneOffset.UTC%22)%0A                            .addImport(%22java.time.ZoneOffset
+        tree, SuggestedFixes.qualifyType(state, fix, ZoneOffset.class.getName()) + %22.UTC
 %22)%0A 

Allow ErrorProne to qualify JavaTimeSystemDefaultTimeZone fix types (#1533)