X-Git-Url: http://git.mutantstargoat.com/user/nuclear/?a=blobdiff_plain;f=src%2Fmain.c;h=fa7dbb1cf15037a2315e380ebbce44ab708a0016;hb=85b4b3f42fd47fbc9baff6f0aa988037f1b73bc1;hp=d6ec65bc64fb4f542cfcd51724ed07408204c074;hpb=2ba6401e39bcfaaccaa45e6b7ef780a7a15b0c48;p=oftp

diff --git a/src/main.c b/src/main.c
index d6ec65b..fa7dbb1 100644
--- a/src/main.c
+++ b/src/main.c
@@ -1,19 +1,41 @@
+#include <stdio.h>
+#include <string.h>
+#include <signal.h>
+#include <errno.h>
+#include <assert.h>
+#include <sys/select.h>
 #include "tgfx.h"
 #include "input.h"
+#include "util.h"
 #include "tui.h"
 #include "ftp.h"
 
-struct ftp *ftp;
-struct tui_widget *uilist;
+void updateui(void);
+int proc_input(void);
+int keypress(int key);
+int parse_args(int argc, char **argv);
 
-int main(void)
+static struct ftp *ftp;
+static struct tui_widget *uilist;
+
+static char *host = "localhost";
+static int port = 21;
+
+int main(int argc, char **argv)
 {
-	union event ev;
+	int i, numsock, maxfd;
+	int ftpsock[16];
+	fd_set rdset;
+	struct timeval tv;
+
+	if(parse_args(argc, argv) == -1) {
+		return 1;
+	}
 
 	if(!(ftp = ftp_alloc())) {
 		return 1;
 	}
-	if(ftp_connect(ftp, "192.168.0.4", 21) == -1) {
+	if(ftp_connect(ftp, host, port) == -1) {
 		ftp_free(ftp);
 		return 1;
 	}
@@ -26,31 +48,198 @@ int main(void)
 	tg_clear();
 
 	uilist = tui_list("Remote", 0, 0, 40, 23, 0, 0);
-	tui_add_list_item(uilist, "first item");
-	tui_add_list_item(uilist, "second item");
-	tui_add_list_item(uilist, "another item");
-	tui_add_list_item(uilist, "foo");
 
-	tg_setcursor(0, 24);
+	tg_setcursor(0, 23);
 
 	tui_draw(uilist);
 
-	while(wait_input(&ev)) {
+	for(;;) {
+		FD_ZERO(&rdset);
+		maxfd = 0;
+
+		numsock = ftp_sockets(ftp, ftpsock, sizeof ftpsock);
+		for(i=0; i<numsock; i++) {
+			FD_SET(ftpsock[i], &rdset);
+			if(ftpsock[i] > maxfd) maxfd = ftpsock[i];
+		}
+
+#ifdef __unix__
+		FD_SET(0, &rdset);
+		tv.tv_sec = 120;
+		tv.tv_usec = 0;
+#else
+		tv.tv_sec = tv.tv_usec = 0;
+#endif
+
+		if(select(maxfd + 1, &rdset, 0, 0, &tv) == -1 && errno == EINTR) {
+			continue;
+		}
+
+#ifdef __unix__
+		if(FD_ISSET(0, &rdset)) {
+			if(proc_input() == -1) {
+				break;
+			}
+		}
+#endif
+
+		for(i=0; i<numsock; i++) {
+			if(FD_ISSET(ftpsock[i], &rdset)) {
+				ftp_handle(ftp, ftpsock[i]);
+			}
+		}
+
+		updateui();
+	}
+
+	tg_cleanup();
+	cleanup_input();
+	ftp_close(ftp);
+	ftp_free(ftp);
+	return 0;
+}
+
+static int cmpnames(const void *a, const void *b)
+{
+	const char *sa = *(const char**)a;
+	const char *sb = *(const char**)b;
+	int isdir_a, isdir_b;
+
+	isdir_a = sa[strlen(sa) - 1] == '/';
+	isdir_b = sb[strlen(sb) - 1] == '/';
+
+	if(isdir_a == isdir_b) {
+		return strcmp(sa, sb);
+	}
+	return isdir_a ? -1 : 1;
+}
+
+void updateui(void)
+{
+	struct ftp_dirent *ent;
+	unsigned int upd = 0;
+	char buf[128];
+
+	if(ftp->curdir_rem && strcmp(tui_get_title(uilist), ftp->curdir_rem) != 0) {
+		tui_set_title(uilist, ftp->curdir_rem);
+		upd |= 1;
+	}
+
+	if(ftp->modified & FTP_MOD_REMDIR) {
+		tui_clear_list(uilist);
+
+		ent = ftp->dent_rem;
+		while(ent) {
+			if(ent->type == FTP_DIR) {
+				sprintf(buf, "%s/", ent->name);
+				tui_add_list_item(uilist, buf);
+			} else {
+				tui_add_list_item(uilist, ent->name);
+			}
+			ent = ent->next;
+		}
+
+		tui_sort_list(uilist, cmpnames);
+		tui_list_select(uilist, 0);
+
+		ftp->modified &= ~FTP_MOD_REMDIR;
+		upd |= 1;
+	}
+
+	if(tui_isdirty(uilist) || upd & 1) {
+		tui_draw(uilist);
+	}
+}
+
+int proc_input(void)
+{
+	union event ev;
+
+	while(poll_input(&ev)) {
 		switch(ev.type) {
 		case EV_KEY:
-			if(ev.key.key == 27) goto done;
+			if(keypress(ev.key.key) == -1) {
+				return -1;
+			}
 			break;
 
 		default:
 			break;
 		}
 	}
+	return 0;
+}
 
-done:
-	tg_cleanup();
+int keypress(int key)
+{
+	switch(key) {
+	case 27:
+	case 'q':
+		return -1;
 
-	cleanup_input();
+	case KB_UP:
+		tui_list_sel_prev(uilist);
+		break;
+	case KB_DOWN:
+		tui_list_sel_next(uilist);
+		break;
+	case KB_LEFT:
+		tui_list_sel_start(uilist);
+		break;
+	case KB_RIGHT:
+		tui_list_sel_end(uilist);
+		break;
+
+	default:
+		break;
+	}
+	return 0;
+}
+
+static const char *usage = "Usage: %s [options] [hostname] [port]\n"
+	"Options:\n"
+	"  -h: print usage information and exit\n";
+
+int parse_args(int argc, char **argv)
+{
+	int i, argidx = 0;
+
+	for(i=1; i<argc; i++) {
+		if(argv[i][0] == '-') {
+			if(argv[i][2] != 0) {
+				goto inval;
+			}
+			switch(argv[i][1]) {
+			case 'h':
+				printf(usage, argv[0]);
+				exit(0);
+
+			default:
+				goto inval;
+			}
+
+		} else {
+			switch(argidx++) {
+			case 0:
+				host = argv[i];
+				break;
+
+			case 1:
+				if((port = atoi(argv[i])) <= 0) {
+					fprintf(stderr, "invalid port number: %s\n", argv[i]);
+					return -1;
+				}
+				break;
+
+			default:
+				goto inval;
+			}
+		}
+	}
 
-	ftp_free(ftp);
 	return 0;
+inval:
+	fprintf(stderr, "invalid argument: %s\n", argv[i]);
+	fprintf(stderr, usage, argv[0]);
+	return -1;
 }