1 files changed, 63 insertions, 0 deletions
diff --git a/gradle/src/main/groovy/com/android/build/gradle/internal/LintGradleRequest.java b/gradle/src/main/groovy/com/android/build/gradle/internal/LintGradleRequest.java
new file mode 100644
index 0000000..b5674f9
--- /dev/null
+++ b/gradle/src/main/groovy/com/android/build/gradle/internal/LintGradleRequest.java
@@ -0,0 +1,63 @@
+package com.android.build.gradle.internal;
+
+import com.android.annotations.NonNull;
+import com.android.annotations.Nullable;
+import com.android.build.gradle.BasePlugin;
+import com.android.builder.model.AndroidProject;
+import com.android.builder.model.Variant;
+import com.android.tools.lint.client.api.LintRequest;
+import com.android.tools.lint.detector.api.Project;
+import com.android.utils.Pair;
+
+import java.io.File;
+import java.util.Collection;
+import java.util.Collections;
+import java.util.List;
+
+class LintGradleRequest extends LintRequest {
+    @NonNull private final LintGradleClient mLintClient;
+    @NonNull private final BasePlugin mPlugin;
+    @NonNull private final String mVariantName;
+    @NonNull private final AndroidProject mModelProject;
+
+    public LintGradleRequest(
+            @NonNull LintGradleClient client,
+            @NonNull AndroidProject modelProject,
+            @NonNull BasePlugin plugin,
+            @Nullable String variantName,
+            @NonNull List<File> files) {
+        super(client, files);
+        mLintClient = client;
+        mModelProject = modelProject;
+        mPlugin = plugin;
+        mVariantName = variantName;
+    }
+
+    @Nullable
+    @Override
+    public Collection<Project> getProjects() {
+        if (mProjects == null) {
+            Variant variant = findVariant(mModelProject, mVariantName);
+            assert variant != null : mVariantName;
+            Pair<LintGradleProject,List<File>> result = LintGradleProject.create(
+                    mLintClient, mModelProject, variant, mPlugin.getProject());
+            mProjects = Collections.<Project>singletonList(result.getFirst());
+            mLintClient.setCustomRules(result.getSecond());
+        }
+
+        return mProjects;
+    }
+
+    private static Variant findVariant(@NonNull AndroidProject project,
+            @NonNull String variantName) {
+        if (variantName != null) {
+            for (Variant variant : project.getVariants()) {
+                if (variantName.equals(variant.getName())) {
+                    return variant;
+                }
+            }
+        }
+
+        return null;
+    }
+}