From 7cdaf988f29adca5ff3e5f4af9df6213c7e31d99 Mon Sep 17 00:00:00 2001
From: Alex Hart <alex@signal.org>
Date: Thu, 13 Oct 2022 10:21:00 -0300
Subject: [PATCH] Allow users to save text stories.

---
 .../stories/dialogs/StoryContextMenu.kt       | 30 +++++++++++++++++++
 .../securesms/stories/my/MyStoriesItem.kt     |  1 -
 .../securesms/util/BitmapUtil.java            |  2 +-
 3 files changed, 31 insertions(+), 2 deletions(-)

diff --git a/app/src/main/java/org/thoughtcrime/securesms/stories/dialogs/StoryContextMenu.kt b/app/src/main/java/org/thoughtcrime/securesms/stories/dialogs/StoryContextMenu.kt
index 00446a7258..e6f401ec15 100644
--- a/app/src/main/java/org/thoughtcrime/securesms/stories/dialogs/StoryContextMenu.kt
+++ b/app/src/main/java/org/thoughtcrime/securesms/stories/dialogs/StoryContextMenu.kt
@@ -10,8 +10,10 @@ import android.view.ViewGroup
 import android.widget.Toast
 import androidx.core.app.ShareCompat
 import androidx.fragment.app.Fragment
+import com.bumptech.glide.load.Options
 import io.reactivex.rxjava3.core.Single
 import org.signal.core.util.DimensionUnit
+import org.signal.core.util.concurrent.SimpleTask
 import org.signal.core.util.logging.Log
 import org.thoughtcrime.securesms.R
 import org.thoughtcrime.securesms.attachments.Attachment
@@ -20,12 +22,17 @@ import org.thoughtcrime.securesms.components.menu.SignalContextMenu
 import org.thoughtcrime.securesms.database.model.MediaMmsMessageRecord
 import org.thoughtcrime.securesms.database.model.MessageRecord
 import org.thoughtcrime.securesms.database.model.databaseprotos.StoryTextPost
+import org.thoughtcrime.securesms.providers.BlobProvider
+import org.thoughtcrime.securesms.stories.StoryTextPostModel
 import org.thoughtcrime.securesms.stories.landing.StoriesLandingItem
 import org.thoughtcrime.securesms.stories.viewer.page.StoryPost
 import org.thoughtcrime.securesms.stories.viewer.page.StoryViewerPageState
 import org.thoughtcrime.securesms.util.Base64
+import org.thoughtcrime.securesms.util.BitmapUtil
 import org.thoughtcrime.securesms.util.DeleteDialog
+import org.thoughtcrime.securesms.util.MediaUtil
 import org.thoughtcrime.securesms.util.SaveAttachmentTask
+import java.io.ByteArrayInputStream
 
 object StoryContextMenu {
 
@@ -45,6 +52,29 @@ object StoryContextMenu {
     val mediaMessageRecord = messageRecord as? MediaMmsMessageRecord
     val uri: Uri? = mediaMessageRecord?.slideDeck?.firstSlide?.uri
     val contentType: String? = mediaMessageRecord?.slideDeck?.firstSlide?.contentType
+
+    if (mediaMessageRecord?.storyType?.isTextStory == true) {
+      SimpleTask.run({
+        val model = StoryTextPostModel.parseFrom(messageRecord)
+        val decoder = StoryTextPostModel.Decoder()
+        val bitmap = decoder.decode(model, 1080, 1920, Options()).get()
+        val jpeg: ByteArrayInputStream = BitmapUtil.toCompressedJpeg(bitmap)
+
+        bitmap.recycle()
+
+        SaveAttachmentTask.Attachment(
+          BlobProvider.getInstance().forData(jpeg.readBytes()).createForSingleUseInMemory(),
+          MediaUtil.IMAGE_JPEG,
+          mediaMessageRecord.dateSent,
+          null
+        )
+      }, { saveAttachment ->
+        SaveAttachmentTask(context)
+          .executeOnExecutor(AsyncTask.THREAD_POOL_EXECUTOR, saveAttachment)
+      })
+      return
+    }
+
     if (uri == null || contentType == null) {
       Log.w(TAG, "Unable to save story media uri: $uri contentType: $contentType")
       Toast.makeText(context, R.string.MyStories__unable_to_save, Toast.LENGTH_SHORT).show()
diff --git a/app/src/main/java/org/thoughtcrime/securesms/stories/my/MyStoriesItem.kt b/app/src/main/java/org/thoughtcrime/securesms/stories/my/MyStoriesItem.kt
index fff8be46c1..86278ebf6c 100644
--- a/app/src/main/java/org/thoughtcrime/securesms/stories/my/MyStoriesItem.kt
+++ b/app/src/main/java/org/thoughtcrime/securesms/stories/my/MyStoriesItem.kt
@@ -186,7 +186,6 @@ object MyStoriesItem {
         .show(
           listOf(
             ActionItem(R.drawable.ic_delete_24_tinted, context.getString(R.string.delete)) { model.onDeleteClick(model) },
-            ActionItem(R.drawable.ic_download_24_tinted, context.getString(R.string.save)) { model.onSaveClick(model) },
             ActionItem(R.drawable.ic_forward_24_tinted, context.getString(R.string.MyStories_forward)) { model.onForwardClick(model) },
             ActionItem(R.drawable.ic_share_24_tinted, context.getString(R.string.StoriesLandingItem__share)) { model.onShareClick(model) },
             ActionItem(R.drawable.ic_info_outline_message_details_24, context.getString(R.string.StoriesLandingItem__info)) { model.onInfoClick(model, storyPreview) }
diff --git a/app/src/main/java/org/thoughtcrime/securesms/util/BitmapUtil.java b/app/src/main/java/org/thoughtcrime/securesms/util/BitmapUtil.java
index 72cebbec4c..e59567e66a 100644
--- a/app/src/main/java/org/thoughtcrime/securesms/util/BitmapUtil.java
+++ b/app/src/main/java/org/thoughtcrime/securesms/util/BitmapUtil.java
@@ -272,7 +272,7 @@ public class BitmapUtil {
     return new Pair<>(options.outWidth, options.outHeight);
   }
 
-  public static InputStream toCompressedJpeg(Bitmap bitmap) {
+  public static ByteArrayInputStream toCompressedJpeg(Bitmap bitmap) {
     ByteArrayOutputStream thumbnailBytes = new ByteArrayOutputStream();
     bitmap.compress(CompressFormat.JPEG, 85, thumbnailBytes);
     return new ByteArrayInputStream(thumbnailBytes.toByteArray());