Popular
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443// Copyright 2020 the Aleph.js authors. All rights reserved. MIT license.
use indexmap::IndexSet;use sha1::{Digest, Sha1};use std::rc::Rc;use swc_common::{SourceMap, Spanned, DUMMY_SP};use swc_ecma_ast::*;use swc_ecma_utils::{private_ident, quote_ident};use swc_ecma_visit::{noop_fold_type, Fold};
pub fn fast_refresh_fold( refresh_reg: &str, refresh_sig: &str, emit_full_signatures: bool, source: Rc<SourceMap>,) -> impl Fold { FastRefreshFold { source, signature_index: 0, registration_index: 0, registrations: vec![], signatures: vec![], refresh_reg: refresh_reg.into(), refresh_sig: refresh_sig.into(), emit_full_signatures, }}
/// aleph.js fast-refresh fold.////// @ref https://github.com/facebook/react/blob/master/packages/react-refresh/src/ReactFreshBabelPlugin.jspub struct FastRefreshFold { source: Rc<SourceMap>, signature_index: u32, registration_index: u32, registrations: Vec<(Ident, String)>, signatures: Vec<Signature>, refresh_reg: String, refresh_sig: String, emit_full_signatures: bool,}
#[derive(Clone, Debug)]struct Signature { parent_ident: Option<Ident>, handle_ident: Ident, hook_calls: Vec<HookCall>,}
#[derive(Clone, Debug)]struct HookCall { obj: Option<Ident>, ident: Ident, key: String, is_builtin: bool,}
impl FastRefreshFold { fn create_registration_handle_ident(&mut self) -> Ident { let mut registration_handle_name = String::from("_c"); self.registration_index += 1; if self.registration_index > 1 { registration_handle_name.push_str(&self.registration_index.to_string()); }; private_ident!(registration_handle_name.as_str()) }
fn get_persistent_fn( &mut self, bindings: &IndexSet<String>, ident: Option<&Ident>, block_stmt: &mut BlockStmt, ) -> (Option<Ident>, Option<Signature>) { let fc_id = match ident { Some(ident) => { if is_componentish_name(ident.as_ref()) { Some(ident.clone()) } else { None } } None => None, }; let mut bindings_scope = IndexSet::<String>::new(); let mut hook_calls = Vec::<HookCall>::new(); let mut exotic_signatures = Vec::<(usize, Signature, Option<Expr>)>::new(); let mut index: usize = 0; let stmts = &mut block_stmt.stmts;
// marge top bindings for id in bindings.iter() { bindings_scope.insert(id.to_string()); }
// collect scope bindings stmts.into_iter().for_each(|stmt| { match stmt { // function useFancyState() {} Stmt::Decl(Decl::Fn(FnDecl { ident, .. })) => { bindings_scope.insert(ident.sym.as_ref().into()); } Stmt::Decl(Decl::Var(VarDecl { decls, .. })) => { decls.into_iter().for_each(|decl| match decl { VarDeclarator { name: Pat::Ident(ident), init: Some(init_expr), .. } => match init_expr.as_ref() { // const useFancyState = function () {} Expr::Fn(_) => { bindings_scope.insert(ident.sym.as_ref().into()); } // const useFancyState = () => {} Expr::Arrow(_) => { bindings_scope.insert(ident.sym.as_ref().into()); } _ => {} }, _ => {} }); } _ => {} } });
stmts.into_iter().for_each(|stmt| { match stmt { // function useFancyState() {} Stmt::Decl(Decl::Fn(FnDecl { ident, function: Function { body: Some(body), .. }, .. })) => { if let (_, Some(signature)) = self.get_persistent_fn(&bindings_scope, Some(ident), body) { exotic_signatures.push((index, signature, None)); } } // var ... Stmt::Decl(Decl::Var(VarDecl { decls, .. })) => { decls.into_iter().for_each(|decl| match decl { VarDeclarator { name, init: Some(init_expr), .. } => match init_expr.as_mut() { // const useFancyState = function () {} Expr::Fn(FnExpr { function: Function { body: Some(body), .. }, .. }) => match name { Pat::Ident(ident) => { if let (_, Some(signature)) = self.get_persistent_fn(&bindings_scope, Some(ident), body) { exotic_signatures.push((index, signature, None)); } } _ => {} }, // const useFancyState = () => {} Expr::Arrow(ArrowExpr { body: BlockStmtOrExpr::BlockStmt(body), .. }) => match name { Pat::Ident(ident) => { if let (_, Some(signature)) = self.get_persistent_fn(&bindings_scope, Some(ident), body) { exotic_signatures.push((index, signature, None)); } } _ => {} }, // cosnt [state, setState] = useSate() Expr::Call(call) => match self.get_hook_call(Some(name), call) { Some(hc) => hook_calls.push(hc), _ => {} }, _ => {} }, _ => {} }); } // useEffect() Stmt::Expr(ExprStmt { expr, .. }) => match expr.as_ref() { Expr::Call(call) => match self.get_hook_call(None, call) { Some(hc) => hook_calls.push(hc), _ => {} }, _ => {} }, // return .. Stmt::Return(ReturnStmt { arg: Some(arg), .. }) => match arg.as_mut() { // return function() {} Expr::Fn(FnExpr { function: Function { body: Some(body), .. }, .. }) => { if let (_, Some(signature)) = self.get_persistent_fn(&bindings_scope, None, body) { exotic_signatures.push((index, signature, Some(arg.as_ref().clone()))); } } // return () => {} Expr::Arrow(ArrowExpr { body: BlockStmtOrExpr::BlockStmt(body), .. }) => { if let (_, Some(signature)) = self.get_persistent_fn(&bindings_scope, None, body) { exotic_signatures.push((index, signature, Some(arg.as_ref().clone()))); } } _ => {} }, _ => {} } index += 1; });
// ! insert // _s(); let mut inserted: usize = 0; let signature = if hook_calls.len() > 0 { let mut handle_ident = String::from("_s"); self.signature_index += 1; if self.signature_index > 1 { handle_ident.push_str(self.signature_index.to_string().as_str()); }; let handle_ident = private_ident!(handle_ident.as_str()); block_stmt.stmts.insert( 0, Stmt::Expr(ExprStmt { span: DUMMY_SP, expr: Box::new(Expr::Call(CallExpr { span: DUMMY_SP, callee: ExprOrSuper::Expr(Box::new(Expr::Ident(handle_ident.clone()))), args: vec![], type_args: None, })), }), ); inserted += 1; Some(Signature { parent_ident: match ident { Some(ident) => Some(ident.clone()), None => None, }, handle_ident, hook_calls, }) } else { None };
if exotic_signatures.len() > 0 { // ! insert // var _s = $RefreshSig$(), _s2 = $RefreshSig$(); block_stmt.stmts.insert( inserted, Stmt::Decl(Decl::Var(VarDecl { span: DUMMY_SP, kind: VarDeclKind::Var, declare: false, decls: exotic_signatures .clone() .into_iter() .map(|signature| VarDeclarator { span: DUMMY_SP, name: Pat::Ident(signature.1.handle_ident), init: Some(Box::new(Expr::Call(CallExpr { span: DUMMY_SP, callee: ExprOrSuper::Expr(Box::new(Expr::Ident(quote_ident!(self .refresh_sig .as_str())))), args: vec![], type_args: None, }))), definite: false, }) .collect(), })), ); inserted += 1;
for (index, exotic_signature, return_expr) in exotic_signatures { let mut args = self.create_arguments_for_signature(&bindings_scope, &exotic_signature); if let Some(return_expr) = return_expr { args.insert( 0, ExprOrSpread { spread: None, expr: Box::new(return_expr), }, ); block_stmt.stmts[index + inserted] = Stmt::Return(ReturnStmt { span: DUMMY_SP, arg: Some(Box::new(Expr::Call(CallExpr { span: DUMMY_SP, callee: ExprOrSuper::Expr(Box::new(Expr::Ident( exotic_signature.handle_ident.clone(), ))), args, type_args: None, }))), }); } else { block_stmt.stmts.insert( index + inserted + 1, Stmt::Expr(ExprStmt { span: DUMMY_SP, expr: Box::new(Expr::Call(CallExpr { span: DUMMY_SP, callee: ExprOrSuper::Expr(Box::new(Expr::Ident( exotic_signature.handle_ident.clone(), ))), args, type_args: None, })), }), ); inserted += 1 } } } (fc_id, signature) }
fn get_hook_call(&self, pat: Option<&Pat>, call: &CallExpr) -> Option<HookCall> { if let Some((obj, ident)) = get_call_callee(call) { let ident_str = ident.sym.as_ref(); let is_builtin = is_builtin_hook( match &obj { Some(obj) => Some(obj), None => None, }, ident_str, ); if is_builtin || (ident_str.len() > 3 && ident_str.starts_with("use") && ident_str[3..].starts_with(char::is_uppercase)) { let mut key = ident_str.to_owned(); match pat { Some(pat) => { let name = self.source.span_to_snippet(pat.span()).unwrap(); key.push('{'); key.push_str(name.as_str()); // `useState` first argument is initial state. if call.args.len() > 0 && is_builtin && ident_str == "useState" { key.push('('); key.push_str( self .source .span_to_snippet(call.args[0].span()) .unwrap() .as_str(), ); key.push(')'); } // `useReducer` second argument is initial state. if call.args.len() > 1 && is_builtin && ident_str == "useReducer" { key.push('('); key.push_str( self .source .span_to_snippet(call.args[1].span()) .unwrap() .as_str(), ); key.push(')'); } key.push('}'); } _ => key.push_str("{}"), }; return Some(HookCall { obj, ident, key, is_builtin, }); } } None }
fn find_inner_component( &mut self, bindings: &IndexSet<String>, parent_name: &str, call: &mut CallExpr, ) -> bool { if !is_componentish_name(parent_name) && !parent_name.starts_with("%default%") { return false; }
if call.args.len() == 0 { return false; }
// first arg should be a function or call match call.args[0].expr.as_ref() { Expr::Fn(_) => {} Expr::Arrow(_) => {} Expr::Call(_) => {} _ => return false, }
if let Some((obj, ident)) = get_call_callee(call) { let mut ident_str = parent_name.to_owned(); ident_str.push('$'); match obj { Some(obj) => { ident_str.push_str(obj.sym.as_ref()); ident_str.push('.'); } _ => {} } ident_str.push_str(ident.sym.as_ref()); match call.args[0].expr.as_mut() { Expr::Call(inner_call) => { if let Some(_) = get_call_callee(inner_call) { let ok = self.find_inner_component(bindings, ident_str.as_str(), inner_call); if ok { let handle_ident = self.create_registration_handle_ident(); self .registrations .push((handle_ident.clone(), ident_str.clone())); call.args[0] = ExprOrSpread { spread: None, expr: Box::new(Expr::Assign(AssignExpr { span: DUMMY_SP, op: AssignOp::Assign, left: PatOrExpr::Expr(Box::new(Expr::Ident(handle_ident))), right: Box::new(Expr::Call(inner_call.clone())), })), } } return ok; } } _ => {} }
let handle_ident = self.create_registration_handle_ident(); self.registrations.push((handle_ident.clone(), ident_str)); match call.args[0].expr.as_mut() { Expr::Fn(fn_expr) => { let mut right = Box::new(Expr::Fn(fn_expr.clone())); match &mut fn_expr.function { Function { body: Some(body), .. } => { if let (_, Some(signature)) = self.get_persistent_fn(bindings, None, body) { let mut args = self.create_arguments_for_signature(bindings, &signature); args.insert( 0, ExprOrSpread { spread: None, expr: Box::new(Expr::Fn(fn_expr.clone())), }, ); right = Box::new(Expr::Call(CallExpr { span: DUMMY_SP, callee: ExprOrSuper::Expr(Box::new(Expr::Ident(signature.handle_ident.clone()))), args, type_args: None, })); self.signatures.push(signature); } } _ => {} }; call.args[0] = ExprOrSpread { spread: None, expr: Box::new(Expr::Assign(AssignExpr { span: DUMMY_SP, op: AssignOp::Assign, left: PatOrExpr::Expr(Box::new(Expr::Ident(handle_ident))), right, })), } } Expr::Arrow(arrow_expr) => { let mut right = Box::new(Expr::Arrow(arrow_expr.clone())); match &mut arrow_expr.body { BlockStmtOrExpr::BlockStmt(body) => { if let (_, Some(signature)) = self.get_persistent_fn(bindings, None, body) { let mut args = self.create_arguments_for_signature(bindings, &signature); args.insert( 0, ExprOrSpread { spread: None, expr: Box::new(Expr::Arrow(arrow_expr.clone())), }, ); right = Box::new(Expr::Call(CallExpr { span: DUMMY_SP, callee: ExprOrSuper::Expr(Box::new(Expr::Ident(signature.handle_ident.clone()))), args, type_args: None, })); self.signatures.push(signature); } } _ => {} }; call.args[0] = ExprOrSpread { spread: None, expr: Box::new(Expr::Assign(AssignExpr { span: DUMMY_SP, op: AssignOp::Assign, left: PatOrExpr::Expr(Box::new(Expr::Ident(handle_ident))), right, })), } } _ => {} } return true; } false }
fn create_arguments_for_signature( &self, bindings: &IndexSet<String>, signature: &Signature, ) -> Vec<ExprOrSpread> { let mut key = Vec::<String>::new(); let mut custom_hooks_in_scope = Vec::<(Option<Ident>, Ident)>::new(); let mut args: Vec<ExprOrSpread> = vec![]; match &signature.parent_ident { Some(parent_ident) => args.push(ExprOrSpread { spread: None, expr: Box::new(Expr::Ident(parent_ident.clone())), }), None => {} } let mut force_reset = false; // todo: parse @refresh reset command signature.hook_calls.clone().into_iter().for_each(|call| { key.push(call.key); if !call.is_builtin { match call.obj { Some(obj) => { if bindings.contains(obj.sym.as_ref().into()) { custom_hooks_in_scope.push((Some(obj.clone()), call.ident.clone())); } else { force_reset = true } } None => { if bindings.contains(call.ident.sym.as_ref().into()) { custom_hooks_in_scope.push((None, call.ident.clone())); } else { force_reset = true; } } } } }); let mut key = key.join("\n"); if !self.emit_full_signatures { let mut hasher = Sha1::new(); hasher.update(key); key = base64::encode(hasher.finalize()); } args.push(ExprOrSpread { spread: None, expr: Box::new(Expr::Lit(Lit::Str(Str { span: DUMMY_SP, value: key.into(), has_escape: false, kind: Default::default(), }))), }); if force_reset || custom_hooks_in_scope.len() > 0 { args.push(ExprOrSpread { spread: None, expr: Box::new(Expr::Lit(Lit::Bool(Bool { span: DUMMY_SP, value: force_reset, }))), }); } if custom_hooks_in_scope.len() > 0 { args.push(ExprOrSpread { spread: None, expr: Box::new(Expr::Arrow(ArrowExpr { span: DUMMY_SP, params: vec![], body: BlockStmtOrExpr::Expr(Box::new(Expr::Array(ArrayLit { span: DUMMY_SP, elems: custom_hooks_in_scope .into_iter() .map(|hook| { let (obj, id) = hook; if let Some(obj) = obj { Some(ExprOrSpread { spread: None, expr: Box::new(Expr::Member(MemberExpr { span: DUMMY_SP, obj: ExprOrSuper::Expr(Box::new(Expr::Ident(obj.clone()))), prop: Box::new(Expr::Ident(id.clone())), computed: false, })), }) } else { Some(ExprOrSpread { spread: None, expr: Box::new(Expr::Ident(id.clone())), }) } }) .collect(), }))), is_async: false, is_generator: false, type_params: None, return_type: None, })), }); } args }}
impl Fold for FastRefreshFold { noop_fold_type!();
fn fold_module_items(&mut self, module_items: Vec<ModuleItem>) -> Vec<ModuleItem> { let mut items = Vec::<ModuleItem>::new(); let mut raw_items = Vec::<ModuleItem>::new(); let mut bindings = IndexSet::<String>::new();
// collect top bindings for item in module_items.clone() { match item { // import React, {useState} from "/react.js" ModuleItem::ModuleDecl(ModuleDecl::Import(ImportDecl { specifiers, .. })) => { specifiers .into_iter() .for_each(|specifier| match specifier { ImportSpecifier::Named(ImportNamedSpecifier { local, .. }) | ImportSpecifier::Default(ImportDefaultSpecifier { local, .. }) | ImportSpecifier::Namespace(ImportStarAsSpecifier { local, .. }) => { bindings.insert(local.sym.as_ref().into()); } }); }
// export function App() {} ModuleItem::ModuleDecl(ModuleDecl::ExportDecl(ExportDecl { decl: Decl::Fn(FnDecl { ident, .. }), .. })) => { bindings.insert(ident.sym.as_ref().into()); }
// export default function App() {} ModuleItem::ModuleDecl(ModuleDecl::ExportDefaultDecl(ExportDefaultDecl { decl: DefaultDecl::Fn(FnExpr { ident: Some(ident), .. }), .. })) => { bindings.insert(ident.sym.as_ref().into()); }
// function App() {} ModuleItem::Stmt(Stmt::Decl(Decl::Fn(FnDecl { ident, .. }))) => { bindings.insert(ident.sym.as_ref().into()); }
// const Foo = () => {} // export const App = () => {} ModuleItem::Stmt(Stmt::Decl(Decl::Var(VarDecl { decls, .. }))) | ModuleItem::ModuleDecl(ModuleDecl::ExportDecl(ExportDecl { decl: Decl::Var(VarDecl { decls, .. }), .. })) => { decls.into_iter().for_each(|decl| match decl { VarDeclarator { name: Pat::Ident(ident), .. } => { bindings.insert(ident.sym.as_ref().into()); } _ => {} }); }
_ => {} }; }
for mut item in module_items { let mut persistent_fns = Vec::<(Option<Ident>, Option<Signature>)>::new(); let mut hocs = Vec::<(Ident, Ident)>::new(); match &mut item { // export function App() {} ModuleItem::ModuleDecl(ModuleDecl::ExportDecl(ExportDecl { decl: Decl::Fn(FnDecl { ident, function: Function { body: Some(body), .. }, .. }), .. })) => persistent_fns.push(self.get_persistent_fn(&bindings, Some(ident), body)),
// export default function App() {} ModuleItem::ModuleDecl(ModuleDecl::ExportDefaultDecl(ExportDefaultDecl { decl: DefaultDecl::Fn(FnExpr { ident: Some(ident), function: Function { body: Some(body), .. }, .. }), .. })) => persistent_fns.push(self.get_persistent_fn(&bindings, Some(ident), body)),
// export default React.memo(() => {}) ModuleItem::ModuleDecl(ModuleDecl::ExportDefaultExpr(ExportDefaultExpr { expr, .. })) => match expr.as_mut() { Expr::Call(call) => { if self.find_inner_component(&bindings, "%default%", call) { let handle_ident = self.create_registration_handle_ident(); self .registrations .push((handle_ident.clone(), "%default%".into())); // export default _c2 = React.memo(_c = () => {}) item = ModuleItem::ModuleDecl(ModuleDecl::ExportDefaultExpr(ExportDefaultExpr { span: DUMMY_SP, expr: Box::new(Expr::Assign(AssignExpr { span: DUMMY_SP, op: AssignOp::Assign, left: PatOrExpr::Expr(Box::new(Expr::Ident(handle_ident))), right: Box::new(Expr::Call(call.clone())), })), })); } } _ => {} },
// function App() {} ModuleItem::Stmt(Stmt::Decl(Decl::Fn(FnDecl { ident, function: Function { body: Some(body), .. }, .. }))) => persistent_fns.push(self.get_persistent_fn(&bindings, Some(ident), body)),
// const Foo = () => {} // export const App = () => {} ModuleItem::Stmt(Stmt::Decl(Decl::Var(VarDecl { decls, .. }))) | ModuleItem::ModuleDecl(ModuleDecl::ExportDecl(ExportDecl { decl: Decl::Var(VarDecl { decls, .. }), .. })) => { decls.into_iter().for_each(|decl| match decl { VarDeclarator { name: Pat::Ident(ident), init: Some(init_expr), .. } => { match init_expr.as_mut() { // const Foo = function () {} Expr::Fn(FnExpr { function: Function { body: Some(body), .. }, .. }) => persistent_fns.push(self.get_persistent_fn(&bindings, Some(ident), body)), // const Foo = () => {} Expr::Arrow(ArrowExpr { body: BlockStmtOrExpr::BlockStmt(body), .. }) => persistent_fns.push(self.get_persistent_fn(&bindings, Some(ident), body)), // const Bar = () => <div /> Expr::Arrow(ArrowExpr { body: BlockStmtOrExpr::Expr(expr), .. }) => match expr.as_ref() { Expr::JSXElement(jsx) => match jsx.as_ref() { JSXElement { .. } => persistent_fns.push((Some(ident.clone()), None)), }, _ => {} }, // const A = forwardRef(function() {}); Expr::Call(call) => { if self.find_inner_component(&bindings, ident.sym.as_ref(), call) { let handle_ident = self.create_registration_handle_ident(); self .registrations .push((handle_ident.clone(), ident.sym.as_ref().into())); hocs.push((ident.clone(), handle_ident)) } } _ => {} } } _ => {} }); }
_ => {} };
raw_items.push(item);
for (fc_id, signature) in persistent_fns { if let Some(fc_id) = fc_id { let registration_handle_id = self.create_registration_handle_ident(); self .registrations .push((registration_handle_id.clone(), fc_id.sym.as_ref().into()));
// ! insert // _c = App; // _c2 = Foo; raw_items.push(ModuleItem::Stmt(Stmt::Expr(ExprStmt { span: DUMMY_SP, expr: Box::new(Expr::Assign(AssignExpr { span: DUMMY_SP, op: AssignOp::Assign, left: PatOrExpr::Pat(Box::new(Pat::Ident(registration_handle_id))), right: Box::new(Expr::Ident(fc_id)), })), }))); }
if let Some(signature) = signature { self.signatures.push(signature); } }
// ! insert (hoc) // _c = App; // _c2 = Foo; for (hoc_handle_id, hoc_id) in hocs { raw_items.push(ModuleItem::Stmt(Stmt::Expr(ExprStmt { span: DUMMY_SP, expr: Box::new(Expr::Assign(AssignExpr { span: DUMMY_SP, op: AssignOp::Assign, left: PatOrExpr::Pat(Box::new(Pat::Ident(hoc_handle_id))), right: Box::new(Expr::Ident(hoc_id)), })), }))); } }
// ! insert // var _c, _c2; if self.registrations.len() > 0 { items.push(ModuleItem::Stmt(Stmt::Decl(Decl::Var(VarDecl { span: DUMMY_SP, kind: VarDeclKind::Var, declare: false, decls: self .registrations .clone() .into_iter() .map(|registration| VarDeclarator { span: DUMMY_SP, name: Pat::Ident(registration.0), init: None, definite: false, }) .collect(), })))); }
// ! insert // var _s = $RefreshSig$(), _s2 = $RefreshSig$(); if self.signatures.len() > 0 { items.push(ModuleItem::Stmt(Stmt::Decl(Decl::Var(VarDecl { span: DUMMY_SP, kind: VarDeclKind::Var, declare: false, decls: self .signatures .clone() .into_iter() .map(|signature| VarDeclarator { span: DUMMY_SP, name: Pat::Ident(signature.handle_ident), init: Some(Box::new(Expr::Call(CallExpr { span: DUMMY_SP, callee: ExprOrSuper::Expr(Box::new(Expr::Ident(quote_ident!(self .refresh_sig .as_str())))), args: vec![], type_args: None, }))), definite: false, }) .collect(), })))); }
// ! insert raw items for item in raw_items { items.push(item); }
// ! insert // _s(App, "useState{[count, setCount](0)}\nuseEffect{}"); for signature in &self.signatures { match signature.parent_ident { Some(_) => { let args = self.create_arguments_for_signature(&bindings, &signature); items.push(ModuleItem::Stmt(Stmt::Expr(ExprStmt { span: DUMMY_SP, expr: Box::new(Expr::Call(CallExpr { span: DUMMY_SP, callee: ExprOrSuper::Expr(Box::new(Expr::Ident(signature.handle_ident.clone()))), args, type_args: None, })), }))); } None => {} } }
// ! insert // $RefreshReg$(_c, "App"); // $RefreshReg$(_c2, "Foo"); for (registration_id, fc_name) in self.registrations.clone() { items.push(ModuleItem::Stmt(Stmt::Expr(ExprStmt { span: DUMMY_SP, expr: Box::new(Expr::Call(CallExpr { span: DUMMY_SP, callee: ExprOrSuper::Expr(Box::new(Expr::Ident(quote_ident!(self .refresh_reg .as_str())))), args: vec![ ExprOrSpread { spread: None, expr: Box::new(Expr::Ident(registration_id)), }, ExprOrSpread { spread: None, expr: Box::new(Expr::Lit(Lit::Str(Str { span: DUMMY_SP, value: fc_name.into(), has_escape: false, kind: Default::default(), }))), }, ], type_args: None, })), }))); } items }}
fn is_componentish_name(name: &str) -> bool { name.starts_with(char::is_uppercase)}
fn is_builtin_hook(obj: Option<&Ident>, id: &str) -> bool { let ok = match id { "useState" | "useReducer" | "useEffect" | "useLayoutEffect" | "useMemo" | "useCallback" | "useRef" | "useContext" | "useImperativeHandle" | "useDebugValue" => true, _ => false, }; match obj { Some(obj) => match obj.sym.as_ref() { "React" => ok, _ => false, }, None => ok, }}
fn get_call_callee(call: &CallExpr) -> Option<(Option<Ident>, Ident)> { let callee = match &call.callee { ExprOrSuper::Super(_) => return None, ExprOrSuper::Expr(callee) => callee.as_ref(), };
match callee { // useState() Expr::Ident(id) => Some((None, id.clone())), // React.useState() Expr::Member(expr) => match &expr.obj { ExprOrSuper::Expr(obj) => match obj.as_ref() { Expr::Ident(obj) => match expr.prop.as_ref() { Expr::Ident(prop) => Some((Some(obj.clone()), prop.clone())), _ => None, }, _ => None, }, _ => None, }, _ => None, }}
#[cfg(test)]mod tests { use super::*; use crate::swc::ParsedModule; use std::cmp::min; use swc_common::Globals;
fn t(specifier: &str, source: &str, expect: &str) -> bool { let module = ParsedModule::parse(specifier, source, None).expect("could not parse module"); let (code, _) = swc_common::GLOBALS.set(&Globals::new(), || { module .apply_transform( fast_refresh_fold( "$RefreshReg$", "$RefreshSig$", true, module.source_map.clone(), ), false, ) .expect("could not transpile module") });
if code != expect { let mut p: usize = 0; for i in 0..min(code.len(), expect.len()) { if code.get(i..i + 1) != expect.get(i..i + 1) { p = i; break; } } println!( "{}\x1b[0;31m{}\x1b[0m", code.get(0..p).unwrap(), code.get(p..).unwrap() ); } code == expect }
#[test] fn test_fast_refresh() { let source = r#" function Hello() { return <h1>Hi</h1>; } Hello = connect(Hello); const Bar = () => { return <Hello />; }; var Baz = () => <div />; export default function App() { const [foo, setFoo] = useState(0); const bar = useState(() => 0); const [state, dispatch] = useReducer(reducer, initialState, init); React.useEffect(() => {}, []); return <h1>{foo}</h1>; } "#; let expect = r#"var _c, _c2, _c3, _c4;var _s = $RefreshSig$();function Hello() { return <h1 >Hi</h1>;}_c = Hello;Hello = connect(Hello);const Bar = ()=>{ return <Hello />;};_c2 = Bar;var Baz = ()=><div />;_c3 = Baz;export default function App() { _s(); const [foo, setFoo] = useState(0); const bar = useState(()=>0 ); const [state, dispatch] = useReducer(reducer, initialState, init); React.useEffect(()=>{ }, []); return <h1 >{foo}</h1>;};_c4 = App;_s(App, "useState{[foo, setFoo](0)}\nuseState{bar(() => 0)}\nuseReducer{[state, dispatch](initialState)}\nuseEffect{}");$RefreshReg$(_c, "Hello");$RefreshReg$(_c2, "Bar");$RefreshReg$(_c3, "Baz");$RefreshReg$(_c4, "App");"#; assert!(t("/app.jsx", source, expect)); }
#[test] fn test_fast_refresh_custom_hooks() { let source = r#" const useFancyEffect = () => { React.useEffect(() => { }); }; function useFancyState() { const [foo, setFoo] = React.useState(0); useFancyEffect(); return foo; } function useFoo() { const [x] = useBar(1, 2, 3); useBarEffect(); } export default function App() { const bar = useFancyState(); return <h1>{bar}</h1>; } "#; let expect = r#"var _c;var _s = $RefreshSig$(), _s2 = $RefreshSig$(), _s3 = $RefreshSig$(), _s4 = $RefreshSig$();const useFancyEffect = ()=>{ _s(); React.useEffect(()=>{ });};function useFancyState() { _s2(); const [foo, setFoo] = React.useState(0); useFancyEffect(); return foo;}function useFoo() { _s3(); const [x] = useBar(1, 2, 3); useBarEffect();}export default function App() { _s4(); const bar = useFancyState(); return <h1 >{bar}</h1>;};_c = App;_s(useFancyEffect, "useEffect{}");_s2(useFancyState, "useState{[foo, setFoo](0)}\nuseFancyEffect{}", false, ()=>[ useFancyEffect ]);_s3(useFoo, "useBar{[x]}\nuseBarEffect{}", true);_s4(App, "useFancyState{bar}", false, ()=>[ useFancyState ]);$RefreshReg$(_c, "App");"#; assert!(t("/app.jsx", source, expect)); }
#[test] fn test_fast_refresh_exotic_signature() { let source = r#" import FancyHook from 'fancy';
export default function App() { const useFancyState = () => { const [foo, setFoo] = React.useState(0); useFancyEffect(); return foo; } const bar = useFancyState(); const baz = FancyHook.useThing(); React.useState(); useThePlatform(); useFancyEffect();
function useFancyEffect() { useEffect(); }
return <h1>{bar}{baz}</h1>; } "#; let expect = r#"var _c;var _s3 = $RefreshSig$();import FancyHook from 'fancy';export default function App() { _s3(); var _s = $RefreshSig$(), _s2 = $RefreshSig$(); const useFancyState = ()=>{ _s(); const [foo, setFoo] = React.useState(0); useFancyEffect(); return foo; }; _s(useFancyState, "useState{[foo, setFoo](0)}\nuseFancyEffect{}", false, ()=>[ useFancyEffect ] ); const bar = useFancyState(); const baz = FancyHook.useThing(); React.useState(); useThePlatform(); useFancyEffect(); function useFancyEffect() { _s2(); useEffect(); } _s2(useFancyEffect, "useEffect{}"); return <h1 >{bar}{baz}</h1>;};_c = App;_s3(App, "useFancyState{bar}\nuseThing{baz}\nuseState{}\nuseThePlatform{}\nuseFancyEffect{}", true, ()=>[ FancyHook.useThing ]);$RefreshReg$(_c, "App");"#; assert!(t("/app.jsx", source, expect)); }
#[test] fn test_fast_refresh_hocs() { let source = r#" const A = forwardRef(function() { return <h1>Foo</h1>; }); const B = memo(React.forwardRef(() => { return <h1>Foo</h1>; })); const C = forwardRef(memo(forwardRef(()=>null))) export const D = React.memo(React.forwardRef((props, ref) => { const [foo, setFoo] = useState(0); React.useEffect(() => {}); return <h1 ref={ref}>{foo}</h1>; })); export const E = React.memo(React.forwardRef(function(props, ref) { const [foo, setFoo] = useState(0); React.useEffect(() => {}); return <h1 ref={ref}>{foo}</h1>; })); function hoc() { return function Inner() { const [foo, setFoo] = useState(0); React.useEffect(() => {}); return <h1 ref={ref}>{foo}</h1>; }; } const F = memo('Foo'); const G = forwardRef(memo(forwardRef())); const I = forwardRef(memo(forwardRef(0, () => {}))); export let H = hoc(); export default React.memo(forwardRef((props, ref) => { return <h1>Foo</h1>; })); "#; let expect = r#"var _c, _c2, _c3, _c4, _c5, _c6, _c7, _c8, _c9, _c10, _c11, _c12, _c13, _c14, _c15, _c16, _c17, _c18;var _s = $RefreshSig$(), _s2 = $RefreshSig$();const A = forwardRef(_c = function() { return <h1 >Foo</h1>;});A = _c2;const B = memo(_c4 = React.forwardRef(_c3 = ()=>{ return <h1 >Foo</h1>;}));B = _c5;const C = forwardRef(_c8 = memo(_c7 = forwardRef(_c6 = ()=>null)));C = _c9;export const D = React.memo(_c11 = React.forwardRef(_c10 = _s((props, ref)=>{ _s(); const [foo, setFoo] = useState(0); React.useEffect(()=>{ }); return <h1 ref={ref}>{foo}</h1>;}, "useState{[foo, setFoo](0)}\nuseEffect{}")));D = _c12;export const E = React.memo(_c14 = React.forwardRef(_c13 = _s2(function(props, ref) { _s2(); const [foo, setFoo] = useState(0); React.useEffect(()=>{ }); return <h1 ref={ref}>{foo}</h1>;}, "useState{[foo, setFoo](0)}\nuseEffect{}")));E = _c15;function hoc() { var _s3 = $RefreshSig$(); return _s3(function Inner() { _s3(); const [foo, setFoo] = useState(0); React.useEffect(()=>{ }); return <h1 ref={ref}>{foo}</h1>; }, "useState{[foo, setFoo](0)}\nuseEffect{}");}const F = memo('Foo');const G = forwardRef(memo(forwardRef()));const I = forwardRef(memo(forwardRef(0, ()=>{})));export let H = hoc();export default _c18 = React.memo(_c17 = forwardRef(_c16 = (props, ref)=>{ return <h1 >Foo</h1>;}));$RefreshReg$(_c, "A$forwardRef");$RefreshReg$(_c2, "A");$RefreshReg$(_c3, "B$memo$React.forwardRef");$RefreshReg$(_c4, "B$memo");$RefreshReg$(_c5, "B");$RefreshReg$(_c6, "C$forwardRef$memo$forwardRef");$RefreshReg$(_c7, "C$forwardRef$memo");$RefreshReg$(_c8, "C$forwardRef");$RefreshReg$(_c9, "C");$RefreshReg$(_c10, "D$React.memo$React.forwardRef");$RefreshReg$(_c11, "D$React.memo");$RefreshReg$(_c12, "D");$RefreshReg$(_c13, "E$React.memo$React.forwardRef");$RefreshReg$(_c14, "E$React.memo");$RefreshReg$(_c15, "E");$RefreshReg$(_c16, "%default%$React.memo$forwardRef");$RefreshReg$(_c17, "%default%$React.memo");$RefreshReg$(_c18, "%default%");"#; assert!(t("/app.jsx", source, expect)); }
#[test] fn test_fast_refresh_ignored() { let source = r#" const NotAComp = 'hi'; export { Baz, NotAComp }; export function sum() {} export const Bad = 42;
let connect = () => { function Comp() { const handleClick = () => {}; return <h1 onClick={handleClick}>Hi</h1>; } return Comp; }; function withRouter() { return function Child() { const handleClick = () => {}; return <h1 onClick={handleClick}>Hi</h1>; } };
let A = foo ? () => { return <h1>Hi</h1>; } : null; const B = (function Foo() { return <h1>Hi</h1>; })(); let C = () => () => { return <h1>Hi</h1>; }; let D = bar && (() => { return <h1>Hi</h1>; });
const throttledAlert = throttle(function () { alert('Hi'); }); const TooComplex = function () { return hello; }(() => {}); if (cond) { const Foo = thing(() => {}); }
export default function() {} "#; let expect = r#"const NotAComp = 'hi';export { Baz, NotAComp };export function sum() {}export const Bad = 42;let connect = ()=>{ function Comp() { const handleClick = ()=>{ }; return <h1 onClick={handleClick}>Hi</h1>; } return Comp;};function withRouter() { return function Child() { const handleClick = ()=>{ }; return <h1 onClick={handleClick}>Hi</h1>; };};let A = foo ? ()=>{ return <h1 >Hi</h1>;} : null;const B = (function Foo() { return <h1 >Hi</h1>;})();let C = ()=>()=>{ return <h1 >Hi</h1>; };let D = bar && (()=>{ return <h1 >Hi</h1>;});const throttledAlert = throttle(function() { alert('Hi');});const TooComplex = function() { return hello;}(()=>{});if (cond) { const Foo = thing(()=>{ });}export default function() {};"#; assert!(t("/app.jsx", source, expect)); }}