X-Git-Url: http://git.mutantstargoat.com/user/nuclear/?p=instimg;a=blobdiff_plain;f=src%2Fmain.c;h=10b3a25bc4d76a3f3d50c732f1e955b3bd42ed8f;hp=68b4c98c64398f2385f57ed018cccf01d05c3675;hb=377e150b95639825a58c759d5ed51406fb14e429;hpb=df3263076df0ad443dd9262b6250820c038fa7f2

diff --git a/src/main.c b/src/main.c
index 68b4c98..10b3a25 100644
--- a/src/main.c
+++ b/src/main.c
@@ -1,29 +1,29 @@
 #include <stdio.h>
 #include <stdlib.h>
+#include <ctype.h>
 #include <windows.h>
 #include "widgets.h"
 #include "rawdisk.h"
 
 static struct wgt_window *win;
 static struct wgt_widget *lb_instto, *bn_inst, *bn_cancel, *cb_devs, *ck_usbonly;
+
 static struct rawdisk_device rawdev[64];
+static const char *items[64];
 static int num_rawdev;
 
+
+static void update_disks(void);
 static void onclick(struct wgt_widget *w);
 static void onmodify(struct wgt_widget *w);
+static void ck_usbonly_handler(struct wgt_widget *w);
+
 
 int WINAPI WinMain(HINSTANCE hinst, HINSTANCE hprevinst, char *cmdline, int showcmd)
 {
-	int i, x, y;
+	int x, y;
 	MSG msg;
-	static const char *items[64];
 
-	if((num_rawdev = rawdisk_detect(rawdev, sizeof rawdev / sizeof *rawdev)) == -1) {
-		return 1;
-	}
-	for(i=0; i<num_rawdev; i++) {
-		items[i] = rawdev[i].name;
-	}
 
 	if(!(win = wgt_window("256boss USB stick installer", 400, 300))) {
 		return 1;
@@ -31,12 +31,11 @@ int WINAPI WinMain(HINSTANCE hinst, HINSTANCE hprevinst, char *cmdline, int show
 	lb_instto = wgt_label(win, "Install to device:", 10, 10);
 
 	x = wgt_xpos_after(lb_instto, WGT_AUTO);
-	cb_devs = wgt_combo(win, items, num_rawdev, 0, x, 10, WGT_AUTO, WGT_AUTO,
-			onmodify);
+	cb_devs = wgt_combo(win, items, num_rawdev, 0, x, 10, 200, WGT_AUTO, onmodify);
 
 	y = wgt_ypos_after(cb_devs, 16);
 	ck_usbonly = wgt_checkbox(win, "only show USB devices", 1, x, y,
-			WGT_AUTO, WGT_AUTO, 0);
+			WGT_AUTO, WGT_AUTO, ck_usbonly_handler);
 
 	y = wgt_ypos_after(ck_usbonly, 16);
 	bn_inst = wgt_button(win, "Install", 10, y, WGT_AUTO, WGT_AUTO, onclick);
@@ -44,6 +43,8 @@ int WINAPI WinMain(HINSTANCE hinst, HINSTANCE hprevinst, char *cmdline, int show
 	x = wgt_xpos_after(bn_inst, WGT_AUTO);
 	bn_cancel = wgt_button(win, "Cancel", x, y, WGT_AUTO, WGT_AUTO, onclick);
 
+	update_disks();
+
 	while(GetMessage(&msg, 0, 0, 0)) {
 		TranslateMessage(&msg);
 		DispatchMessage(&msg);
@@ -53,9 +54,71 @@ int WINAPI WinMain(HINSTANCE hinst, HINSTANCE hprevinst, char *cmdline, int show
 	return 0;
 }
 
+static int isusbdev(struct rawdisk_device *dev)
+{
+	char *ptr = dev->path;
+	while(*ptr) {
+		int i;
+		char buf[3];
+		for(i=0; i<3; i++) {
+			buf[i] = tolower(ptr[i]);
+		}
+		if(memcmp(buf, "usb", 3) == 0) {
+			return 1;
+		}
+		ptr++;
+	}
+	return 0;
+}
+
+static void update_disks(void)
+{
+	int i;
+
+	for(i=0; i<num_rawdev; i++) {
+		free(rawdev[i].name);
+		free(rawdev[i].path);
+	}
+
+	if((num_rawdev = rawdisk_detect(rawdev, sizeof rawdev / sizeof *rawdev)) == -1) {
+		MessageBox(0, "Failed to detect storage devices!", 0, MB_OK);
+		exit(1);
+	}
+
+	if(wgt_checkbox_checked(ck_usbonly)) {
+		for(i=0; i<num_rawdev; i++) {
+			if(!isusbdev(rawdev + i)) {
+				free(rawdev[i].name);
+				free(rawdev[i].path);
+				rawdev[i] = rawdev[--num_rawdev];
+			}
+		}
+	}
+
+	for(i=0; i<num_rawdev; i++) {
+		items[i] = rawdev[i].name;
+	}
+
+	wgt_combo_setitems(cb_devs, items, num_rawdev);
+
+	if(num_rawdev) {
+		wgt_enable_widget(cb_devs);
+		wgt_enable_widget(bn_inst);
+	} else {
+		wgt_disable_widget(cb_devs);
+		wgt_disable_widget(bn_inst);
+	}
+}
+
 static void onclick(struct wgt_widget *w)
 {
-	MessageBox(0, "clicked", "clicked", MB_OK);
+	if(w == bn_inst) {
+		MessageBox(0, "clicked", "clicked", MB_OK);
+	} else if(w == bn_cancel) {
+		PostQuitMessage(0);
+	} else if(w == ck_usbonly) {
+		update_disks();
+	}
 }
 
 static void onmodify(struct wgt_widget *w)
@@ -65,4 +128,10 @@ static void onmodify(struct wgt_widget *w)
 		const char *selstr = wgt_get_combo_item(w, sel);
 		MessageBox(0, selstr, "selected item", MB_OK);
 	}
-}
\ No newline at end of file
+}
+
+static void ck_usbonly_handler(struct wgt_widget *w)
+{
+	printf("FOO\n");
+	update_disks();
+}