mblaze/mscan.c
Ivy Foster ca8cef1660 Add $MBLAZE environment variable to set profile location
As a side benefit, all callers of blaze822_home_file need only pass
the filename, so the base directory is set in only one place.
2016-09-01 18:50:20 -05:00

549 lines
10 KiB
C

#ifndef _XOPEN_SOURCE
#define _XOPEN_SOURCE 700
#endif
#include <sys/ioctl.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <errno.h>
#include <fcntl.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <strings.h>
#include <time.h>
#include <unistd.h>
#include <wchar.h>
#include <locale.h>
#include "blaze822.h"
static int cols;
static wchar_t replacement = L'?';
static char *cur;
static char *aliases[32];
static int alias_idx;
static int Iflag;
static int nflag;
static int curyear;
static int curyday;
static char default_fflag[] = "%c%u%r %-3n %10d %17f %t %2i%s";
static char *fflag = default_fflag;
int
u8putstr(FILE *out, char *s, ssize_t l, int pad)
{
ssize_t ol = l;
while (*s && l > 0) {
if (*s == '\t')
*s = ' ';
if (*s >= 32 && *s < 127) {
putc(*s, out);
s++;
l--;
} else {
wchar_t wc;
int r = mbtowc(&wc, s, 4);
if (r < 0) {
r = 1;
wc = replacement;
}
s += r;
l -= wcwidth(wc);
if (l >= 0)
fprintf(out, "%lc", wc);
}
}
if (pad)
while (l-- > 0)
putc(' ', out);
if (l < 0)
l = 0;
return ol - l;
}
int
itsme(char *v)
{
int i;
char *disp, *addr;
while ((v = blaze822_addr(v, &disp, &addr)))
for (i = 0; addr && i < alias_idx; i++)
if (strcmp(aliases[i], addr) == 0)
return 1;
return 0;
}
static int init;
void
numline(char *file)
{
if (!init) {
// delay loading of the seq until we need to scan the first
// file, in case someone in the pipe updated the map before
char *seq = blaze822_seq_open(0);
blaze822_seq_load(seq);
cur = blaze822_seq_cur();
init = 1;
}
while (*file == ' ' || *file == '\t')
file++;
long lineno = blaze822_seq_find(file);
if (lineno)
printf("%ld\n", lineno);
else
printf("%s\n", file);
}
static char *
fmt_date(struct message *msg, int w, int iso)
{
static char date[32];
char *v;
if (!msg)
return "";
v = blaze822_hdr(msg, "date");
if (!v)
return "(unknown)";
time_t t = blaze822_date(v);
if (t == -1)
return "(invalid)";
struct tm *tm;
tm = localtime(&t);
if (iso) {
if (w >= 19)
strftime(date, sizeof date, "%Y-%m-%d %H:%M:%S", tm);
else if (w >= 16)
strftime(date, sizeof date, "%Y-%m-%d %H:%M", tm);
else
strftime(date, sizeof date, "%Y-%m-%d", tm);
} else if (w < 10) {
if (tm->tm_year != curyear)
strftime(date, sizeof date, "%b%y", tm);
else if (tm->tm_yday != curyday)
strftime(date, sizeof date, "%d%b", tm);
else
strftime(date, sizeof date, "%H:%M", tm);
} else {
if (tm->tm_year != curyear)
strftime(date, sizeof date, "%Y-%m-%d", tm);
else if (tm->tm_yday != curyday)
strftime(date, sizeof date, "%a %b %e", tm);
else
strftime(date, sizeof date, "%a %H:%M", tm);
}
return date;
}
static char *
fmt_subject(struct message *msg, char *file, int strip)
{
static char subjdec[100];
char *subj = "(no subject)";
char *v;
if (!msg) {
snprintf(subjdec, sizeof subjdec, "\\_ %s", file);
return subjdec;
}
if ((v = blaze822_hdr(msg, "subject")))
subj = v;
blaze822_decode_rfc2047(subjdec, subj, sizeof subjdec - 1, "UTF-8");
if (strip) {
size_t i;
for (i = 0; subjdec[i]; ) {
if (subjdec[i] == ' ') {
i++;
continue;
} else if (strncasecmp("re:", subjdec+i, 3) == 0 ||
strncasecmp("aw:", subjdec+i, 3) == 0) {
i += 3;
continue;
} else if (strncasecmp("fwd:", subjdec+i, 4) == 0) {
i += 4;
continue;
}
break;
}
return subjdec + i;
}
return subjdec;
}
static char *
fmt_from(struct message *msg)
{
static char fromdec[64];
char *from = "(unknown)";
char to[256];
char *v, *w;
if (!msg)
return "";
if ((v = blaze822_hdr(msg, "from"))) {
if (itsme(v) && ((w = blaze822_hdr(msg, "to")))) {
snprintf(to, sizeof to, "TO:%s", w);
from = to;
} else {
char *disp, *addr;
blaze822_addr(v, &disp, &addr);
if (disp)
from = disp;
else if (addr)
from = addr;
}
}
blaze822_decode_rfc2047(fromdec, from, sizeof fromdec - 1, "UTF-8");
fromdec[sizeof fromdec - 1] = 0;
return fromdec;
}
static char *
fmt_to_flag(struct message *msg)
{
char *v;
if (!msg || !alias_idx)
return " ";
if ((v = blaze822_hdr(msg, "to")) && itsme(v))
return ">";
else if ((v = blaze822_hdr(msg, "cc")) && itsme(v))
return "+";
else if ((v = blaze822_hdr(msg, "resent-to")) && itsme(v))
return ":";
else if ((v = blaze822_hdr(msg, "from")) && itsme(v))
return "<";
else
return " ";
}
static ssize_t
print_human(intmax_t i, int w)
{
double d = i / 1024.0;
const char *u = "\0\0M\0G\0T\0P\0E\0Z\0Y\0";
while (d >= 1024) {
u += 2;
d /= 1024.0;
}
if (d < 1.0)
return printf("%*.2f", w, d);
else if (!*u)
return printf("%*.0f", w, d);
else if (d < 10.0)
return printf("%*.1f%s", w-1, d, u);
else
return printf("%*.0f%s", w-1, d, u);
}
void
oneline(char *file)
{
if (!init) {
// delay loading of the seq until we need to scan the first
// file, in case someone in the pipe updated the map before
char *seq = blaze822_seq_open(0);
blaze822_seq_load(seq);
cur = blaze822_seq_cur();
init = 1;
}
int indent = 0;
while (*file == ' ' || *file == '\t') {
indent++;
file++;
}
struct message *msg = blaze822(file);
char *flags = msg ? strstr(file, ":2,") : 0;
if (!flags)
flags = "";
else
flags += 3;
int wleft = cols;
char *f;
for (f = fflag; *f; f++) {
if (*f == '\\') {
f++;
switch (*f) {
case 'n': putchar('\n'); wleft = cols; break;
case 't': putchar('\t'); wleft -= (8 - wleft % 8); break;
default:
putchar('\\'); wleft--;
putchar(*f); wleft--;
}
continue;
}
if (*f != '%') {
putchar(*f);
wleft--;
continue;
}
f++;
int w = 0;
if ((*f >= '0' && *f <= '9') || *f == '-') {
errno = 0;
char *e;
w = strtol(f, &e, 10);
if (errno != 0)
w = 0;
else
f = e;
}
if (!*f)
break;
switch (*f) {
case '%':
putchar('%');
wleft--;
break;
case 'c':
if (cur && strcmp(cur, file) == 0)
putchar('>');
else
putchar(' ');
wleft--;
break;
case 'u': // unseen
if (strchr(flags, 'F'))
putchar('*');
else if (msg && !strchr(flags, 'S'))
putchar('.');
else if (strchr(flags, 'T'))
putchar('x');
else
putchar(' ');
wleft--;
break;
case 'r': // replied
if (strchr(flags, 'R'))
putchar('-');
else if (strchr(flags, 'P'))
putchar(':');
else
putchar(' ');
wleft--;
break;
case 't': // to-flag
wleft -= printf("%s", fmt_to_flag(msg));
break;
case 'M': // raw Maildir flags
if (!w) w = -3;
wleft -= printf("%*s", w, flags);
break;
case 'n':
{
long lineno = msg ? blaze822_seq_find(file) : 0;
if (lineno)
wleft -= printf("%*ld", w, lineno);
else
wleft -= printf("%*s", w, "");
}
break;
case 'd':
case 'D':
if (!w) w = 10;
wleft -= printf("%*s", w,
fmt_date(msg, w, Iflag || *f == 'D'));
break;
case 'f':
if (w < 0)
w += wleft;
if (w)
wleft -= u8putstr(stdout,
fmt_from(msg), w, 1);
else
wleft -= u8putstr(stdout,
fmt_from(msg), wleft, 0);
break;
case 'i':
{
int z;
if (!w)
w = 1;
if (indent >= 10) {
wleft -= printf("..%2d..", indent);
indent = 10 - 6/w;
}
for (z = 0; z < w*indent; z++)
putchar(' ');
wleft -= w*indent;
}
break;
case 's':
case 'S':
if (w < 0)
w += wleft;
if (w)
wleft -= u8putstr(stdout,
fmt_subject(msg, file, *f == 'S'), w, 1);
else
wleft -= u8putstr(stdout,
fmt_subject(msg, file, *f == 'S'), wleft, 0);
break;
case 'b':
{
struct stat st;
if (msg) {
if (stat(file, &st) != 0)
st.st_size = 0;
wleft -= print_human(st.st_size, w);
} else {
wleft -= printf("%.*s", w, "");
}
}
break;
case 'F':
{
char *e = file + strlen(file);
while (file < e && *e != '/')
e--;
e--;
while (file < e && *e != '/')
e--;
char *b = e - 1;
while (file < b && *b != '/')
b--;
if (*b == '/')
b++;
if (*b == '.')
b++;
if (w) {
if (w < 0)
w = -w;
wleft -= printf("%*.*s",
-w, (int)(e-b < w ? e-b : w), b);
} else {
wleft -= printf("%.*s", (int)(e-b), b);
}
}
break;
case 'R':
if (w)
wleft -= printf("%*.*s", w, w, file);
else
wleft -= printf("%s", file);
break;
case 'I':
{
char *m = msg ? blaze822_hdr(msg, "message-id") : 0;
if (!m)
m = "(unknown)";
if (w)
wleft -= printf("%*.*s", w, w, m);
else
wleft -= printf("%s", m);
}
break;
default:
putchar('%');
putchar(*f);
wleft -= 2;
}
}
printf("\n");
blaze822_free(msg);
}
int
main(int argc, char *argv[])
{
int c;
while ((c = getopt(argc, argv, "If:n")) != -1)
switch(c) {
case 'I': Iflag++; break;
case 'f': fflag = optarg; break;
case 'n': nflag = 1; break;
default:
fprintf(stderr, "Usage: mscan [-n] [-f format] [-I] [msgs...]\n");
exit(1);
}
if (nflag) {
if (argc == optind && isatty(0))
blaze822_loop1(":", numline);
else
blaze822_loop(argc-optind, argv+optind, numline);
return 0;
}
time_t now = time(0);
struct tm *tm = localtime(&now);
curyear = tm->tm_year;
curyday = tm->tm_yday;
setlocale(LC_ALL, ""); // for wcwidth later
if (wcwidth(0xFFFD) > 0)
replacement = 0xFFFD;
struct winsize w;
if (ioctl(1, TIOCGWINSZ, &w) == 0)
cols = w.ws_col;
if (getenv("COLUMNS"))
cols = atoi(getenv("COLUMNS"));
if (cols <= 40)
cols = 80;
char *f = blaze822_home_file("profile");
struct message *config = blaze822(f);
if (config) {
char *v, *d, *a;
if ((v = blaze822_hdr(config, "local-mailbox")))
while (alias_idx < (int)(sizeof aliases / sizeof aliases[0]) &&
(v = blaze822_addr(v, &d, &a)))
if (a)
aliases[alias_idx++] = strdup(a);
if ((v = blaze822_hdr(config, "alternate-mailboxes")))
while (alias_idx < (int)(sizeof aliases / sizeof aliases[0]) &&
(v = blaze822_addr(v, &d, &a)))
if (a)
aliases[alias_idx++] = strdup(a);
if ((v = blaze822_hdr(config, "scan-format")))
if (fflag == default_fflag) // NB. ==
fflag = v;
}
long i;
if (argc == optind && isatty(0))
i = blaze822_loop1(":", oneline);
else
i = blaze822_loop(argc-optind, argv+optind, oneline);
fprintf(stderr, "%ld mails scanned\n", i);
return 0;
}