feat: made LSP work

Author: Zffu

compiler/diagnostics/src/lib.rs

@@ -70,6 +70,10 @@ pub fn dump_diagnostics() {
     })
 }
 
+pub fn clear_diagnostics() {
+    DIAGNOSTIC_CONTAINER.with_borrow_mut(|f| f.diagnostics.clear())
+}
+
 pub fn has_diagnostics() -> bool {
     DIAGNOSTIC_CONTAINER.with_borrow(|f| return !f.diagnostics.is_empty())
 }

compiler/lexer/src/lexer.rs

@@ -39,12 +39,16 @@ const USE_KEYWORD_HASH: u64 = hash!("use");
 /// ```
 /// let result: LexerParseResult<Vec<LexerToken>> = lexer_parse_file("test_file.qf").expect("Lexer didn't work");
 /// ```
-pub fn lexer_parse_file(file_path: &String) -> DiagnosticResult<Vec<LexerToken>> {
-    let contents: String = match fs::read_to_string(file_path) {
+pub fn lexer_parse_file(path: &String) -> DiagnosticResult<Vec<LexerToken>> {
+    let contents: String = match fs::read_to_string(path) {
         Ok(v) => v,
-        Err(_) => panic!("Couldn't read the file {}", file_path),
+        Err(_) => panic!("Couldn't read the file {}", path),
     };
 
+    lexer_parse(contents, path)
+}
+
+pub fn lexer_parse(contents: String, file_path: &String) -> DiagnosticResult<Vec<LexerToken>> {
     let mut tokens: Vec<LexerToken> = Vec::new();
 
     let mut i: usize = 0;

lsp/src/bridge.rs

@@ -1,10 +1,12 @@
 use ast_parser::parse_ast_ctx;
 use astoir::run_astoir_mir;
-use diagnostics::{diagnostic::Diagnostic, get_diagnostics, has_diagnostics};
-use lexer::lexer::lexer_parse_file;
+use diagnostics::{clear_diagnostics, diagnostic::Diagnostic, get_diagnostics, has_diagnostics};
+use lexer::lexer::lexer_parse;
 
-pub fn check_for_file(url: String) -> Vec<Diagnostic> {
-    let lexer = lexer_parse_file(&url);
+pub fn check_for_file(url: String, text: String) -> Vec<Diagnostic> {
+    clear_diagnostics();
+
+    let lexer = lexer_parse(text, &url);
     if has_diagnostics() {
         return get_diagnostics();
     }
@@ -14,7 +16,7 @@ pub fn check_for_file(url: String) -> Vec<Diagnostic> {
         return get_diagnostics();
     }
 
-    run_astoir_mir(ast.unwrap());
+    let _ = run_astoir_mir(ast.unwrap());
 
     return get_diagnostics();
 }

lsp/src/diags.rs

@@ -3,8 +3,6 @@ use tower_lsp::lsp_types::{
     Diagnostic, DiagnosticRelatedInformation, DiagnosticSeverity, Location, Position, Range, Url,
 };
 
-use crate::main;
-
 pub fn to_tower_diag(diagnostic: diagnostics::diagnostic::Diagnostic) -> Diagnostic {
     let severity = match diagnostic.level {
         Level::Error => DiagnosticSeverity::ERROR,

lsp/src/lsp.rs

@@ -1,14 +1,14 @@
 use std::{
-    collections::{HashMap, HashSet},
+    collections::HashSet,
     sync::{Arc, Mutex},
 };
 
 use tower_lsp::{
     Client, LanguageServer,
     lsp_types::{
-        CompletionOptions, DidOpenTextDocumentParams, HoverProviderCapability, InitializeParams,
-        InitializeResult, InitializedParams, MessageType, ServerCapabilities, ServerInfo,
-        TextDocumentSyncCapability, TextDocumentSyncKind,
+        DidChangeTextDocumentParams, DidOpenTextDocumentParams, InitializeParams, InitializeResult,
+        MessageType, ServerCapabilities, ServerInfo, TextDocumentSyncCapability,
+        TextDocumentSyncKind,
     },
 };
 
@@ -28,8 +28,8 @@ impl LanguageServer for LSPBackend {
     ) -> tower_lsp::jsonrpc::Result<InitializeResult> {
         Ok(InitializeResult {
             capabilities: ServerCapabilities {
-                hover_provider: Some(HoverProviderCapability::Simple(true)),
-                completion_provider: Some(CompletionOptions::default()),
+                hover_provider: None,
+                completion_provider: None,
                 text_document_sync: Some(TextDocumentSyncCapability::Kind(
                     TextDocumentSyncKind::FULL,
                 )),
@@ -50,14 +50,31 @@ impl LanguageServer for LSPBackend {
 
         let mut diags = vec![];
 
+        for diag in check_for_file(uri.clone().replace("file://", ""), text.clone()) {
+            diags.push(to_tower_diag(diag))
+        }
+
+        self.lsp_client.log_message(MessageType::ERROR, text).await;
+
         self.lsp_client
-            .log_message(MessageType::ERROR, format!("Path: {}", uri))
+            .publish_diagnostics(uri.parse().unwrap(), diags, None)
             .await;
+    }
 
-        for diag in check_for_file(uri.clone().replace("file://", "")) {
+    async fn did_change(&self, params: DidChangeTextDocumentParams) {
+        let uri = params.text_document.uri.to_string();
+        let text = params.content_changes[0].text.clone();
+
+        self.documents.lock().unwrap().insert(uri.clone());
+
+        let mut diags = vec![];
+
+        for diag in check_for_file(uri.clone().replace("file://", ""), text.clone()) {
             diags.push(to_tower_diag(diag))
         }
 
+        self.lsp_client.log_message(MessageType::ERROR, text).await;
+
         self.lsp_client
             .publish_diagnostics(uri.parse().unwrap(), diags, None)
             .await;