From 16b04cd68813e4e7e75ccb1a6553df39527b3651 Mon Sep 17 00:00:00 2001
From: Andrew <andrew@digerati.co.nz>
Date: Mon, 27 Nov 2023 19:43:08 +1300
Subject: [PATCH] Add `--keep display` equivalent to `--strip safe`

---
 src/main.rs | 16 +++++++++++++---
 1 file changed, 13 insertions(+), 3 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 94e2f9b4..24ee1280 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -614,10 +614,20 @@ fn parse_opts_into_struct(
     }
 
     if let Some(keep) = matches.get_one::<String>("keep") {
-        let names = keep
+        let mut keep_display = false;
+        let mut names = keep
             .split(',')
-            .map(parse_chunk_name)
-            .collect::<Result<_, _>>()?;
+            .filter_map(|x| {
+                if x == "display" {
+                    keep_display = true;
+                    return None;
+                }
+                Some(parse_chunk_name(x))
+            })
+            .collect::<Result<IndexSet<_>, _>>()?;
+        if keep_display {
+            names.extend(StripChunks::KEEP_SAFE.iter().cloned());
+        }
         opts.strip = StripChunks::Keep(names)
     }