91 lines
2.7 KiB
C#
91 lines
2.7 KiB
C#
using NubLang.Ast;
|
|
using NubLang.Generation;
|
|
using NubLang.Syntax;
|
|
using OmniSharp.Extensions.LanguageServer.Protocol;
|
|
using OmniSharp.Extensions.LanguageServer.Protocol.Server;
|
|
|
|
namespace NubLang.LSP;
|
|
|
|
public class WorkspaceManager(DiagnosticsPublisher diagnosticsPublisher, ILanguageServerFacade server)
|
|
{
|
|
private readonly Dictionary<string, SyntaxTree> _syntaxTrees = new();
|
|
private readonly Dictionary<string, CompilationUnit> _compilationUnits = new();
|
|
|
|
public void Init(string rootPath)
|
|
{
|
|
var files = Directory.GetFiles(rootPath, "*.nub", SearchOption.AllDirectories);
|
|
foreach (var path in files)
|
|
{
|
|
var text = File.ReadAllText(path);
|
|
var tokenizer = new Tokenizer(path, text);
|
|
|
|
tokenizer.Tokenize();
|
|
diagnosticsPublisher.Publish(path, tokenizer.Diagnostics);
|
|
|
|
var parser = new Parser();
|
|
var parseResult = parser.Parse(tokenizer.Tokens);
|
|
diagnosticsPublisher.Publish(path, parser.Diagnostics);
|
|
|
|
_syntaxTrees[path] = parseResult;
|
|
}
|
|
|
|
Generate();
|
|
}
|
|
|
|
public void UpdateFile(DocumentUri path)
|
|
{
|
|
var fsPath = path.GetFileSystemPath();
|
|
var text = File.ReadAllText(fsPath);
|
|
var tokenizer = new Tokenizer(fsPath, text);
|
|
|
|
tokenizer.Tokenize();
|
|
diagnosticsPublisher.Publish(path, tokenizer.Diagnostics);
|
|
|
|
var parser = new Parser();
|
|
var parseResult = parser.Parse(tokenizer.Tokens);
|
|
diagnosticsPublisher.Publish(path, parser.Diagnostics);
|
|
|
|
_syntaxTrees[fsPath] = parseResult;
|
|
|
|
Generate();
|
|
}
|
|
|
|
private void Generate()
|
|
{
|
|
var modules = Module.Collect(_syntaxTrees.Select(x => x.Value).ToList());
|
|
|
|
foreach (var (fsPath, syntaxTree) in _syntaxTrees)
|
|
{
|
|
var typeChecker = new TypeChecker(syntaxTree, modules);
|
|
var result = typeChecker.Check();
|
|
diagnosticsPublisher.Publish(fsPath, typeChecker.Diagnostics);
|
|
_compilationUnits[fsPath] = result;
|
|
|
|
var generator = new Generator(result);
|
|
var c = generator.Emit();
|
|
|
|
server.SendNotification("nub/output", new
|
|
{
|
|
content = c,
|
|
path = fsPath
|
|
});
|
|
}
|
|
}
|
|
|
|
public void RemoveFile(DocumentUri path)
|
|
{
|
|
var fsPath = path.GetFileSystemPath();
|
|
_syntaxTrees.Remove(fsPath);
|
|
_compilationUnits.Remove(fsPath);
|
|
}
|
|
|
|
public Dictionary<string, CompilationUnit> GetCompilationUnits()
|
|
{
|
|
return _compilationUnits;
|
|
}
|
|
|
|
public CompilationUnit? GetCompilationUnit(DocumentUri path)
|
|
{
|
|
return _compilationUnits.GetValueOrDefault(path.GetFileSystemPath());
|
|
}
|
|
} |