Add account switcher

api/src/commonMain/graphql/com/materiiapps/gloom/gql/fragments/UserAccount.fragment.graphql

@@ -0,0 +1,9 @@
+fragment UserAccount on User {
+    login
+    name
+    avatarUrl
+    id
+    notificationListsWithThreadCount(statuses: UNREAD) {
+        totalCount
+    }
+}

api/src/commonMain/graphql/com/materiiapps/gloom/gql/queries/AccountInfo.query.graphql

@@ -0,0 +1,5 @@
+query AccountInfo {
+    viewer {
+        ...UserAccount
+    }
+}

api/src/commonMain/graphql/com/materiiapps/gloom/gql/queries/Accounts.query.graphql

@@ -0,0 +1,5 @@
+query Accounts($ids: [ID!]!) {
+    nodes(ids: $ids) {
+        ...UserAccount
+    }
+}

api/src/commonMain/kotlin/com/materiiapps/gloom/api/repository/GraphQLRepository.kt

@@ -11,6 +11,13 @@ class GraphQLRepository(
     private val service: GraphQLService
 ) {
 
+    suspend fun getAccountInfo(token: String) =
+        service.getAccountInfo(token).transform { it.viewer.userAccount }
+
+    suspend fun getAccountsByIds(ids: List<String>) = service.getAccountsByIds(ids).transform {
+        it.nodes.mapNotNull { it?.userAccount }
+    }
+
     suspend fun getCurrentProfile() =
         service.getCurrentProfile().transform { ModelUser.fromProfileQuery(it) }
 

api/src/commonMain/kotlin/com/materiiapps/gloom/api/service/GraphQLService.kt

@@ -4,9 +4,12 @@ import com.apollographql.apollo3.ApolloCall
 import com.apollographql.apollo3.ApolloClient
 import com.apollographql.apollo3.api.Operation
 import com.apollographql.apollo3.api.Optional
+import com.apollographql.apollo3.cache.normalized.doNotStore
 import com.materiiapps.gloom.api.utils.response
 import com.materiiapps.gloom.api.utils.toOptional
 import com.materiiapps.gloom.domain.manager.AuthManager
+import com.materiiapps.gloom.gql.AccountInfoQuery
+import com.materiiapps.gloom.gql.AccountsQuery
 import com.materiiapps.gloom.gql.DefaultBranchQuery
 import com.materiiapps.gloom.gql.FeedQuery
 import com.materiiapps.gloom.gql.FollowUserMutation
@@ -46,6 +49,19 @@ class GraphQLService(
     private fun <D : Operation.Data> ApolloCall<D>.addToken() =
         addHttpHeader(HttpHeaders.Authorization, "Bearer ${authManager.authToken}")
 
+    suspend fun getAccountInfo(token: String) = withContext(Dispatchers.IO) {
+        client.query(AccountInfoQuery())
+            .doNotStore(true)
+            .addHttpHeader(HttpHeaders.Authorization, "Bearer ${token}")
+            .response()
+    }
+
+    suspend fun getAccountsByIds(ids: List<String>) = withContext(Dispatchers.IO) {
+        client.query(AccountsQuery(ids))
+            .addToken()
+            .response()
+    }
+
     suspend fun getCurrentProfile() = withContext(Dispatchers.IO) {
         client.query(ProfileQuery())
             .addToken()

app/src/main/java/com/materiiapps/gloom/ui/GloomActivity.kt

@@ -1,10 +1,10 @@
 package com.materiiapps.gloom.ui
 
+import android.content.Context
 import android.content.Intent
 import android.os.Bundle
 import androidx.activity.ComponentActivity
 import androidx.activity.compose.setContent
-import androidx.compose.animation.ExperimentalAnimationApi
 import androidx.compose.foundation.isSystemInDarkTheme
 import androidx.compose.runtime.CompositionLocalProvider
 import androidx.compose.runtime.SideEffect
@@ -17,6 +17,8 @@ import cafe.adriel.voyager.navigator.Navigator
 import cafe.adriel.voyager.navigator.NavigatorDisposeBehavior
 import com.google.accompanist.systemuicontroller.rememberSystemUiController
 import com.materiiapps.gloom.api.repository.GithubAuthRepository
+import com.materiiapps.gloom.api.repository.GraphQLRepository
+import com.materiiapps.gloom.api.utils.fold
 import com.materiiapps.gloom.api.utils.ifSuccessful
 import com.materiiapps.gloom.domain.manager.AuthManager
 import com.materiiapps.gloom.domain.manager.PreferenceManager
@@ -25,27 +27,31 @@ import com.materiiapps.gloom.ui.screens.auth.LandingScreen
 import com.materiiapps.gloom.ui.screens.root.RootScreen
 import com.materiiapps.gloom.ui.theme.GloomTheme
 import com.materiiapps.gloom.ui.transitions.SlideTransition
+import com.materiiapps.gloom.ui.utils.clearRootNavigation
 import com.materiiapps.gloom.utils.LinkHandler
 import com.materiiapps.gloom.utils.LocalLinkHandler
 import com.materiiapps.gloom.utils.deeplinks.DeepLinkWrapper
 import com.materiiapps.gloom.utils.deeplinks.addAllRoutes
-import kotlinx.coroutines.Dispatchers
+import com.materiiapps.gloom.utils.getOAuthCode
+import com.materiiapps.gloom.utils.isOAuthUri
 import kotlinx.coroutines.launch
+import org.koin.android.ext.android.getKoin
 import org.koin.android.ext.android.inject
 
 class GloomActivity : ComponentActivity() {
 
     private val authRepo: GithubAuthRepository by inject()
+    private val gqlRepo: GraphQLRepository by inject()
     private val auth: AuthManager by inject()
     private val prefs: PreferenceManager by inject()
     private lateinit var navigator: Navigator
 
-    @OptIn(ExperimentalAnimationApi::class)
     override fun onCreate(savedInstanceState: Bundle?) {
         installSplashScreen()
         super.onCreate(savedInstanceState)
 
         WindowCompat.setDecorFitsSystemWindows(window, false)
+        getKoin().declare<Context>(this@GloomActivity)
 
         setContent {
             val isDark = when (prefs.theme) {
@@ -94,20 +100,48 @@ class GloomActivity : ComponentActivity() {
 
     override fun onNewIntent(intent: Intent?) {
         super.onNewIntent(intent)
+        this.intent = intent
+        println("Activity re-entered")
         intent?.let {
-            if (it.data.toString().startsWith("github://com.github.android/oauth")) {
-                println(it.data.toString())
-                it.data!!.getQueryParameter("code")?.let {
-                    lifecycleScope.launch(Dispatchers.IO) {
+            if (it.isOAuthUri()) {
+                if (auth.awaitingAuthType == null) return
+                it.getOAuthCode()?.let {
+                    lifecycleScope.launch {
+                        auth.setAuthState(loading = true)
                         val res = authRepo.getAccessToken(it)
-                        res.ifSuccessful { token ->
-                            auth.authToken = token.accessToken
-                            navigator.replace(RootScreen())
-                        }
+                        res.fold(
+                            success = { token ->
+                                gqlRepo.getAccountInfo(token.accessToken).ifSuccessful { account ->
+                                    auth.addAccount(
+                                        id = account.id,
+                                        token = token.accessToken,
+                                        type = auth.awaitingAuthType!!,
+                                        username = account.login,
+                                        avatarUrl = account.avatarUrl,
+                                        displayName = account.name,
+                                        notificationCount = account.notificationListsWithThreadCount.totalCount
+                                    )
+                                    auth.switchToAccount(account.id)
+                                    clearRootNavigation()
+                                    navigator.replaceAll(RootScreen())
+                                }
+                                auth.setAuthState(authType = null, loading = false)
+                            },
+                            error = { auth.setAuthState(authType = null, loading = false) },
+                            failure = { auth.setAuthState(authType = null, loading = false) },
+                            empty = { auth.setAuthState(authType = null, loading = false) }
+                        )
                     }
                 }
             }
         }
     }
 
+    override fun onResume() {
+        super.onResume()
+        if (!intent.isOAuthUri()) {
+            auth.setAuthState(authType = null)
+        }
+    }
+
 }

app/src/main/java/com/materiiapps/gloom/utils/OAuthUtils.kt

@@ -0,0 +1,10 @@
+package com.materiiapps.gloom.utils
+
+import android.content.Intent
+
+fun Intent.isOAuthUri() = data.toString().startsWith("github://com.github.android/oauth")
+
+fun Intent.getOAuthCode(): String? {
+    if(!isOAuthUri()) return null
+    return data?.getQueryParameter("code")
+}

shared/build.gradle.kts

@@ -3,6 +3,7 @@ plugins {
     alias(libs.plugins.android.library)
     alias(libs.plugins.compose)
     alias(libs.plugins.kotlin.multiplatform)
+    alias(libs.plugins.kotlin.serialization)
 }
 
 android {

shared/src/androidMain/kotlin/com/materiiapps/gloom/utils/CustomTabUtils.kt

@@ -29,7 +29,7 @@ fun Context.openCustomTab(url: String, force: Boolean) = CustomTabsIntent.Builde
     launchUrl(this@openCustomTab, Uri.parse(url))
 }
 
-fun Context.openLink(url: Uri) {
+fun Context.openLink(url: Uri, forceCustomTab: Boolean = false) {
     // TODO: Setting to disable custom tabs
-    openCustomTab(url.toString(), force = false)
+    openCustomTab(url.toString(), force = forceCustomTab)
 }

shared/src/androidMain/kotlin/com/materiiapps/gloom/utils/LinkHandler.kt

@@ -7,9 +7,9 @@ import androidx.core.net.toUri
 
 actual class LinkHandler(private val ctx: Context) {
 
-    actual fun openLink(link: String) {
+    actual fun openLink(link: String, forceCustomTab: Boolean) {
         try {
-            ctx.openLink(link.toUri())
+            ctx.openLink(link.toUri(), forceCustomTab)
         } catch (e: Throwable) {
             Log.e("LinkHandler", "Failed to open link", e)
         }

shared/src/commonMain/kotlin/com/materiiapps/gloom/di/modules/ManagerModule.kt

@@ -1,27 +1,31 @@
 package com.materiiapps.gloom.di.modules
 
+import com.apollographql.apollo3.ApolloClient
 import com.materiiapps.gloom.domain.manager.AuthManager
 import com.materiiapps.gloom.domain.manager.DialogManager
 import com.materiiapps.gloom.domain.manager.DownloadManager
 import com.materiiapps.gloom.domain.manager.PreferenceManager
 import com.materiiapps.gloom.domain.manager.ShareManager
 import com.materiiapps.gloom.domain.manager.ToastManager
+import com.materiiapps.gloom.utils.Logger
 import com.materiiapps.gloom.utils.SettingsProvider
+import kotlinx.serialization.json.Json
 import org.koin.core.module.dsl.singleOf
 import org.koin.core.qualifier.named
 import org.koin.dsl.module
 
 fun managerModule() = module {
 
-    singleOf(::AuthManager)
     singleOf(::DownloadManager)
     singleOf(::ShareManager)
     singleOf(::ToastManager)
 
     fun providePreferenceManager(settings: SettingsProvider) = PreferenceManager(settings)
     fun provideDialogManager(settings: SettingsProvider) = DialogManager(settings)
+    fun provideAuthManager(settings: SettingsProvider, apollo: ApolloClient, json: Json, logger: Logger) = AuthManager(settings, apollo, json, logger)
 
     single { providePreferenceManager(get(named("prefs"))) }
     single { provideDialogManager(get(named("dialogs"))) }
+    single { provideAuthManager(get(named("auth")), get(), get(), get()) }
 
 }

shared/src/commonMain/kotlin/com/materiiapps/gloom/di/modules/SettingsModule.kt

@@ -14,4 +14,8 @@ fun settingsModule() = module {
         SettingsProvider(get(), "dialogs")
     }
 
+    single(named("auth")) {
+        SettingsProvider(get(), "auth")
+    }
+
 }