From b3a140658b715ccb46cd32aa7f9a1131c06ec718 Mon Sep 17 00:00:00 2001
From: tibbi <tibor@kaputa.sk>
Date: Fri, 8 Mar 2019 20:22:07 +0100
Subject: [PATCH] fix #131, handle Edit intent

---
 app/src/main/AndroidManifest.xml              |  6 +++++
 .../draw/pro/activities/MainActivity.kt       | 24 +++++++++++++++----
 2 files changed, 25 insertions(+), 5 deletions(-)

diff --git a/app/src/main/AndroidManifest.xml b/app/src/main/AndroidManifest.xml
index 1fecc91..8849430 100644
--- a/app/src/main/AndroidManifest.xml
+++ b/app/src/main/AndroidManifest.xml
@@ -55,6 +55,12 @@
                 <action android:name="android.media.action.IMAGE_CAPTURE"/>
                 <category android:name="android.intent.category.DEFAULT"/>
             </intent-filter>
+
+            <intent-filter>
+                <action android:name="android.intent.action.EDIT"/>
+                <category android:name="android.intent.category.DEFAULT"/>
+                <data android:mimeType="image/*"/>
+            </intent-filter>
         </activity>
 
         <activity
diff --git a/app/src/main/kotlin/com/simplemobiletools/draw/pro/activities/MainActivity.kt b/app/src/main/kotlin/com/simplemobiletools/draw/pro/activities/MainActivity.kt
index 3a716a6..b703db3 100644
--- a/app/src/main/kotlin/com/simplemobiletools/draw/pro/activities/MainActivity.kt
+++ b/app/src/main/kotlin/com/simplemobiletools/draw/pro/activities/MainActivity.kt
@@ -35,7 +35,6 @@ import java.io.ByteArrayOutputStream
 import java.io.File
 import java.io.OutputStream
 
-
 class MainActivity : SimpleActivity(), CanvasListener {
     private val PICK_IMAGE_INTENT = 1
 
@@ -52,6 +51,7 @@ class MainActivity : SimpleActivity(), CanvasListener {
     private var brushSize = 0f
     private var isEraserOn = false
     private var isImageCaptureIntent = false
+    private var isEditIntent = false
     private var lastBitmapPath = ""
 
     override fun onCreate(savedInstanceState: Bundle?) {
@@ -112,9 +112,10 @@ class MainActivity : SimpleActivity(), CanvasListener {
     override fun onCreateOptionsMenu(menu: Menu): Boolean {
         menuInflater.inflate(R.menu.menu, menu)
         menu.apply {
-            findItem(R.id.menu_confirm).isVisible = isImageCaptureIntent
-            findItem(R.id.menu_save).isVisible = !isImageCaptureIntent
-            findItem(R.id.menu_share).isVisible = !isImageCaptureIntent
+            findItem(R.id.menu_confirm).isVisible = isImageCaptureIntent || isEditIntent
+            findItem(R.id.menu_save).isVisible = !isImageCaptureIntent && !isEditIntent
+            findItem(R.id.menu_share).isVisible = !isImageCaptureIntent && !isEditIntent
+            findItem(R.id.open_file).isVisible = !isEditIntent
         }
 
         return true
@@ -196,6 +197,16 @@ class MainActivity : SimpleActivity(), CanvasListener {
                 invalidateOptionsMenu()
             }
         }
+
+        if (intent?.action == Intent.ACTION_EDIT) {
+            val data = intent.data
+            val output = intent.extras?.get(MediaStore.EXTRA_OUTPUT)
+            if (data != null && output != null && output is Uri) {
+                tryOpenUri(data, intent)
+                isEditIntent = true
+                intentUri = output
+            }
+        }
     }
 
     private fun getStoragePermission(callback: () -> Unit) {
@@ -276,7 +287,10 @@ class MainActivity : SimpleActivity(), CanvasListener {
     }
 
     private fun confirmImage() {
-        if (intentUri?.scheme == "content") {
+        if (isEditIntent) {
+            val outputStream = contentResolver.openOutputStream(intentUri)
+            saveToOutputStream(outputStream, defaultPath.getCompressionFormat())
+        } else if (intentUri?.scheme == "content") {
             val outputStream = contentResolver.openOutputStream(intentUri)
             saveToOutputStream(outputStream, defaultPath.getCompressionFormat())
         } else {