Blame


1 4a4a7c3e 2004-04-25 devnull /*
2 4a4a7c3e 2004-04-25 devnull * mc - columnate
3 4a4a7c3e 2004-04-25 devnull *
4 4a4a7c3e 2004-04-25 devnull * mc[-][-LINEWIDTH][-t][file...]
5 4a4a7c3e 2004-04-25 devnull * - causes break on colon
6 4a4a7c3e 2004-04-25 devnull * -LINEWIDTH sets width of line in which to columnate(default 80)
7 4a4a7c3e 2004-04-25 devnull * -t suppresses expanding multiple blanks into tabs
8 4a4a7c3e 2004-04-25 devnull *
9 4a4a7c3e 2004-04-25 devnull */
10 4a4a7c3e 2004-04-25 devnull #include <u.h>
11 4a4a7c3e 2004-04-25 devnull #include <sys/ioctl.h>
12 4a4a7c3e 2004-04-25 devnull #include <sys/termios.h>
13 4a4a7c3e 2004-04-25 devnull #include <libc.h>
14 4a4a7c3e 2004-04-25 devnull #include <draw.h>
15 4a4a7c3e 2004-04-25 devnull #include <bio.h>
16 4a4a7c3e 2004-04-25 devnull #include <fcall.h>
17 8cc98332 2005-01-04 devnull #include <9pclient.h>
18 4a4a7c3e 2004-04-25 devnull #include <thread.h>
19 4a4a7c3e 2004-04-25 devnull
20 4a4a7c3e 2004-04-25 devnull #define WIDTH 80
21 4a4a7c3e 2004-04-25 devnull #define TAB 4
22 4a4a7c3e 2004-04-25 devnull #define WORD_ALLOC_QUANTA 1024
23 4a4a7c3e 2004-04-25 devnull #define ALLOC_QUANTA 4096
24 4a4a7c3e 2004-04-25 devnull
25 4a4a7c3e 2004-04-25 devnull int wordsize(Rune*, int);
26 4a4a7c3e 2004-04-25 devnull int nexttab(int);
27 4a4a7c3e 2004-04-25 devnull
28 4a4a7c3e 2004-04-25 devnull int tabwid;
29 fb5ca0b9 2004-04-25 devnull int mintab = 1;
30 4a4a7c3e 2004-04-25 devnull int linewidth=WIDTH;
31 4a4a7c3e 2004-04-25 devnull int colonflag=0;
32 4a4a7c3e 2004-04-25 devnull int tabflag=0; /* -t flag turned off forever, except in acme */
33 4a4a7c3e 2004-04-25 devnull Rune *cbuf, *cbufp;
34 4a4a7c3e 2004-04-25 devnull Rune **word;
35 4a4a7c3e 2004-04-25 devnull int maxwidth=0;
36 4a4a7c3e 2004-04-25 devnull int nalloc=ALLOC_QUANTA;
37 4a4a7c3e 2004-04-25 devnull int nwalloc=WORD_ALLOC_QUANTA;
38 4a4a7c3e 2004-04-25 devnull int nchars=0;
39 4a4a7c3e 2004-04-25 devnull int nwords=0;
40 4a4a7c3e 2004-04-25 devnull Biobuf bin;
41 4a4a7c3e 2004-04-25 devnull Biobuf bout;
42 4a4a7c3e 2004-04-25 devnull
43 4a4a7c3e 2004-04-25 devnull void getwidth(void), readbuf(int), error(char *);
44 4a4a7c3e 2004-04-25 devnull void scanwords(void), columnate(void), morechars(void);
45 4a4a7c3e 2004-04-25 devnull
46 4a4a7c3e 2004-04-25 devnull void
47 4a4a7c3e 2004-04-25 devnull threadmain(int argc, char *argv[])
48 4a4a7c3e 2004-04-25 devnull {
49 4a4a7c3e 2004-04-25 devnull int i;
50 4a4a7c3e 2004-04-25 devnull int lineset;
51 4a4a7c3e 2004-04-25 devnull int ifd;
52 4a4a7c3e 2004-04-25 devnull
53 4a4a7c3e 2004-04-25 devnull lineset = 0;
54 4a4a7c3e 2004-04-25 devnull Binit(&bout, 1, OWRITE);
55 4a4a7c3e 2004-04-25 devnull while(argc > 1 && argv[1][0] == '-'){
56 4a4a7c3e 2004-04-25 devnull --argc; argv++;
57 4a4a7c3e 2004-04-25 devnull switch(argv[0][1]){
58 4a4a7c3e 2004-04-25 devnull case '\0':
59 4a4a7c3e 2004-04-25 devnull colonflag = 1;
60 4a4a7c3e 2004-04-25 devnull break;
61 4a4a7c3e 2004-04-25 devnull case 't':
62 4a4a7c3e 2004-04-25 devnull tabflag = 0;
63 4a4a7c3e 2004-04-25 devnull break;
64 4a4a7c3e 2004-04-25 devnull default:
65 4a4a7c3e 2004-04-25 devnull linewidth = atoi(&argv[0][1]);
66 4a4a7c3e 2004-04-25 devnull if(linewidth <= 1)
67 4a4a7c3e 2004-04-25 devnull linewidth = WIDTH;
68 4a4a7c3e 2004-04-25 devnull lineset = 1;
69 4a4a7c3e 2004-04-25 devnull break;
70 4a4a7c3e 2004-04-25 devnull }
71 4a4a7c3e 2004-04-25 devnull }
72 4a4a7c3e 2004-04-25 devnull if(lineset == 0)
73 4a4a7c3e 2004-04-25 devnull getwidth();
74 4a4a7c3e 2004-04-25 devnull cbuf = cbufp = malloc(ALLOC_QUANTA*(sizeof *cbuf));
75 4a4a7c3e 2004-04-25 devnull word = malloc(WORD_ALLOC_QUANTA*(sizeof *word));
76 4a4a7c3e 2004-04-25 devnull if(word == 0 || cbuf == 0)
77 4a4a7c3e 2004-04-25 devnull error("out of memory");
78 4a4a7c3e 2004-04-25 devnull if(argc == 1)
79 4a4a7c3e 2004-04-25 devnull readbuf(0);
80 4a4a7c3e 2004-04-25 devnull else{
81 4a4a7c3e 2004-04-25 devnull for(i = 1; i < argc; i++){
82 4a4a7c3e 2004-04-25 devnull if((ifd = open(*++argv, OREAD)) == -1)
83 4a4a7c3e 2004-04-25 devnull fprint(2, "mc: can't open %s (%r)\n", *argv);
84 4a4a7c3e 2004-04-25 devnull else{
85 4a4a7c3e 2004-04-25 devnull readbuf(ifd);
86 4a4a7c3e 2004-04-25 devnull Bflush(&bin);
87 4a4a7c3e 2004-04-25 devnull close(ifd);
88 4a4a7c3e 2004-04-25 devnull }
89 4a4a7c3e 2004-04-25 devnull }
90 4a4a7c3e 2004-04-25 devnull }
91 4a4a7c3e 2004-04-25 devnull columnate();
92 ecdecae3 2004-04-25 devnull Bflush(&bout);
93 8cc98332 2005-01-04 devnull threadexitsall(0);
94 4a4a7c3e 2004-04-25 devnull }
95 4a4a7c3e 2004-04-25 devnull void
96 4a4a7c3e 2004-04-25 devnull error(char *s)
97 4a4a7c3e 2004-04-25 devnull {
98 4a4a7c3e 2004-04-25 devnull fprint(2, "mc: %s\n", s);
99 8cc98332 2005-01-04 devnull threadexitsall(s);
100 4a4a7c3e 2004-04-25 devnull }
101 4a4a7c3e 2004-04-25 devnull void
102 4a4a7c3e 2004-04-25 devnull readbuf(int fd)
103 4a4a7c3e 2004-04-25 devnull {
104 4a4a7c3e 2004-04-25 devnull int lastwascolon = 0;
105 4a4a7c3e 2004-04-25 devnull long c;
106 4a4a7c3e 2004-04-25 devnull int linesiz = 0;
107 4a4a7c3e 2004-04-25 devnull
108 4a4a7c3e 2004-04-25 devnull Binit(&bin, fd, OREAD);
109 4a4a7c3e 2004-04-25 devnull do{
110 4a4a7c3e 2004-04-25 devnull if(nchars++ >= nalloc)
111 4a4a7c3e 2004-04-25 devnull morechars();
112 4a4a7c3e 2004-04-25 devnull *cbufp++ = c = Bgetrune(&bin);
113 4a4a7c3e 2004-04-25 devnull linesiz++;
114 4a4a7c3e 2004-04-25 devnull if(c == '\t') {
115 4a4a7c3e 2004-04-25 devnull cbufp[-1] = L' ';
116 4a4a7c3e 2004-04-25 devnull while(linesiz%TAB != 0) {
117 4a4a7c3e 2004-04-25 devnull if(nchars++ >= nalloc)
118 4a4a7c3e 2004-04-25 devnull morechars();
119 4a4a7c3e 2004-04-25 devnull *cbufp++ = L' ';
120 4a4a7c3e 2004-04-25 devnull linesiz++;
121 4a4a7c3e 2004-04-25 devnull }
122 4a4a7c3e 2004-04-25 devnull }
123 4a4a7c3e 2004-04-25 devnull if(colonflag && c == ':')
124 4a4a7c3e 2004-04-25 devnull lastwascolon++;
125 4a4a7c3e 2004-04-25 devnull else if(lastwascolon){
126 4a4a7c3e 2004-04-25 devnull if(c == '\n'){
127 4a4a7c3e 2004-04-25 devnull --nchars; /* skip newline */
128 4a4a7c3e 2004-04-25 devnull *cbufp = L'\0';
129 4a4a7c3e 2004-04-25 devnull while(nchars > 0 && cbuf[--nchars] != '\n')
130 4a4a7c3e 2004-04-25 devnull ;
131 4a4a7c3e 2004-04-25 devnull if(nchars)
132 4a4a7c3e 2004-04-25 devnull nchars++;
133 4a4a7c3e 2004-04-25 devnull columnate();
134 4a4a7c3e 2004-04-25 devnull if (nchars)
135 4a4a7c3e 2004-04-25 devnull Bputc(&bout, '\n');
136 4a4a7c3e 2004-04-25 devnull Bprint(&bout, "%S", cbuf+nchars);
137 4a4a7c3e 2004-04-25 devnull nchars = 0;
138 4a4a7c3e 2004-04-25 devnull cbufp = cbuf;
139 4a4a7c3e 2004-04-25 devnull }
140 4a4a7c3e 2004-04-25 devnull lastwascolon = 0;
141 4a4a7c3e 2004-04-25 devnull }
142 4a4a7c3e 2004-04-25 devnull if(c == '\n')
143 4a4a7c3e 2004-04-25 devnull linesiz = 0;
144 4a4a7c3e 2004-04-25 devnull }while(c >= 0);
145 4a4a7c3e 2004-04-25 devnull }
146 4a4a7c3e 2004-04-25 devnull void
147 4a4a7c3e 2004-04-25 devnull scanwords(void)
148 4a4a7c3e 2004-04-25 devnull {
149 4a4a7c3e 2004-04-25 devnull Rune *p, *q;
150 4a4a7c3e 2004-04-25 devnull int i, w;
151 4a4a7c3e 2004-04-25 devnull
152 4a4a7c3e 2004-04-25 devnull nwords=0;
153 4a4a7c3e 2004-04-25 devnull maxwidth=0;
154 4a4a7c3e 2004-04-25 devnull for(p = q = cbuf, i = 0; i < nchars; i++){
155 4a4a7c3e 2004-04-25 devnull if(*p++ == L'\n'){
156 4a4a7c3e 2004-04-25 devnull if(nwords >= nwalloc){
157 4a4a7c3e 2004-04-25 devnull nwalloc += WORD_ALLOC_QUANTA;
158 4a4a7c3e 2004-04-25 devnull if((word = realloc(word, nwalloc*sizeof(*word)))==0)
159 4a4a7c3e 2004-04-25 devnull error("out of memory");
160 4a4a7c3e 2004-04-25 devnull }
161 4a4a7c3e 2004-04-25 devnull word[nwords++] = q;
162 4a4a7c3e 2004-04-25 devnull p[-1] = L'\0';
163 4a4a7c3e 2004-04-25 devnull w = wordsize(q, p-q-1);
164 4a4a7c3e 2004-04-25 devnull if(w > maxwidth)
165 4a4a7c3e 2004-04-25 devnull maxwidth = w;
166 4a4a7c3e 2004-04-25 devnull q = p;
167 4a4a7c3e 2004-04-25 devnull }
168 4a4a7c3e 2004-04-25 devnull }
169 4a4a7c3e 2004-04-25 devnull }
170 4a4a7c3e 2004-04-25 devnull
171 4a4a7c3e 2004-04-25 devnull void
172 4a4a7c3e 2004-04-25 devnull columnate(void)
173 4a4a7c3e 2004-04-25 devnull {
174 4a4a7c3e 2004-04-25 devnull int i, j;
175 4a4a7c3e 2004-04-25 devnull int words_per_line;
176 4a4a7c3e 2004-04-25 devnull int nlines;
177 4a4a7c3e 2004-04-25 devnull int col;
178 4a4a7c3e 2004-04-25 devnull int endcol;
179 4a4a7c3e 2004-04-25 devnull
180 4a4a7c3e 2004-04-25 devnull
181 4a4a7c3e 2004-04-25 devnull scanwords();
182 4a4a7c3e 2004-04-25 devnull if(nwords==0)
183 4a4a7c3e 2004-04-25 devnull return;
184 4a4a7c3e 2004-04-25 devnull maxwidth = nexttab(maxwidth+mintab-1);
185 4a4a7c3e 2004-04-25 devnull words_per_line = linewidth/maxwidth;
186 4a4a7c3e 2004-04-25 devnull if(words_per_line <= 0)
187 4a4a7c3e 2004-04-25 devnull words_per_line = 1;
188 4a4a7c3e 2004-04-25 devnull nlines=(nwords+words_per_line-1)/words_per_line;
189 4a4a7c3e 2004-04-25 devnull for(i = 0; i < nlines; i++){
190 4a4a7c3e 2004-04-25 devnull col = endcol = 0;
191 4a4a7c3e 2004-04-25 devnull for(j = i; j < nwords; j += nlines){
192 4a4a7c3e 2004-04-25 devnull endcol += maxwidth;
193 4a4a7c3e 2004-04-25 devnull Bprint(&bout, "%S", word[j]);
194 4a4a7c3e 2004-04-25 devnull col += wordsize(word[j], runestrlen(word[j]));
195 4a4a7c3e 2004-04-25 devnull if(j+nlines < nwords){
196 4a4a7c3e 2004-04-25 devnull if(tabflag) {
197 4a4a7c3e 2004-04-25 devnull while(col < endcol){
198 4a4a7c3e 2004-04-25 devnull Bputc(&bout, '\t');
199 4a4a7c3e 2004-04-25 devnull col = nexttab(col);
200 4a4a7c3e 2004-04-25 devnull }
201 4a4a7c3e 2004-04-25 devnull }else{
202 4a4a7c3e 2004-04-25 devnull while(col < endcol){
203 4a4a7c3e 2004-04-25 devnull Bputc(&bout, ' ');
204 4a4a7c3e 2004-04-25 devnull col++;
205 4a4a7c3e 2004-04-25 devnull }
206 4a4a7c3e 2004-04-25 devnull }
207 4a4a7c3e 2004-04-25 devnull }
208 4a4a7c3e 2004-04-25 devnull }
209 4a4a7c3e 2004-04-25 devnull Bputc(&bout, '\n');
210 4a4a7c3e 2004-04-25 devnull }
211 4a4a7c3e 2004-04-25 devnull }
212 4a4a7c3e 2004-04-25 devnull
213 4a4a7c3e 2004-04-25 devnull int
214 4a4a7c3e 2004-04-25 devnull wordsize(Rune *w, int nw)
215 4a4a7c3e 2004-04-25 devnull {
216 4a4a7c3e 2004-04-25 devnull if(nw < 0)
217 4a4a7c3e 2004-04-25 devnull abort();
218 4a4a7c3e 2004-04-25 devnull if(font)
219 4a4a7c3e 2004-04-25 devnull return runestringnwidth(font, w, nw);
220 4a4a7c3e 2004-04-25 devnull return nw;
221 4a4a7c3e 2004-04-25 devnull }
222 4a4a7c3e 2004-04-25 devnull
223 4a4a7c3e 2004-04-25 devnull int
224 4a4a7c3e 2004-04-25 devnull nexttab(int col)
225 4a4a7c3e 2004-04-25 devnull {
226 4a4a7c3e 2004-04-25 devnull if(tabwid){
227 4a4a7c3e 2004-04-25 devnull col += tabwid;
228 4a4a7c3e 2004-04-25 devnull col -= col%tabwid;
229 4a4a7c3e 2004-04-25 devnull return col;
230 4a4a7c3e 2004-04-25 devnull }
231 fb5ca0b9 2004-04-25 devnull return col+1;
232 4a4a7c3e 2004-04-25 devnull }
233 4a4a7c3e 2004-04-25 devnull
234 4a4a7c3e 2004-04-25 devnull void
235 4a4a7c3e 2004-04-25 devnull morechars(void)
236 4a4a7c3e 2004-04-25 devnull {
237 4a4a7c3e 2004-04-25 devnull nalloc += ALLOC_QUANTA;
238 4a4a7c3e 2004-04-25 devnull if((cbuf = realloc(cbuf, nalloc*sizeof(*cbuf))) == 0)
239 4a4a7c3e 2004-04-25 devnull error("out of memory");
240 4a4a7c3e 2004-04-25 devnull cbufp = cbuf+nchars-1;
241 4a4a7c3e 2004-04-25 devnull }
242 4a4a7c3e 2004-04-25 devnull
243 4a4a7c3e 2004-04-25 devnull /*
244 4a4a7c3e 2004-04-25 devnull * These routines discover the width of the display.
245 4a4a7c3e 2004-04-25 devnull * It takes some work. If we do the easy calls to the
246 4a4a7c3e 2004-04-25 devnull * draw library, the screen flashes due to repainting
247 4a4a7c3e 2004-04-25 devnull * when mc exits.
248 4a4a7c3e 2004-04-25 devnull */
249 4a4a7c3e 2004-04-25 devnull int
250 4a4a7c3e 2004-04-25 devnull windowrect(struct winsize *ws)
251 4a4a7c3e 2004-04-25 devnull {
252 4a4a7c3e 2004-04-25 devnull int tty;
253 4a4a7c3e 2004-04-25 devnull
254 4a4a7c3e 2004-04-25 devnull if((tty = open("/dev/tty", OWRITE)) < 0)
255 4a4a7c3e 2004-04-25 devnull tty = 1;
256 4a4a7c3e 2004-04-25 devnull
257 4a4a7c3e 2004-04-25 devnull if(ioctl(tty, TIOCGWINSZ, ws) < 0){
258 4a4a7c3e 2004-04-25 devnull if(tty != 1)
259 4a4a7c3e 2004-04-25 devnull close(tty);
260 4a4a7c3e 2004-04-25 devnull return -1;
261 4a4a7c3e 2004-04-25 devnull }
262 4a4a7c3e 2004-04-25 devnull if(tty != 1)
263 4a4a7c3e 2004-04-25 devnull close(tty);
264 4a4a7c3e 2004-04-25 devnull return 0;
265 4a4a7c3e 2004-04-25 devnull }
266 4a4a7c3e 2004-04-25 devnull
267 4a4a7c3e 2004-04-25 devnull void
268 4a4a7c3e 2004-04-25 devnull getwidth(void)
269 4a4a7c3e 2004-04-25 devnull {
270 8cc98332 2005-01-04 devnull CFsys *fs;
271 4a4a7c3e 2004-04-25 devnull char buf[500], *p, *f[10];
272 4a4a7c3e 2004-04-25 devnull int fd, n, nf;
273 4a4a7c3e 2004-04-25 devnull struct winsize ws;
274 4a4a7c3e 2004-04-25 devnull
275 4a4a7c3e 2004-04-25 devnull if((p = getenv("winid")) != nil){
276 4a4a7c3e 2004-04-25 devnull fs = nsmount("acme", "");
277 4a4a7c3e 2004-04-25 devnull if(fs == nil)
278 4a4a7c3e 2004-04-25 devnull return;
279 4a4a7c3e 2004-04-25 devnull snprint(buf, sizeof buf, "acme/%d/ctl", atoi(p));
280 4a4a7c3e 2004-04-25 devnull if((fd = fsopenfd(fs, buf, OREAD)) < 0)
281 4a4a7c3e 2004-04-25 devnull return;
282 4a4a7c3e 2004-04-25 devnull if((n=readn(fd, buf, sizeof buf-1)) <= 0)
283 4a4a7c3e 2004-04-25 devnull return;
284 4a4a7c3e 2004-04-25 devnull buf[n] = 0;
285 4a4a7c3e 2004-04-25 devnull if((nf=tokenize(buf, f, nelem(f))) < 7)
286 4a4a7c3e 2004-04-25 devnull return;
287 4a4a7c3e 2004-04-25 devnull tabwid = 0;
288 4a4a7c3e 2004-04-25 devnull if(nf >= 8 && (tabwid = atoi(f[7])) == 0)
289 4a4a7c3e 2004-04-25 devnull return;
290 4a4a7c3e 2004-04-25 devnull if((font = openfont(nil, f[6])) == nil)
291 4a4a7c3e 2004-04-25 devnull return;
292 4a4a7c3e 2004-04-25 devnull mintab = stringwidth(font, "0");
293 4a4a7c3e 2004-04-25 devnull if(tabwid == 0)
294 4a4a7c3e 2004-04-25 devnull tabwid = mintab*4;
295 4a4a7c3e 2004-04-25 devnull linewidth = atoi(f[5]);
296 4a4a7c3e 2004-04-25 devnull tabflag = 1;
297 4a4a7c3e 2004-04-25 devnull return;
298 4a4a7c3e 2004-04-25 devnull }
299 4a4a7c3e 2004-04-25 devnull
300 fb5ca0b9 2004-04-25 devnull if((p = getenv("TERM")) != nil && strcmp(p, "9term") == 0)
301 4a4a7c3e 2004-04-25 devnull if((p = getenv("font")) != nil)
302 4a4a7c3e 2004-04-25 devnull font = openfont(nil, p);
303 fb5ca0b9 2004-04-25 devnull
304 4a4a7c3e 2004-04-25 devnull if(windowrect(&ws) < 0)
305 4a4a7c3e 2004-04-25 devnull return;
306 fb5ca0b9 2004-04-25 devnull if(ws.ws_xpixel == 0)
307 4a4a7c3e 2004-04-25 devnull font = nil;
308 fb5ca0b9 2004-04-25 devnull if(font){
309 fb5ca0b9 2004-04-25 devnull mintab = stringwidth(font, "0");
310 fb5ca0b9 2004-04-25 devnull if((p = getenv("tabstop")) != nil)
311 fb5ca0b9 2004-04-25 devnull tabwid = atoi(p)*mintab;
312 fb5ca0b9 2004-04-25 devnull else
313 fb5ca0b9 2004-04-25 devnull tabwid = 4*mintab;
314 fb5ca0b9 2004-04-25 devnull tabflag = 1;
315 fb5ca0b9 2004-04-25 devnull linewidth = ws.ws_xpixel;
316 fb5ca0b9 2004-04-25 devnull }else
317 fb5ca0b9 2004-04-25 devnull linewidth = ws.ws_col;
318 4a4a7c3e 2004-04-25 devnull }
319 4a4a7c3e 2004-04-25 devnull