Skip to content

Commit

Permalink
refactor(codegen)!: remove Codegen::enableSourceMap API
Browse files Browse the repository at this point in the history
  • Loading branch information
Boshen committed Oct 11, 2024
1 parent d7d6d64 commit 114c4ce
Show file tree
Hide file tree
Showing 7 changed files with 66 additions and 51 deletions.
7 changes: 3 additions & 4 deletions crates/oxc/src/compiler.rs
Original file line number Diff line number Diff line change
Expand Up @@ -283,11 +283,10 @@ pub trait CompilerInterface {
mangler: Option<Mangler>,
options: CodegenOptions,
) -> CodegenReturn {
let mut codegen = CodeGenerator::new().with_options(options).with_mangler(mangler);
let mut options = options;
if self.enable_sourcemap() {
codegen = codegen
.enable_source_map(source_path.to_string_lossy().as_ref(), program.source_text);
options.source_map_path = Some(source_path.to_path_buf());
}
codegen.build(program)
CodeGenerator::new().with_options(options).with_mangler(mangler).build(program)
}
}
7 changes: 5 additions & 2 deletions crates/oxc_codegen/examples/sourcemap.rs
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@ use std::{env, path::Path};

use base64::{prelude::BASE64_STANDARD, Engine};
use oxc_allocator::Allocator;
use oxc_codegen::{CodeGenerator, CodegenReturn};
use oxc_codegen::{CodeGenerator, CodegenOptions, CodegenReturn};
use oxc_parser::Parser;
use oxc_span::SourceType;

Expand All @@ -28,7 +28,10 @@ fn main() -> std::io::Result<()> {
}

let CodegenReturn { code, map } = CodeGenerator::new()
.enable_source_map(path.to_string_lossy().as_ref(), &source_text)
.with_options(CodegenOptions {
source_map_path: Some(path.to_path_buf()),
..CodegenOptions::default()
})
.build(&ret.program);

if let Some(source_map) = map {
Expand Down
27 changes: 16 additions & 11 deletions crates/oxc_codegen/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@ mod gen;
mod operator;
mod sourcemap_builder;

use std::borrow::Cow;
use std::{borrow::Cow, path::PathBuf};

use oxc_ast::ast::{
BindingIdentifier, BlockStatement, Expression, IdentifierReference, Program, Statement,
Expand All @@ -35,7 +35,7 @@ pub use crate::{
/// Code generator without whitespace removal.
pub type CodeGenerator<'a> = Codegen<'a>;

#[derive(Clone, Copy)]
#[derive(Debug, Clone)]
pub struct CodegenOptions {
/// Use single quotes instead of double quotes.
///
Expand All @@ -58,11 +58,19 @@ pub struct CodegenOptions {
///
/// Default is `false`.
pub annotation_comments: bool,

pub source_map_path: Option<PathBuf>,
}

impl Default for CodegenOptions {
fn default() -> Self {
Self { single_quote: false, minify: false, comments: true, annotation_comments: false }
Self {
single_quote: false,
minify: false,
comments: true,
annotation_comments: false,
source_map_path: None,
}
}
}

Expand All @@ -78,6 +86,8 @@ pub struct CodegenReturn {
pub code: String,

/// The source map from the input source code to the generated source code.
///
/// You must set [`CodegenOptions::source_map_path`] for this to be [`Some`].
pub map: Option<oxc_sourcemap::SourceMap>,
}

Expand Down Expand Up @@ -185,14 +195,6 @@ impl<'a> Codegen<'a> {
self
}

#[must_use]
pub fn enable_source_map(mut self, source_name: &str, source_text: &str) -> Self {
let mut sourcemap_builder = SourcemapBuilder::default();
sourcemap_builder.with_name_and_source(source_name, source_text);
self.sourcemap_builder = Some(sourcemap_builder);
self
}

#[must_use]
pub fn with_mangler(mut self, mangler: Option<Mangler>) -> Self {
self.mangler = mangler;
Expand All @@ -207,6 +209,9 @@ impl<'a> Codegen<'a> {
if self.options.print_annotation_comments() {
self.build_comments(&program.comments);
}
if let Some(path) = &self.options.source_map_path {
self.sourcemap_builder = Some(SourcemapBuilder::new(path, program.source_text));
}

program.print(&mut self, Context::default());
let code = self.into_source_text();
Expand Down
38 changes: 15 additions & 23 deletions crates/oxc_codegen/src/sourcemap_builder.rs
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
use std::sync::Arc;
use std::{path::Path, sync::Arc};

use nonmax::NonMaxU32;
use oxc_index::{Idx, IndexVec};
Expand Down Expand Up @@ -73,27 +73,23 @@ pub struct SourcemapBuilder {
generated_column: u32,
}

impl Default for SourcemapBuilder {
fn default() -> Self {
impl SourcemapBuilder {
pub fn new(path: &Path, source_text: &str) -> Self {
let mut sourcemap_builder = oxc_sourcemap::SourceMapBuilder::default();
let line_offset_tables = Self::generate_line_offset_tables(source_text);
let source_id =
sourcemap_builder.set_source_and_content(path.to_string_lossy().as_ref(), source_text);
Self {
source_id: 0,
original_source: "".into(),
source_id,
original_source: Arc::from(source_text),
last_generated_update: 0,
last_position: None,
line_offset_tables: LineOffsetTables::default(),
sourcemap_builder: oxc_sourcemap::SourceMapBuilder::default(),
line_offset_tables,
sourcemap_builder,
generated_line: 0,
generated_column: 0,
}
}
}

impl SourcemapBuilder {
pub fn with_name_and_source(&mut self, name: &str, source: &str) {
self.line_offset_tables = Self::generate_line_offset_tables(source);
self.source_id = self.sourcemap_builder.set_source_and_content(name, source);
self.original_source = source.into();
}

pub fn into_sourcemap(self) -> oxc_sourcemap::SourceMap {
self.sourcemap_builder.into_sourcemap()
Expand Down Expand Up @@ -392,8 +388,7 @@ mod test {
}

fn assert_mapping(source: &str, mappings: &[(u32, u32, u32)]) {
let mut builder = SourcemapBuilder::default();
builder.with_name_and_source("x.js", source);
let mut builder = SourcemapBuilder::new(Path::new("x.js"), source);
for (position, expected_line, expected_col) in mappings.iter().copied() {
let (line, col) = builder.search_original_line_and_column(position);
assert_eq!(
Expand All @@ -407,8 +402,7 @@ mod test {
#[test]
fn add_source_mapping() {
fn create_mappings(source: &str, line: u32, column: u32) {
let mut builder = SourcemapBuilder::default();
builder.with_name_and_source("x.js", source);
let mut builder = SourcemapBuilder::new(Path::new("x.js"), source);
let output: Vec<u8> = source.as_bytes().into();
for (i, _ch) in source.char_indices() {
#[allow(clippy::cast_possible_truncation)]
Expand Down Expand Up @@ -444,8 +438,7 @@ mod test {
#[test]
fn add_source_mapping_for_name() {
let output = "ac".as_bytes();
let mut builder = SourcemapBuilder::default();
builder.with_name_and_source("x.js", "ab");
let mut builder = SourcemapBuilder::new(Path::new("x.js"), "ab");
builder.add_source_mapping_for_name(output, Span::new(0, 1), "a");
builder.add_source_mapping_for_name(output, Span::new(1, 2), "c");
let sm = builder.into_sourcemap();
Expand All @@ -464,8 +457,7 @@ mod test {
#[test]
fn add_source_mapping_for_unordered_position() {
let output = "".as_bytes();
let mut builder = SourcemapBuilder::default();
builder.with_name_and_source("x.js", "ab");
let mut builder = SourcemapBuilder::new(Path::new("x.js"), "ab");
builder.add_source_mapping(output, 1, None);
builder.add_source_mapping(output, 0, None);
let sm = builder.into_sourcemap();
Expand Down
13 changes: 7 additions & 6 deletions napi/transform/src/isolated_declaration.rs
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@ use napi_derive::napi;

use oxc::{
allocator::Allocator,
codegen::CodeGenerator,
codegen::{CodeGenerator, CodegenOptions},
isolated_declarations::IsolatedDeclarations,
napi::{
isolated_declarations::{IsolatedDeclarationsOptions, IsolatedDeclarationsResult},
Expand Down Expand Up @@ -39,11 +39,12 @@ pub fn isolated_declaration(
)
.build(&ret.program);

let mut codegen = CodeGenerator::new();
if options.sourcemap == Some(true) {
codegen = codegen.enable_source_map(&filename, &source_text);
}
let codegen_ret = codegen.build(&transformed_ret.program);
let codegen_ret = CodeGenerator::new()
.with_options(CodegenOptions {
source_map_path: Some(source_path.to_path_buf()),
..CodegenOptions::default()
})
.build(&transformed_ret.program);

let errors = ret.errors.into_iter().chain(transformed_ret.errors).collect();
let errors = wrap_diagnostics(source_path, source_type, &source_text, errors);
Expand Down
11 changes: 9 additions & 2 deletions tasks/benchmark/benches/codegen.rs
Original file line number Diff line number Diff line change
@@ -1,6 +1,8 @@
use std::path::PathBuf;

use oxc_allocator::Allocator;
use oxc_benchmark::{criterion_group, criterion_main, BenchmarkId, Criterion};
use oxc_codegen::CodeGenerator;
use oxc_codegen::{CodeGenerator, CodegenOptions};
use oxc_parser::Parser;
use oxc_span::SourceType;
use oxc_tasks_common::TestFiles;
Expand All @@ -22,7 +24,12 @@ fn bench_codegen(criterion: &mut Criterion) {
let mut group = criterion.benchmark_group("codegen_sourcemap");
group.bench_with_input(id, &ret.program, |b, program| {
b.iter_with_large_drop(|| {
CodeGenerator::new().enable_source_map(&file.file_name, source_text).build(program)
CodeGenerator::new()
.with_options(CodegenOptions {
source_map_path: Some(PathBuf::from(&file.file_name)),
..CodegenOptions::default()
})
.build(program)
});
});
group.finish();
Expand Down
14 changes: 11 additions & 3 deletions tasks/benchmark/benches/sourcemap.rs
Original file line number Diff line number Diff line change
@@ -1,6 +1,8 @@
use std::path::PathBuf;

use oxc_allocator::Allocator;
use oxc_benchmark::{criterion_group, criterion_main, BenchmarkId, Criterion};
use oxc_codegen::{CodeGenerator, CodegenReturn};
use oxc_codegen::{CodeGenerator, CodegenOptions, CodegenReturn};
use oxc_parser::Parser;
use oxc_sourcemap::ConcatSourceMapBuilder;
use oxc_span::SourceType;
Expand All @@ -18,13 +20,19 @@ fn bench_sourcemap(criterion: &mut Criterion) {
let ret = Parser::new(&allocator, source_text, source_type).parse();

let CodegenReturn { code: output_txt, .. } = CodeGenerator::new()
.enable_source_map(file.file_name.as_str(), source_text)
.with_options(CodegenOptions {
source_map_path: Some(PathBuf::from(&file.file_name)),
..CodegenOptions::default()
})
.build(&ret.program);
let lines = output_txt.matches('\n').count() as u32;

b.iter(|| {
let CodegenReturn { map, .. } = CodeGenerator::new()
.enable_source_map(file.file_name.as_str(), source_text)
.with_options(CodegenOptions {
source_map_path: Some(PathBuf::from(&file.file_name)),
..CodegenOptions::default()
})
.build(&ret.program);
if let Some(sourcemap) = map {
let concat_sourcemap_builder = ConcatSourceMapBuilder::from_sourcemaps(&[
Expand Down

0 comments on commit 114c4ce

Please sign in to comment.