X-Git-Url: https://diplodocus.org/git/nmh/blobdiff_plain/6125f5f8e7f544ccdcecb3d216c9c1035d693b4e..cdbb097c8f061dfea2e92f0beafc64fdf50a4eb7:/sbr/fmt_compile.c

diff --git a/sbr/fmt_compile.c b/sbr/fmt_compile.c
index 2e1a0ff5..970e71e5 100644
--- a/sbr/fmt_compile.c
+++ b/sbr/fmt_compile.c
@@ -87,6 +87,8 @@ extern struct mailname fmt_mnull;
 #define TF_STNDOUT 14       /* special - enter underline mode     */
 #define TF_RESET   15       /* special - reset terminal modes     */
 #define TF_HASCLR  16       /* special - terminal have color?     */
+#define TF_FGCOLR  17       /* special - foreground term color    */
+#define TF_BGCOLR  18       /* special - background term color    */
 
 /* ftable->flags */
 /* NB that TFL_PUTS is also used to decide whether the test
@@ -105,7 +107,9 @@ extern struct mailname fmt_mnull;
  *	    what maps a particular function name into a format instruction.
  * type -   The type of argument this function expects.  Those types are
  *	    listed above (with the TF_ prefix).  This affects what gets
- *	    placed in the format instruction (the f_un union).
+ *	    placed in the format instruction (the f_un union).  Also,
+ *	    instructions that require special handling are distinguished
+ *	    here (TF_MYMBOX is one example).
  * f_type - The instruction corresponding to this function (from the list
  *	    in fmt_compile.h).
  * extra  - Used by some functions to provide extra data to the compiler.
@@ -225,10 +229,40 @@ static struct ftable functable[] = {
      { "standout",   TF_STNDOUT,FT_LS_LIT,	0,		TFL_PUTS },
      { "resetterm",  TF_RESET,	FT_LS_LIT,	0,		TFL_PUTS },
      { "hascolor",   TF_HASCLR, FT_LV_LIT,	0,		0 },
+     { "fgcolor",    TF_FGCOLR, FT_LS_LIT,	0,		TFL_PUTS },
+     { "bgcolor",    TF_BGCOLR, FT_LS_LIT,	0,		TFL_PUTS },
 
      { NULL,         0,		0,		0,		0 }
 };
 
+/*
+ * A mapping of color names to terminfo color numbers.
+ *
+ * There are two sets of terminal-setting codes: 'setaf/setab' (ANSI) and
+ * 'setf/setb'.  Different terminals support different capabilities, so
+ * we provide a mapping for both.  I'm not crazy about putting numbers
+ * directly in here, but it seems these are well defined by terminfo
+ * so it should be okay.
+ */
+
+struct colormap {
+    char *colorname;	/* Name of color */
+    int ansinum;	/* The ANSI escape color number */
+    int nonansinum;	/* The non-ANSI escape color number */
+};
+
+static struct colormap colortable[] = {
+    { "black",		0,	0 },
+    { "red",		1,	4 },
+    { "green",		2,	2 },
+    { "yellow",		3,	6 },
+    { "blue",		4,	1 },
+    { "magenta",	5,	5 },
+    { "cyan",		6,	3 },
+    { "white",		7,	7 },
+    { NULL,		0,	0 }
+};
+
 /* 
  * Hash function for component name.  The function should be
  * case independent and probably shouldn't involve a routine
@@ -330,11 +364,7 @@ compile_error(char *str, char *cp)
     usr_fstring[errpos] = '\0';
 
     for (i = errpos-errctx; i < errpos; i++) {
-#ifdef LOCALE
 	if (iscntrl((unsigned char) usr_fstring[i]))
-#else
-	if (usr_fstring[i] < 32)
-#endif
 	    usr_fstring[i] = '_';
     }
 
@@ -662,6 +692,42 @@ do_func(char *sp)
     	LV(t->f_type, get_term_numcap("colors") > 1);
 	break;
 
+    case TF_FGCOLR:
+    case TF_BGCOLR: {
+	struct colormap *cmap = colortable;
+    	char *code;
+
+	sp = cp - 1;
+	while (c && c != ')')
+	    c = *cp++;
+	cp[-1] = '\0';
+
+	while (cmap->colorname != NULL) {
+	    if (strcasecmp(sp, cmap->colorname) == 0)
+	    	break;
+	    cmap++;
+	}
+
+	if (cmap->colorname == NULL) {
+	    CERROR("Unknown color name");
+	    break;
+	}
+
+	code = get_term_stringparm(t->type == TF_FGCOLR ? "setaf" : "setab",
+				   cmap->ansinum, 0);
+
+	/*
+	 * If this doesn't have anything, try falling back to setf/setb
+	 */
+
+	if (! code)
+	    code = get_term_stringparm(t->type == TF_FGCOLR ? "setf" : "setb",
+	    			       cmap->nonansinum, 0);
+
+	LS(t->f_type, code);
+	break;
+    }
+
     case TF_NOW:
 	LV(t->f_type, time((time_t *) 0));
 	break;