First pass at formatter

1 files changed, 42 insertions, 20 deletions

diff --git a/src/tomo.c b/src/tomo.c
index d02cce74..36a47b7f 100644
--- a/src/tomo.c
+++ b/src/tomo.c
@@ -19,6 +19,7 @@
 #include "compile/files.h"
 #include "compile/headers.h"
 #include "config.h"
+#include "formatter.h"
 #include "modules.h"
 #include "naming.h"
 #include "parse/files.h"
@@ -74,9 +75,9 @@ static const char *paths_str(List_t paths) {
 
 static OptionalList_t files = NONE_LIST, args = NONE_LIST, uninstall = NONE_LIST, libraries = NONE_LIST;
 static OptionalBool_t verbose = false, quiet = false, show_version = false, show_parse_tree = false,
-                      show_prefix = false, stop_at_transpile = false, stop_at_obj_compilation = false,
-                      compile_exe = false, should_install = false, clean_build = false, source_mapping = true,
-                      show_changelog = false;
+                      do_format_code = false, show_prefix = false, stop_at_transpile = false,
+                      stop_at_obj_compilation = false, compile_exe = false, should_install = false, clean_build = false,
+                      source_mapping = true, show_changelog = false;
 
 static OptionalText_t show_codegen = NONE_TEXT,
                       cflags = Text("-Werror -fdollars-in-identifiers -std=c2x -Wno-trigraphs "
@@ -185,23 +186,38 @@ int main(int argc, char *argv[]) {
                         "  --optimization|-O <level>: set optimization level\n"
                         "  --run|-r: run a program from " TOMO_PREFIX "/share/tomo_" TOMO_VERSION "/installed\n");
     Text_t help = Texts(Text("\x1b[1mtomo\x1b[m: a compiler for the Tomo programming language"), Text("\n\n"), usage);
-    tomo_parse_args(
-        argc, argv, usage, help, TOMO_VERSION, {"files", true, List$info(&Path$info), &files},
-        {"args", true, List$info(&Text$info), &args}, {"verbose", false, &Bool$info, &verbose},
-        {"v", false, &Bool$info, &verbose}, {"version", false, &Bool$info, &show_version},
-        {"parse", false, &Bool$info, &show_parse_tree}, {"p", false, &Bool$info, &show_parse_tree},
-        {"prefix", false, &Bool$info, &show_prefix}, {"quiet", false, &Bool$info, &quiet},
-        {"q", false, &Bool$info, &quiet}, {"transpile", false, &Bool$info, &stop_at_transpile},
-        {"t", false, &Bool$info, &stop_at_transpile}, {"compile-obj", false, &Bool$info, &stop_at_obj_compilation},
-        {"c", false, &Bool$info, &stop_at_obj_compilation}, {"compile-exe", false, &Bool$info, &compile_exe},
-        {"e", false, &Bool$info, &compile_exe}, {"uninstall", false, List$info(&Text$info), &uninstall},
-        {"u", false, List$info(&Text$info), &uninstall}, {"library", false, List$info(&Path$info), &libraries},
-        {"L", false, List$info(&Path$info), &libraries}, {"show-codegen", false, &Text$info, &show_codegen},
-        {"C", false, &Text$info, &show_codegen}, {"install", false, &Bool$info, &should_install},
-        {"I", false, &Bool$info, &should_install}, {"optimization", false, &Text$info, &optimization},
-        {"O", false, &Text$info, &optimization}, {"force-rebuild", false, &Bool$info, &clean_build},
-        {"f", false, &Bool$info, &clean_build}, {"source-mapping", false, &Bool$info, &source_mapping},
-        {"m", false, &Bool$info, &source_mapping}, {"changelog", false, &Bool$info, &show_changelog}, );
+    tomo_parse_args(argc, argv, usage, help, TOMO_VERSION, //
+                    {"files", true, List$info(&Path$info), &files}, //
+                    {"args", true, List$info(&Text$info), &args}, //
+                    {"verbose", false, &Bool$info, &verbose}, //
+                    {"v", false, &Bool$info, &verbose}, //
+                    {"version", false, &Bool$info, &show_version}, //
+                    {"parse", false, &Bool$info, &show_parse_tree}, //
+                    {"p", false, &Bool$info, &show_parse_tree}, //
+                    {"format", false, &Bool$info, &do_format_code}, //
+                    {"prefix", false, &Bool$info, &show_prefix}, //
+                    {"quiet", false, &Bool$info, &quiet}, //
+                    {"q", false, &Bool$info, &quiet}, //
+                    {"transpile", false, &Bool$info, &stop_at_transpile}, //
+                    {"t", false, &Bool$info, &stop_at_transpile}, //
+                    {"compile-obj", false, &Bool$info, &stop_at_obj_compilation}, //
+                    {"c", false, &Bool$info, &stop_at_obj_compilation}, //
+                    {"compile-exe", false, &Bool$info, &compile_exe}, //
+                    {"e", false, &Bool$info, &compile_exe}, //
+                    {"uninstall", false, List$info(&Text$info), &uninstall}, //
+                    {"u", false, List$info(&Text$info), &uninstall}, //
+                    {"library", false, List$info(&Path$info), &libraries}, //
+                    {"L", false, List$info(&Path$info), &libraries}, //
+                    {"show-codegen", false, &Text$info, &show_codegen}, //
+                    {"C", false, &Text$info, &show_codegen}, //
+                    {"install", false, &Bool$info, &should_install}, //
+                    {"I", false, &Bool$info, &should_install}, //
+                    {"optimization", false, &Text$info, &optimization}, //
+                    {"O", false, &Text$info, &optimization}, //
+                    {"force-rebuild", false, &Bool$info, &clean_build}, {"f", false, &Bool$info, &clean_build}, //
+                    {"source-mapping", false, &Bool$info, &source_mapping},
+                    {"m", false, &Bool$info, &source_mapping}, //
+                    {"changelog", false, &Bool$info, &show_changelog}, );
 
     if (show_prefix) {
         print(TOMO_PREFIX);
@@ -299,6 +315,12 @@ int main(int argc, char *argv[]) {
             continue;
         }
 
+        if (do_format_code) {
+            Text_t formatted = format_file(Path$as_c_string(path));
+            print(formatted);
+            continue;
+        }
+
         Path_t exe_path = compile_exe ? Path$with_extension(path, Text(""), true)
                                       : build_file(Path$with_extension(path, Text(""), true), "");