added hideCancel flag to FrameSaveErrorDialog

Automattic · Nov 4, 2020 · a1c5a80 · a1c5a80
1 parent 179b016
commit a1c5a80
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/stories/src/main/java/com/wordpress/stories/compose/FrameSaveErrorDialog.kt b/stories/src/main/java/com/wordpress/stories/compose/FrameSaveErrorDialog.kt
@@ -17,7 +17,10 @@ class FrameSaveErrorDialog : DialogFragment() {
             .setMessage(arguments?.getString(ARG_MESSAGE))
 
         okListener?.let {
-            builder.setNegativeButton(android.R.string.cancel) { _, _ -> dismiss() }
+            val hideCancel = arguments?.getBoolean(ARG_HIDE_CANCEL) ?: false
+            if (!hideCancel) {
+                builder.setNegativeButton(android.R.string.cancel) { _, _ -> dismiss() }
+            }
             builder.setPositiveButton(
                     arguments?.getString(ARG_OK_LABEL) ?: activity?.getString(android.R.string.ok)) {
                     _, _ -> okListener?.OnOkClicked(this)
@@ -33,18 +36,21 @@ class FrameSaveErrorDialog : DialogFragment() {
         @JvmStatic private val ARG_MESSAGE = "message"
         @JvmStatic private val ARG_TITLE = "title"
         @JvmStatic private val ARG_OK_LABEL = "ok_label"
+        @JvmStatic private val ARG_HIDE_CANCEL = "hide_cancel"
 
         @JvmStatic fun newInstance(
             title: String,
             message: String,
             okButtonLabel: String? = null,
-            listener: FrameSaveErrorDialogOk? = null
+            listener: FrameSaveErrorDialogOk? = null,
+            hideCancelButton: Boolean = false
         ): FrameSaveErrorDialog =
             FrameSaveErrorDialog().apply {
                 arguments = Bundle().apply {
                     putString(ARG_TITLE, title)
                     putString(ARG_MESSAGE, message)
                     putString(ARG_OK_LABEL, okButtonLabel)
+                    putBoolean(ARG_HIDE_CANCEL, hideCancelButton)
                 }
                 okListener = listener
             }