Add support for the android studio DB inspector

gradle/libs.versions.toml

@@ -76,7 +76,7 @@ kotlin-android = { id = "org.jetbrains.kotlin.android", version.ref = "kotlin" }
 kotlin-jvm = { id = "org.jetbrains.kotlin.jvm", version.ref = "kotlin" }
 kotlin-js = { id = "org.jetbrains.kotlin.js", version.ref = "kotlin" }
 dokka = { id = "org.jetbrains.dokka", version.ref = "dokka" }
-intellij = { id = "org.jetbrains.intellij", version = "1.5.3" }
+intellij = { id = "org.jetbrains.intellij", version = "1.5.2" }
 grammarKitComposer = { id = "com.alecstrong.grammar.kit.composer", version = "0.1.9" }
 publish = { id = "com.vanniktech.maven.publish", version = "0.19.0" }
 spotless = { id = "com.diffplug.spotless", version = "6.4.2" } # When updating, check if we can remove base plugin from root build.gradle

sqldelight-idea-plugin/build.gradle

@@ -18,6 +18,7 @@ intellij {
       "org.jetbrains.kotlin",
       "com.intellij.gradle",
       "com.intellij.java",
+      "org.jetbrains.android"
   ]
   if (!project.version.endsWith("SNAPSHOT")) {
     patchPluginXml {

sqldelight-idea-plugin/src/main/kotlin/app/cash/sqldelight/intellij/run/android/SqlDelightRunAnnotator.kt

@@ -0,0 +1,78 @@
+package app.cash.sqldelight.intellij.run.android
+
+import app.cash.sqldelight.core.lang.util.rawSqlText
+import com.alecstrong.sql.psi.core.psi.SqlStmt
+import com.alecstrong.sql.psi.core.psi.SqlStmtList
+import com.android.tools.idea.sqlite.DatabaseInspectorProjectService
+import com.android.tools.idea.sqlite.annotator.RunSqliteStatementGutterIconAction
+import com.android.tools.idea.sqlite.ui.DatabaseInspectorViewsFactoryImpl
+import com.intellij.lang.annotation.AnnotationHolder
+import com.intellij.lang.annotation.Annotator
+import com.intellij.lang.annotation.HighlightSeverity
+import com.intellij.lang.java.JavaLanguage
+import com.intellij.openapi.actionSystem.AnAction
+import com.intellij.openapi.editor.markup.GutterIconRenderer
+import com.intellij.psi.PsiElement
+import com.intellij.psi.PsiFileFactory
+import com.intellij.psi.PsiLanguageInjectionHost
+import com.intellij.psi.util.PsiTreeUtil
+import com.intellij.util.ui.EmptyIcon
+import icons.StudioIcons
+import javax.swing.Icon
+
+class SqlDelightRunAnnotator : Annotator {
+  override fun annotate(element: PsiElement, holder: AnnotationHolder) {
+    if (element.parent !is SqlStmtList || element !is SqlStmt) return
+
+    val text = element.rawSqlText().trim().replace("\\s+".toRegex(), " ")
+    val s = """
+      |package com.example;
+      |
+      |import android.database.sqlite.SQLiteDatabase;
+      |
+      |class Util {
+      |    void f(SQLiteDatabase db) {
+      |        db.execSQL("$text");
+      |    }
+      |}
+    """.trimMargin()
+
+    val psiFile = PsiFileFactory.getInstance(element.project)
+      .createFileFromText("Util.java", JavaLanguage.INSTANCE, s)
+
+    val host = PsiTreeUtil.findChildOfType(psiFile, PsiLanguageInjectionHost::class.java) ?: return
+
+    if (!PsiTreeUtil.hasErrorElements(psiFile)) {
+      holder.newAnnotation(HighlightSeverity.INFORMATION, "")
+        .gutterIconRenderer(RunSqliteStatementGutterIconRenderer(host))
+        .create()
+    }
+  }
+
+  /**
+   * Shows an icon in the gutter when a SQLite statement is recognized. eg. Room @Query annotations.
+   */
+  private data class RunSqliteStatementGutterIconRenderer(private val element: PsiElement) : GutterIconRenderer() {
+    private val sqliteExplorerProjectService = DatabaseInspectorProjectService.getInstance(element.project)
+
+    override fun getIcon(): Icon {
+      return if (sqliteExplorerProjectService.hasOpenDatabase()) {
+        StudioIcons.DatabaseInspector.NEW_QUERY
+      } else {
+        EmptyIcon.ICON_0
+      }
+    }
+
+    override fun getTooltipText(): String {
+      return "Run Sqlite statement in Database Inspector"
+    }
+
+    override fun isNavigateAction(): Boolean {
+      return sqliteExplorerProjectService.hasOpenDatabase()
+    }
+
+    override fun getClickAction(): AnAction {
+      return RunSqliteStatementGutterIconAction(element.project, element, DatabaseInspectorViewsFactoryImpl.getInstance())
+    }
+  }
+}

sqldelight-idea-plugin/src/main/resources/META-INF/com.squareup.sqldelight-withAndroid.xml

@@ -0,0 +1,6 @@
+<idea-plugin>
+  <extensions defaultExtensionNs="com.intellij">
+    <annotator language="SqlDelight"
+        implementationClass="app.cash.sqldelight.intellij.run.android.SqlDelightRunAnnotator"/>
+  </extensions>
+</idea-plugin>

sqldelight-idea-plugin/src/main/resources/META-INF/plugin.xml

@@ -8,6 +8,7 @@
   <depends>com.intellij.java</depends>
   <depends>org.jetbrains.kotlin</depends>
   <depends>com.intellij.gradle</depends>
+  <depends optional="true" config-file="com.squareup.sqldelight-withAndroid.xml">org.jetbrains.android</depends>
 
   <description><![CDATA[
       Generates typesafe Kotlin APIs from SQL, and provides language