fix: empty redactions, #217

tellerops · May 15, 2024 · 38f5965 · 38f5965
1 parent 5d0d60a
commit 38f5965
Show file tree

Hide file tree

Showing 6 changed files with 32 additions and 7 deletions.
diff --git a/teller-cli/tests/cmd/redact.in/.teller.yml b/teller-cli/tests/cmd/redact.in/.teller.yml
@@ -0,0 +1,11 @@
+providers:
+  dot1:
+    kind: dotenv
+    maps:
+      - id: one
+        path: one.env
+  dot2:
+    kind: dotenv
+    maps:
+      - id: two
+        path: two.env
diff --git a/teller-cli/tests/cmd/redact.in/one.env b/teller-cli/tests/cmd/redact.in/one.env
@@ -0,0 +1,3 @@
+PRINT_NAME=linus
+FOO_BAR=foo
+EMPTY=""
diff --git a/teller-cli/tests/cmd/redact.in/text.txt b/teller-cli/tests/cmd/redact.in/text.txt
@@ -0,0 +1,2 @@
+hello linus,
+I just read that you made linux.
diff --git a/teller-cli/tests/cmd/redact.in/two.env b/teller-cli/tests/cmd/redact.in/two.env
@@ -0,0 +1,2 @@
+PRINT_MOOD=happy
+FOO_BAZ=baz
diff --git a/teller-cli/tests/cmd/redact.trycmd b/teller-cli/tests/cmd/redact.trycmd
@@ -0,0 +1,4 @@
+```console
+$ teller redact --in text.txt
+
+```
diff --git a/teller-core/src/redact.rs b/teller-core/src/redact.rs
@@ -39,13 +39,16 @@ impl Redactor {
         if self.has_match(message, kvs) {
             let mut redacted = message.to_string();
             for kv in kvs {
-                redacted = redacted.replace(
-                    &kv.value,
-                    kv.meta
-                        .as_ref()
-                        .and_then(|m| m.redact_with.as_ref())
-                        .map_or("[REDACTED]", |s| s.as_str()),
-                );
+                // only replace values with at least 2 chars
+                if kv.value.len() >= 2 {
+                    redacted = redacted.replace(
+                        &kv.value,
+                        kv.meta
+                            .as_ref()
+                            .and_then(|m| m.redact_with.as_ref())
+                            .map_or("[REDACTED]", |s| s.as_str()),
+                    );
+                }
             }
             Cow::Owned(redacted)
         } else {