[feat] Search project by project url and set branch for linting.

Author: Blarc

src/main/kotlin/com/github/blarc/gitlab/template/lint/plugin/GitlabLintRunner.kt

@@ -2,8 +2,6 @@ package com.github.blarc.gitlab.template.lint.plugin
 
 import com.github.blarc.gitlab.template.lint.plugin.gitlab.GitLabFactory
 import com.intellij.dvcs.DvcsUtil
-import com.intellij.notification.NotificationGroupManager
-import com.intellij.notification.NotificationType
 import com.intellij.openapi.components.Service
 import com.intellij.openapi.project.Project
 import git4idea.GitUtil
@@ -19,26 +17,28 @@ class GitlabLintRunner(private val project: Project) {
         val repository = DvcsUtil.guessCurrentRepositoryQuick(project, repositoryManager, filePath)
 
         if (repository != null) {
-            val url = repository.remotes.first().urls[0].toHttpUrlOrNull()
-            val projectName = repository.project.name
-            val gitlab = GitLabFactory.getInstance(project)!!.getGitLab("${url?.scheme}://${url?.host}")
-
             var result: GitlabLintResponse? = null
-            gitlab.searchProject(projectName).thenAccept {
-                val projectId = it[0].id
-                gitlab.lintContent(content, projectId).thenAccept { gitlabLintResponse ->
-                    result = gitlabLintResponse
+            try {
+
+                val url = repository.remotes.first().firstUrl?.toHttpUrlOrNull()
+                val gitlab = GitLabFactory.getInstance(project)!!.getGitLab("${url?.scheme}://${url?.host}")
+
+                gitlab.searchProject(url.toString().removeSuffix(".git")).thenAccept {
+                    val projectId = it.id
+                    val branch = repository.currentBranch!!.name
+                    gitlab.lintContent(content, projectId, branch).thenAccept { gitlabLintResponse ->
+                        result = gitlabLintResponse
+                    }.get()
                 }.get()
-            }.get()
+            }
+            catch (e: Exception) {
+                GitlabLintUtils.createNotification(project, e.localizedMessage)
+            }
 
             return result
         }
 
-        NotificationGroupManager.getInstance()
-            .getNotificationGroup("GitlabLint")
-            .createNotification("Gitlab Lint", "Project is not a gitlab repo.", NotificationType.ERROR)
-            .notify(project)
-
+        GitlabLintUtils.createNotification(project, "Project is not a gitlab repository.")
         return null
     }
 }

src/main/kotlin/com/github/blarc/gitlab/template/lint/plugin/GitlabLintUtils.kt

@@ -0,0 +1,16 @@
+package com.github.blarc.gitlab.template.lint.plugin
+
+import com.intellij.notification.NotificationGroupManager
+import com.intellij.notification.NotificationType
+import com.intellij.openapi.project.Project
+
+class GitlabLintUtils {
+    companion object{
+        fun createNotification(project: Project, message: String, type: NotificationType = NotificationType.ERROR) {
+            NotificationGroupManager.getInstance()
+                .getNotificationGroup("GitlabLint")
+                .createNotification("Gitlab Lint", message, type)
+                .notify(project)
+        }
+    }
+}

src/main/kotlin/com/github/blarc/gitlab/template/lint/plugin/gitlab/Gitlab.kt

@@ -71,13 +71,14 @@ open class GitLab @JvmOverloads constructor(
     }
 
     @OptIn(ExperimentalSerializationApi::class)
-    fun searchProject(projectName: String): CompletableFuture<Array<GitlabProject>> {
+    fun searchProject(projectUrl: String): CompletableFuture<GitlabProject> {
 
+        val projectName = projectUrl.split("/").last()
         val request: Request = prepareRequest("/projects?search=$projectName")
             .get()
             .build()
 
-        val result = CompletableFuture<Array<GitlabProject>>()
+        val result = CompletableFuture<GitlabProject>()
 
         httpClient.newCall(request)
             .enqueue(object: Callback{
@@ -89,8 +90,9 @@ open class GitLab @JvmOverloads constructor(
                     response.use {
                         if (it.isSuccessful) {
                             val responseString = it.body!!.string()
-                            val decodeFromString = json.decodeFromString<Array<GitlabProject>>(responseString)
-                            result.complete(decodeFromString)
+                            val gitlabProjects = json.decodeFromString<Array<GitlabProject>>(responseString)
+                            val gitlabProject = gitlabProjects.find { gitlabProject -> gitlabProject.webUrl.equals(projectUrl, true) }
+                            result.complete(gitlabProject)
                         }
                         else {
                             result.completeExceptionally(RuntimeException(it.body?.string() ?: "Request was unsuccessful!"))
@@ -103,10 +105,12 @@ open class GitLab @JvmOverloads constructor(
     }
 
     @OptIn(ExperimentalSerializationApi::class)
-    fun lintContent(content: String, projectId: Long): CompletableFuture<GitlabLintResponse> {
+    fun lintContent(content: String, projectId: Long, branch: String): CompletableFuture<GitlabLintResponse> {
 
         val formBody = FormBody.Builder()
             .add("content", content)
+            .add("ref", branch)
+            .add("dry_run", "true")
             .build()
 
         val request = prepareRequest("/projects/${projectId}/ci/lint")

src/main/kotlin/com/github/blarc/gitlab/template/lint/plugin/gitlab/GitlabProject.kt

@@ -1,6 +1,11 @@
 package com.github.blarc.gitlab.template.lint.plugin.gitlab
 
-@kotlinx.serialization.Serializable
+import kotlinx.serialization.SerialName
+import kotlinx.serialization.Serializable
+
+@Serializable
 data class GitlabProject(
-    val id: Long
+    val id: Long,
+    @SerialName("web_url")
+    val webUrl: String
 )