using System.Reflection;
using NubLang.CLI;
using NubLang.Code;
using NubLang.Diagnostics;
using NubLang.Generation;
using NubLang.Generation.QBE;
using NubLang.Parsing;
using NubLang.Parsing.Syntax;
using NubLang.Tokenization;
using NubLang.TypeChecking;
using NubLang.TypeChecking.Node;

var options = new Options();

for (var i = 0; i < args.Length; i++)
{
    var arg = args[i];
    switch (arg)
    {
        case "-o":
        {
            ++i;
            if (i >= args.Length)
            {
                return 1;
            }

            options.OutputPath = args[i];
            break;
        }
        default:
        {
            options.Files.Add(new SourceFile(arg));
            break;
        }
    }
}

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

foreach (var file in options.Files)
{
    if (!File.Exists(file.Path))
    {
        Console.Error.WriteLine($"File '{file}' does not exist");
        return 1;
    }
}

foreach (var file in options.Files)
{
    var tokenizer = new Tokenizer(file);
    var tokens = tokenizer.Tokenize().ToList();
    diagnostics.AddRange(tokenizer.GetDiagnostics());

    var parser = new Parser();
    var syntaxTree = parser.Parse(tokens);

    diagnostics.AddRange(parser.GetDiagnostics());

    syntaxTrees.Add(syntaxTree);
}

var definitionTable = new DefinitionTable(syntaxTrees);

var typedSyntaxTrees = new List<TypedSyntaxTree>();

foreach (var syntaxTree in syntaxTrees)
{
    var typeChecker = new TypeChecker(syntaxTree, definitionTable);
    var typedSyntaxTree = typeChecker.Check();
    diagnostics.AddRange(typeChecker.GetDiagnostics());
    typedSyntaxTrees.Add(typedSyntaxTree);
}

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

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

var typedDefinitionTable = new TypedDefinitionTable(typedSyntaxTrees);

var objectFiles = new List<string>();

for (var i = 0; i < typedSyntaxTrees.Count; i++)
{
    var syntaxTree = typedSyntaxTrees[i];
    var outFileName = Path.Combine("build", "code", Path.ChangeExtension(options.Files[i].Path, null));

    var outFileDir = Path.GetDirectoryName(outFileName);
    if (!string.IsNullOrEmpty(outFileDir))
    {
        Directory.CreateDirectory(outFileDir);
    }

    var generator = new QBEGenerator(syntaxTree, typedDefinitionTable);
    var ssa = generator.Emit();

    var ssaFilePath = Path.ChangeExtension(outFileName, "ssa");
    File.WriteAllText(ssaFilePath, ssa);

    var asmFilePath = Path.ChangeExtension(outFileName, "s");
    var qbeSuccess = await QBE.Invoke(ssaFilePath, asmFilePath);

    if (!qbeSuccess)
    {
        return 1;
    }

    var objFilePath = Path.ChangeExtension(outFileName, "o");
    var asmSuccess = await GCC.Assemble(asmFilePath, objFilePath);

    if (!asmSuccess)
    {
        return 1;
    }

    objectFiles.Add(objFilePath);
}

const string runtimeName = "libruntime_x64.a";

var resources = Assembly.GetExecutingAssembly().GetManifestResourceNames();
var runtime = resources.First(r => r.EndsWith(runtimeName));

await using var reader = Assembly.GetExecutingAssembly().GetManifestResourceStream(runtime);

if (reader == null)
{
    Console.Error.WriteLine($"Cannot open stream to '{runtimeName}'");
    return 1;
}

var runtimePath = Path.Combine("build", "runtime", runtimeName);
var runtimeDir = Path.GetDirectoryName(runtimePath);
if (!string.IsNullOrEmpty(runtimeDir))
{
    Directory.CreateDirectory(runtimeDir);
}

await using var writer = new FileStream(runtimePath, FileMode.Create);

reader.CopyTo(writer);

objectFiles.Add(runtimePath);

var outPath = options.OutputPath ?? Path.Combine("build", "out.a");
var outDir = Path.GetDirectoryName(outPath);
if (!string.IsNullOrEmpty(outDir))
{
    Directory.CreateDirectory(outDir);
}

var archiveResult = await Archive.Invoke(outPath, objectFiles);
if (!archiveResult)
{
    return 1;
}

return 0;