fix(rome_service): disallow large files from being parsed and analyzed

crates/rome_cli/src/traversal.rs

@@ -9,8 +9,8 @@ use rome_diagnostics::{
     v2::{
         self,
         adapters::{IoError, StdError},
-        category, Advices, Category, Diagnostic, DiagnosticExt, Error, FilePath, LogCategory,
-        PrintDescription, PrintDiagnostic, Severity, Visit,
+        category, Advices, Category, Diagnostic, DiagnosticExt, Error, FilePath, PrintDescription,
+        PrintDiagnostic, Severity, Visit,
     },
     MAXIMUM_DISPLAYABLE_DIAGNOSTICS,
 };
@@ -238,14 +238,8 @@ struct FormatDiffAdvice<'a> {
 
 impl Advices for FormatDiffAdvice<'_> {
     fn record(&self, visitor: &mut dyn Visit) -> io::Result<()> {
-        // Skip printing the diff for files over 1Mb (probably a minified file)
-        let max_len = self.old.len().max(self.new.len());
-        if max_len >= 1_000_000 {
-            visitor.record_log(LogCategory::Info, &"[Diff not printed for file over 1Mb]")
-        } else {
-            let diff = TextEdit::from_unicode_words(self.old, self.new);
-            visitor.record_diff(&diff)
-        }
+        let diff = TextEdit::from_unicode_words(self.old, self.new);
+        visitor.record_diff(&diff)
     }
 }
 

crates/rome_cli/tests/main.rs

@@ -674,6 +674,34 @@ mod check {
             result,
         ));
     }
+
+    #[test]
+    fn file_too_large() {
+        let mut fs = MemoryFileSystem::default();
+        let mut console = BufferConsole::default();
+
+        let file_path = Path::new("check.js");
+        fs.insert(file_path.into(), "statement();\n".repeat(80660).as_bytes());
+
+        let result = run_cli(
+            DynRef::Borrowed(&mut fs),
+            DynRef::Borrowed(&mut console),
+            Arguments::from_vec(vec![OsString::from("check"), file_path.as_os_str().into()]),
+        );
+
+        assert!(result.is_ok(), "run_cli returned {result:?}");
+
+        // Do not store the content of the file in the snapshot
+        fs.remove(file_path);
+
+        assert_cli_snapshot(SnapshotPayload::new(
+            module_path!(),
+            "file_too_large",
+            fs,
+            console,
+            result,
+        ));
+    }
 }
 
 mod ci {
@@ -884,6 +912,34 @@ mod ci {
             result,
         ));
     }
+
+    #[test]
+    fn file_too_large() {
+        let mut fs = MemoryFileSystem::default();
+        let mut console = BufferConsole::default();
+
+        let file_path = Path::new("ci.js");
+        fs.insert(file_path.into(), "statement();\n".repeat(80660).as_bytes());
+
+        let result = run_cli(
+            DynRef::Borrowed(&mut fs),
+            DynRef::Borrowed(&mut console),
+            Arguments::from_vec(vec![OsString::from("ci"), file_path.as_os_str().into()]),
+        );
+
+        assert!(result.is_ok(), "run_cli returned {result:?}");
+
+        // Do not store the content of the file in the snapshot
+        fs.remove(file_path);
+
+        assert_cli_snapshot(SnapshotPayload::new(
+            module_path!(),
+            "file_too_large",
+            fs,
+            console,
+            result,
+        ));
+    }
 }
 
 mod format {
@@ -1690,6 +1746,38 @@ mod format {
             result,
         ));
     }
+
+    #[test]
+    fn file_too_large() {
+        let mut fs = MemoryFileSystem::default();
+        let mut console = BufferConsole::default();
+
+        let file_path = Path::new("format.js");
+        fs.insert(file_path.into(), "statement();\n".repeat(80660).as_bytes());
+
+        let result = run_cli(
+            DynRef::Borrowed(&mut fs),
+            DynRef::Borrowed(&mut console),
+            Arguments::from_vec(vec![
+                OsString::from("format"),
+                file_path.as_os_str().into(),
+                OsString::from("--write"),
+            ]),
+        );
+
+        assert!(result.is_ok(), "run_cli returned {result:?}");
+
+        // Do not store the content of the file in the snapshot
+        fs.remove(file_path);
+
+        assert_cli_snapshot(SnapshotPayload::new(
+            module_path!(),
+            "file_too_large",
+            fs,
+            console,
+            result,
+        ));
+    }
 }
 
 mod help {

crates/rome_cli/tests/snapshots/main_check/file_too_large.snap

@@ -0,0 +1,19 @@
+---
+source: crates/rome_cli/tests/snap_test.rs
+expression: content
+---
+# Emitted Messages
+
+```block
+check.js lint ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
+
+  × The file check.js could not be parsed because it's too large (the file is 1.0 MiB long, but the size limit is 1.0 MiB)
+  
+
+```
+
+```block
+Skipped 1 files
+```
+
+

crates/rome_cli/tests/snapshots/main_ci/file_too_large.snap

@@ -0,0 +1,19 @@
+---
+source: crates/rome_cli/tests/snap_test.rs
+expression: content
+---
+# Emitted Messages
+
+```block
+ci.js lint ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
+
+  × The file ci.js could not be parsed because it's too large (the file is 1.0 MiB long, but the size limit is 1.0 MiB)
+  
+
+```
+
+```block
+Skipped 1 files
+```
+
+

crates/rome_cli/tests/snapshots/main_format/file_too_large.snap

@@ -0,0 +1,19 @@
+---
+source: crates/rome_cli/tests/snap_test.rs
+expression: content
+---
+# Emitted Messages
+
+```block
+format.js lint ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
+
+  × The file format.js could not be parsed because it's too large (the file is 1.0 MiB long, but the size limit is 1.0 MiB)
+  
+
+```
+
+```block
+Skipped 1 files
+```
+
+

crates/rome_console/src/fmt.rs

@@ -233,3 +233,67 @@ impl Display for Duration {
         })
     }
 }
+
+#[repr(transparent)]
+#[derive(Clone, Copy, Debug)]
+pub struct Bytes(pub usize);
+
+impl std::fmt::Display for Bytes {
+    fn fmt(&self, fmt: &mut fmt::Formatter<'_>) -> fmt::Result {
+        let Self(mut value) = *self;
+
+        if value < 1024 {
+            return write!(fmt, "{value} B");
+        }
+
+        const PREFIX: [char; 4] = ['K', 'M', 'G', 'T'];
+        let prefix = PREFIX
+            .into_iter()
+            .find(|_| {
+                let next_value = value / 1024;
+                if next_value < 1024 {
+                    return true;
+                }
+
+                value = next_value;
+                false
+            })
+            .unwrap_or('T');
+
+        write!(fmt, "{:.1} {prefix}iB", value as f32 / 1024.0)
+    }
+}
+
+impl Display for Bytes {
+    fn fmt(&self, fmt: &mut Formatter) -> io::Result<()> {
+        write!(fmt, "{self}")
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use crate::fmt::Bytes;
+
+    #[test]
+    fn display_bytes() {
+        // Examples taken from https://stackoverflow.com/a/3758880
+        assert_eq!(Bytes(0).to_string(), "0 B");
+        assert_eq!(Bytes(27).to_string(), "27 B");
+        assert_eq!(Bytes(999).to_string(), "999 B");
+        assert_eq!(Bytes(1_000).to_string(), "1000 B");
+        assert_eq!(Bytes(1_023).to_string(), "1023 B");
+        assert_eq!(Bytes(1_024).to_string(), "1.0 KiB");
+        assert_eq!(Bytes(1_728).to_string(), "1.7 KiB");
+        assert_eq!(Bytes(110_592).to_string(), "108.0 KiB");
+        assert_eq!(Bytes(999_999).to_string(), "976.6 KiB");
+        assert_eq!(Bytes(7_077_888).to_string(), "6.8 MiB");
+        assert_eq!(Bytes(452_984_832).to_string(), "432.0 MiB");
+        assert_eq!(Bytes(28_991_029_248).to_string(), "27.0 GiB");
+        assert_eq!(Bytes(1_855_425_871_872).to_string(), "1.7 TiB");
+
+        #[cfg(target_pointer_width = "32")]
+        assert_eq!(Bytes(usize::MAX).to_string(), "4.0 GiB");
+        #[cfg(target_pointer_width = "64")]
+        assert_eq!(Bytes(usize::MAX).to_string(), "16384.0 TiB");
+    }
+}

crates/rome_fs/src/fs/memory.rs

@@ -43,6 +43,11 @@ impl MemoryFileSystem {
         files.insert(path, Arc::new(Mutex::new(content.into())));
     }
 
+    /// Remove a file from the filesystem
+    pub fn remove(&mut self, path: &Path) {
+        self.files.0.write().remove(path);
+    }
+
     pub fn files(self) -> IntoIter<PathBuf, FileEntry> {
         let files = self.files.0.into_inner();
         files.into_iter()

crates/rome_service/src/lib.rs

@@ -1,3 +1,4 @@
+use rome_console::fmt::Bytes;
 use rome_console::{Console, EnvConsole};
 use rome_formatter::FormatError;
 use rome_fs::{FileSystem, OsFileSystem, RomePath};
@@ -71,6 +72,12 @@ pub enum RomeError {
     TransportError(TransportError),
     /// Emitted when the file is ignored and should not be processed
     FileIgnored(PathBuf),
+    /// Emitted when a file could not be parsed because it's larger than the size limite
+    FileTooLarge {
+        path: PathBuf,
+        size: usize,
+        limit: usize,
+    },
 }
 
 impl Debug for RomeError {
@@ -163,6 +170,9 @@ impl Display for RomeError {
             RomeError::FileIgnored(path) => {
                 write!(f, "The file {} was ignored", path.display())
             }
+            RomeError::FileTooLarge { path, size, limit } => {
+                write!(f, "The file {} could not be parsed because it's too large (the file is {} long, but the size limit is {})", path.display(), Bytes(*size), Bytes(*limit))
+            }
         }
     }
 }

crates/rome_service/src/workspace/server.rs

@@ -169,6 +169,18 @@ impl WorkspaceServer {
                     .parse
                     .ok_or_else(self.build_capability_error(rome_path))?;
 
+                /// Limit the size of files to 1.0 MiB
+                const SIZE_LIMIT: usize = 1024 * 1024;
+
+                let size = document.content.as_bytes().len();
+                if size >= SIZE_LIMIT {
+                    return Err(RomeError::FileTooLarge {
+                        path: rome_path.to_path_buf(),
+                        size,
+                        limit: SIZE_LIMIT,
+                    });
+                }
+
                 let parsed = parse(rome_path, document.language_hint, &document.content);
 
                 Ok(entry.insert(parsed).clone())