refactor: 拆分 oxide_emit 为独立 emit pass crate（依赖倒置）

2026-10-25 11:11:11 +08:00 · 2026-08-05 20:13:18 +08:00
parent 4f670eb7fd
commit 7c8f19f61a
62 changed files with 1045 additions and 993 deletions
@@ -0,0 +1,23 @@
+use crate::{CompileCtx, Emitter};
+use oxide_ir::inst::Inst;
+use oxide_parser::Statement;
+
+impl Emitter {
+    pub(crate) fn emit_do_while_statement(&self, stmt: &Statement, ctx: &mut CompileCtx) -> Result<Option<u8>, String> {
+        let Statement::DoWhileStatement(dw) = stmt else {
+            return Ok(None);
+        };
+        let start_label = ctx.next_label_id();
+        let end_label = ctx.next_label_id();
+        ctx.labels.set_label_pos(start_label, ctx.insts.len());
+        ctx.push_loop(end_label, start_label);
+        let n_labeled = ctx.take_pending_loop_labels(end_label, start_label);
+        self.emit_statement(&dw.body, ctx)?;
+        let test_reg = self.emit_expression(&dw.test, ctx)?;
+        ctx.inst(Inst::jmp_if_true(test_reg, start_label));
+        ctx.labels.set_label_pos(end_label, ctx.insts.len());
+        ctx.pop_label_scopes(n_labeled);
+        ctx.pop_loop();
+        Ok(None)
+    }
+}
@@ -0,0 +1,63 @@
+use crate::{CompileCtx, Emitter};
+use oxide_ir::inst::Inst;
+use oxide_ir::operand::Operand;
+use oxide_bytecode::module::Constant;
+use oxide_bytecode::opcode::OpCode;
+use oxide_parser::{BindingPattern, ForStatementInit, Statement, VariableDeclarationKind};
+
+impl Emitter {
+    pub(crate) fn emit_for_statement(&self, stmt: &Statement, ctx: &mut CompileCtx) -> Result<Option<u8>, String> {
+        let Statement::ForStatement(fr) = stmt else {
+            return Ok(None);
+        };
+        let start_label = ctx.next_label_id();
+        let update_label = ctx.next_label_id();
+        let end_label = ctx.next_label_id();
+        ctx.push_loop(end_label, update_label);
+        let n_labeled = ctx.take_pending_loop_labels(end_label, update_label);
+        if let Some(init) = &fr.init {
+            if let Some(expr) = init.as_expression() {
+                self.emit_expression(expr, ctx)?;
+            } else if let ForStatementInit::VariableDeclaration(decl) = init {
+                for d in &decl.declarations {
+                    let is_const = matches!(decl.kind, VariableDeclarationKind::Const);
+                    if let Some(init_expr) = &d.init {
+                        let val_reg = self.emit_expression(init_expr, ctx)?;
+                        self.emit_binding_pattern(&d.id, val_reg, decl.kind, is_const, ctx)?;
+                    } else if let BindingPattern::BindingIdentifier(bi) = &d.id {
+                        let idx = ctx.add_constant(Constant::Undefined);
+                        let tmp = ctx.alloc_reg();
+                        ctx.inst(Inst::load_const(Operand::Reg(tmp as u32), idx));
+                        let var_reg = ctx.alloc_reg();
+                        let target_reg = if matches!(decl.kind, VariableDeclarationKind::Var) {
+                            match ctx.declare(bi.name.as_str(), var_reg, decl.kind, is_const) {
+                                Ok(()) => var_reg,
+                                Err(_) => ctx.lookup(bi.name.as_str()).unwrap_or(var_reg),
+                            }
+                        } else {
+                            ctx.declare(bi.name.as_str(), var_reg, decl.kind, is_const)?;
+                            var_reg
+                        };
+                        ctx.inst(Inst::new(OpCode::STORE_VAR, Operand::Reg(target_reg as u32), Operand::Reg(tmp as u32), Operand::None));
+                        ctx.init_var(bi.name.as_str());
+                    }
+                }
+            }
+        }
+        ctx.labels.set_label_pos(start_label, ctx.insts.len());
+        if let Some(test) = &fr.test {
+            let test_reg = self.emit_expression(test, ctx)?;
+            ctx.inst(Inst::jmp_if_false(test_reg, end_label));
+        }
+        self.emit_statement(&fr.body, ctx)?;
+        ctx.labels.set_label_pos(update_label, ctx.insts.len());
+        if let Some(update) = &fr.update {
+            self.emit_expression(update, ctx)?;
+        }
+        ctx.inst(Inst::jmp(start_label));
+        ctx.labels.set_label_pos(end_label, ctx.insts.len());
+        ctx.pop_label_scopes(n_labeled);
+        ctx.pop_loop();
+        Ok(None)
+    }
+}
@@ -0,0 +1,56 @@
+use crate::{CompileCtx, Emitter};
+use oxide_ir::inst::Inst;
+use oxide_ir::operand::Operand;
+use oxide_bytecode::opcode::OpCode;
+use oxide_parser::{ForStatementLeft, Statement, VariableDeclarationKind};
+
+impl Emitter {
+    pub(crate) fn emit_for_in_statement(&self, stmt: &Statement, ctx: &mut CompileCtx) -> Result<Option<u8>, String> {
+        let Statement::ForInStatement(fi) = stmt else {
+            return Ok(None);
+        };
+        let start_label = ctx.next_label_id();
+        let end_label = ctx.next_label_id();
+        let obj_reg = self.emit_expression(&fi.right, ctx)?;
+        ctx.inst(Inst::new(OpCode::FOR_IN_INIT, Operand::None, Operand::Reg(obj_reg as u32), Operand::None));
+        ctx.labels.set_label_pos(start_label, ctx.insts.len());
+        ctx.push_loop(end_label, start_label);
+        let n_labeled = ctx.take_pending_loop_labels(end_label, start_label);
+        let done_reg = ctx.alloc_reg();
+        ctx.inst(Inst::new(OpCode::FOR_IN_DONE, Operand::Reg(done_reg as u32), Operand::None, Operand::None));
+        // done=true 表示迭代结束：done=false 时跳过 end jmp 继续迭代
+        let continue_label = ctx.next_label_id();
+        ctx.inst(Inst::jmp_if_false(done_reg, continue_label));
+        ctx.inst(Inst::jmp(end_label));
+        ctx.labels.set_label_pos(continue_label, ctx.insts.len());
+        let key_reg = ctx.alloc_reg();
+        ctx.inst(Inst::new(OpCode::FOR_IN_NEXT, Operand::Reg(key_reg as u32), Operand::None, Operand::None));
+        match &fi.left {
+            ForStatementLeft::VariableDeclaration(decl) => {
+                for d in &decl.declarations {
+                    let name = match &d.id {
+                        oxide_parser::BindingPattern::BindingIdentifier(bi) => bi.name.as_str(),
+                        _ => return Err("destructuring not supported".into()),
+                    };
+                    let var_reg = ctx.alloc_reg();
+                    ctx.declare(name, var_reg, decl.kind, matches!(decl.kind, VariableDeclarationKind::Const))?;
+                    ctx.inst(Inst::new(OpCode::STORE_VAR, Operand::Reg(var_reg as u32), Operand::Reg(key_reg as u32), Operand::None));
+                    ctx.init_var(name);
+                }
+            }
+            ForStatementLeft::AssignmentTargetIdentifier(id_ref) => {
+                let name = id_ref.name.as_str();
+                let var_reg = ctx.lookup_or_global(name);
+                ctx.inst(Inst::new(OpCode::STORE_VAR, Operand::Reg(var_reg as u32), Operand::Reg(key_reg as u32), Operand::None));
+            }
+            _ => return Err("unsupported for-in left-hand side".into()),
+        }
+        self.emit_statement(&fi.body, ctx)?;
+        ctx.inst(Inst::jmp(start_label));
+        ctx.labels.set_label_pos(end_label, ctx.insts.len());
+        ctx.inst(Inst::new(OpCode::FOR_IN_CLEANUP, Operand::None, Operand::None, Operand::None));
+        ctx.pop_label_scopes(n_labeled);
+        ctx.pop_loop();
+        Ok(None)
+    }
+}
@@ -0,0 +1,51 @@
+use crate::{CompileCtx, Emitter};
+use oxide_ir::inst::Inst;
+use oxide_ir::operand::Operand;
+use oxide_bytecode::opcode::OpCode;
+use oxide_parser::{ForStatementLeft, Statement};
+
+impl Emitter {
+    pub(crate) fn emit_for_of_statement(&self, stmt: &Statement, ctx: &mut CompileCtx) -> Result<Option<u8>, String> {
+        let Statement::ForOfStatement(fo) = stmt else {
+            return Ok(None);
+        };
+        let start_label = ctx.next_label_id();
+        let end_label = ctx.next_label_id();
+        let iter_src_reg = self.emit_expression(&fo.right, ctx)?;
+        ctx.inst(Inst::new(OpCode::FOR_OF_INIT, Operand::None, Operand::Reg(iter_src_reg as u32), Operand::None));
+        ctx.labels.set_label_pos(start_label, ctx.insts.len());
+        ctx.push_loop(end_label, start_label);
+        let n_labeled = ctx.take_pending_loop_labels(end_label, start_label);
+        let has_reg = ctx.alloc_reg();
+        ctx.inst(Inst::new(OpCode::FOR_OF_DONE, Operand::Reg(has_reg as u32), Operand::None, Operand::None));
+        ctx.inst(Inst::jmp_if_false(has_reg, end_label));
+        let val_reg = ctx.alloc_reg();
+        ctx.inst(Inst::new(OpCode::FOR_OF_NEXT, Operand::Reg(val_reg as u32), Operand::None, Operand::None));
+        match &fo.left {
+            ForStatementLeft::VariableDeclaration(decl) => {
+                for d in &decl.declarations {
+                    self.emit_binding_pattern(&d.id, val_reg, decl.kind, false, ctx)?;
+                }
+            }
+            ForStatementLeft::AssignmentTargetIdentifier(id_ref) => {
+                let name = id_ref.name.as_str();
+                let var_reg = ctx.lookup_or_global(name);
+                ctx.inst(Inst::new(OpCode::STORE_VAR, Operand::Reg(var_reg as u32), Operand::Reg(val_reg as u32), Operand::None));
+            }
+            ForStatementLeft::ArrayAssignmentTarget(ap) => {
+                self.emit_array_assignment(ap, val_reg, ctx)?;
+            }
+            ForStatementLeft::ObjectAssignmentTarget(op) => {
+                self.emit_object_assignment(op, val_reg, ctx)?;
+            }
+            _ => return Err("unsupported for-of left-hand side".into()),
+        }
+        self.emit_statement(&fo.body, ctx)?;
+        ctx.inst(Inst::jmp(start_label));
+        ctx.labels.set_label_pos(end_label, ctx.insts.len());
+        ctx.inst(Inst::new(OpCode::FOR_OF_CLOSE, Operand::None, Operand::None, Operand::None));
+        ctx.pop_label_scopes(n_labeled);
+        ctx.pop_loop();
+        Ok(None)
+    }
+}
@@ -0,0 +1,21 @@
+use crate::{CompileCtx, Emitter};
+use oxide_parser::Statement;
+
+mod do_while;
+mod for_;
+mod for_in;
+mod for_of;
+mod while_loop;
+
+impl Emitter {
+    pub(crate) fn emit_iteration_domain(&self, stmt: &Statement, ctx: &mut CompileCtx) -> Result<Option<u8>, String> {
+        match stmt {
+            Statement::WhileStatement(_) => self.emit_while_statement(stmt, ctx),
+            Statement::DoWhileStatement(_) => self.emit_do_while_statement(stmt, ctx),
+            Statement::ForStatement(_) => self.emit_for_statement(stmt, ctx),
+            Statement::ForInStatement(_) => self.emit_for_in_statement(stmt, ctx),
+            Statement::ForOfStatement(_) => self.emit_for_of_statement(stmt, ctx),
+            _ => Ok(None),
+        }
+    }
+}
@@ -0,0 +1,24 @@
+use crate::{CompileCtx, Emitter};
+use oxide_ir::inst::Inst;
+use oxide_parser::Statement;
+
+impl Emitter {
+    pub(crate) fn emit_while_statement(&self, stmt: &Statement, ctx: &mut CompileCtx) -> Result<Option<u8>, String> {
+        let Statement::WhileStatement(wh) = stmt else {
+            return Ok(None);
+        };
+        let start_label = ctx.next_label_id();
+        let end_label = ctx.next_label_id();
+        ctx.labels.set_label_pos(start_label, ctx.insts.len());
+        ctx.push_loop(end_label, start_label);
+        let n_labeled = ctx.take_pending_loop_labels(end_label, start_label);
+        let test_reg = self.emit_expression(&wh.test, ctx)?;
+        ctx.inst(Inst::jmp_if_false(test_reg, end_label));
+        self.emit_statement(&wh.body, ctx)?;
+        ctx.inst(Inst::jmp(start_label));
+        ctx.labels.set_label_pos(end_label, ctx.insts.len());
+        ctx.pop_label_scopes(n_labeled);
+        ctx.pop_loop();
+        Ok(None)
+    }
+}