Add ktlint (#213)

* Add ktlint

* Simplify `initClient` function and clean imports in SseIntegrationTest.

Author: devcrocod

.editorconfig

@@ -0,0 +1,37 @@
+root = true
+
+[*]
+charset = utf-8
+end_of_line = lf
+insert_final_newline = true
+indent_style = space
+indent_size = 4
+max_line_length = 120
+
+[*.json]
+indent_size = 2
+
+[{*.yaml,*.yml}]
+indent_size = 2
+
+[*.{kt,kts}]
+ij_kotlin_code_style_defaults = KOTLIN_OFFICIAL
+
+#  Disable wildcard imports entirely
+ij_kotlin_name_count_to_use_star_import = 2147483647
+ij_kotlin_name_count_to_use_star_import_for_members = 2147483647
+ij_kotlin_packages_to_use_import_on_demand = unset
+
+ktlint_code_style = intellij_idea
+ktlint_experimental = enabled
+ktlint_standard_filename = disabled
+ktlint_standard_no-empty-first-line-in-class-body = disabled
+ktlint_class_signature_rule_force_multiline_when_parameter_count_greater_or_equal_than = 4
+ktlint_function_signature_rule_force_multiline_when_parameter_count_greater_or_equal_than = 4
+ktlint_standard_chain-method-continuation = disabled
+ktlint_ignore_back_ticked_identifier = true
+ktlint_standard_multiline-expression-wrapping = disabled
+ktlint_standard_when-entry-bracing = disabled
+
+[*/build/**/*]
+ktlint = disabled

build.gradle.kts

@@ -1,4 +1,12 @@
+plugins {
+    alias(libs.plugins.ktlint)
+}
+
 allprojects {
     group = "io.modelcontextprotocol"
     version = "0.6.0"
-}
+}
+
+subprojects {
+    apply(plugin = "org.jlleitschuh.gradle.ktlint")
+}

gradle/libs.versions.toml

@@ -3,6 +3,7 @@
 kotlin = "2.2.0"
 dokka = "2.0.0"
 atomicfu = "0.29.0"
+ktlint = "13.0.0"
 
 # libraries version
 serialization = "1.9.0"
@@ -57,6 +58,7 @@ ktor-serialization-kotlinx-json = { group = "io.ktor", name = "ktor-serializatio
 
 [plugins]
 kotlinx-binary-compatibility-validator = { id = "org.jetbrains.kotlinx.binary-compatibility-validator", version.ref = "binaryCompatibilityValidatorPlugin" }
+ktlint = { id = "org.jlleitschuh.gradle.ktlint", version.ref = "ktlint" }
 
 # Samples
 kotlin-jvm = { id = "org.jetbrains.kotlin.jvm", version.ref = "kotlin" }

kotlin-sdk-client/build.gradle.kts

@@ -11,9 +11,15 @@ plugins {
 }
 
 kotlin {
-    iosArm64(); iosX64(); iosSimulatorArm64()
-    watchosX64(); watchosArm64(); watchosSimulatorArm64()
-    tvosX64(); tvosArm64(); tvosSimulatorArm64()
+    iosArm64()
+    iosX64()
+    iosSimulatorArm64()
+    watchosX64()
+    watchosArm64()
+    watchosSimulatorArm64()
+    tvosX64()
+    tvosArm64()
+    tvosSimulatorArm64()
     js {
         browser()
         nodejs()
@@ -40,4 +46,4 @@ kotlin {
             }
         }
     }
-}
+}

kotlin-sdk-client/src/commonMain/kotlin/io/modelcontextprotocol/kotlin/sdk/client/Client.kt

@@ -80,10 +80,8 @@ public class ClientOptions(
  * @param clientInfo Information about the client implementation (name, version).
  * @param options Configuration options for this client.
  */
-public open class Client(
-    private val clientInfo: Implementation,
-    options: ClientOptions = ClientOptions(),
-) : Protocol(options) {
+public open class Client(private val clientInfo: Implementation, options: ClientOptions = ClientOptions()) :
+    Protocol(options) {
 
     /**
      * Retrieves the server's reported capabilities after the initialization process completes.
@@ -144,13 +142,13 @@ public open class Client(
             val message = InitializeRequest(
                 protocolVersion = LATEST_PROTOCOL_VERSION,
                 capabilities = capabilities,
-                clientInfo = clientInfo
+                clientInfo = clientInfo,
             )
             val result = request<InitializeResult>(message)
 
             if (!SUPPORTED_PROTOCOL_VERSIONS.contains(result.protocolVersion)) {
                 throw IllegalStateException(
-                    "Server's protocol version is not supported: ${result.protocolVersion}"
+                    "Server's protocol version is not supported: ${result.protocolVersion}",
                 )
             }
 
@@ -165,11 +163,9 @@ public open class Client(
             }
 
             throw error
-
         }
     }
 
-
     override fun assertCapabilityForMethod(method: Method) {
         when (method) {
             Method.Defined.LoggingSetLevel -> {
@@ -181,7 +177,7 @@ public open class Client(
             Method.Defined.PromptsGet,
             Method.Defined.PromptsList,
             Method.Defined.CompletionComplete,
-                -> {
+            -> {
                 if (serverCapabilities?.prompts == null) {
                     throw IllegalStateException("Server does not support prompts (required for $method)")
                 }
@@ -192,28 +188,28 @@ public open class Client(
             Method.Defined.ResourcesRead,
             Method.Defined.ResourcesSubscribe,
             Method.Defined.ResourcesUnsubscribe,
-                -> {
+            -> {
                 val resCaps = serverCapabilities?.resources
                     ?: error("Server does not support resources (required for $method)")
 
                 if (method == Method.Defined.ResourcesSubscribe && resCaps.subscribe != true) {
                     throw IllegalStateException(
-                        "Server does not support resource subscriptions (required for $method)"
+                        "Server does not support resource subscriptions (required for $method)",
                     )
                 }
             }
 
             Method.Defined.ToolsCall,
             Method.Defined.ToolsList,
-                -> {
+            -> {
                 if (serverCapabilities?.tools == null) {
                     throw IllegalStateException("Server does not support tools (required for $method)")
                 }
             }
 
             Method.Defined.Initialize,
             Method.Defined.Ping,
-                -> {
+            -> {
                 // No specific capability required
             }
 
@@ -228,15 +224,15 @@ public open class Client(
             Method.Defined.NotificationsRootsListChanged -> {
                 if (capabilities.roots?.listChanged != true) {
                     throw IllegalStateException(
-                        "Client does not support roots list changed notifications (required for $method)"
+                        "Client does not support roots list changed notifications (required for $method)",
                     )
                 }
             }
 
             Method.Defined.NotificationsInitialized,
             Method.Defined.NotificationsCancelled,
             Method.Defined.NotificationsProgress,
-                -> {
+            -> {
                 // Always allowed
             }
 
@@ -251,23 +247,23 @@ public open class Client(
             Method.Defined.SamplingCreateMessage -> {
                 if (capabilities.sampling == null) {
                     throw IllegalStateException(
-                        "Client does not support sampling capability (required for $method)"
+                        "Client does not support sampling capability (required for $method)",
                     )
                 }
             }
 
             Method.Defined.RootsList -> {
                 if (capabilities.roots == null) {
                     throw IllegalStateException(
-                        "Client does not support roots capability (required for $method)"
+                        "Client does not support roots capability (required for $method)",
                     )
                 }
             }
 
             Method.Defined.ElicitationCreate -> {
                 if (capabilities.elicitation == null) {
                     throw IllegalStateException(
-                        "Client does not support elicitation capability (required for $method)"
+                        "Client does not support elicitation capability (required for $method)",
                     )
                 }
             }
@@ -280,16 +276,14 @@ public open class Client(
         }
     }
 
-
     /**
      * Sends a ping request to the server to check connectivity.
      *
      * @param options Optional request options.
      * @throws IllegalStateException If the server does not support the ping method (unlikely).
      */
-    public suspend fun ping(options: RequestOptions? = null): EmptyRequestResult {
-        return request<EmptyRequestResult>(PingRequest(), options)
-    }
+    public suspend fun ping(options: RequestOptions? = null): EmptyRequestResult =
+        request<EmptyRequestResult>(PingRequest(), options)
 
     /**
      * Sends a completion request to the server, typically to generate or complete some content.
@@ -299,9 +293,8 @@ public open class Client(
      * @return The completion result returned by the server, or `null` if none.
      * @throws IllegalStateException If the server does not support prompts or completion.
      */
-    public suspend fun complete(params: CompleteRequest, options: RequestOptions? = null): CompleteResult {
-        return request(params, options)
-    }
+    public suspend fun complete(params: CompleteRequest, options: RequestOptions? = null): CompleteResult =
+        request(params, options)
 
     /**
      * Sets the logging level on the server.
@@ -310,9 +303,8 @@ public open class Client(
      * @param options Optional request options.
      * @throws IllegalStateException If the server does not support logging.
      */
-    public suspend fun setLoggingLevel(level: LoggingLevel, options: RequestOptions? = null): EmptyRequestResult {
-        return request<EmptyRequestResult>(SetLevelRequest(level), options)
-    }
+    public suspend fun setLoggingLevel(level: LoggingLevel, options: RequestOptions? = null): EmptyRequestResult =
+        request<EmptyRequestResult>(SetLevelRequest(level), options)
 
     /**
      * Retrieves a prompt by name from the server.
@@ -322,9 +314,8 @@ public open class Client(
      * @return The requested prompt details, or `null` if not found.
      * @throws IllegalStateException If the server does not support prompts.
      */
-    public suspend fun getPrompt(request: GetPromptRequest, options: RequestOptions? = null): GetPromptResult {
-        return request(request, options)
-    }
+    public suspend fun getPrompt(request: GetPromptRequest, options: RequestOptions? = null): GetPromptResult =
+        request(request, options)
 
     /**
      * Lists all available prompts from the server.
@@ -337,9 +328,7 @@ public open class Client(
     public suspend fun listPrompts(
         request: ListPromptsRequest = ListPromptsRequest(),
         options: RequestOptions? = null,
-    ): ListPromptsResult {
-        return request(request, options)
-    }
+    ): ListPromptsResult = request(request, options)
 
     /**
      * Lists all available resources from the server.
@@ -352,9 +341,7 @@ public open class Client(
     public suspend fun listResources(
         request: ListResourcesRequest = ListResourcesRequest(),
         options: RequestOptions? = null,
-    ): ListResourcesResult {
-        return request(request, options)
-    }
+    ): ListResourcesResult = request(request, options)
 
     /**
      * Lists resource templates available on the server.
@@ -367,9 +354,7 @@ public open class Client(
     public suspend fun listResourceTemplates(
         request: ListResourceTemplatesRequest,
         options: RequestOptions? = null,
-    ): ListResourceTemplatesResult {
-        return request(request, options)
-    }
+    ): ListResourceTemplatesResult = request(request, options)
 
     /**
      * Reads a resource from the server by its URI.
@@ -382,9 +367,7 @@ public open class Client(
     public suspend fun readResource(
         request: ReadResourceRequest,
         options: RequestOptions? = null,
-    ): ReadResourceResult {
-        return request(request, options)
-    }
+    ): ReadResourceResult = request(request, options)
 
     /**
      * Subscribes to resource changes on the server.
@@ -396,9 +379,7 @@ public open class Client(
     public suspend fun subscribeResource(
         request: SubscribeRequest,
         options: RequestOptions? = null,
-    ): EmptyRequestResult {
-        return request(request, options)
-    }
+    ): EmptyRequestResult = request(request, options)
 
     /**
      * Unsubscribes from resource changes on the server.
@@ -410,9 +391,7 @@ public open class Client(
     public suspend fun unsubscribeResource(
         request: UnsubscribeRequest,
         options: RequestOptions? = null,
-    ): EmptyRequestResult {
-        return request(request, options)
-    }
+    ): EmptyRequestResult = request(request, options)
 
     /**
      * Calls a tool on the server by name, passing the specified arguments.
@@ -443,7 +422,7 @@ public open class Client(
 
         val request = CallToolRequest(
             name = name,
-            arguments = JsonObject(jsonArguments)
+            arguments = JsonObject(jsonArguments),
         )
         return callTool(request, compatibility, options)
     }
@@ -461,12 +440,10 @@ public open class Client(
         request: CallToolRequest,
         compatibility: Boolean = false,
         options: RequestOptions? = null,
-    ): CallToolResultBase? {
-        return if (compatibility) {
-            request<CompatibilityCallToolResult>(request, options)
-        } else {
-            request<CallToolResult>(request, options)
-        }
+    ): CallToolResultBase? = if (compatibility) {
+        request<CompatibilityCallToolResult>(request, options)
+    } else {
+        request<CallToolResult>(request, options)
     }
 
     /**
@@ -480,9 +457,7 @@ public open class Client(
     public suspend fun listTools(
         request: ListToolsRequest = ListToolsRequest(),
         options: RequestOptions? = null,
-    ): ListToolsResult {
-        return request(request, options)
-    }
+    ): ListToolsResult = request(request, options)
 
     /**
      * Registers a single root.
@@ -491,10 +466,7 @@ public open class Client(
      * @param name A human-readable name for the root.
      * @throws IllegalStateException If the client does not support roots.
      */
-    public fun addRoot(
-        uri: String,
-        name: String,
-    ) {
+    public fun addRoot(uri: String, name: String) {
         if (capabilities.roots == null) {
             logger.error { "Failed to add root '$name': Client does not support roots capability" }
             throw IllegalStateException("Client does not support roots capability.")