1#include <stdio.h>
2#include <string.h>
3#include <signal.h>
4#include <sys/wait.h>
5#include <ctype.h>
6#include <stdlib.h>
7#include <unistd.h>
8#include <dirent.h>
9#include <limits.h>
10#include <locale.h>
11
12#include <histedit.h>
13
14
15static int continuation;
16volatile sig_atomic_t gotsig;
17
18static wchar_t *
19prompt(EditLine *el)
20{
21	static wchar_t a[] = L"\1\033[7m\1Edit$\1\033[0m\1 ";
22	static wchar_t b[] = L"Edit> ";
23
24	return continuation ? b : a;
25}
26
27
28static void
29sig(int i)
30{
31	gotsig = i;
32}
33
34const char *
35my_wcstombs(const wchar_t *wstr)
36{
37	static struct {
38		char *str;
39		int len;
40	} buf;
41
42	int needed = wcstombs(0, wstr, 0) + 1;
43	if (needed > buf.len) {
44		buf.str = malloc(needed);
45		buf.len = needed;
46	}
47	wcstombs(buf.str, wstr, needed);
48	buf.str[needed - 1] = 0;
49
50	return buf.str;
51}
52
53
54static unsigned char
55complete(EditLine *el, int ch)
56{
57	DIR *dd = opendir(".");
58	struct dirent *dp;
59	const wchar_t *ptr;
60	char *buf, *bptr;
61	const LineInfoW *lf = el_wline(el);
62	int len, mblen, i;
63	unsigned char res;
64
65	/* Find the last word */
66	for (ptr = lf->cursor -1; !iswspace(*ptr) && ptr > lf->buffer; --ptr)
67		continue;
68	len = lf->cursor - ++ptr;
69
70	/* Convert last word to multibyte encoding, so we can compare to it */
71	wctomb(NULL, 0); /* Reset shift state */
72	mblen = MB_LEN_MAX * len + 1;
73	buf = bptr =(char *)malloc(mblen);
74	for (i = 0; i < len; ++i) {
75		/* Note: really should test for -1 return from wctomb */
76		bptr += wctomb(bptr, ptr[i]);
77	}
78	*bptr = 0; /* Terminate multibyte string */
79	mblen = bptr - buf;
80
81	/* Scan directory for matching name */
82	for (dp = readdir(dd); dp != NULL; dp = readdir(dd)) {
83		if (mblen > strlen(dp->d_name))
84			continue;
85		if (strncmp(dp->d_name, buf, mblen) == 0) {
86			if (el_insertstr(el, &dp->d_name[mblen]) == -1)
87				res = CC_ERROR;
88			else
89				res = CC_REFRESH;
90			break;
91		}
92	}
93
94	closedir(dd);
95	free(buf);
96	return res;
97}
98
99
100int
101main(int argc, char *argv[])
102{
103	EditLine *el = NULL;
104	int numc, ncontinuation;
105	const wchar_t *line;
106	TokenizerW *tok;
107	HistoryW *hist;
108	HistEventW ev;
109#ifdef DEBUG
110	int i;
111#endif
112
113	setlocale(LC_ALL, "");
114
115	(void)signal(SIGINT,  sig);
116	(void)signal(SIGQUIT, sig);
117	(void)signal(SIGHUP,  sig);
118	(void)signal(SIGTERM, sig);
119
120	hist = history_winit();		/* Init built-in history     */
121	history_w(hist, &ev, H_SETSIZE, 100);	/* Remember 100 events	     */
122
123	tok = tok_winit(NULL);			/* Init the tokenizer	     */
124
125	el = el_init(argv[0], stdin, stdout, stderr);
126
127	el_wset(el, EL_EDITOR, L"vi");		/* Default editor is vi	     */
128	el_wset(el, EL_SIGNAL, 1);		/* Handle signals gracefully */
129	el_wset(el, EL_PROMPT_ESC, prompt, '\1'); /* Set the prompt function */
130
131	el_wset(el, EL_HIST, history_w, hist);	/* FIXME - history_w? */
132
133					/* Add a user-defined function	*/
134	el_wset(el, EL_ADDFN, L"ed-complete", L"Complete argument", complete);
135
136					/* Bind <tab> to it */
137	el_wset(el, EL_BIND, L"^I", L"ed-complete", NULL);
138
139	/*
140	* Bind j, k in vi command mode to previous and next line, instead
141	* of previous and next history.
142	*/
143	el_wset(el, EL_BIND, L"-a", L"k", L"ed-prev-line", NULL);
144	el_wset(el, EL_BIND, L"-a", L"j", L"ed-next-line", NULL);
145
146	/* Source the user's defaults file. */
147	el_source(el, NULL);
148
149	while((line = el_wgets(el, &numc)) != NULL && numc != 0) {
150		int ac, cc, co, rc;
151		const wchar_t **av;
152
153		const LineInfoW *li;
154		li = el_wline(el);
155
156#ifdef DEBUG
157		(void)fwprintf(stderr, L"==> got %d %ls", numc, line);
158		(void)fwprintf(stderr, L"  > li `%.*ls_%.*ls'\n",
159		    (li->cursor - li->buffer), li->buffer,
160		    (li->lastchar - 1 - li->cursor),
161		    (li->cursor >= li->lastchar) ? L"" : li->cursor);
162#endif
163
164		if (gotsig) {
165			(void)fprintf(stderr, "Got signal %d.\n", gotsig);
166			gotsig = 0;
167			el_reset(el);
168		}
169
170		if(!continuation && numc == 1)
171			continue;	/* Only got a linefeed */
172
173		ac = cc = co = 0;
174		ncontinuation = tok_wline(tok, li, &ac, &av, &cc, &co);
175		if (ncontinuation < 0) {
176			(void) fprintf(stderr, "Internal error\n");
177			continuation = 0;
178			continue;
179		}
180
181#ifdef DEBUG
182		(void)fprintf(stderr, "  > nc %d ac %d cc %d co %d\n",
183			ncontinuation, ac, cc, co);
184#endif
185		history_w(hist, &ev, continuation ? H_APPEND : H_ENTER, line);
186
187		continuation = ncontinuation;
188		ncontinuation = 0;
189		if(continuation)
190			continue;
191
192#ifdef DEBUG
193		for (i = 0; i < ac; ++i) {
194			(void)fwprintf(stderr, L"  > arg# %2d ", i);
195			if (i != cc)
196				(void)fwprintf(stderr, L"`%ls'\n", av[i]);
197			else
198				(void)fwprintf(stderr, L"`%.*ls_%ls'\n",
199				    co, av[i], av[i] + co);
200		}
201#endif
202
203		if (wcscmp (av[0], L"history") == 0) {
204			switch(ac) {
205			case 1:
206				for(rc = history_w(hist, &ev, H_LAST);
207				     rc != -1;
208				     rc = history_w(hist, &ev, H_PREV))
209					(void)fwprintf(stdout, L"%4d %ls",
210					     ev.num, ev.str);
211				break;
212			case 2:
213				if (wcscmp(av[1], L"clear") == 0)
214					history_w(hist, &ev, H_CLEAR);
215				else
216					goto badhist;
217				break;
218			case 3:
219				if (wcscmp(av[1], L"load") == 0)
220					history_w(hist, &ev, H_LOAD,
221					    my_wcstombs(av[2]));
222				else if (wcscmp(av[1], L"save") == 0)
223					history_w(hist, &ev, H_SAVE,
224					    my_wcstombs(av[2]));
225				else
226					goto badhist;
227				break;
228			badhist:
229			default:
230				(void)fprintf(stderr,
231				    "Bad history arguments\n");
232				break;
233			}
234		} else if (el_wparse(el, ac, av) == -1) {
235			switch (fork()) {
236			case 0: {
237				Tokenizer *ntok = tok_init(NULL);
238				int nargc;
239				const char **nav;
240				tok_str(ntok, my_wcstombs(line), &nargc, &nav);
241				execvp(nav[0],(char **)nav);
242				perror(nav[0]);
243				_exit(1);
244				/* NOTREACHED */
245				break;
246			}
247			case -1:
248				perror("fork");
249				break;
250			default:
251				if (wait(&rc) == -1)
252					perror("wait");
253				(void)fprintf(stderr, "Exit %x\n", rc);
254				break;
255			}
256		}
257
258		tok_wreset(tok);
259	}
260
261	el_end(el);
262	tok_wend(tok);
263	history_wend(hist);
264
265	fprintf(stdout, "\n");
266	return 0;
267}
268
269
270