nub-lang-archive/compiler/NubLang.CLI/Program.cs

using System.Diagnostics;
using NubLang.Ast;
using NubLang.Diagnostics;
using NubLang.Generation;
using NubLang.Syntax;

var diagnostics = new List<Diagnostic>();
var syntaxTrees = new List<SyntaxTree>();

var nubFiles = args.Where(x => Path.GetExtension(x) == ".nub").ToArray();
var objectFileArgs = args.Where(x => Path.GetExtension(x) is ".o" or ".a").ToArray();

foreach (var file in nubFiles)
{
    var tokenizer = new Tokenizer(file, File.ReadAllText(file));
    tokenizer.Tokenize();
    diagnostics.AddRange(tokenizer.Diagnostics);

    var parser = new Parser();
    var syntaxTree = parser.Parse(tokenizer.Tokens);
    diagnostics.AddRange(parser.Diagnostics);

    syntaxTrees.Add(syntaxTree);
}

var modules = Module.Collect(syntaxTrees);
var compilationUnits = new List<CompilationUnit>();

for (var i = 0; i < nubFiles.Length; i++)
{
    var typeChecker = new TypeChecker(syntaxTrees[i], modules);
    var compilationUnit = typeChecker.Check();

    compilationUnits.Add(compilationUnit);
    diagnostics.AddRange(typeChecker.Diagnostics);
}

foreach (var diagnostic in diagnostics)
{
    Console.Error.WriteLine(diagnostic.FormatANSI());
}

if (diagnostics.Any(diagnostic => diagnostic.Severity == DiagnosticSeverity.Error))
{
    return 1;
}

for (var i = 0; i < nubFiles.Length; i++)
{
    var compilationUnit = compilationUnits[i];

    var generator = new Generator(compilationUnit);
    Console.WriteLine(generator.Generate());
}

return 0;

// var cPaths = new List<string>();
//
// Directory.CreateDirectory(".build");
//
// for (var i = 0; i < nubFiles.Length; i++)
// {
//     var file = nubFiles[i];
//     var compilationUnit = compilationUnits[i];
//
//     var generator = new Generator(compilationUnit);
//     var directory = Path.GetDirectoryName(file);
//     if (!string.IsNullOrWhiteSpace(directory))
//     {
//         Directory.CreateDirectory(Path.Combine(".build", directory));
//     }
//
//     var path = Path.Combine(".build", Path.ChangeExtension(file, "c"));
//     File.WriteAllText(path, generator.Emit());
//     cPaths.Add(path);
// }
//
// var objectPaths = new List<string>();
//
// foreach (var cPath in cPaths)
// {
//     var objectPath = Path.ChangeExtension(cPath, "o");
//     using var compileProcess = Process.Start("gcc", [
//         "-ffreestanding", "-nostartfiles", "-std=c23",
//         "-g", "-lm",
//         "-c", "-o", objectPath,
//         cPath,
//     ]);
//
//     compileProcess.WaitForExit();
//
//     if (compileProcess.ExitCode != 0)
//     {
//         Console.Error.WriteLine($"gcc failed with exit code {compileProcess.ExitCode}");
//         return 1;
//     }
//
//     objectPaths.Add(objectPath);
// }
//
// if (modules.TryGetValue("main", out var mainModule))
// {
//     var mainFunction = mainModule
//         .Functions(true)
//         .FirstOrDefault(x => x.Prototype.ExternSymbol == "main");
//
//     if (mainFunction is { Prototype.ExternSymbol: not null })
//     {
//         var runtime = $"""
//                        .intel_syntax noprefix
//
//                        .text
//                        .globl _start
//                        _start:
//                            mov rdi, [rsp]       # argc
//                            mov rsi, [rsp + 8]   # argv
//                            call {mainFunction.Prototype.ExternSymbol}
//                            mov rdi, rax         # Move return value into rdi
//                            mov rax, 60          # syscall: exit
//                            syscall
//
//                        """;
//
//         var runtimePath = Path.Combine(".build", "runtime.s");
//         File.WriteAllText(runtimePath, runtime);
//
//         using var assembleProcess = Process.Start(new ProcessStartInfo("as", ["-g", "-c", runtimePath, "-o", Path.Combine(".build", "runtime.o")]));
//         if (assembleProcess == null) return 1;
//         assembleProcess.WaitForExit();
//
//         if (assembleProcess.ExitCode != 0)
//         {
//             Console.Error.WriteLine($"gcc failed with exit code {assembleProcess.ExitCode}");
//             return 1;
//         }
//
//         if (assembleProcess.ExitCode != 0) return 1;
//
//         using var linkProcess = Process.Start(new ProcessStartInfo("gcc", [
//             "-ffreestanding", "-nostartfiles", "-std=c23",
//             "-g", "-lm",
//             "-o", Path.Combine(".build", "out"),
//             ..objectPaths,
//             Path.Combine(".build", "runtime.o"),
//             ..objectFileArgs
//         ]));
//
//         if (linkProcess == null) return 1;
//         linkProcess.WaitForExit();
//
//         if (linkProcess.ExitCode != 0)
//         {
//             Console.Error.WriteLine($"gcc failed with exit code {linkProcess.ExitCode}");
//             return 1;
//         }
//
//         Console.WriteLine("Build successful: .build/out");
//     }
//     else
//     {
//         Console.WriteLine("No main function found in module main, skipping link step");
//     }
// }
// else
// {
//     Console.WriteLine("No main function found in module main, skipping link step");
// }
//
// return 0;