Add auto-select existing languages feature

Author: Airsaid

CHANGELOG.md

@@ -4,6 +4,9 @@
 
 ## [Unreleased]
 
+### Added
+- Auto-select existing languages option in the Select Languages dialog to automatically detect and select languages that already exist in the project.
+
 ### Fixed
 - ExceptionInInitializerError when opening plguin.
 

src/main/kotlin/com/airsaid/localization/action/TranslateAction.kt

@@ -98,6 +98,11 @@ private fun isTranslatable(values: List<PsiElement>, valueService: AndroidValues
 
 private fun showSelectLanguageDialog(project: Project, values: List<PsiElement>, valueFile: PsiFile) {
   val dialog = SelectLanguagesDialog(project)
+
+  // Set resource directory for auto-selecting existing languages
+  val resourceDir = valueFile.virtualFile.parent.parent
+  dialog.setResourceDir(resourceDir)
+
   dialog.setOnClickListener(object : SelectLanguagesDialog.OnClickListener {
     override fun onClickListener(selectedLanguage: List<Lang>) {
       val translationTask = TranslateTask(project, "Translating...", selectedLanguage, values, valueFile)

src/main/kotlin/com/airsaid/localization/constant/Constants.kt

@@ -29,4 +29,5 @@ object Constants {
   const val KEY_FAVORITE_LANGUAGES = "$PLUGIN_ID.favorite_languages"
   const val KEY_IS_OVERWRITE_EXISTING_STRING = "$PLUGIN_ID.is_overwrite_existing_string"
   const val KEY_IS_OPEN_TRANSLATED_FILE = "$PLUGIN_ID.is_open_translated_file"
+  const val KEY_IS_AUTO_SELECT_EXISTING = "$PLUGIN_ID.is_auto_select_existing"
 }

src/main/kotlin/com/airsaid/localization/ui/SelectLanguagesDialog.kt

@@ -52,6 +52,7 @@ import com.airsaid.localization.utils.LanguageUtil
 import com.intellij.ide.util.PropertiesComponent
 import com.intellij.openapi.options.ShowSettingsUtil
 import com.intellij.openapi.project.Project
+import com.intellij.openapi.vfs.VirtualFile
 import java.awt.Dimension
 import java.awt.Toolkit
 import kotlin.math.roundToInt
@@ -84,8 +85,10 @@ class SelectLanguagesDialog(private val project: Project) : ComposeDialog(projec
   private val selectedLanguages = mutableStateListOf<Lang>()
   private val overwriteExistingState = mutableStateOf(false)
   private val openTranslatedFileState = mutableStateOf(false)
+  private val autoSelectExistingState = mutableStateOf(false)
 
   private var stateInitialized by mutableStateOf(false)
+  private var resourceDir: VirtualFile? = null
 
   private var onClickListener: OnClickListener? = null
 
@@ -102,6 +105,15 @@ class SelectLanguagesDialog(private val project: Project) : ComposeDialog(projec
     onClickListener = listener
   }
 
+  /**
+   * Sets the resource directory to enable auto-selecting existing languages.
+   *
+   * @param resourceDir The resource directory containing values folders.
+   */
+  fun setResourceDir(resourceDir: VirtualFile) {
+    this.resourceDir = resourceDir
+  }
+
   @Composable
   override fun Content() {
     LaunchedEffect(Unit) {
@@ -110,6 +122,13 @@ class SelectLanguagesDialog(private val project: Project) : ComposeDialog(projec
       }
     }
 
+    // Monitor changes to auto-select setting
+    LaunchedEffect(autoSelectExistingState.value) {
+      if (stateInitialized && autoSelectExistingState.value) {
+        selectExistingLanguages()
+      }
+    }
+
     if (!stateInitialized) {
       Box(modifier = Modifier.fillMaxSize(), contentAlignment = Alignment.Center) {
         CircularProgressIndicator()
@@ -132,10 +151,13 @@ class SelectLanguagesDialog(private val project: Project) : ComposeDialog(projec
         selectedLanguages = selectedLanguages,
         overwriteExistingChecked = overwriteExistingState.value,
         openTranslatedFileChecked = openTranslatedFileState.value,
+        autoSelectExistingChecked = autoSelectExistingState.value,
+        hasResourceDir = resourceDir != null,
         onSelectAllChanged = { selectAll(languages, it) },
         onFavoriteSelectAllChanged = { selectAll(favoriteLanguages, it) },
         onOverwriteChanged = { checked -> overwriteExistingState.value = checked },
         onOpenTranslatedFileChanged = { checked -> openTranslatedFileState.value = checked },
+        onAutoSelectExistingChanged = { checked -> autoSelectExistingState.value = checked },
         onLanguageToggled = { lang, checked -> selectLanguage(lang, checked) },
         onFavoriteToggle = { lang, isFavorite -> setFavoriteLanguage(lang, isFavorite) },
         onOpenSettings = { openPluginSettings() },
@@ -146,6 +168,7 @@ class SelectLanguagesDialog(private val project: Project) : ComposeDialog(projec
       LanguageUtil.saveSelectedLanguages(project, selectedLanguages)
       properties().setValue(Constants.KEY_IS_OVERWRITE_EXISTING_STRING, overwriteExistingState.value)
       properties().setValue(Constants.KEY_IS_OPEN_TRANSLATED_FILE, openTranslatedFileState.value)
+      properties().setValue(Constants.KEY_IS_AUTO_SELECT_EXISTING, autoSelectExistingState.value)
       onClickListener?.onClickListener(selectedLanguages.toList())
     }
 
@@ -183,6 +206,12 @@ class SelectLanguagesDialog(private val project: Project) : ComposeDialog(projec
 
     overwriteExistingState.value = properties.getBoolean(Constants.KEY_IS_OVERWRITE_EXISTING_STRING)
     openTranslatedFileState.value = properties.getBoolean(Constants.KEY_IS_OPEN_TRANSLATED_FILE)
+    autoSelectExistingState.value = properties.getBoolean(Constants.KEY_IS_AUTO_SELECT_EXISTING)
+
+    // If auto-select existing languages is enabled and resource directory is available, auto-select existing languages
+    if (autoSelectExistingState.value) {
+      selectExistingLanguages()
+    }
 
     stateInitialized = true
   }
@@ -216,6 +245,15 @@ class SelectLanguagesDialog(private val project: Project) : ComposeDialog(projec
     LanguageUtil.saveFavoriteLanguages(project, favoriteLanguages)
   }
 
+  private fun selectExistingLanguages() {
+    resourceDir?.let { resDir ->
+      val existingLanguages = LanguageUtil.getExistingProjectLanguages(resDir, supportedLanguages)
+      existingLanguages.forEach { lang ->
+        selectLanguage(lang, true)
+      }
+    }
+  }
+
   private fun properties(): PropertiesComponent {
     return PropertiesComponent.getInstance(project)
   }
@@ -254,10 +292,13 @@ private fun SelectLanguagesContent(
   selectedLanguages: SnapshotStateList<Lang>,
   overwriteExistingChecked: Boolean,
   openTranslatedFileChecked: Boolean,
+  autoSelectExistingChecked: Boolean,
+  hasResourceDir: Boolean,
   onSelectAllChanged: (Boolean) -> Unit,
   onFavoriteSelectAllChanged: (Boolean) -> Unit,
   onOverwriteChanged: (Boolean) -> Unit,
   onOpenTranslatedFileChanged: (Boolean) -> Unit,
+  onAutoSelectExistingChanged: (Boolean) -> Unit,
   onLanguageToggled: (Lang, Boolean) -> Unit,
   onFavoriteToggle: (Lang, Boolean) -> Unit,
   onOpenSettings: () -> Unit,
@@ -282,10 +323,13 @@ private fun SelectLanguagesContent(
       favoriteSelectAll = favoriteSelectAllChecked,
       overwriteExisting = overwriteExistingChecked,
       openTranslatedFile = openTranslatedFileChecked,
+      autoSelectExisting = autoSelectExistingChecked,
+      hasResourceDir = hasResourceDir,
       onSelectAllChanged = onSelectAllChanged,
       onFavoriteSelectAllChanged = onFavoriteSelectAllChanged,
       onOverwriteChanged = onOverwriteChanged,
       onOpenTranslatedFileChanged = onOpenTranslatedFileChanged,
+      onAutoSelectExistingChanged = onAutoSelectExistingChanged,
       onLanguageToggled = onLanguageToggled,
       onFavoriteToggle = onFavoriteToggle,
       modifier = Modifier.weight(1f, fill = true),
@@ -307,10 +351,13 @@ private fun LanguagesCard(
   favoriteSelectAll: Boolean,
   overwriteExisting: Boolean,
   openTranslatedFile: Boolean,
+  autoSelectExisting: Boolean,
+  hasResourceDir: Boolean,
   onSelectAllChanged: (Boolean) -> Unit,
   onFavoriteSelectAllChanged: (Boolean) -> Unit,
   onOverwriteChanged: (Boolean) -> Unit,
   onOpenTranslatedFileChanged: (Boolean) -> Unit,
+  onAutoSelectExistingChanged: (Boolean) -> Unit,
   onLanguageToggled: (Lang, Boolean) -> Unit,
   onFavoriteToggle: (Lang, Boolean) -> Unit,
   modifier: Modifier = Modifier,
@@ -361,6 +408,9 @@ private fun LanguagesCard(
         onOverwriteChanged = onOverwriteChanged,
         openTranslatedFile = openTranslatedFile,
         onOpenTranslatedFileChanged = onOpenTranslatedFileChanged,
+        autoSelectExisting = autoSelectExisting,
+        hasResourceDir = hasResourceDir,
+        onAutoSelectExistingChanged = onAutoSelectExistingChanged,
       )
 
       OutlinedTextField(
@@ -530,6 +580,9 @@ private fun OptionsSection(
   onOverwriteChanged: (Boolean) -> Unit,
   openTranslatedFile: Boolean,
   onOpenTranslatedFileChanged: (Boolean) -> Unit,
+  autoSelectExisting: Boolean,
+  hasResourceDir: Boolean,
+  onAutoSelectExistingChanged: (Boolean) -> Unit,
 ) {
   FlowRow(
     horizontalArrangement = Arrangement.spacedBy(16.dp),
@@ -547,6 +600,14 @@ private fun OptionsSection(
       checked = openTranslatedFile,
       onCheckedChange = onOpenTranslatedFileChanged,
     )
+    if (hasResourceDir) {
+      OptionItem(
+        text = "Auto-select existing languages",
+        tooltip = "Automatically select languages that already exist in your project when opening this dialog.",
+        checked = autoSelectExisting,
+        onCheckedChange = onAutoSelectExistingChanged,
+      )
+    }
   }
 }
 

src/main/kotlin/com/airsaid/localization/utils/LanguageUtil.kt

@@ -22,6 +22,7 @@ import com.airsaid.localization.translate.lang.Lang
 import com.airsaid.localization.translate.lang.Languages
 import com.intellij.ide.util.PropertiesComponent
 import com.intellij.openapi.project.Project
+import com.intellij.openapi.vfs.VirtualFile
 import org.apache.http.util.TextUtils
 
 /**
@@ -85,6 +86,41 @@ object LanguageUtil {
     return language.joinToString(SEPARATOR_SELECTED_LANGUAGES_CODE) { it.code }
   }
 
+  /**
+   * Get languages that already exist in the project by scanning the resource directories.
+   *
+   * @param resourceDir the resource directory (parent of values directories).
+   * @param supportedLanguages the list of supported languages from translator.
+   * @return list of languages that have corresponding values directories in the project.
+   */
+  fun getExistingProjectLanguages(resourceDir: VirtualFile, supportedLanguages: List<Lang>): List<Lang> {
+    val existingLanguages = mutableListOf<Lang>()
+
+    for (child in resourceDir.children) {
+      if (child.isDirectory) {
+        val dirName = child.name
+
+        if (dirName == "values") {
+          // Default values directory represents the default language (usually English)
+          // We'll look for a language with empty directoryName or "en" code
+          val defaultLang = supportedLanguages.find { it.directoryName.isBlank() || it.code == "en" }
+          defaultLang?.let { existingLanguages.add(it) }
+        } else if (dirName.startsWith("values-")) {
+          val languageCode = dirName.substring(7) // Remove "values-" prefix
+
+          // Find matching language by directory name
+          val matchingLang = supportedLanguages.find { lang ->
+            lang.directoryName == languageCode
+          }
+
+          matchingLang?.let { existingLanguages.add(it) }
+        }
+      }
+    }
+
+    return existingLanguages.distinct()
+  }
+
   private fun parseStoredLanguageCodes(value: String?): List<String> {
     if (value.isNullOrEmpty()) return emptyList()