fix: reload JDK distributions when returning from terminal to onboarding activity

Author: itsaky

app/src/main/java/com/itsaky/androidide/activities/OnboardingActivity.kt

@@ -19,6 +19,8 @@ package com.itsaky.androidide.activities
 
 import android.content.Intent
 import android.os.Bundle
+import android.util.Log
+import androidx.activity.result.contract.ActivityResultContracts
 import androidx.core.content.ContextCompat
 import androidx.fragment.app.Fragment
 import com.github.appintro.AppIntro2
@@ -32,18 +34,42 @@ import com.itsaky.androidide.fragments.onboarding.IdeSetupConfigurationFragment
 import com.itsaky.androidide.fragments.onboarding.OnboardingInfoFragment
 import com.itsaky.androidide.fragments.onboarding.PermissionsFragment
 import com.itsaky.androidide.fragments.onboarding.StatisticsFragment
+import com.itsaky.androidide.models.JdkDistribution
 import com.itsaky.androidide.preferences.internal.prefManager
 import com.itsaky.androidide.preferences.internal.statConsentDialogShown
+import com.itsaky.androidide.tasks.launchAsyncWithProgress
 import com.itsaky.androidide.ui.themes.IThemeManager
 import com.itsaky.androidide.utils.Environment
 import com.termux.shared.android.PackageUtils
 import com.termux.shared.markdown.MarkdownUtils
 import com.termux.shared.termux.TermuxConstants
+import kotlinx.coroutines.CoroutineName
+import kotlinx.coroutines.CoroutineScope
+import kotlinx.coroutines.Dispatchers
+import kotlinx.coroutines.Job
+import kotlinx.coroutines.cancel
+import kotlinx.coroutines.withContext
 
 class OnboardingActivity : AppIntro2() {
 
+  private val terminalActivityCallback = registerForActivityResult(
+    ActivityResultContracts.StartActivityForResult()) {
+    Log.d(TAG, "TerminalActivity: resultCode=${it.resultCode}")
+    if (!isFinishing) {
+      reloadJdkDistInfo {
+        tryNavigateToMainIfSetupIsCompleted()
+      }
+    }
+  }
+
+  private val activityScope =
+    CoroutineScope(Dispatchers.Main + CoroutineName("OnboardingActivity"))
+
+  private var listJdkInstallationsJob: Job? = null
+
   companion object {
 
+    private const val TAG = "OnboardingActivity"
     private const val KEY_ARCHCONFIG_WARNING_IS_SHOWN = "ide.archConfig.experimentalWarning.isShown"
   }
 
@@ -52,9 +78,8 @@ class OnboardingActivity : AppIntro2() {
 
     super.onCreate(savedInstanceState)
 
-    if (isSetupDone()) {
-      openActivity(MainActivity::class.java)
-      finish()
+    if (tryNavigateToMainIfSetupIsCompleted()) {
+      return
     }
 
     setSwipeLock(true)
@@ -110,12 +135,16 @@ class OnboardingActivity : AppIntro2() {
 
   override fun onResume() {
     super.onResume()
-    if (isSetupDone()) {
-      openActivity(MainActivity::class.java)
-      finish()
+    reloadJdkDistInfo {
+      tryNavigateToMainIfSetupIsCompleted()
     }
   }
 
+  override fun onDestroy() {
+    super.onDestroy()
+    activityScope.cancel("Activity is being destroyed")
+  }
+
   override fun onNextPressed(currentFragment: Fragment?) {
     (currentFragment as? StatisticsFragment?)?.updateStatOptInStatus()
   }
@@ -129,31 +158,58 @@ class OnboardingActivity : AppIntro2() {
     }
 
     if (!checkToolsIsInstalled() && currentFragment is IdeSetupConfigurationFragment) {
-      val intenet = Intent(this, TerminalActivity::class.java)
+      val intent = Intent(this, TerminalActivity::class.java)
       if (currentFragment.isAutoInstall()) {
-        intenet.putExtra(TerminalActivity.EXTRA_ONBOARDING_RUN_IDESETUP, true)
-        intenet.putExtra(TerminalActivity.EXTRA_ONBOARDING_RUN_IDESETUP_ARGS,
+        intent.putExtra(TerminalActivity.EXTRA_ONBOARDING_RUN_IDESETUP, true)
+        intent.putExtra(TerminalActivity.EXTRA_ONBOARDING_RUN_IDESETUP_ARGS,
           currentFragment.buildIdeSetupArguments())
       }
-      startActivity(intenet)
+      terminalActivityCallback.launch(intent)
       return
     }
 
-    openActivity(MainActivity::class.java)
-  }
-
-  private fun openActivity(cls: Class<*>) {
-    startActivity(Intent(this, cls))
+    tryNavigateToMainIfSetupIsCompleted()
   }
 
   private fun checkToolsIsInstalled(): Boolean {
     return IJdkDistributionProvider.getInstance().installedDistributions.isNotEmpty()
         && Environment.ANDROID_HOME.exists()
   }
 
-  private fun isSetupDone() =
-    (checkToolsIsInstalled() && statConsentDialogShown && PermissionsFragment.areAllPermissionsGranted(
-      this))
+  private fun isSetupCompleted(): Boolean {
+    return checkToolsIsInstalled()
+        && statConsentDialogShown
+        && PermissionsFragment.areAllPermissionsGranted(this)
+  }
+
+  private fun tryNavigateToMainIfSetupIsCompleted(): Boolean {
+    if (isSetupCompleted()) {
+      startActivity(Intent(this, MainActivity::class.java))
+      finish()
+      return true
+    }
+
+    return false
+  }
+
+  private inline fun reloadJdkDistInfo(crossinline distConsumer: (List<JdkDistribution>) -> Unit) {
+    listJdkInstallationsJob?.cancel("Reloading JDK distributions")
+
+    listJdkInstallationsJob = activityScope.launchAsyncWithProgress(Dispatchers.Default,
+      configureFlashbar = { builder, _ ->
+        builder.message(string.please_wait)
+      }) { _, _ ->
+      val distributionProvider = IJdkDistributionProvider.getInstance()
+      distributionProvider.loadDistributions()
+      withContext(Dispatchers.Main) {
+        distConsumer(distributionProvider.installedDistributions)
+      }
+    }.also {
+      it?.invokeOnCompletion {
+        listJdkInstallationsJob = null
+      }
+    }
+  }
 
   private fun isInstalledOnSdCard(): Boolean {
     // noinspection SdCardPath

app/src/main/java/com/itsaky/androidide/app/configuration/JdkDistributionProviderImpl.kt

@@ -36,8 +36,17 @@ class JdkDistributionProviderImpl : IJdkDistributionProvider {
     private val log = ILogger.newInstance("JdkDistributionProviderImpl")
   }
 
-  override val installedDistributions: List<JdkDistribution> by lazy {
-    JdkUtils.findJavaInstallations().also { distributions ->
+  private var _installedDistributions: List<JdkDistribution>? = null
+
+  override val installedDistributions: List<JdkDistribution>
+    get() = _installedDistributions ?: emptyList()
+
+  override fun loadDistributions() {
+    _installedDistributions = doLoadDistributions()
+  }
+
+  private fun doLoadDistributions(): List<JdkDistribution> {
+    return JdkUtils.findJavaInstallations().also { distributions ->
 
       // set the default value for the 'javaHome' preference
       if (javaHome.isBlank() && distributions.isNotEmpty()) {

common/src/main/java/com/itsaky/androidide/app/configuration/IJdkDistributionProvider.kt

@@ -17,6 +17,7 @@
 
 package com.itsaky.androidide.app.configuration
 
+import androidx.annotation.WorkerThread
 import com.itsaky.androidide.models.JdkDistribution
 import com.itsaky.androidide.utils.ServiceLoader
 
@@ -32,6 +33,13 @@ interface IJdkDistributionProvider {
    */
   val installedDistributions: List<JdkDistribution>
 
+  /**
+   * Reloads the installed JDK distributions. This function is synchronous and should not be called
+   * on the UI thread.
+   */
+  @WorkerThread
+  fun loadDistributions()
+
   /**
    * Get the [JdkDistribution] instance for the given java version.
    *

common/src/main/java/com/itsaky/androidide/managers/ToolsManager.java

@@ -53,7 +53,7 @@ public static void init(@NonNull BaseApplication app, Runnable onFinish) {
 
     CompletableFuture.runAsync(() -> {
       // Load installed JDK distributions
-      IJdkDistributionProvider.getInstance().getInstalledDistributions();
+      IJdkDistributionProvider.getInstance().loadDistributions();
 
       writeNoMediaFile();
       extractAapt2();

termux/termux-app/src/main/java/com/itsaky/androidide/activities/TerminalActivity.kt

@@ -26,12 +26,14 @@ import android.view.View
 import android.widget.Button
 import androidx.core.content.ContextCompat
 import androidx.core.view.WindowCompat
+import com.itsaky.androidide.terminal.IdeTerminalSessionClient
 import com.itsaky.androidide.terminal.IdesetupSession
 import com.itsaky.androidide.utils.Environment
 import com.itsaky.androidide.utils.ILogger
 import com.itsaky.androidide.utils.flashError
 import com.termux.R
 import com.termux.app.TermuxActivity
+import com.termux.app.terminal.TermuxTerminalSessionActivityClient
 import com.termux.shared.termux.shell.command.runner.terminal.TermuxSession
 
 /**
@@ -70,6 +72,10 @@ class TerminalActivity : TermuxActivity() {
       KEY_TERMINAL_CAN_ADD_SESSIONS, true) ?: true
   }
 
+  override fun onCreateTerminalSessionClient(): TermuxTerminalSessionActivityClient {
+    return IdeTerminalSessionClient(this)
+  }
+
   override fun onSaveInstanceState(savedInstanceState: Bundle) {
     super.onSaveInstanceState(savedInstanceState)
     savedInstanceState.putBoolean(KEY_TERMINAL_CAN_ADD_SESSIONS, canAddNewSessions)

termux/termux-app/src/main/java/com/itsaky/androidide/terminal/IdeTerminalSessionClient.kt

@@ -0,0 +1,45 @@
+/*
+ *  This file is part of AndroidIDE.
+ *
+ *  AndroidIDE is free software: you can redistribute it and/or modify
+ *  it under the terms of the GNU General Public License as published by
+ *  the Free Software Foundation, either version 3 of the License, or
+ *  (at your option) any later version.
+ *
+ *  AndroidIDE is distributed in the hope that it will be useful,
+ *  but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ *  GNU General Public License for more details.
+ *
+ *  You should have received a copy of the GNU General Public License
+ *   along with AndroidIDE.  If not, see <https://www.gnu.org/licenses/>.
+ */
+
+package com.itsaky.androidide.terminal
+
+import com.itsaky.androidide.activities.TerminalActivity
+import com.termux.app.terminal.TermuxTerminalSessionActivityClient
+import com.termux.terminal.TerminalSession
+import com.termux.terminal.TerminalSessionClient
+
+/**
+ * [TerminalSessionClient] delegate for AndroidIDE.
+ *
+ * @author Akash Yadav
+ */
+class IdeTerminalSessionClient(
+  activity: TerminalActivity
+) : TermuxTerminalSessionActivityClient(activity) {
+
+  override fun onSessionFinished(finishedSession: TerminalSession) {
+    val termuxSession = mActivity?.termuxService?.getTermuxSessionForTerminalSession(
+      finishedSession)
+    if (termuxSession != null && termuxSession is IdesetupSession) {
+      // if the finished session was performing tools installation
+      // then set the result code for the installation process
+      mActivity.setResult(finishedSession.exitStatus)
+    }
+
+    super.onSessionFinished(finishedSession)
+  }
+}