feat(formatter): support printing Formatter IR

Author: Dunqing

crates/oxc_formatter/examples/formatter.rs

@@ -60,17 +60,16 @@ fn main() -> Result<(), String> {
     };
 
     let formatter = Formatter::new(&allocator, options);
+    let formatted = formatter.format(&ret.program);
     if show_ir {
-        let doc = formatter.doc(&ret.program);
-        println!("[");
-        for el in doc.iter() {
-            println!("  {el:?},");
-        }
-        println!("]");
-    } else {
-        let code = formatter.build(&ret.program);
-        println!("{code}");
+        println!("--- IR ---");
+        println!("{}", &formatted.document().to_string());
+        println!("--- End IR ---\n");
     }
 
+    println!("--- Formatted Code ---");
+    let code = formatted.print().map_err(|e| e.to_string())?.into_code();
+    println!("{code}");
+    println!("--- End Formatted Code ---");
     Ok(())
 }

crates/oxc_formatter/src/formatter/context.rs

@@ -42,16 +42,18 @@ impl std::fmt::Debug for FormatContext<'_> {
 
 impl<'ast> FormatContext<'ast> {
     pub fn new(
-        program: &'ast Program<'ast>,
+        source_text: &'ast str,
+        source_type: SourceType,
+        comments: &'ast [Comment],
         allocator: &'ast Allocator,
         options: FormatOptions,
     ) -> Self {
-        let source_text = SourceText::new(program.source_text);
+        let source_text = SourceText::new(source_text);
         Self {
             options,
             source_text,
-            source_type: program.source_type,
-            comments: Comments::new(source_text, &program.comments),
+            source_type,
+            comments: Comments::new(source_text, comments),
             allocator,
             cached_elements: FxHashMap::default(),
         }