Improve the build script

1 files changed, 66 insertions, 9 deletions

diff --git a/nob.c b/nob.c
index a4f5755..4e42699 100644
--- a/nob.c
+++ b/nob.c
@@ -4,11 +4,17 @@
 #include "stdbool.h"
 
 #define BUILD_FOLDER "build/"
-#define SRC_FOLDER   "src/"
+#define SRC_FOLDER "src/"
 
-#define nob_cc_include_path(cmd, path) cmd_append(cmd, temp_sprintf("-I%s", path))
+#define nob_cc_include_path(cmd, path)                                         \
+  cmd_append(cmd, temp_sprintf("-I%s", path))
 #define nob_cc_no_link(cmd) cmd_append(cmd, "-c")
 
+typedef struct {
+  Nob_String_View *items;
+  size_t count;
+  size_t capacity;
+} File_Names;
 
 bool build_object(Cmd *cmd, const char *src_path, const char *o_path) {
   nob_cc(cmd);
@@ -22,23 +28,74 @@ bool build_object(Cmd *cmd, const char *src_path, const char *o_path) {
   return cmd_run_sync_and_reset(cmd);
 }
 
+bool build_all_src_files(Cmd *cmd, const char *src_path, const char *build_path,
+                         File_Names *filenames) {
+  File_Paths src_files = {0};
+  if (!read_entire_dir(src_path, &src_files))
+    return false;
+
+  for (int i = 0; i < src_files.count; i++) {
+    const char *file = src_files.items[i];
+    Nob_String_View file_sv = nob_sv_from_cstr(file);
+    if (!nob_sv_end_with(file_sv, ".c")) {
+      continue;
+    }
+
+    Nob_String_View filename = file_sv;
+    filename.count -= 2;
+    nob_da_append(filenames, filename);
+
+    Nob_String_Builder src_sb = {0};
+    nob_sb_append_cstr(&src_sb, src_path);
+    nob_sb_append_cstr(&src_sb, file);
+    nob_sb_append_null(&src_sb);
+
+    Nob_String_Builder out_sb = {0};
+    nob_sb_append_cstr(&out_sb, build_path);
+    nob_sb_appendf(&out_sb, SV_Fmt ".o", SV_Arg(filename));
+    nob_sb_append_null(&out_sb);
+
+    if (!build_object(cmd, src_sb.items, out_sb.items))
+      return false;
+
+    nob_sb_free(src_sb);
+    nob_sb_free(out_sb);
+  }
+  return true;
+}
+
 int main(int argc, char **argv) {
   NOB_GO_REBUILD_URSELF(argc, argv);
 
-  if (!nob_mkdir_if_not_exists(BUILD_FOLDER)) return 1;
+  if (!mkdir_if_not_exists(BUILD_FOLDER))
+    return 1;
 
-  Nob_Cmd cmd = {0};
+  Cmd cmd = {0};
 
-  if (!build_object(&cmd, SRC_FOLDER"main.c", BUILD_FOLDER"main.o")) return 1;
+  File_Names filenames = {0};
+  if (!build_all_src_files(&cmd, SRC_FOLDER, BUILD_FOLDER, &filenames))
+    return 1;
 
   nob_cc(&cmd);
   nob_cc_flags(&cmd);
   nob_cmd_append(&cmd, "-ggdb");
   nob_cc_include_path(&cmd, SRC_FOLDER);
-  nob_cc_inputs(&cmd, BUILD_FOLDER"main.o");
+
+  for (int i = 0; i < filenames.count; i++) {
+    Nob_String_View name = filenames.items[i];
+    Nob_String_Builder input_sb = {0};
+    nob_sb_append_cstr(&input_sb, BUILD_FOLDER);
+    nob_sb_append_buf(&input_sb, name.data, name.count);
+    nob_sb_appendf(&input_sb, ".o");
+    nob_sb_append_null(&input_sb);
+
+    nob_cmd_append(&cmd, input_sb.items);
+  }
+
   nob_cmd_append(&cmd, "-lm", "-lasound", "-lraylib");
-  nob_cc_output(&cmd, BUILD_FOLDER"crynth");
-  
-  if (!nob_cmd_run(&cmd)) return 1;
+  nob_cc_output(&cmd, BUILD_FOLDER "crynth");
+
+  if (!nob_cmd_run(&cmd))
+    return 1;
   return 0;
 }