listamsb: support COLORIZE_AMSB environment var.

1 files changed, 8 insertions, 2 deletions

diff --git a/atascii.c b/atascii.c
index 19d7404..d4e4a91 100644
--- a/atascii.c
+++ b/atascii.c
@@ -32,7 +32,6 @@ static int affects_inv(int c) {
 	}
 }
 
-
 int atascii_context_init(atascii_ctx *ctx, int mode, int flags) {
 	memset(ctx, 0, sizeof(atascii_ctx));
 
@@ -62,7 +61,7 @@ char *atascii_a2utf(atascii_ctx *ctx, int src, char *dest) {
 
 	if(src == ATA_CHR_FINISH) {
 		if(ctx->inv)
-			strcpy(dest, atascii_inverse_off);
+			strcpy(dest, ctx->mode == ATA_MODE_MAGAZINE ? "{norm}": atascii_inverse_off);
 		return dest;
 	}
 
@@ -149,3 +148,10 @@ char *atascii_a2utf(atascii_ctx *ctx, int src, char *dest) {
 
 	return dest;
 }
+
+const char *atascii_finish(atascii_ctx *ctx) {
+	if(ctx->inv)
+		return ATA_MODE_MAGAZINE ? "{norm}": atascii_inverse_off;
+	else
+		return "";
+};