1 892de798 2004-04-19 devnull #include <u.h>
  2 892de798 2004-04-19 devnull #include <libc.h>
  3 892de798 2004-04-19 devnull #include <bio.h>
  4 892de798 2004-04-19 devnull #include <ctype.h>
  5 892de798 2004-04-19 devnull 
  6 892de798 2004-04-19 devnull /*
  7 892de798 2004-04-19 devnull  * block up paragraphs, possibly with indentation
  8 892de798 2004-04-19 devnull  */
  9 892de798 2004-04-19 devnull 
 10 892de798 2004-04-19 devnull int extraindent = 0;		/* how many spaces to indent all lines */
 11 892de798 2004-04-19 devnull int indent = 0;			/* current value of indent, before extra indent */
 12 892de798 2004-04-19 devnull int length = 70;		/* how many columns per output line */
 13 892de798 2004-04-19 devnull int join = 1;			/* can lines be joined? */
 14 892de798 2004-04-19 devnull int maxtab = 8;
 15 892de798 2004-04-19 devnull Biobuf bin;
 16 892de798 2004-04-19 devnull Biobuf bout;
 17 892de798 2004-04-19 devnull 
 18 892de798 2004-04-19 devnull typedef struct Word Word;
 19 892de798 2004-04-19 devnull struct Word{
 20 892de798 2004-04-19 devnull 	int	bol;
 21 892de798 2004-04-19 devnull 	int	indent;
 22 892de798 2004-04-19 devnull 	char	text[1];
 23 892de798 2004-04-19 devnull };
 24 892de798 2004-04-19 devnull 
 25 892de798 2004-04-19 devnull void	fmt(void);
 26 892de798 2004-04-19 devnull 
 27 892de798 2004-04-19 devnull void
 28 892de798 2004-04-19 devnull usage(void)
 29 892de798 2004-04-19 devnull {
 30 892de798 2004-04-19 devnull 	fprint(2, "usage: %s [-j] [-i indent] [-l length] [file...]\n", argv0);
 31 892de798 2004-04-19 devnull 	exits("usage");
 32 892de798 2004-04-19 devnull }
 33 892de798 2004-04-19 devnull 
 34 892de798 2004-04-19 devnull void
 35 892de798 2004-04-19 devnull main(int argc, char **argv)
 36 892de798 2004-04-19 devnull {
 37 892de798 2004-04-19 devnull 	int i, f;
 38 892de798 2004-04-19 devnull 	char *s, *err;
 39 892de798 2004-04-19 devnull 
 40 892de798 2004-04-19 devnull 	ARGBEGIN{
 41 892de798 2004-04-19 devnull 	case 'i':
 42 892de798 2004-04-19 devnull 		extraindent = atoi(EARGF(usage()));
 43 892de798 2004-04-19 devnull 		break;
 44 892de798 2004-04-19 devnull 	case 'j':
 45 892de798 2004-04-19 devnull 		join = 0;
 46 892de798 2004-04-19 devnull 		break;
 47 892de798 2004-04-19 devnull 	case 'w':
 48 892de798 2004-04-19 devnull 	case 'l':
 49 892de798 2004-04-19 devnull 		length = atoi(EARGF(usage()));
 50 892de798 2004-04-19 devnull 		break;
 51 892de798 2004-04-19 devnull 	default:
 52 892de798 2004-04-19 devnull 		usage();
 53 892de798 2004-04-19 devnull 	}ARGEND
 54 892de798 2004-04-19 devnull 
 55 892de798 2004-04-19 devnull 	if(length <= indent){
 56 892de798 2004-04-19 devnull 		fprint(2, "%s: line length<=indentation\n", argv0);
 57 892de798 2004-04-19 devnull 		exits("length");
 58 892de798 2004-04-19 devnull 	}
 59 892de798 2004-04-19 devnull 
 60 892de798 2004-04-19 devnull 	s=getenv("tabstop");
 61 892de798 2004-04-19 devnull 	if(s!=nil && atoi(s)>0)
 62 892de798 2004-04-19 devnull 		maxtab=atoi(s);
 63 892de798 2004-04-19 devnull 	err = nil;
 64 892de798 2004-04-19 devnull 	Binit(&bout, 1, OWRITE);
 65 892de798 2004-04-19 devnull 	if(argc <= 0){
 66 892de798 2004-04-19 devnull 		Binit(&bin, 0, OREAD);
 67 892de798 2004-04-19 devnull 		fmt();
 68 892de798 2004-04-19 devnull 	}else{
 69 892de798 2004-04-19 devnull 		for(i=0; i<argc; i++){
 70 892de798 2004-04-19 devnull 			f = open(argv[i], OREAD);
 71 892de798 2004-04-19 devnull 			if(f < 0){
 72 892de798 2004-04-19 devnull 				fprint(2, "%s: can't open %s: %r\n", argv0, argv[i]);
 73 892de798 2004-04-19 devnull 				err = "open";
 74 892de798 2004-04-19 devnull 			}else{
 75 892de798 2004-04-19 devnull 				Binit(&bin, f, OREAD);
 76 892de798 2004-04-19 devnull 				fmt();
 77 892de798 2004-04-19 devnull 				Bterm(&bin);
 78 892de798 2004-04-19 devnull 				if(i != argc-1)
 79 892de798 2004-04-19 devnull 					Bputc(&bout, '\n');
 80 892de798 2004-04-19 devnull 			}
 81 892de798 2004-04-19 devnull 		}
 82 892de798 2004-04-19 devnull 	}
 83 892de798 2004-04-19 devnull 	exits(err);
 84 892de798 2004-04-19 devnull }
 85 892de798 2004-04-19 devnull 
 86 892de798 2004-04-19 devnull int
 87 892de798 2004-04-19 devnull indentof(char **linep)
 88 892de798 2004-04-19 devnull {
 89 892de798 2004-04-19 devnull 	int i, ind;
 90 892de798 2004-04-19 devnull 	char *line;
 91 892de798 2004-04-19 devnull 
 92 892de798 2004-04-19 devnull 	ind = 0;
 93 892de798 2004-04-19 devnull 	line = *linep;
 94 892de798 2004-04-19 devnull 	for(i=0; line[i]; i++)
 95 892de798 2004-04-19 devnull 		switch(line[i]){
 96 892de798 2004-04-19 devnull 		default:
 97 892de798 2004-04-19 devnull 			*linep = line;
 98 892de798 2004-04-19 devnull 			return ind;
 99 892de798 2004-04-19 devnull 		case ' ':
100 892de798 2004-04-19 devnull 			ind++;
101 892de798 2004-04-19 devnull 			break;
102 892de798 2004-04-19 devnull 		case '\t':
103 892de798 2004-04-19 devnull 			ind += maxtab;
104 892de798 2004-04-19 devnull 			ind -= ind%maxtab;
105 892de798 2004-04-19 devnull 			break;
106 892de798 2004-04-19 devnull 		}
107 892de798 2004-04-19 devnull 			
108 892de798 2004-04-19 devnull 	/* plain white space doesn't change the indent */
109 892de798 2004-04-19 devnull 	*linep = "";
110 892de798 2004-04-19 devnull 	return indent;
111 892de798 2004-04-19 devnull }
112 892de798 2004-04-19 devnull 
113 892de798 2004-04-19 devnull Word**
114 892de798 2004-04-19 devnull addword(Word **words, int *nwordp, char *s, int l, int indent, int bol)
115 892de798 2004-04-19 devnull {
116 892de798 2004-04-19 devnull 	Word *w;
117 892de798 2004-04-19 devnull 
118 892de798 2004-04-19 devnull 	w = malloc(sizeof(Word)+l+1);
119 892de798 2004-04-19 devnull 	memmove(w->text, s, l);
120 892de798 2004-04-19 devnull 	w->text[l] = '\0';
121 892de798 2004-04-19 devnull 	w->indent = indent;
122 892de798 2004-04-19 devnull 	w->bol = bol;
123 892de798 2004-04-19 devnull 	words = realloc(words, (*nwordp+1)*sizeof(Word*));
124 892de798 2004-04-19 devnull 	words[(*nwordp)++] = w;
125 892de798 2004-04-19 devnull 	return words;
126 892de798 2004-04-19 devnull }
127 892de798 2004-04-19 devnull 
128 892de798 2004-04-19 devnull Word**
129 892de798 2004-04-19 devnull parseline(char *line, Word **words, int *nwordp)
130 892de798 2004-04-19 devnull {
131 892de798 2004-04-19 devnull 	int ind, l, bol;
132 892de798 2004-04-19 devnull 
133 892de798 2004-04-19 devnull 	ind = indentof(&line);
134 892de798 2004-04-19 devnull 	indent = ind;
135 892de798 2004-04-19 devnull 	bol = 1;
136 892de798 2004-04-19 devnull 	for(;;){
137 892de798 2004-04-19 devnull 		/* find next word */
138 892de798 2004-04-19 devnull 		while(*line==' ' || *line=='\t')
139 892de798 2004-04-19 devnull 			line++;
140 892de798 2004-04-19 devnull 		if(*line == '\0'){
141 892de798 2004-04-19 devnull 			if(bol)
142 892de798 2004-04-19 devnull 				return addword(words, nwordp, "", 0, -1, bol);
143 892de798 2004-04-19 devnull 			break;
144 892de798 2004-04-19 devnull 		}
145 892de798 2004-04-19 devnull 		/* how long is this word? */
146 892de798 2004-04-19 devnull 		for(l=0; line[l]; l++)
147 565fd89e 2006-02-14 devnull 			if(line[l]==' ' || line[l]=='\t'){
148 565fd89e 2006-02-14 devnull 				if(join)
149 565fd89e 2006-02-14 devnull 					while(line[l]==' ' || line[l]=='\t')
150 565fd89e 2006-02-14 devnull 						l++;
151 892de798 2004-04-19 devnull 				break;
152 565fd89e 2006-02-14 devnull 			}
153 892de798 2004-04-19 devnull 		words = addword(words, nwordp, line, l, indent, bol);
154 892de798 2004-04-19 devnull 		bol = 0;
155 892de798 2004-04-19 devnull 		line += l;
156 892de798 2004-04-19 devnull 	}
157 892de798 2004-04-19 devnull 	return words;
158 892de798 2004-04-19 devnull }
159 892de798 2004-04-19 devnull 
160 892de798 2004-04-19 devnull void
161 892de798 2004-04-19 devnull printindent(int w)
162 892de798 2004-04-19 devnull {
163 892de798 2004-04-19 devnull 	while(w >= maxtab){
164 892de798 2004-04-19 devnull 		Bputc(&bout, '\t');
165 892de798 2004-04-19 devnull 		w -= maxtab;
166 892de798 2004-04-19 devnull 	}
167 892de798 2004-04-19 devnull 	while(w > 0){
168 892de798 2004-04-19 devnull 		Bputc(&bout, ' ');
169 892de798 2004-04-19 devnull 		w--;
170 892de798 2004-04-19 devnull 	}
171 892de798 2004-04-19 devnull }
172 892de798 2004-04-19 devnull 
173 892de798 2004-04-19 devnull /* give extra space if word ends with period, etc. */
174 892de798 2004-04-19 devnull int
175 892de798 2004-04-19 devnull nspaceafter(char *s)
176 892de798 2004-04-19 devnull {
177 892de798 2004-04-19 devnull 	int n;
178 892de798 2004-04-19 devnull 
179 892de798 2004-04-19 devnull 	n = strlen(s);
180 892de798 2004-04-19 devnull 	if(n < 2)
181 892de798 2004-04-19 devnull 		return 1;
182 1c171e3a 2005-07-19 devnull 	if(isupper((uchar)s[0]) && n < 4)
183 892de798 2004-04-19 devnull 		return 1;
184 892de798 2004-04-19 devnull 	if(strchr(".!?", s[n-1]) != nil)
185 892de798 2004-04-19 devnull 		return 2;
186 892de798 2004-04-19 devnull 	return 1;
187 892de798 2004-04-19 devnull }
188 892de798 2004-04-19 devnull 	
189 892de798 2004-04-19 devnull 
190 892de798 2004-04-19 devnull void
191 892de798 2004-04-19 devnull printwords(Word **w, int nw)
192 892de798 2004-04-19 devnull {
193 892de798 2004-04-19 devnull 	int i, j, n, col, nsp;
194 892de798 2004-04-19 devnull 
195 892de798 2004-04-19 devnull 	/* one output line per loop */
196 892de798 2004-04-19 devnull 	for(i=0; i<nw; ){
197 892de798 2004-04-19 devnull 		/* if it's a blank line, print it */
198 892de798 2004-04-19 devnull 		if(w[i]->indent == -1){
199 892de798 2004-04-19 devnull 			Bputc(&bout, '\n');
200 892de798 2004-04-19 devnull 			if(++i == nw)	/* out of words */
201 892de798 2004-04-19 devnull 				break;
202 892de798 2004-04-19 devnull 		}
203 892de798 2004-04-19 devnull 		/* emit leading indent */
204 892de798 2004-04-19 devnull 		col = extraindent+w[i]->indent;
205 892de798 2004-04-19 devnull 		printindent(col);
206 892de798 2004-04-19 devnull 		/* emit words until overflow; always emit at least one word */
207 892de798 2004-04-19 devnull 		for(n=0;; n++){
208 892de798 2004-04-19 devnull 			Bprint(&bout, "%s", w[i]->text);
209 892de798 2004-04-19 devnull 			col += utflen(w[i]->text);
210 892de798 2004-04-19 devnull 			if(++i == nw)
211 892de798 2004-04-19 devnull 				break;	/* out of words */
212 892de798 2004-04-19 devnull 			if(w[i]->indent != w[i-1]->indent)
213 892de798 2004-04-19 devnull 				break;	/* indent change */
214 892de798 2004-04-19 devnull 			nsp = nspaceafter(w[i-1]->text);
215 892de798 2004-04-19 devnull 			if(col+nsp+utflen(w[i]->text) > extraindent+length)
216 892de798 2004-04-19 devnull 				break;	/* fold line */
217 2d8bbdea 2006-01-18 devnull 			if(!join && w[i]->bol)
218 892de798 2004-04-19 devnull 				break;
219 892de798 2004-04-19 devnull 			for(j=0; j<nsp; j++)
220 892de798 2004-04-19 devnull 				Bputc(&bout, ' ');	/* emit space; another word will follow */
221 892de798 2004-04-19 devnull 			col += nsp;
222 892de798 2004-04-19 devnull 		}
223 892de798 2004-04-19 devnull 		/* emit newline */
224 892de798 2004-04-19 devnull 		Bputc(&bout, '\n');
225 892de798 2004-04-19 devnull 	}
226 892de798 2004-04-19 devnull }
227 892de798 2004-04-19 devnull 
228 892de798 2004-04-19 devnull void
229 892de798 2004-04-19 devnull fmt(void)
230 892de798 2004-04-19 devnull {
231 892de798 2004-04-19 devnull 	char *s;
232 892de798 2004-04-19 devnull 	int i, nw;
233 892de798 2004-04-19 devnull 	Word **w;
234 892de798 2004-04-19 devnull 
235 892de798 2004-04-19 devnull 	nw = 0;
236 892de798 2004-04-19 devnull 	w = nil;
237 892de798 2004-04-19 devnull 	while((s = Brdstr(&bin, '\n', 1)) != nil){
238 892de798 2004-04-19 devnull 		w = parseline(s, w, &nw);
239 892de798 2004-04-19 devnull 		free(s);
240 892de798 2004-04-19 devnull 	}
241 892de798 2004-04-19 devnull 	printwords(w, nw);
242 892de798 2004-04-19 devnull 	for(i=0; i<nw; i++)
243 892de798 2004-04-19 devnull 		free(w[i]);
244 892de798 2004-04-19 devnull 	free(w);
245 892de798 2004-04-19 devnull }