escape prompt colors properly for readline

1 files changed, 12 insertions, 1 deletions

diff --git a/bin/fancy-prompt b/bin/fancy-prompt
index d9f9809..484b5fc 100755
--- a/bin/fancy-prompt
+++ b/bin/fancy-prompt
@@ -3,6 +3,13 @@
 # use strict;
 # use warnings;
 
+# readline gets confused if non-printable characters in prompts aren't escaped
+my $prompt_escape;
+if ($ARGV[0] eq '--prompt-escape') {
+    shift;
+    $prompt_escape = 1;
+}
+
 # collect information
 chomp(my $hostname = `hostname`);
 my $cols           = `tput cols`;
@@ -90,7 +97,11 @@ $colornames{battery}  = $battery > 40 ? $colornames{background}
                       : $battery > 15 ? $colornames{battwarn}
                       : $battery >  5 ? $colornames{battcrit}
                       :                 $colornames{battemerg};
-sub color { $colors{$colornames{$_[0]}} }
+sub color {
+    my $code = $colors{$colornames{$_[0]}};
+    $code = '\[' . $code . '\]' if $prompt_escape;
+    $code;
+}
 
 # define some display things
 my $border       = '-';