Minix1.5/commands/pretty.c
/* pretty - MINIX prettyprinter Author: Andy Tanenbaum */
/* This program is just a post processor for indent. Unfortunately,
* indent can't quite produce MINIX format output. This program takes
* the output from indent and fixes up a couple of items:
*
* - the first tab stop is at line_length 3 instead of 9
* - short 'if', 'while' and 'for' statements should be on a single line
* - placement of comments
* - return (value) ==> return(value)
* - etc.
*
* To use pretty, install "indent" in /bin or /usr/bin.
*/
#include <stdio.h>
#define NAME_SIZE 14
#define PATH_MAX 255
#define MAX_LINE 78
#define BUF_SIZE 1024
#define TAB 8
#define COM_COL 33
#define skip_whitespace(p) while (*p == ' ' || *p == '\t') p++;
char buf[BUF_SIZE];
char buf2[BUF_SIZE];
char buf3[BUF_SIZE];
int loops;
main(argc, argv)
int argc;
char *argv[];
{
int i, n;
char *name, intermediate[PATH_MAX+1], *p;
if (argc < 2) usage();
for (i = 1; i < argc; i++) {
/* Invent a name for the intermediate file. */
name = argv[i];
p = name + strlen(name) - 1;
while (p >= name && *p != '/') p--;
p++; /* p points to last component */
strncpy(intermediate, p, NAME_SIZE);
if (strlen(intermediate) < NAME_SIZE) {
/* Name is less than 14 chars. Just add "+" to name. */
strcat(intermediate, "+");
} else {
/* Name is 14 chars. Invert case of last character. */
n = strlen(intermediate);
intermediate[n - 1] ^= 040;
}
indent(name, intermediate);
process(name, intermediate);
unlink(intermediate);
}
}
indent(s, intermediate)
char *s; /* name of file to prettyprint */
char *intermediate; /* name of intermediate file */
{
/* Call indent to indent the file. Then rename it. */
int n;
char buf[BUF_SIZE];
strcpy(buf, "indent ");
strcat(buf, s);
strcat(buf," -npro -bap -bbb -br -ncdb -cli0.5 -di1 -lp -npsl -nfc1 -nip\n");
n = system(buf);
if (n < 0) {
fprintf(stderr, "pretty: cannot indent %s\n", s);
exit(1);
}
/* Rename the intermediate file. */
strcpy(buf, "mv ");
strcat(buf, s);
strcat(buf, " ");
strcat(buf, intermediate);
strcat(buf, "\n");
n = system(buf);
if (n < 0) {
fprintf(stderr, "pretty: mv %s %s failed\n", s, intermediate);
exit(1);
}
}
process(s, intermediate)
char *s; /* name of file to prettyprint */
char *intermediate; /* name of intermediate file */
{
/* File is now indented. Post process it. */
int t, m, tabcol;
FILE *in, *out;
char *p;
in = fopen(intermediate, "r"); /* open intermediate file */
if (in == NULL) {
fprintf(stderr, "pretty: cannot open %s\n", intermediate);
exit(1);
}
/* Create the output file. */
out = fopen(s, "w");
if (out == NULL) {
fprintf(stderr, "pretty: cannot open %s for writing\n", s);
exit(1);
}
/* Process the file a line at a time. */
loops = 0;
fgets(buf, BUF_SIZE, in);
fgets(buf2, BUF_SIZE, in);
while (1) {
if (fgets(buf3, BUF_SIZE, in) == NULL) {
if (buf[0] == '\0') {
fclose(in);
fclose(out);
return;
} else {
buf3[0] = '\0';
}
}
do_return(buf3);
/* Lines beginning with \t ==> " "; \t\t ==> \t */
if (buf[0] == '\t') {
/* Check for any subsequent tabs; add 1 if found. */
p = &buf[0];
if (buf[1] == '\t') {
/* Line starts with two tabs. */
shift_left(buf);
/* If there are more tabs, compensate for lost tab. */
skip_whitespace(p);
while (*p != '\t' && *p != '\n') p++;
if (*p == '\t') {
shift_right(p);
*p = '\t'; /* insert new tab */
}
} else {
/* Line starts with one tab. */
shift_right(buf);
buf[0] = ' ';
buf[1] = ' ';
}
}
ifwhilefor(in, out);
do_return(buf);
do_case(in);
place_comment(buf); /* align comment right */
splice_comment(in);
three_liner(in);
capitalize();
add_blank();
fputs(buf, out);
strcpy(buf, buf2);
strcpy(buf2, buf3);
loops++; /* keep track of loop iterations */
}
}
join()
{
char *p, *q;
int col, line_length();
/* Don't join two if statements. */
p = buf2;
skip_whitespace(p);
if (strncmp(p, "if ", 3) == 0) return(MAX_LINE + 100);
/* Don't join if the third line is 'else'. */
p = buf3;
skip_whitespace(p);
if (strncmp(p, "else", 4) == 0) return(MAX_LINE + 100);
/* Determine how long the joined statements will be. */
q = buf2;
skip_whitespace(q);
col = line_length(buf) + line_length(q) + 1;
return(col);
}
shift_left(p)
register char *p;
{
/* Copy a string to the left 1 line_length. */
do {
*p = *(p + 1);
p++;
} while (*p != 0);
}
shift_right(p)
register char *p;
{
/* Copy a string to the right 1 line_length. */
register char *q;
q = p + strlen(p); /* points to 0 byte */
while (q >= p) {
*(q + 1) = *q; /* move character to the right */
q--;
}
}
int line_length(p)
char *p;
{
/* Determine the length of the line, handling tabs correctly. */
int col = 1;
while (*p != '\n' && *p != '\0') {
if (*p != '\t') {
col++;
} else {
col += TAB - ((col - 1) % TAB);
}
p++;
}
return(col - 1);
}
place_comment(buf)
char *buf;
{
/* See if there is a comment that has to be readjusted to get it right. */
int col, col2, comcol, last, quotes;
char tmp[BUF_SIZE];
char *p, *q, *rb, *rt;
col = 1;
p = buf;
/* Skip any initial white space in buf. */
while (*p == ' ' || *p == '\t') {
if (*p == ' ')
col++;
else
col += TAB - (col - 1) % TAB;
p++;
}
if (*p == '\n') return; /* empty line */
if (*p == '/' && *(p + 1) == '*')
return; /* line starts with comment */
/* Scan forward looking for a comment. */
quotes = 0;
while (1) {
if (*p == '\n') return;
if (*p == '"') quotes++; /* careful about strings */
if (*p == '/' && *(p + 1) == '*') break;
if (*p != '\t')
col++;
else
col += TAB - (col - 1) % TAB;
if (*p != ' ' && *p != '\t') last = col;
p++;
}
/* We found a comment. Where should it go? */
if (quotes & 1) return; /* if this is inside a string, return */
if (last <= COM_COL) {
comcol = COM_COL;
} else {
comcol = ((last - 1) / TAB) * TAB + TAB + 1;
}
if (comcol == col) return; /* it is ok as is */
if (comcol > col) {
/* Move comment to the right. */
shift_right(p);
*p = '\t';
return;
}
/* Move comment to the left. */
q = p - 1; /* q points to char before comment */
while (*q == ' ' || *q == '\t') q--;
rb = buf;
rt = tmp;
col2 = 1;
while (rb <= q) {
if (*rb != '\t')
col2++;
else
col2 += TAB - (col2 - 1) % TAB;
*rt++ = *rb++; /* copy one character */
}
while (col2 < comcol) {
*rt++ = '\t';
col2 += TAB - (col2 - 1) % TAB;
}
*rt = 0;
strcat(rt, p);
strcpy(buf, tmp);
}
ifwhilefor(in, out)
FILE *in, *out;
{
/* Check for 'if' or 'while' split over two lines. */
int is_if, is_while, is_for;
char *first1, *first2, *second1;
/* Check for 'if', 'while' or 'for' statements split over two lines. */
first1 = &buf[0];
skip_whitespace(first1);
first2 = first1 + strlen(first1) - 2; /* last char */
while (first2 > &buf[0] && (*first2 == ' ' || *first2 == '\t')) first2--;
/* First1 and first2 now point to first/last nonblank chars. */
is_if = (strncmp(first1, "if ", 3) == 0 ? 1 : 0);
is_for = (strncmp(first1, "for ", 4) == 0 ? 1 : 0);
is_while = (strncmp(first1, "while ", 6) == 0 ? 1 : 0);
if ((is_if || is_for || is_while) && *first2 == ')') {
/* This is an 'if' or 'while' statement ending with ')'. */
if (join() < MAX_LINE) {
*(first2 + 1) = ' ';
*(first2 + 2) = 0;
second1 = &buf2[0];
while (*second1 == ' ' || *second1 == '\t') second1++;
strcat(first2, second1);
scrunch(in);
}
}
}
do_return(b)
char *b;
{
/* Remove the space after return, i.e., return (0) ==> return(0). */
char *p;
p = b;
skip_whitespace(p);
if (strncmp(p, "return (", 8) != 0) return;
shift_left(p + 6);
}
splice_comment(in)
FILE *in;
{
/* Indent has the problem that it sometimes breaks one line comments over two
* lines, even though the original comment fits quite well on one line. Fix.
*/
char *p, *q;
p = buf2; /* affected lines are followed by * something */
skip_whitespace(p);
if (*p != '*' || *(p + 1) != ' ') return;
/* Second line starts with * something. See if first one contains
* comment. */
q = buf;
while (1) {
while (*q != '/' && *q != '\n') q++;
if (*q == '\n') return;
if (*(q + 1) == '*') break;
q++;
}
/* We found a comment here. See if the two lines can be merged. */
if (line_length(buf) + line_length(p + 1) > MAX_LINE) return;
q = buf + strlen(buf) - 1; /* q points to '\n' */
*q = 0;
strcat(buf, p + 1);
scrunch(in);
}
add_blank()
{
/* Add a blank line between closing bracket and comment. */
char *p, *q;
p = buf;
skip_whitespace(p);
q = buf2;
skip_whitespace(q);
if (strcmp(p, "}\n") == 0 && strncmp(q, "/*", 2) == 0) {
strcat(buf, "\n");
}
}
usage()
{
fprintf(stderr, "Usage: pretty file ...\n");
}
do_case(in)
FILE *in;
{
/* Try to get short cases all on 1 line. */
char *p, *q, *r;
p = buf;
skip_whitespace(p);
if (strncmp(p, "case ", 5) != 0 && strncmp(p, "default", 7) != 0) return;
shift_right(p);
*p++ = ' ';
shift_right(p);
*p++ = ' ';
shift_right(p);
*p++ = ' ';
shift_right(p);
*p++ = ' ';
q = buf2;
skip_whitespace(q);
r = buf3;
skip_whitespace(r);
if (strncmp(r, "break;", 6) != 0 && strcmp(r, "}\n") != 0) return;
/* This is a case statement and the case is only 1 line long (+ break). */
if (line_length(buf) + 3 * TAB + line_length(q) + line_length(r) > MAX_LINE)
return;
p = buf + strlen(buf) - 1;
*p = 0;
strcat(buf, "\t");
strcat(buf, q);
p = buf + strlen(buf) - 1;
*p = 0;
if (strncmp(r, "break;", 6) == 0) {
strcat(buf, "\t"); /* 'break' case */
} else {
strcat(buf, "\n"); /* 'default' case */
}
if (strncmp(r, "}", 1) == 0) {
/* The "}' at the end of the switch has to be fudged. */
p = buf;
while (*p++ == '\t') strcat(buf, "\t");
}
strcat(buf, r);
if (fgets(buf2, BUF_SIZE, in) == NULL) buf2[0] = '\0';
if (fgets(buf3, BUF_SIZE, in) == NULL) buf3[0] = '\0';
}
capitalize()
{
/* Capitalize the first word of comments that begin a section. */
char *p, *q;
if (loops == 0) return;
p = buf;
skip_whitespace(p);
if (strncmp(p, "/* ", 3) != 0) return;
q = p + 3;
if (*q >= 'a' && *q <= 'z') *q = *q - 'a' + 'A';
}
scrunch(in)
FILE *in;
{
/* Move buf3 to buf2 and reload buf3. */
strcpy(buf2, buf3);
if (fgets(buf3, BUF_SIZE, in) == NULL) buf3[0] = '\0';
}
three_liner(in)
FILE *in;
{
/* Handle three line comments that get munged. */
char *p, *q;
p = buf;
q = buf2;
skip_whitespace(p);
skip_whitespace(q);
if (*p == '/' && *(p + 1) == '*' && *q == '*' && *(q + 1) == '/') {
q = p + strlen(p) - 1;
*q = 0;
strcat(p, " */\n");
scrunch(in);
}
}