From 3af19de95fff6cc2b6cd8944fd2221c322f39708 Mon Sep 17 00:00:00 2001
From: Dan-J-D <52302961+Dan-J-D@users.noreply.github.com>
Date: Thu, 7 Dec 2023 13:06:49 -0500
Subject: [PATCH] added the ability to -w and rebuild on change

---
 README.md       |   7 +-
 cmakeauto.c     | 200 +++++++++++++++++++++++++++++++++++++++++++++---
 cmakeauto.h     |  19 ++++-
 cmakeautoargs.c |  52 +++++++++++--
 4 files changed, 258 insertions(+), 20 deletions(-)

diff --git a/README.md b/README.md
index f51c996..83b42aa 100644
--- a/README.md
+++ b/README.md
@@ -3,7 +3,7 @@ A tool to simplify CMake Toolchain Commands
 
 ### Usage:
 ```bash
-Usage: cmakeauto help|build|configure|template -b <build path> -s <source path> [-g <Generator>] [-m <Mode>] [-a <Architecture>] [-ei <Extra Init Params>] [-eb <Extra Build Params>] [-ar] [-t <Template>]
+makeauto help|build|configure|template -b <build path> -s <source path> [-g <Generator>] [-m <Mode>] [-a <Architecture>] [-ei <Extra Init Params>] [-eb <Extra Build Params>] [-w <Directories>] [-t <Template>]
         help: print help message
         build: build project
         configure: configure project
@@ -21,11 +21,12 @@ Parameters:
                 x64: 64-bit
         -ei: (Optional) This specifies extra init params to be passed to cmake.
         -eb: (Optional) This specifies extra build params to be passed to cmake.
-        -ar: (Optional) This specifies whether to auto reload the project when source files changed.
+        -w: (Optional) This specifies the directories to be watched and will auto rebuild if a change was found.
         -t: (Only for Template Action) This specifies the template to be used when creating a template project.
-				
+
 Templates:
         helloworld
+        helloworlddll
 ```
 
 ### Build with CMake For Release X64:
diff --git a/cmakeauto.c b/cmakeauto.c
index f5f9363..71988dc 100644
--- a/cmakeauto.c
+++ b/cmakeauto.c
@@ -35,6 +35,19 @@ bool cma_file_exists(const char *path)
 #endif
 }
 
+bool cma_folder_exists(const char *path)
+{
+#ifdef _WIN32
+	DWORD attr = GetFileAttributesA(path);
+	return attr != INVALID_FILE_ATTRIBUTES && (attr & FILE_ATTRIBUTE_DIRECTORY);
+#elif __linux__
+	struct stat st = {0};
+	return stat(path, &st) != -1 && S_ISDIR(st.st_mode);
+#else
+#error unsupported platform
+#endif
+}
+
 bool cma_copy_file(const char *src, const char *dst)
 {
 #ifdef _WIN32
@@ -252,22 +265,174 @@ bool cma_create_process(char *filename,
 	return true;
 }
 
+#ifdef __linux__
+bool cma_watch_add_folder_callback(const char *abspath, // return false to stop iterating
+																	 const char *relpath,
+																	 const char *name,
+																	 bool isfolder /* 0 = file, 1 = folder */,
+																	 void *userdata)
+{
+	CMakeAutoConfig *config = (CMakeAutoConfig *)userdata;
+	if (isfolder)
+	{
+		if (config->watchfolderhandles_count >= WATCHFOLDER_MAX_LEN)
+		{
+			printf("max watch folders reached\n");
+			return false;
+		}
+
+		config->watchfolderhandles[config->watchfolderhandles_count] =
+				inotify_add_watch(config->watchfolderhandles[0], abspath, IN_CREATE | IN_MODIFY | IN_DELETE);
+		config->watchfolderhandles_count++;
+	}
+
+	return true;
+}
+
+void cma_watch_remove_all_subfolders(CMakeAutoConfig *config)
+{
+	for (unsigned int i = config->watchfolders_count + 1; i < config->watchfolderhandles_count; i++)
+		inotify_rm_watch(config->watchfolderhandles[0], config->watchfolderhandles[i]);
+	config->watchfolderhandles_count = config->watchfolders_count + 1;
+}
+#endif
+
 bool cma_watch_folder_init(CMakeAutoConfig *config)
 {
-	/* Unimplemented */
-	return false;
+#ifdef _WIN32
+	config->watchfolderhandles_count = config->watchfolders_count;
+	config->watchfolderhandles = (watchfolderhandle_t *)malloc(sizeof(watchfolderhandle_t) * config->watchfolderhandles_count);
+	for (unsigned int i = 0; i < config->watchfolderhandles_count; i++)
+	{
+		config->watchfolderhandles[i] = FindFirstChangeNotificationA(config->watchfolders[i],
+																																 TRUE,
+																																 FILE_NOTIFY_CHANGE_LAST_WRITE |
+																																		 FILE_NOTIFY_CHANGE_CREATION |
+																																		 FILE_NOTIFY_CHANGE_FILE_NAME |
+																																		 FILE_NOTIFY_CHANGE_DIR_NAME);
+		if (config->watchfolderhandles[i] == INVALID_HANDLE_VALUE)
+		{
+			printf("failed to watch folder %s\n", config->srcdir);
+			return false;
+		}
+	}
+
+	return true;
+#elif __linux__
+	config->watchfolderhandles_count = config->watchfolders_count + 1;
+	config->watchfolderhandles = (watchfolderhandle_t *)malloc(sizeof(watchfolderhandle_t) * config->watchfolderhandles_count);
+	config->watchfolderhandles[0] = inotify_init();
+	if (config->watchfolderhandles[0] == -1)
+	{
+		printf("failed to watch folder %s\n", config->srcdir);
+		return false;
+	}
+
+	for (unsigned int i = 0; i < config->watchfolders_count; i++)
+	{
+		config->watchfolderhandles[i + 1] = inotify_add_watch(config->watchfolderhandles[0], config->watchfolders[i], IN_CREATE | IN_MODIFY | IN_DELETE);
+		if (config->watchfolderhandles[i + 1] < 0)
+		{
+			printf("failed to watch folder %s\n", config->srcdir);
+			return false;
+		}
+
+		cma_iterate_dir(config->watchfolders[i], ".", config, true, cma_watch_add_folder_callback);
+	}
+
+	return true;
+#else
+#error unsupported platform
+#endif
 }
 
 bool cma_watch_folder_wait_for_next_change(CMakeAutoConfig *config)
 {
-	/* Unimplemented */
-	return false;
+#ifdef _WIN32
+	while (true)
+	{
+		unsigned long wait_result = WaitForMultipleObjects(config->watchfolderhandles_count, config->watchfolderhandles, FALSE, INFINITE);
+		if (wait_result == WAIT_FAILED)
+		{
+			printf("failed to wait for folder change\n");
+			return false;
+		}
+
+		if (wait_result == WAIT_TIMEOUT)
+			continue;
+
+		if (wait_result >= WAIT_OBJECT_0 && wait_result < WAIT_OBJECT_0 + config->watchfolderhandles_count)
+		{
+			if (FindNextChangeNotification(config->watchfolderhandles[wait_result - WAIT_OBJECT_0]) == FALSE)
+			{
+				printf("failed to watch folder %s\n", config->srcdir);
+				return false;
+			}
+			break;
+		}
+	}
+
+	return true;
+#elif __linux__
+	while (true)
+	{
+		char buf[4096] ALIGNAS(struct inotify_event);
+		const struct inotify_event *event;
+		unsigned long long len;
+		char *ptr;
+
+		len = read(config->watchfolderhandles[0], buf, sizeof(buf));
+		if (len == -1 && errno != EAGAIN)
+		{
+			printf("failed to read inotify event\n");
+			return false;
+		}
+
+		if (len <= 0)
+		{
+			printf("inotify event len <= 0\n");
+			return false;
+		}
+
+		for (ptr = buf; ptr < buf + len; ptr += sizeof(struct inotify_event) + event->len)
+		{
+			event = (const struct inotify_event *)ptr;
+
+			if (event->mask & (IN_CREATE | IN_DELETE))
+			{
+				if (event->mask & IN_ISDIR)
+				{
+					cma_watch_remove_all_subfolders(config);
+
+					for (unsigned int i = 0; i < config->watchfolders_count; i++)
+						cma_iterate_dir(config->watchfolders[i], ".", config, true, cma_watch_add_folder_callback);
+				}
+			}
+		}
+
+		break;
+	}
+	return true;
+#else
+#error unsupported platform
+#endif
 }
 
 bool cma_watch_folder_close(CMakeAutoConfig *config)
 {
-	/* Unimplemented */
-	return false;
+#ifdef _WIN32
+	for (unsigned int i = 0; i < config->watchfolderhandles_count; i++)
+		FindCloseChangeNotification(config->watchfolderhandles[i]);
+	return true;
+#elif __linux__
+	printf("closing watch folder\n");
+	for (unsigned int i = 1; i < config->watchfolderhandles_count; i++)
+		inotify_rm_watch(config->watchfolderhandles[0], config->watchfolderhandles[i]);
+	close(config->watchfolderhandles[0]);
+	return true;
+#else
+#error unsupported platform
+#endif
 }
 
 bool cma_init_proj(CMakeAutoConfig *config)
@@ -479,11 +644,26 @@ int main(int argc, char **argv)
 	{
 	case CMAKE_AUTO_ACTION_BUILD:
 	{
-		if (cma_init_proj(&config))
-			cma_build(&config);
+		if (config.options & CMAKE_AUTO_OPTION_AUTO_RELOAD)
+			if (!cma_watch_folder_init(&config))
+			{
+				printf("failed to init watch folder\n");
+				break;
+			}
+
+		do
+		{
+			if (cma_init_proj(&config))
+				cma_build(&config);
+
+			if (config.options & CMAKE_AUTO_OPTION_AUTO_RELOAD)
+				if (!cma_watch_folder_wait_for_next_change(&config))
+					printf("failed to wait for next change\n");
+		} while (config.options & CMAKE_AUTO_OPTION_AUTO_RELOAD);
 
 		if (config.options & CMAKE_AUTO_OPTION_AUTO_RELOAD)
-			printf("auto reload is not implemented yet\n");
+			if (!cma_watch_folder_close(&config))
+				printf("failed to close watch folder\n");
 		break;
 	}
 	case CMAKE_AUTO_ACTION_CONFIGURE:
@@ -500,7 +680,7 @@ int main(int argc, char **argv)
 		strcat_s(buf, FILE_MAX_PATH, "templates" FILE_SEPERATOR);
 		strcat_s(buf, FILE_MAX_PATH, config.template);
 
-		if (!cma_file_exists)
+		if (!cma_folder_exists(buf))
 		{
 			printf("template not found\n");
 			break;
diff --git a/cmakeauto.h b/cmakeauto.h
index f1be5b1..dd429e9 100644
--- a/cmakeauto.h
+++ b/cmakeauto.h
@@ -9,10 +9,13 @@
 
 #ifdef _WIN32
 #include <Windows.h>
+#include <fileapi.h>
 
 #define FILE_MAX_PATH _MAX_PATH
 #define FILE_SEPERATOR "\\"
 #define FILE_SEPERATOR_CHAR '\\'
+
+typedef void *watchfolderhandle_t;
 #elif __linux__
 #include <sys/types.h>
 #include <sys/stat.h>
@@ -22,15 +25,25 @@
 #include <sys/wait.h>
 #include <fcntl.h>
 #include <errno.h>
+#include <sys/inotify.h>
+
+#ifdef __GNUC__
+#define ALIGNAS(TYPE) __attribute__((aligned(__alignof__(TYPE))))
+#else
+#define ALIGNAS(TYPE) /* empty */
+#endif
 
 #define FILE_MAX_PATH PATH_MAX
 #define FILE_SEPERATOR "/"
 #define FILE_SEPERATOR_CHAR '/'
+
+typedef int watchfolderhandle_t;
 #else
 #error unsupported platform
 #endif
 
 #define GERNERATOR_MAX_LEN 64
+#define WATCHFOLDER_MAX_LEN 64
 
 #ifdef __linux__
 #define stricmp strcasecmp
@@ -105,7 +118,10 @@ typedef struct
 	char *extra_init_args;
 	char *extra_build_args;
 
-	void *watchfolderhandle;
+	char watchfolders[FILE_MAX_PATH + 1][WATCHFOLDER_MAX_LEN];
+	unsigned int watchfolders_count;
+	watchfolderhandle_t *watchfolderhandles;
+	unsigned int watchfolderhandles_count;
 } CMakeAutoConfig;
 
 /**
@@ -125,6 +141,7 @@ void cma_print_usage();
 bool cma_abspath(char *buf, size_t size, const char *path);
 bool cma_create_dir_include_existing(const char *path);
 bool cma_file_exists(const char *path);
+bool cma_folder_exists(const char *path);
 bool cma_copy_file(const char *src, const char *dst);
 bool cma_get_current_process_absfilepath(char *buf, size_t size);
 bool cma_get_workdir(char *buf, size_t size);
diff --git a/cmakeautoargs.c b/cmakeautoargs.c
index 440e04f..620de0c 100644
--- a/cmakeautoargs.c
+++ b/cmakeautoargs.c
@@ -10,7 +10,7 @@ bool print_templates(const char *abspath, const char *relpath, const char *name,
 
 void cma_print_usage()
 {
-	printf("Usage: cmakeauto help|build|configure|template -b <build path> -s <source path> [-g <Generator>] [-m <Mode>] [-a <Architecture>] [-ei <Extra Init Params>] [-eb <Extra Build Params>] [-ar] [-t <Template>]\n");
+	printf("Usage: cmakeauto help|build|configure|template -b <build path> -s <source path> [-g <Generator>] [-m <Mode>] [-a <Architecture>] [-ei <Extra Init Params>] [-eb <Extra Build Params>] [-w <Directories>] [-t <Template>]\n");
 	printf("\thelp: print help message\n");
 	printf("\tbuild: build project\n");
 	printf("\tconfigure: configure project\n");
@@ -26,7 +26,7 @@ void cma_print_usage()
 	printf("\t\tx64: 64-bit\n");
 	printf("\t-ei: (Optional) This specifies extra init params to be passed to cmake.\n");
 	printf("\t-eb: (Optional) This specifies extra build params to be passed to cmake.\n");
-	printf("\t-ar: (Optional) This specifies whether to auto reload the project when source files changed.\n");
+	printf("\t-w: (Optional) This specifies the directories to be watched and will auto rebuild if a change was found.\n");
 	printf("\t-t: (Only for Template Action) This specifies the template to be used when creating a template project.\n");
 
 	char buf[FILE_MAX_PATH + 1];
@@ -170,15 +170,55 @@ bool cma_parse_args(int argc, char **argv, CMakeAutoConfig *config)
 					i++;
 				}
 			}
-			else if (stricmp(argv[i], "-ar") == 0)
-			{
-				config->options |= CMAKE_AUTO_OPTION_AUTO_RELOAD;
-			}
 			else if (stricmp(argv[i], "-h") == 0)
 			{
 				config->action = CMAKE_AUTO_ACTION_HELP;
 				return true;
 			}
+			else if (stricmp(argv[i], "-w") == 0)
+			{
+				config->options |= CMAKE_AUTO_OPTION_AUTO_RELOAD;
+
+				if (i + 1 < argc)
+				{
+					char temppath[FILE_MAX_PATH + 1];
+					char *paths = argv[i + 1];
+					while (*paths)
+					{
+						if (config->watchfolderhandles_count >= WATCHFOLDER_MAX_LEN)
+						{
+							printf("max watch folders reached\n");
+							break;
+						}
+
+						char *endpath = paths;
+						while (*endpath && *endpath != ';')
+							endpath++;
+
+						memset(temppath, 0, FILE_MAX_PATH + 1);
+						memcpy(temppath, paths, endpath - paths);
+
+						if (!cma_abspath(config->watchfolders[config->watchfolders_count], FILE_MAX_PATH, temppath))
+						{
+							printf("invalid or error happened when parsing watch folder\n");
+							return false;
+						}
+						config->watchfolders_count++;
+
+						while (*paths && *paths != ';')
+							paths++;
+						if (*paths == ';')
+							paths++;
+					}
+
+					i++;
+				}
+				else
+				{
+					printf("no watch folder specified\n");
+					return false;
+				}
+			}
 			else
 			{
 				printf("unknown argument: %s\n", argv[i]);