X-Git-Url: https://diplodocus.org/git/nmh/blobdiff_plain/d66378909931a74ed83e29d592982eaa3e7e0107..b83f328aa3deb6a1ee889e54dcec6981d2ac3498:/sbr/fmt_compile.c

diff --git a/sbr/fmt_compile.c b/sbr/fmt_compile.c
index 23e07263..4ae85bc9 100644
--- a/sbr/fmt_compile.c
+++ b/sbr/fmt_compile.c
@@ -16,7 +16,7 @@
  * There is a NOT a one-to-one correspondence between format strings and
  * format instructions; some functions have side effects that can result
  * in multiple instructions being generated.  The exact list of instructions
- * generated by a format string can be seem with the nmh fmtdump utility.
+ * generated by a format string can be seem with the nmh fmttest utility.
  *
  * A list of format instructions can be found in fmt_compile.h.
  *
@@ -35,7 +35,7 @@
  *
  * - Add the code in fmt_scan.c to handle your new function.
  *
- * - Add code to fmtdump.c to display your new function.
+ * - Add code to fmttest.c to display your new function.
  *
  * - Document the new function in the mh-format(5) man page.
  *
@@ -49,6 +49,9 @@
 #include <h/mts.h>
 #include <h/utils.h>
 
+#include <curses.h>
+#include <term.h>
+
 #ifdef HAVE_SYS_TIME_H
 # include <sys/time.h>
 #endif
@@ -66,6 +69,10 @@ static struct comp *cm;			/* most recent comp ref  */
 static struct ftable *ftbl;		/* most recent func ref  */
 static int ncomp;
 static int infunction;			/* function nesting cnt  */
+static int termstatus = 0;		/* 0=uninit,1=ok,-1=fail */
+static char *termcbuf = NULL;		/* Capability output str */
+static char *termcbufp = NULL;		/* Capability buf ptr    */
+static size_t termcbufsz = 0;		/* Size of termcbuf      */
 
 extern struct mailname fmt_mnull;
 
@@ -82,6 +89,9 @@ extern struct mailname fmt_mnull;
 #define TF_MYNAME  9        /* special - get current name of user */
 #define TF_MYHOST  10       /* special - get "local" hostname     */
 #define TF_LMBOX   11       /* special - get full local mailbox   */
+#define TF_BOLD    12	    /* special - enter terminal bold mode */
+#define TF_UNDERLN 13       /* special - enter underline mode     */
+#define TF_RESET   14       /* special - reset terminal modes     */
 
 /* ftable->flags */
 /* NB that TFL_PUTS is also used to decide whether the test
@@ -213,7 +223,11 @@ static struct ftable functable[] = {
 
      { "mymbox",     TF_COMP,	FT_LV_COMPFLAG,	FT_MYMBOX,	TFL_PUTN },
 
-     { "unquote",   TF_EXPR, 	FT_LS_UNQUOTE,	0,		TFL_PUTS},
+     { "unquote",    TF_EXPR, 	FT_LS_UNQUOTE,	0,		TFL_PUTS },
+
+     { "bold",       TF_BOLD,	FT_LS_LIT,	0,		TFL_PUTS },
+     { "underline",  TF_UNDERLN,FT_LS_LIT,	0,		TFL_PUTS },
+     { "resetterm",  TF_RESET,	FT_LS_LIT,	0,		TFL_PUTS },
 
      { NULL,         0,		0,		0,		0 }
 };
@@ -288,7 +302,9 @@ static char *do_loop(char *);
 static char *do_if(char *);
 static void free_component(struct comp *);
 static void free_comptable(void);
-
+static void initialize_terminfo(void);
+static void get_term_stringcap(struct ftable *, char *);
+static int termbytes(int);
 
 /*
  * Lookup a function name in the functable
@@ -632,6 +648,18 @@ do_func(char *sp)
     	LS(t->f_type, getlocalmbox());
 	break;
 
+    case TF_BOLD:
+    	get_term_stringcap(t, "bold");
+	break;
+
+    case TF_UNDERLN:
+   	get_term_stringcap(t, "smul");
+	break;
+
+    case TF_RESET:
+    	get_term_stringcap(t, "sgr0");
+	break;
+
     case TF_NOW:
 	LV(t->f_type, time((time_t *) 0));
 	break;
@@ -1035,3 +1063,101 @@ free_component(struct comp *cm)
     	free(cm);
     }
 }
+
+/*
+ * These functions handles the case of outputting terminal strings depending
+ * on the terminfo setting.
+ */
+
+/*
+ * We should only be called if we haven't yet called setupterm()
+ */
+
+void
+initialize_terminfo(void)
+{
+    int errret, rc;
+
+    rc = setupterm(NULL, fileno(stdout), &errret);
+
+    if (rc != 0 || errret != 1)
+    	termstatus = -1;
+    else
+    	termstatus = 1;
+}
+
+/*
+ * Place the results of the specified string entry into the str register.
+ *
+ * Arguments are:
+ *
+ * t	- Pointer to instruction table entry (used to create fmt instruction)
+ * cap	- Name of terminfo capability to insert (e.g., "bold", or "sgr0").
+ *
+ * This ended up being more complicated than I hoped.  You need to fetch the
+ * entry via tigetstr(), but there MAY be a padding format embedded in what
+ * gets returned by tigetstr(), so you have to run it through tputs().
+ * And of course tputs() is designed to output to a terminal, so you have
+ * capture every byte output by the tputs() callback to get the final
+ * string to write to the format engine.
+ *
+ * If padding bytes are NULs that will be a problem for us, but some quick
+ * experimentation suggests that padding bytes are mostly a non-issue anymore.
+ * If they still crop up we'll have to figure out how to deal with them.
+ */
+
+void
+get_term_stringcap(struct ftable *t, char *cap)
+{
+    char *parm;
+
+    /*
+     * Common to all functions; initialize the termcap if we need it.
+     * If it didn't initialize successfully, return silently
+     */
+
+    if (termstatus == 0)
+    	initialize_terminfo();
+
+    if (termstatus == -1)
+    	return;
+
+    parm = tigetstr(cap);
+
+    if (parm == (char *) -1 || parm == NULL) {
+    	return;
+    }
+
+    termcbufp = termcbuf;
+
+    tputs(parm, 1, termbytes);
+
+    termcbufp = '\0';
+
+    LS(t->f_type, termcbuf);
+}
+
+/*
+ * Store a sequence of characters in our local buffer
+ */
+
+static int
+termbytes(int c)
+{
+    size_t offset;
+
+    /*
+     * Bump up the buffer size if we've reached the end (leave room for
+     * a trailing NUL)
+     */
+
+    if ((offset = termcbufp - termcbuf) - 1 >= termcbufsz) {
+        termcbufsz += 64;
+    	termcbuf = mh_xrealloc(termcbuf, termcbufsz);
+	termcbufp = termcbuf + offset;
+    }
+
+    *termcbufp++ = c;
+
+    return 0;
+}