task: handle different base URLs dependent on router

Author: jwilkinson87

src/main/kotlin/com/ctrlhub/core/Api.kt

@@ -3,6 +3,12 @@ package com.ctrlhub.core
 import com.ctrlhub.core.api.KtorApiClient
 import com.ctrlhub.core.router.AuthRouter
 
-class Api(val apiClient: KtorApiClient) {
+class Api private constructor(apiClient: KtorApiClient) {
     val auth: AuthRouter = AuthRouter(apiClient)
+
+    companion object {
+        fun create(): Api {
+            return Api(KtorApiClient.create(Config.apiBaseUrl))
+        }
+    }
 }

src/main/kotlin/com/ctrlhub/core/Config.kt

@@ -0,0 +1,22 @@
+package com.ctrlhub.core
+
+enum class Environment {
+    STAGING,
+    PRODUCTION
+}
+
+object Config {
+    var environment: Environment = Environment.PRODUCTION
+
+    val authBaseUrl: String
+        get() = when (environment) {
+            Environment.STAGING -> "auth.ctrl-hub.dev"
+            Environment.PRODUCTION -> "auth.ctrl-hub.com"
+        }
+
+    val apiBaseUrl: String
+        get() = when (environment) {
+            Environment.STAGING -> "api.ctrl-hub.dev"
+            Environment.PRODUCTION -> "api.ctrl-hub.com"
+        }
+}

src/main/kotlin/com/ctrlhub/core/api/ApiException.kt

@@ -1,4 +1,11 @@
 package com.ctrlhub.core.api
 
-// TODO - flesh this out more with different types of exceptions
-class ApiException(message: String, e: Throwable): Exception(message, e)
+import io.ktor.client.statement.HttpResponse
+
+open class ApiException(message: String, e: Throwable) : Exception(message, e)
+
+class ApiClientException(message: String, val response: HttpResponse, e: Throwable) : ApiException(message, e) {
+    fun statusCode(): Int {
+        return response.status.value
+    }
+}

src/main/kotlin/com/ctrlhub/core/router/AuthRouter.kt

@@ -1,24 +1,31 @@
 package com.ctrlhub.core.router
 
+import com.ctrlhub.core.Config
+import com.ctrlhub.core.api.ApiClientException
 import com.ctrlhub.core.api.ApiException
 import com.ctrlhub.core.api.KtorApiClient
 import com.ctrlhub.core.api.payload.auth.LoginPayload
 import com.ctrlhub.core.api.response.auth.AuthFlowResponse
 import com.ctrlhub.core.api.response.auth.CompleteResponse
 import io.ktor.client.call.*
+import io.ktor.client.plugins.ClientRequestException
 
-class AuthRouter(apiClient: KtorApiClient) : Router(apiClient) {
+class AuthRouter(apiClient: KtorApiClient) : Router(apiClient = apiClient) {
     suspend fun initiate(): AuthFlowResponse {
         return try {
-            return apiClient.get(url = "self-service/login/api").body()
+            return apiClient.get(url = "${Config.authBaseUrl}/self-service/login/api").body()
+        } catch (e: ClientRequestException) {
+            throw ApiClientException("Failed to initiate auth", e.response, e)
         } catch (e: Exception) {
             throw ApiException("Failed to initiate auth", e)
         }
     }
 
     suspend fun complete(flowId: String, payload: LoginPayload): CompleteResponse {
         return try {
-            apiClient.post(url = "self-service/login?flow=$flowId", body = payload).body()
+            apiClient.post(url = "${Config.authBaseUrl}/self-service/login?flow=$flowId", body = payload).body()
+        } catch (e: ClientRequestException) {
+            throw ApiClientException("Failed to complete auth", e.response, e)
         } catch (e: Exception) {
             throw ApiException("Failed to complete auth", e)
         }

src/main/kotlin/com/ctrlhub/core/router/Router.kt

@@ -1,7 +1,6 @@
 package com.ctrlhub.core.router
 
+import com.ctrlhub.core.Config
 import com.ctrlhub.core.api.KtorApiClient
 
-abstract class Router(
-    protected val apiClient: KtorApiClient
-)
+abstract class Router(val apiClient: KtorApiClient)

src/test/kotlin/com/ctrlhub/core/ConfigTest.kt

@@ -0,0 +1,41 @@
+package com.ctrlhub.core
+
+import kotlin.test.AfterTest
+import kotlin.test.Test
+import kotlin.test.assertEquals
+
+class ConfigTest {
+    @AfterTest
+    fun tearDown() {
+        // Reset the environment to its default after each test
+        Config.environment = Environment.PRODUCTION
+    }
+
+    @Test
+    fun `authBaseUrl should return correct value for PRODUCTION`() {
+        Config.environment = Environment.PRODUCTION
+
+        assertEquals("auth.ctrl-hub.com", Config.authBaseUrl)
+    }
+
+    @Test
+    fun `authBaseUrl should return correct value for STAGING`() {
+        Config.environment = Environment.STAGING
+
+        assertEquals("auth.ctrl-hub.dev", Config.authBaseUrl)
+    }
+
+    @Test
+    fun `apiBaseUrl should return correct value for PRODUCTION`() {
+        Config.environment = Environment.PRODUCTION
+
+        assertEquals("api.ctrl-hub.com", Config.apiBaseUrl)
+    }
+
+    @Test
+    fun `apiBaseUrl should return correct value for STAGING`() {
+        Config.environment = Environment.STAGING
+
+        assertEquals("api.ctrl-hub.dev", Config.apiBaseUrl)
+    }
+}