sbase/unexpand.c

168 lines
3.0 KiB
C
Raw Normal View History

2014-06-09 15:52:20 +00:00
/* See LICENSE file for copyright and license details. */
#include <stdint.h>
2014-06-09 15:52:20 +00:00
#include <stdlib.h>
2014-11-21 17:52:09 +00:00
#include "utf.h"
2014-06-09 15:52:20 +00:00
#include "util.h"
static int aflag = 0;
static size_t *tablist = NULL;
static size_t tablistlen = 8;
2014-06-09 15:52:20 +00:00
static size_t
parselist(const char *s)
2014-06-09 15:52:20 +00:00
{
2015-02-20 12:26:36 +00:00
size_t i;
char *p, *tmp;
tmp = estrdup(s);
for (i = 0; (p = strsep(&tmp, " ,")); i++) {
if (*p == '\0')
eprintf("empty field in tablist\n");
tablist = ereallocarray(tablist, i + 1, sizeof(*tablist));
2015-02-20 12:26:36 +00:00
tablist[i] = estrtonum(p, 1, MIN(LLONG_MAX, SIZE_MAX));
if (i > 0 && tablist[i - 1] >= tablist[i])
eprintf("tablist must be ascending\n");
}
tablist = ereallocarray(tablist, i + 1, sizeof(*tablist));
2015-02-20 12:26:36 +00:00
return i;
2014-06-09 15:52:20 +00:00
}
static void
unexpandspan(size_t last, size_t col)
2014-06-09 15:52:20 +00:00
{
size_t off, i, j;
Rune r;
2014-06-09 15:52:20 +00:00
if (tablistlen == 1) {
i = 0;
off = last % tablist[i];
2014-06-09 15:52:20 +00:00
if ((col - last) + off >= tablist[i] && last < col)
last -= off;
r = '\t';
for (; last + tablist[i] <= col; last += tablist[i])
efputrune(&r, stdout, "<stdout>");
r = ' ';
for (; last < col; last++)
efputrune(&r, stdout, "<stdout>");
2014-06-09 15:52:20 +00:00
} else {
for (i = 0; i < tablistlen; i++)
if (col < tablist[i])
break;
for (j = 0; j < tablistlen; j++)
if (last < tablist[j])
break;
r = '\t';
for (; j < i; j++) {
efputrune(&r, stdout, "<stdout>");
last = tablist[j];
2014-06-09 15:52:20 +00:00
}
r = ' ';
for (; last < col; last++)
efputrune(&r, stdout, "<stdout>");
2014-06-09 15:52:20 +00:00
}
}
static void
2014-11-21 17:52:09 +00:00
unexpand(const char *file, FILE *fp)
2014-06-09 15:52:20 +00:00
{
2014-11-21 17:52:09 +00:00
Rune r;
size_t last = 0, col = 0, i;
int bol = 1;
2014-06-09 15:52:20 +00:00
while (efgetrune(&r, fp, file)) {
2014-11-21 17:52:09 +00:00
switch (r) {
2014-06-09 15:52:20 +00:00
case ' ':
if (!bol && !aflag)
last++;
2014-06-09 15:52:20 +00:00
col++;
break;
case '\t':
if (tablistlen == 1) {
if (!bol && !aflag)
last += tablist[0] - col % tablist[0];
col += tablist[0] - col % tablist[0];
} else {
for (i = 0; i < tablistlen; i++)
if (col < tablist[i])
break;
if (!bol && !aflag)
last = tablist[i];
col = tablist[i];
}
2014-06-09 15:52:20 +00:00
break;
case '\b':
if (bol || aflag)
unexpandspan(last, col);
2014-06-09 15:52:20 +00:00
col -= (col > 0);
last = col;
bol = 0;
2014-06-09 15:52:20 +00:00
break;
case '\n':
if (bol || aflag)
unexpandspan(last, col);
last = col = 0;
bol = 1;
2014-06-09 15:52:20 +00:00
break;
default:
if (bol || aflag)
unexpandspan(last, col);
last = ++col;
bol = 0;
break;
2014-06-09 15:52:20 +00:00
}
2014-11-21 17:52:09 +00:00
if ((r != ' ' && r != '\t') || (!aflag && !bol))
efputrune(&r, stdout, "<stdout>");
2014-06-09 15:52:20 +00:00
}
if (last < col && (bol || aflag))
unexpandspan(last, col);
}
static void
usage(void)
{
eprintf("usage: %s [-a] [-t tablist] [file ...]\n", argv0);
}
int
main(int argc, char *argv[])
{
FILE *fp;
int ret = 0;
char *tl = "8";
ARGBEGIN {
case 't':
tl = EARGF(usage());
if (!*tl)
eprintf("tablist cannot be empty\n");
/* Fallthrough: -t implies -a */
case 'a':
aflag = 1;
break;
default:
usage();
} ARGEND;
tablistlen = parselist(tl);
if (!argc) {
unexpand("<stdin>", stdin);
} else {
for (; *argv; argc--, argv++) {
if (!(fp = fopen(*argv, "r"))) {
weprintf("fopen %s:", *argv);
ret = 1;
continue;
}
unexpand(*argv, fp);
fclose(fp);
}
}
return ret;
2014-06-09 15:52:20 +00:00
}