Add -y option

bayashi · Mar 20, 2024 · 1cedd4e · 1cedd4e
1 parent e4c35a8
commit 1cedd4e
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -12,6 +12,12 @@
 $ pwd | highlightrepo
 ```
 
+Use `-y` option for `PS1`
+
+```sh
+PS1="[\u@\h \$(pwd | highlightrepo -y)]\n\[\e[32m\]\\$\[\e[0m\] "
+```
+
 ## Installation
 
 ```cmd

diff --git a/arg.go b/arg.go
@@ -24,6 +24,7 @@ var (
 type options struct {
 	color   string
 	noTilde bool
+	nonTTY   bool
 }
 
 func (cli *runner) parseArgs() *options {
@@ -33,6 +34,7 @@ func (cli *runner) parseArgs() *options {
 	var flagVersion bool
 	flag.StringVarP(&o.color, "color", "c", "cyan", "Color name to highlight")
 	flag.BoolVarP(&o.noTilde, "no-tilde", "t", false, "Not replace home dir path to tilda")
+	flag.BoolVarP(&o.nonTTY, "non-tty", "y", false, "Bypass the check for non-tty output streams")
 	flag.BoolVarP(&flagHelp, "help", "h", false, "Show help (This message) and exit")
 	flag.BoolVarP(&flagVersion, "version", "v", false, "Show version and build command info and exit")
 

diff --git a/highlightrepo.go b/highlightrepo.go
@@ -28,6 +28,11 @@ func NewHighlightrepo(o *options) *highlightrepo {
 	h := &highlightrepo{
 		options: o,
 	}
+
+	if o.noTTY {
+		color.NoColor = false
+	}
+
 	h.hlColor = colorpalette.Get(o.color)
 
 	return h