app: add viewmodel and fragment to import gpg keys

Signed-off-by: Aditya Wasan <adityawasan55@gmail.com>

Author: Skrilltrax

.idea/deploymentTargetDropDown.xml

@@ -7,8 +7,10 @@ package dev.msfjarvis.aps.ui.onboarding.activity
 
 import android.os.Bundle
 import androidx.appcompat.app.AppCompatActivity
+import dagger.hilt.android.AndroidEntryPoint
 import dev.msfjarvis.aps.R
 
+@AndroidEntryPoint
 class OnboardingActivity : AppCompatActivity(R.layout.activity_onboarding) {
 
   override fun onCreate(savedInstanceState: Bundle?) {

app/src/main/java/dev/msfjarvis/aps/ui/onboarding/activity/OnboardingActivity.kt

@@ -0,0 +1,77 @@
+/*
+ * Copyright © 2014-2021 The Android Password Store Authors. All Rights Reserved.
+ * SPDX-License-Identifier: GPL-3.0-only
+ */
+
+package dev.msfjarvis.aps.ui.onboarding.fragments
+
+import android.os.Bundle
+import android.view.View
+import android.widget.Toast
+import androidx.activity.result.contract.ActivityResultContracts
+import androidx.core.content.edit
+import androidx.fragment.app.Fragment
+import androidx.fragment.app.viewModels
+import com.github.michaelbull.result.onFailure
+import com.github.michaelbull.result.onSuccess
+import dagger.hilt.android.AndroidEntryPoint
+import dev.msfjarvis.aps.R
+import dev.msfjarvis.aps.databinding.FragmentKeySelectionBinding
+import dev.msfjarvis.aps.ui.onboarding.viewmodel.KeySelectionViewModel
+import dev.msfjarvis.aps.util.extensions.commitChange
+import dev.msfjarvis.aps.util.extensions.finish
+import dev.msfjarvis.aps.util.extensions.sharedPrefs
+import dev.msfjarvis.aps.util.extensions.unsafeLazy
+import dev.msfjarvis.aps.util.extensions.viewBinding
+import dev.msfjarvis.aps.util.settings.PreferenceKeys
+import kotlinx.coroutines.flow.onEach
+
+@AndroidEntryPoint
+class GopenpgpKeySelectionFragment : Fragment(R.layout.fragment_key_selection) {
+
+  private val settings by unsafeLazy { requireActivity().applicationContext.sharedPrefs }
+  private val binding by viewBinding(FragmentKeySelectionBinding::bind)
+  private val viewModel: KeySelectionViewModel by viewModels()
+
+  private val gpgKeySelectAction = registerForActivityResult(ActivityResultContracts.GetContent()) { uri ->
+    if (uri == null) {
+      // TODO: Use string resources here
+      showError("No files chosen")
+      return@registerForActivityResult
+    }
+
+    val fis = requireContext().contentResolver.openInputStream(uri)
+    if (fis == null) {
+      showError("Error resolving content uri")
+      return@registerForActivityResult
+    }
+
+    viewModel.importKey(fis)
+  }
+
+  override fun onViewCreated(view: View, savedInstanceState: Bundle?) {
+    super.onViewCreated(view, savedInstanceState)
+    binding.selectKey.setOnClickListener {
+      gpgKeySelectAction.launch("*/*")
+    }
+
+    viewModel.importKeyStatus.onEach { result ->
+      result.onSuccess {
+        settings.edit { putBoolean(PreferenceKeys.REPOSITORY_INITIALIZED, true) }
+        requireActivity().commitChange(getString(R.string.git_commit_gpg_id, getString(R.string.app_name)))
+        finish()
+      }.onFailure {
+        showError(it.message!!)
+      }
+    }
+  }
+
+  private fun showError(message: String) {
+    Toast.makeText(requireContext(), message, Toast.LENGTH_LONG).show()
+  }
+
+  companion object {
+
+    fun newInstance() = GopenpgpKeySelectionFragment()
+  }
+}

app/src/main/java/dev/msfjarvis/aps/ui/onboarding/fragments/GopenpgpKeySelectionFragment.kt

@@ -157,7 +157,7 @@ class RepoLocationFragment : Fragment(R.layout.fragment_repo_location) {
       if (!PasswordRepository.isInitialized) {
         PasswordRepository.initialize()
       }
-      parentFragmentManager.performTransactionWithBackStack(KeySelectionFragment.newInstance())
+      parentFragmentManager.performTransactionWithBackStack(GopenpgpKeySelectionFragment.newInstance())
     }
       .onFailure { e ->
         e(e)

app/src/main/java/dev/msfjarvis/aps/ui/onboarding/fragments/RepoLocationFragment.kt

@@ -0,0 +1,67 @@
+package dev.msfjarvis.aps.ui.onboarding.viewmodel
+
+import androidx.lifecycle.ViewModel
+import androidx.lifecycle.viewModelScope
+import com.github.michaelbull.result.Err
+import com.github.michaelbull.result.Ok
+import com.github.michaelbull.result.Result
+import com.github.michaelbull.result.map
+import com.github.michaelbull.result.onFailure
+import dagger.hilt.android.lifecycle.HiltViewModel
+import dev.msfjarvis.aps.data.crypto.KeyManager
+import dev.msfjarvis.aps.data.repo.PasswordRepository
+import java.io.File
+import java.io.InputStream
+import javax.inject.Inject
+import kotlinx.coroutines.Dispatchers
+import kotlinx.coroutines.flow.MutableSharedFlow
+import kotlinx.coroutines.flow.asSharedFlow
+import kotlinx.coroutines.launch
+import kotlinx.coroutines.withContext
+
+@HiltViewModel
+class KeySelectionViewModel @Inject constructor(private val keyManager: KeyManager) : ViewModel() {
+
+  private val _importKeyStatus = MutableSharedFlow<Result<Unit, Throwable>>()
+  val importKeyStatus = _importKeyStatus.asSharedFlow()
+  fun importKey(keyInputStream: InputStream) {
+    viewModelScope.launch {
+      val lines = keyInputStream.bufferedReader().readLines()
+
+      // Validate the incoming InputStream
+      if (!validateKey(lines)) {
+        _importKeyStatus.emit(Err<Throwable>(IllegalStateException("Selected file does not appear to be an GPG private key.")))
+        return@launch
+      }
+      // Join InputStream and add key to KeyManager
+      val fileContent = lines.joinToString("\n")
+      keyManager.addKey(fileContent).onFailure { throwable ->
+        _importKeyStatus.emit(Err(throwable))
+        return@launch
+      }
+      // Create `.gpg-id` file
+      createGpgIdFile().onFailure { throwable ->
+        _importKeyStatus.emit(Err(throwable))
+        return@launch
+      }
+
+      _importKeyStatus.emit(Ok(Unit))
+    }
+  }
+
+  private fun validateKey(lines: List<String>): Boolean {
+    // The file must have more than 2 lines, and the first and last line must have private key
+    // markers.
+    return (lines.size > 2 ||
+      Regex("BEGIN .* PRIVATE KEY").containsMatchIn(lines.first()) &&
+      Regex("END .* PRIVATE KEY").containsMatchIn(lines.last())
+      )
+  }
+
+  private suspend fun createGpgIdFile(): Result<Unit, Throwable> = withContext(Dispatchers.IO) {
+    return@withContext keyManager.listKeyIds().map { keys ->
+      val idFile = File(PasswordRepository.getRepositoryDirectory(), ".gpg-id")
+      idFile.writeText((keys + "").joinToString("\n"))
+    }
+  }
+}