scripts: Add color to repo-add

Signed-off-by: William Giokas <1007380@gmail.com>
Signed-off-by: Allan McRae <allan@archlinux.org>

2 files changed, 8 insertions, 0 deletions

diff --git a/doc/repo-add.8.txt b/doc/repo-add.8.txt
index 91f3ac6d..049195a7 100644
--- a/doc/repo-add.8.txt
+++ b/doc/repo-add.8.txt
@@ -57,6 +57,9 @@ Common Options
 	If the signature is invalid, an error is produced and the update does not
 	proceed.
 
+*\--nocolor*::
+	Remove color from repo-add and repo-remove output.
+
 repo-add Options
 ----------------
 *-d, \--delta*::
diff --git a/scripts/repo-add.sh.in b/scripts/repo-add.sh.in
index 3e18a1a1..d4d72e4f 100644
--- a/scripts/repo-add.sh.in
+++ b/scripts/repo-add.sh.in
@@ -37,6 +37,7 @@ VERIFY=0
 REPO_DB_FILE=
 LOCKFILE=
 CLEAN_LOCK=0
+USE_COLOR='y'
 
 # ensure we have a sane umask set
 umask 0022
@@ -71,6 +72,7 @@ packages to remove can be specified on the command line.\n")"
 		printf -- "$(gettext "Please move along, there is nothing to see here.\n")"
 		return
 	fi
+	printf -- "$(gettext "  --nocolor         turn off color in output\n")"
 	printf -- "$(gettext "  -q, --quiet       minimize output\n")"
 	printf -- "$(gettext "  -s, --sign        sign database with GnuPG after update\n")"
 	printf -- "$(gettext "  -k, --key <key>   use the specified key to sign the database\n")"
@@ -610,6 +612,7 @@ while (( $# )); do
 		-d|--delta) DELTA=1;;
 		-n|--new) ONLYADDNEW=1;;
 		-f|--files) WITHFILES=1;;
+		--nocolor) USE_COLOR='n';;
 		-s|--sign)
 			check_gpg
 			SIGN=1
@@ -642,6 +645,8 @@ while (( $# )); do
 	shift
 done
 
+m4_include(library/term_colors.sh)
+
 REPO_DB_FILE=${args[0]}
 if [[ -z $REPO_DB_FILE ]]; then
 	usage