refactor(langauge_server): introdude LSPFileSystem

Author: Sysix

crates/oxc_language_server/src/file_system.rs

@@ -0,0 +1,32 @@
+use std::sync::Arc;
+
+use tower_lsp_server::lsp_types::Uri;
+
+use crate::ConcurrentHashMap;
+
+pub struct LSPFileSystem {
+    files: Arc<ConcurrentHashMap<Uri, String>>,
+}
+
+impl LSPFileSystem {
+    pub fn new() -> Self {
+        Self { files: Arc::new(ConcurrentHashMap::default()) }
+    }
+
+    pub fn clear(&self) {
+        self.files.pin().clear();
+    }
+
+    pub fn set(&self, uri: &Uri, content: String) {
+        self.files.pin().insert(uri.clone(), content);
+    }
+
+    #[expect(dead_code)] // used for the oxc_formatter in the future
+    pub fn get(&self, uri: &Uri) -> Option<String> {
+        self.files.pin().get(uri).cloned()
+    }
+
+    pub fn remove(&self, uri: &Uri) {
+        self.files.pin().remove(uri);
+    }
+}

crates/oxc_language_server/src/main.rs

@@ -22,6 +22,7 @@ use tower_lsp_server::{
 mod capabilities;
 mod code_actions;
 mod commands;
+mod file_system;
 mod formatter;
 mod linter;
 mod options;
@@ -36,6 +37,8 @@ use linter::server_linter::ServerLinterRun;
 use options::{Options, WorkspaceOption};
 use worker::WorkspaceWorker;
 
+use crate::file_system::LSPFileSystem;
+
 type ConcurrentHashMap<K, V> = papaya::HashMap<K, V, FxBuildHasher>;
 
 const OXC_CONFIG_FILE: &str = ".oxlintrc.json";
@@ -51,6 +54,7 @@ struct Backend {
     // 2. `workspace/didChangeWorkspaceFolders` request
     workspace_workers: Arc<RwLock<Vec<WorkspaceWorker>>>,
     capabilities: OnceCell<Capabilities>,
+    file_system: Arc<RwLock<LSPFileSystem>>,
 }
 
 impl LanguageServer for Backend {
@@ -203,6 +207,7 @@ impl LanguageServer for Backend {
 
     async fn shutdown(&self) -> Result<()> {
         self.clear_all_diagnostics().await;
+        self.file_system.write().await.clear();
         Ok(())
     }
 
@@ -444,6 +449,10 @@ impl LanguageServer for Backend {
         let Some(worker) = workers.iter().find(|worker| worker.is_responsible_for_uri(uri)) else {
             return;
         };
+
+        // saving the file means we can read again from the file system
+        self.file_system.write().await.remove(uri);
+
         if let Some(diagnostics) = worker.lint_file(uri, None, ServerLinterRun::OnSave).await {
             self.client
                 .publish_diagnostics(
@@ -464,6 +473,11 @@ impl LanguageServer for Backend {
             return;
         };
         let content = params.content_changes.first().map(|c| c.text.clone());
+
+        if let Some(content) = &content {
+            self.file_system.write().await.set(uri, content.to_string());
+        }
+
         if let Some(diagnostics) = worker.lint_file(uri, content, ServerLinterRun::OnType).await {
             self.client
                 .publish_diagnostics(
@@ -483,6 +497,9 @@ impl LanguageServer for Backend {
         };
 
         let content = params.text_document.text;
+
+        self.file_system.write().await.set(uri, content.to_string());
+
         if let Some(diagnostics) =
             worker.lint_file(uri, Some(content), ServerLinterRun::Always).await
         {
@@ -502,6 +519,7 @@ impl LanguageServer for Backend {
         let Some(worker) = workers.iter().find(|worker| worker.is_responsible_for_uri(uri)) else {
             return;
         };
+        self.file_system.write().await.remove(uri);
         worker.remove_diagnostics(&params.text_document.uri).await;
     }
 
@@ -626,6 +644,7 @@ async fn main() {
         client,
         workspace_workers: Arc::new(RwLock::new(vec![])),
         capabilities: OnceCell::new(),
+        file_system: Arc::new(RwLock::new(LSPFileSystem::new())),
     })
     .finish();