denoland-deno/cli/tools/lint/ast.rs

use deno_ast::{
  swc::{
    ast::{
      AssignTarget, AssignTargetPat, BlockStmtOrExpr, Callee, Decl,
      ExportSpecifier, Expr, ExprOrSpread, FnExpr, ForHead, Function, Ident,
      IdentName, JSXAttrName, JSXAttrOrSpread, JSXAttrValue, JSXElement,
      JSXElementChild, JSXElementName, JSXEmptyExpr, JSXExpr, JSXExprContainer,
      JSXFragment, JSXMemberExpr, JSXNamespacedName, JSXObject,
      JSXOpeningElement, Lit, MemberProp, ModuleDecl, ModuleExportName,
      ModuleItem, ObjectPatProp, Param, Pat, Program, Prop, PropName,
      PropOrSpread, SimpleAssignTarget, Stmt, SuperProp, TsType, VarDeclOrExpr,
    },
    common::{Span, Spanned, SyntaxContext, DUMMY_SP},
  },
  view::{AssignOp, BinaryOp, UnaryOp, UpdateOp, VarDeclKind},
  ParsedSource,
};
use indexmap::IndexMap;

// Keep in sync with JS
enum AstNode {
  Invalid,
  //
  Program,

  // Module declarations
  Import,
  ImportDecl,
  ExportDecl,
  ExportNamedDeclaration,
  ExportDefaultDecl,
  ExportDefaultExpr,
  ExportAll,
  TsImportEquals,
  TsExportAssignment,
  TsNamespaceExport,

  // Decls
  Class,
  Fn,
  Var,
  Using,
  TsInterface,
  TsTypeAlias,
  TsEnum,
  TsModule,

  // Statements
  Block,
  Empty,
  Debugger,
  With,
  Return,
  Labeled,
  Break,
  Continue,
  If,
  Switch,
  SwitchCase,
  Throw,
  TryStatement,
  While,
  DoWhile,
  For,
  ForInStatement,
  ForOf,
  Decl,
  Expr,

  // Expressions
  This,
  Array,
  Object,
  FnExpr,
  Unary,
  UpdateExpression,
  BinaryExpression,
  Assign,
  MemberExpression,
  Super,
  Cond,
  Call,
  New,
  Paren,
  Seq,
  Identifier,
  TemplateLiteral,
  TaggedTpl,
  ArrowFunctionExpression,
  ClassExpr,
  Yield,
  MetaProp,
  AwaitExpression,
  LogicalExpression,
  TsTypeAssertion,
  TsConstAssertion,
  TsNonNull,
  TsAs,
  TsInstantiation,
  TsSatisfies,
  PrivateIdentifier,
  OptChain,

  // Literals
  StringLiteral,
  Bool,
  Null,
  NumericLiteral,
  BigIntLiteral,
  RegExpLiteral,

  // Custom
  EmptyExpr,
  Spread,
  Property,
  VariableDeclarator,
  CatchClause,
  RestElement,
  ExportSpecifier,
  TemplateElement,

  // Patterns
  ArrayPattern,
  AssignmentPattern,
  ObjectPattern,

  // JSX
  JSXAttribute,
  JSXClosingElement,
  JSXClosingFragment,
  JSXElement,
  JSXEmptyExpression,
  JSXExpressionContainer,
  JSXFragment,
  JSXIdentifier,
  JSXMemberExpression,
  JSXNamespacedName,
  JSXOpeningElement,
  JSXOpeningFragment,
  JSXSpreadAttribute,
  JSXSpreadChild,
  JSXText,
}

impl From<AstNode> for u8 {
  fn from(m: AstNode) -> u8 {
    m as u8
  }
}

const MASK_U32_1: u32 = 0b11111111_00000000_00000000_00000000;
const MASK_U32_2: u32 = 0b00000000_11111111_00000000_00000000;
const MASK_U32_3: u32 = 0b00000000_00000000_11111111_00000000;
const MASK_U32_4: u32 = 0b00000000_00000000_00000000_11111111;

fn append_u32(result: &mut Vec<u8>, value: u32) {
  let v1: u8 = ((value & MASK_U32_1) >> 24) as u8;
  let v2: u8 = ((value & MASK_U32_2) >> 16) as u8;
  let v3: u8 = ((value & MASK_U32_3) >> 8) as u8;
  let v4: u8 = (value & MASK_U32_4) as u8;

  result.push(v1);
  result.push(v2);
  result.push(v3);
  result.push(v4);
}

fn append_usize(result: &mut Vec<u8>, value: usize) {
  let raw = u32::try_from(value).unwrap();
  append_u32(result, raw);
}

fn write_usize(result: &mut Vec<u8>, value: usize, idx: usize) {
  let raw = u32::try_from(value).unwrap();

  let v1: u8 = ((raw & MASK_U32_1) >> 24) as u8;
  let v2: u8 = ((raw & MASK_U32_2) >> 16) as u8;
  let v3: u8 = ((raw & MASK_U32_3) >> 8) as u8;
  let v4: u8 = (raw & MASK_U32_4) as u8;

  result[idx] = v1;
  result[idx + 1] = v2;
  result[idx + 2] = v3;
  result[idx + 3] = v4;
}

enum Flag {
  ProgramModule,
  FnAsync,
  FnGenerator,
  FnDeclare,
  MemberComputed,
  PropShorthand,
  PropComputed,
  PropGetter,
  PropSetter,
  PropMethod,
  VarVar,
  VarConst,
  VarLet,
  VarDeclare,
  ExportType,
  TplTail,
  ForAwait,
  LogicalOr,
  LogicalAnd,
  LogicalNullishCoalescin,
  JSXSelfClosing,

  BinEqEq,
  BinNotEq,
  BinEqEqEq,
  BinNotEqEq,
  BinLt,
  BinLtEq,
  BinGt,
  BinGtEq,
  BinLShift,
  BinRShift,
  BinZeroFillRShift,
  BinAdd,
  BinSub,
  BinMul,
  BinDiv,
  BinMod,
  BinBitOr,
  BinBitXor,
  BinBitAnd,
  BinIn,
  BinInstanceOf,
  BinExp,

  UnaryMinus,
  UnaryPlus,
  UnaryBang,
  UnaryTilde,
  UnaryTypeOf,
  UnaryVoid,
  UnaryDelete,

  UpdatePrefix,
  UpdatePlusPlus,
  UpdateMinusMinus,
}

fn assign_op_to_flag(m: AssignOp) -> u8 {
  match m {
    AssignOp::Assign => 0,
    AssignOp::AddAssign => 1,
    AssignOp::SubAssign => 2,
    AssignOp::MulAssign => 3,
    AssignOp::DivAssign => 4,
    AssignOp::ModAssign => 5,
    AssignOp::LShiftAssign => 6,
    AssignOp::RShiftAssign => 7,
    AssignOp::ZeroFillRShiftAssign => 8,
    AssignOp::BitOrAssign => 9,
    AssignOp::BitXorAssign => 10,
    AssignOp::BitAndAssign => 11,
    AssignOp::ExpAssign => 12,
    AssignOp::AndAssign => 13,
    AssignOp::OrAssign => 14,
    AssignOp::NullishAssign => 15,
  }
}

impl From<Flag> for u8 {
  fn from(m: Flag) -> u8 {
    match m {
      Flag::ProgramModule => 0b00000001,
      Flag::FnAsync => 0b00000001,
      Flag::FnGenerator => 0b00000010,
      Flag::FnDeclare => 0b00000100,
      Flag::MemberComputed => 0b00000001,
      Flag::PropShorthand => 0b00000001,
      Flag::PropComputed => 0b00000010,
      Flag::PropGetter => 0b00000100,
      Flag::PropSetter => 0b00001000,
      Flag::PropMethod => 0b00010000,
      Flag::VarVar => 0b00000001,
      Flag::VarConst => 0b00000010,
      Flag::VarLet => 0b00000100,
      Flag::VarDeclare => 0b00001000,
      Flag::ExportType => 0b000000001,
      Flag::TplTail => 0b000000001,
      Flag::ForAwait => 0b000000001,
      Flag::LogicalOr => 0b000000001,
      Flag::LogicalAnd => 0b000000010,
      Flag::LogicalNullishCoalescin => 0b000000100,
      Flag::JSXSelfClosing => 0b000000001,
      Flag::BinEqEq => 1,
      Flag::BinNotEq => 2,
      Flag::BinEqEqEq => 3,
      Flag::BinNotEqEq => 4,
      Flag::BinLt => 5,
      Flag::BinLtEq => 6,
      Flag::BinGt => 7,
      Flag::BinGtEq => 8,
      Flag::BinLShift => 9,
      Flag::BinRShift => 10,
      Flag::BinZeroFillRShift => 11,
      Flag::BinAdd => 12,
      Flag::BinSub => 13,
      Flag::BinMul => 14,
      Flag::BinDiv => 15,
      Flag::BinMod => 16,
      Flag::BinBitOr => 17,
      Flag::BinBitXor => 18,
      Flag::BinBitAnd => 19,
      Flag::BinIn => 20,
      Flag::BinInstanceOf => 21,
      Flag::BinExp => 22,

      Flag::UnaryMinus => 1,
      Flag::UnaryPlus => 2,
      Flag::UnaryBang => 3,
      Flag::UnaryTilde => 4,
      Flag::UnaryTypeOf => 5,
      Flag::UnaryVoid => 6,
      Flag::UnaryDelete => 7,

      Flag::UpdatePrefix => 0b000000001,
      Flag::UpdatePlusPlus => 0b000000010,
      Flag::UpdateMinusMinus => 0b000000100,
    }
  }
}

#[derive(Debug, Clone)]
struct FlagValue(u8);

impl FlagValue {
  fn new() -> Self {
    Self(0)
  }

  fn set(&mut self, flag: Flag) {
    let value: u8 = flag.into();
    self.0 |= value;
  }
}

#[derive(Debug)]
struct StringTable {
  id: usize,
  table: IndexMap<String, usize>,
}

impl StringTable {
  fn new() -> Self {
    Self {
      id: 0,
      table: IndexMap::new(),
    }
  }

  fn insert(&mut self, s: &str) -> usize {
    if let Some(id) = self.table.get(s) {
      return id.clone();
    }

    let id = self.id;
    self.id += 1;
    self.table.insert(s.to_string(), id);
    id
  }

  fn serialize(&mut self) -> Vec<u8> {
    let mut result: Vec<u8> = vec![];
    append_u32(&mut result, self.table.len() as u32);

    // Assume that it's sorted by id
    for (s, _id) in &self.table {
      let bytes = s.as_bytes();
      append_u32(&mut result, bytes.len() as u32);
      result.append(&mut bytes.to_vec());
    }

    // eprintln!("Serialized string table: {:#?}", result);

    result
  }
}

struct SerializeCtx {
  id: usize,
  id_to_offset: IndexMap<usize, usize>,
  result: Vec<u8>,
  str_table: StringTable,
}

impl SerializeCtx {
  fn new() -> Self {
    let mut ctx = Self {
      id: 0,
      id_to_offset: IndexMap::new(),
      result: vec![],
      str_table: StringTable::new(),
    };

    ctx.str_table.insert("");

    ctx.push_node(AstNode::Empty, 0, &DUMMY_SP);

    ctx
  }

  fn reserve_child_ids_with_count(&mut self, count: usize) -> usize {
    append_usize(&mut self.result, count);
    self.reserve_child_ids(count)
  }

  fn reserve_child_ids(&mut self, count: usize) -> usize {
    let offset = self.result.len();

    for _ in 0..count {
      append_usize(&mut self.result, 0);
    }

    offset
  }

  fn push_node(
    &mut self,
    kind: AstNode,
    parent_id: usize,
    span: &Span,
  ) -> usize {
    let id = self.id;
    self.id_to_offset.insert(id, self.result.len());
    self.id += 1;

    let kind_value: u8 = kind.into();
    self.result.push(kind_value);
    append_usize(&mut self.result, parent_id);

    // Span
    append_u32(&mut self.result, span.lo.0);
    append_u32(&mut self.result, span.hi.0);

    id
  }

  fn set_child(&mut self, offset: usize, child_id: usize, idx: usize) {
    let pos = offset + (idx * 4);
    write_usize(&mut self.result, child_id, pos);
  }

  fn reserve_flag(&mut self) -> usize {
    self.result.push(0);
    self.result.len() - 1
  }
}

pub fn serialize_ast_bin(parsed_source: &ParsedSource) -> Vec<u8> {
  let mut ctx = SerializeCtx::new();

  let parent_id = 0;

  let program = &parsed_source.program();
  let mut flags = FlagValue::new();

  // eprintln!("SWC {:#?}", program);

  match program.as_ref() {
    Program::Module(module) => {
      let id = ctx.push_node(AstNode::Program, parent_id, &module.span);

      flags.set(Flag::ProgramModule);
      ctx.result.push(flags.0);

      let offset = ctx.reserve_child_ids_with_count(module.body.len());

      for (i, item) in module.body.iter().enumerate() {
        let child_id = match item {
          ModuleItem::ModuleDecl(module_decl) => {
            serialize_module_decl(&mut ctx, module_decl, parent_id)
          }
          ModuleItem::Stmt(stmt) => serialize_stmt(&mut ctx, stmt, id),
        };

        ctx.set_child(offset, child_id, i);
      }
    }
    Program::Script(script) => {
      let id = ctx.push_node(AstNode::Program, parent_id, &script.span);

      ctx.result.push(flags.0);
      let offset = ctx.reserve_child_ids_with_count(script.body.len());

      for (i, stmt) in script.body.iter().enumerate() {
        let child_id = serialize_stmt(&mut ctx, stmt, id);
        ctx.set_child(offset, child_id, i);
      }
    }
  }

  let mut result: Vec<u8> = vec![];

  // Serialize string table
  // eprintln!("STRING {:#?}", ctx.str_table);
  result.append(&mut ctx.str_table.serialize());

  // Serialize ids
  append_usize(&mut result, ctx.id_to_offset.len());

  let offset = result.len() + (ctx.id_to_offset.len() * 4);

  for (_i, value) in ctx.id_to_offset {
    append_usize(&mut result, value + offset);
  }

  // Append serialized AST
  result.append(&mut ctx.result);
  result
}

fn serialize_module_decl(
  ctx: &mut SerializeCtx,
  module_decl: &ModuleDecl,
  parent_id: usize,
) -> usize {
  match module_decl {
    ModuleDecl::Import(node) => {
      ctx.push_node(AstNode::Import, parent_id, &node.span)
    }
    ModuleDecl::ExportDecl(node) => {
      ctx.push_node(AstNode::ExportDecl, parent_id, &node.span)
    }
    ModuleDecl::ExportNamed(node) => {
      let id =
        ctx.push_node(AstNode::ExportNamedDeclaration, parent_id, &node.span);

      let mut flags = FlagValue::new();
      flags.set(Flag::ExportType);
      ctx.result.push(flags.0);

      let offset = ctx.reserve_child_ids(2);
      if let Some(src) = &node.src {
        let child_id = serialize_lit(ctx, &Lit::Str(*src.clone()), id);
        ctx.set_child(offset, child_id, 0);
      }

      // FIXME: I don't think these are valid
      if let Some(attributes) = &node.with {
        let child_id =
          serialize_expr(ctx, &Expr::Object(*attributes.clone()), id);
        ctx.set_child(offset, child_id, 1);
      }

      for (i, spec) in node.specifiers.iter().enumerate() {
        match spec {
          ExportSpecifier::Named(child) => {
            let export_id =
              ctx.push_node(AstNode::ExportSpecifier, id, &child.span);

            let mut flags = FlagValue::new();
            flags.set(Flag::ExportType);
            ctx.result.push(flags.0);

            let export_offset = ctx.reserve_child_ids(2);

            let org_id =
              serialize_module_exported_name(ctx, &child.orig, export_id);
            ctx.set_child(export_offset, org_id, 0);

            if let Some(exported) = &child.exported {
              let exported_id =
                serialize_module_exported_name(ctx, exported, export_id);
              ctx.set_child(export_offset, exported_id, 1);
            }

            ctx.set_child(offset, export_id, i);
          }

          // These two aren't syntactically valid
          ExportSpecifier::Namespace(_) => todo!(),
          ExportSpecifier::Default(_) => todo!(),
        };
      }

      id
    }
    ModuleDecl::ExportDefaultDecl(node) => {
      ctx.push_node(AstNode::ExportDefaultDecl, parent_id, &node.span)
    }
    ModuleDecl::ExportDefaultExpr(node) => {
      ctx.push_node(AstNode::ExportDefaultExpr, parent_id, &node.span)
    }
    ModuleDecl::ExportAll(node) => {
      ctx.push_node(AstNode::ExportAll, parent_id, &node.span)
    }
    ModuleDecl::TsImportEquals(node) => {
      ctx.push_node(AstNode::TsImportEquals, parent_id, &node.span)
    }
    ModuleDecl::TsExportAssignment(node) => {
      ctx.push_node(AstNode::TsExportAssignment, parent_id, &node.span)
    }
    ModuleDecl::TsNamespaceExport(node) => {
      ctx.push_node(AstNode::TsNamespaceExport, parent_id, &node.span)
    }
  }
}

fn serialize_module_exported_name(
  ctx: &mut SerializeCtx,
  name: &ModuleExportName,
  parent_id: usize,
) -> usize {
  match &name {
    ModuleExportName::Ident(ident) => serialize_ident(ctx, &ident, parent_id),
    ModuleExportName::Str(lit) => {
      serialize_lit(ctx, &Lit::Str(lit.clone()), parent_id)
    }
  }
}

fn serialize_stmt(
  ctx: &mut SerializeCtx,
  stmt: &Stmt,
  parent_id: usize,
) -> usize {
  match stmt {
    Stmt::Block(node) => {
      let id = ctx.push_node(AstNode::Block, parent_id, &node.span);
      let offset = ctx.reserve_child_ids_with_count(node.stmts.len());

      for (i, child) in node.stmts.iter().enumerate() {
        let child_id = serialize_stmt(ctx, child, id);
        ctx.set_child(offset, child_id, i);
      }

      id
    }
    Stmt::Empty(_) => 0,
    Stmt::Debugger(node) => {
      ctx.push_node(AstNode::Debugger, parent_id, &node.span)
    }
    Stmt::With(_) => todo!(),
    Stmt::Return(node) => {
      let id = ctx.push_node(AstNode::Return, parent_id, &node.span);
      let offset = ctx.reserve_child_ids(1);

      if let Some(arg) = &node.arg {
        let child_id = serialize_expr(ctx, &arg, id);
        ctx.set_child(offset, child_id, 0);
      };

      id
    }
    Stmt::Labeled(node) => {
      let id = ctx.push_node(AstNode::Labeled, parent_id, &node.span);
      let offset = ctx.reserve_child_ids(2);

      let ident_id = serialize_ident(ctx, &node.label, id);
      let stmt_id = serialize_stmt(ctx, &node.body, id);

      ctx.set_child(offset, ident_id, 0);
      ctx.set_child(offset, stmt_id, 1);

      id
    }
    Stmt::Break(node) => {
      let id = ctx.push_node(AstNode::Break, parent_id, &node.span);
      let offset = ctx.reserve_child_ids(1);

      if let Some(arg) = &node.label {
        let child_id = serialize_ident(ctx, &arg, id);
        ctx.set_child(offset, child_id, 0);
      };

      id
    }
    Stmt::Continue(node) => {
      let id = ctx.push_node(AstNode::Continue, parent_id, &node.span);
      let offset = ctx.reserve_child_ids(1);

      if let Some(arg) = &node.label {
        let child_id = serialize_ident(ctx, &arg, id);
        ctx.set_child(offset, child_id, 0);
      };

      id
    }
    Stmt::If(node) => {
      let id = ctx.push_node(AstNode::If, parent_id, &node.span);
      let offset = ctx.reserve_child_ids(3); // Test + Consequent + Alternate

      let test_id = serialize_expr(ctx, node.test.as_ref(), id);
      let cons_id = serialize_stmt(ctx, node.cons.as_ref(), id);

      ctx.set_child(offset, test_id, 0);
      ctx.set_child(offset, cons_id, 1);

      if let Some(alt) = &node.alt {
        let child_id = serialize_stmt(ctx, &alt, id);
        ctx.set_child(offset, child_id, 2);
      }

      id
    }
    Stmt::Switch(node) => {
      let id = ctx.push_node(AstNode::Switch.into(), parent_id, &node.span);

      let offset = ctx.reserve_child_ids_with_count(node.cases.len());

      for (i, case) in node.cases.iter().enumerate() {
        let child_id =
          ctx.push_node(AstNode::SwitchCase.into(), parent_id, &case.span);

        let child_offset =
          ctx.reserve_child_ids_with_count(case.cons.len() + 1);

        for (j, cons) in case.cons.iter().enumerate() {
          let cons_id = serialize_stmt(ctx, cons, child_id);
          ctx.set_child(child_offset, cons_id, j);
        }

        ctx.set_child(offset, child_id, i);
      }

      id
    }
    Stmt::Throw(node) => {
      let id = ctx.push_node(AstNode::Throw.into(), parent_id, &node.span);
      let offset = ctx.reserve_child_ids(1);

      let child_id = serialize_expr(ctx, &node.arg, id);
      ctx.set_child(offset, child_id, 0);

      id
    }
    Stmt::Try(node) => {
      let try_id =
        ctx.push_node(AstNode::TryStatement.into(), parent_id, &node.span);
      let offset = ctx.reserve_child_ids(3); // Block + Catch + Finalizer

      let block_id =
        serialize_stmt(ctx, &Stmt::Block(node.block.clone()), try_id);
      ctx.set_child(offset, block_id, 0);

      if let Some(catch_clause) = &node.handler {
        let clause_id = ctx.push_node(
          AstNode::CatchClause.into(),
          try_id,
          &catch_clause.span,
        );

        let clause_offset = ctx.reserve_child_ids(2); // Param + Body

        if let Some(param) = &catch_clause.param {
          let param_id = serialize_pat(ctx, param, clause_id);
          ctx.set_child(clause_offset, param_id, 0);
        }

        let body_id =
          serialize_stmt(ctx, &Stmt::Block(catch_clause.body.clone()), try_id);
        ctx.set_child(clause_offset, body_id, 1);

        ctx.set_child(offset, clause_id, 1);
      }

      if let Some(finalizer) = &node.finalizer {
        let child_id =
          serialize_stmt(ctx, &Stmt::Block(finalizer.clone()), try_id);
        ctx.set_child(offset, child_id, 2);
      }

      try_id
    }
    Stmt::While(node) => {
      let id = ctx.push_node(AstNode::While, parent_id, &node.span);
      let offset = ctx.reserve_child_ids(2);

      let test_id = serialize_expr(ctx, node.test.as_ref(), id);
      let stmt_id = serialize_stmt(ctx, node.body.as_ref(), id);

      ctx.set_child(offset, test_id, 0);
      ctx.set_child(offset, stmt_id, 1);

      id
    }
    Stmt::DoWhile(node) => {
      let id = ctx.push_node(AstNode::DoWhile, parent_id, &node.span);

      let offset = ctx.reserve_child_ids(2);

      let expr_id = serialize_expr(ctx, node.test.as_ref(), id);
      let stmt_id = serialize_stmt(ctx, node.body.as_ref(), id);

      ctx.set_child(offset, expr_id, 0);
      ctx.set_child(offset, stmt_id, 0);

      id
    }
    Stmt::For(node) => {
      let id = ctx.push_node(AstNode::For, parent_id, &node.span);
      let offset = ctx.reserve_child_ids(4); // Init + test + update + body

      if let Some(init) = &node.init {
        match init {
          VarDeclOrExpr::VarDecl(var_decl) => {
            serialize_stmt(ctx, &Stmt::Decl(Decl::Var(var_decl.clone())), id);
          }
          VarDeclOrExpr::Expr(expr) => {
            serialize_expr(ctx, expr, id);
          }
        }
      }

      if let Some(test_expr) = &node.test {
        let child_id = serialize_expr(ctx, test_expr.as_ref(), parent_id);
        ctx.set_child(offset, child_id, 1);
      }

      if let Some(update_expr) = &node.update {
        let child_id = serialize_expr(ctx, update_expr.as_ref(), id);
        ctx.set_child(offset, child_id, 2);
      }

      let child_id = serialize_stmt(ctx, &node.body.as_ref(), id);
      ctx.set_child(offset, child_id, 3);

      id
    }
    Stmt::ForIn(node) => {
      let id = ctx.push_node(AstNode::ForInStatement, parent_id, &node.span);
      let offset = ctx.reserve_child_ids(3); // Left + Right + Body

      let left_id = serialize_for_head(ctx, &node.left, id);
      let right_id = serialize_expr(ctx, node.right.as_ref(), id);
      let body_id = serialize_stmt(ctx, node.body.as_ref(), id);

      ctx.set_child(offset, left_id, 0);
      ctx.set_child(offset, right_id, 1);
      ctx.set_child(offset, body_id, 2);

      id
    }
    Stmt::ForOf(node) => {
      let id = ctx.push_node(AstNode::ForOf, parent_id, &node.span);

      let mut flags = FlagValue::new();
      flags.set(Flag::ForAwait);
      ctx.result.push(flags.0);

      let offset = ctx.reserve_child_ids(3);
      let left_id = serialize_for_head(ctx, &node.left, id);
      let right_id = serialize_expr(ctx, node.right.as_ref(), id);
      let body_id = serialize_stmt(ctx, node.body.as_ref(), id);

      ctx.set_child(offset, left_id, 0);
      ctx.set_child(offset, right_id, 1);
      ctx.set_child(offset, body_id, 2);

      id
    }
    Stmt::Decl(node) => serialize_decl(ctx, node, parent_id),
    Stmt::Expr(node) => {
      let id = ctx.push_node(AstNode::Expr, parent_id, &node.span);
      let offset = ctx.reserve_child_ids(1);

      let child_id = serialize_expr(ctx, node.expr.as_ref(), id);
      ctx.set_child(offset, child_id, 0);

      id
    }
  }
}

fn serialize_decl(
  ctx: &mut SerializeCtx,
  decl: &Decl,
  parent_id: usize,
) -> usize {
  match decl {
    Decl::Class(node) => {
      let id = ctx.push_node(AstNode::Class, parent_id, &node.class.span);

      // FIXME

      id
    }
    Decl::Fn(node) => {
      let id = ctx.push_node(AstNode::Fn, parent_id, &node.function.span);
      let mut flags = FlagValue::new();
      if node.declare {
        flags.set(Flag::FnDeclare)
      }
      if node.function.is_async {
        flags.set(Flag::FnAsync)
      }
      if node.function.is_generator {
        flags.set(Flag::FnGenerator)
      }
      ctx.result.push(flags.0);

      let offset = ctx.reserve_child_ids(4);
      let param_offset =
        ctx.reserve_child_ids_with_count(node.function.params.len());

      let ident_id = serialize_ident(ctx, &node.ident, parent_id);
      ctx.set_child(offset, ident_id, 0);

      if let Some(type_params) = &node.function.type_params {
        // FIXME
        // ctx.set_child(offset, type_param_id, 1);
      }

      if let Some(return_type) = &node.function.return_type {
        // FIXME
        // ctx.set_child(offset, return_type_id, 2);
      }

      if let Some(body) = &node.function.body {
        let body_id = serialize_stmt(ctx, &Stmt::Block(body.clone()), id);
        ctx.set_child(offset, body_id, 3);
      }

      for (i, param) in node.function.params.iter().enumerate() {
        let child_id = serialize_pat(ctx, &param.pat, id);
        ctx.set_child(param_offset, child_id, i);
      }

      id
    }
    Decl::Var(node) => {
      let id = ctx.push_node(AstNode::Var, parent_id, &node.span);
      let mut flags = FlagValue::new();
      if node.declare {
        flags.set(Flag::VarDeclare)
      }
      match node.kind {
        VarDeclKind::Var => flags.set(Flag::VarVar),
        VarDeclKind::Let => flags.set(Flag::VarLet),
        VarDeclKind::Const => flags.set(Flag::VarConst),
      }
      ctx.result.push(flags.0);

      let offset = ctx.reserve_child_ids_with_count(node.decls.len());

      for (i, decl) in node.decls.iter().enumerate() {
        let child_id =
          ctx.push_node(AstNode::VariableDeclarator, id, &decl.span);

        ctx.set_child(offset, child_id, i);
        // FIXME: Definite?

        let child_offset = ctx.reserve_child_ids(2); // Name + init

        let decl_id = serialize_pat(ctx, &decl.name, child_id);
        ctx.set_child(child_offset, decl_id, 0);

        if let Some(init) = &decl.init {
          let expr_id = serialize_expr(ctx, init.as_ref(), child_id);
          ctx.set_child(child_offset, expr_id, 1);
        }
      }

      id
    }
    Decl::Using(node) => {
      let id = ctx.push_node(AstNode::Using, parent_id, &node.span);
      let offset = ctx.reserve_child_ids_with_count(node.decls.len());

      for (i, decl) in node.decls.iter().enumerate() {
        // FIXME
      }

      id
    }
    Decl::TsInterface(node) => {
      // ident + body + type_ann + extends(Vec)
      let count = 3 + node.extends.len();

      let id =
        ctx.push_node(AstNode::TsInterface.into(), parent_id, &node.span);

      // FIXME

      id
    }
    Decl::TsTypeAlias(node) => {
      // FIXME: Declare flag
      let id =
        ctx.push_node(AstNode::TsTypeAlias.into(), parent_id, &node.span);

      let offset = ctx.reserve_child_ids(1);

      let ident_id = serialize_ident(ctx, &node.id, id);
      ctx.set_child(offset, ident_id, 0);

      // FIXME
      // let foo = ts_type_alias_decl.type_ann

      id
    }
    Decl::TsEnum(node) => {
      let id = ctx.push_node(AstNode::TsEnum.into(), parent_id, &node.span);

      // Ident + member count
      let count = 1 + node.members.len();
      let offset = ctx.reserve_child_ids_with_count(count);

      let ident_id = serialize_ident(ctx, &node.id, parent_id);
      ctx.set_child(offset, ident_id, id);

      for (i, member) in node.members.iter().enumerate() {
        // FIXME
        // let member_id = member.
        // ctx.set_child(id, member_id, id);
      }

      id
    }
    Decl::TsModule(ts_module_decl) => {
      ctx.push_node(AstNode::TsModule, parent_id, &ts_module_decl.span)
    }
  }
}

fn serialize_expr(
  ctx: &mut SerializeCtx,
  expr: &Expr,
  parent_id: usize,
) -> usize {
  match expr {
    Expr::This(node) => ctx.push_node(AstNode::This, parent_id, &node.span),
    Expr::Array(node) => {
      let id = ctx.push_node(AstNode::Array.into(), parent_id, &node.span);
      let offset = ctx.reserve_child_ids_with_count(node.elems.len());

      for (i, maybe_item) in node.elems.iter().enumerate() {
        if let Some(item) = maybe_item {
          let child_id = serialize_expr_or_spread(ctx, item, id);
          ctx.set_child(offset, child_id, i);
        }
      }

      id
    }
    Expr::Object(node) => {
      let id = ctx.push_node(AstNode::Object, parent_id, &node.span);
      let offset = ctx.reserve_child_ids_with_count(node.props.len());

      for (i, prop) in node.props.iter().enumerate() {
        let child_id = match prop {
          PropOrSpread::Spread(spread_element) => serialize_spread(
            ctx,
            spread_element.expr.as_ref(),
            &spread_element.dot3_token,
            parent_id,
          ),
          PropOrSpread::Prop(prop) => {
            let mut flags = FlagValue::new();
            let prop_id = ctx.push_node(AstNode::Property, id, &prop.span());

            let flag_offset = ctx.reserve_flag();
            let child_offset = ctx.reserve_child_ids(2);

            // FIXME: optional
            match prop.as_ref() {
              Prop::Shorthand(ident) => {
                flags.set(Flag::PropShorthand);

                let child_id = serialize_ident(ctx, ident, prop_id);
                ctx.set_child(child_offset, child_id, 0);
                ctx.set_child(child_offset, child_id, 1);
              }
              Prop::KeyValue(key_value_prop) => {
                if let PropName::Computed(_) = key_value_prop.key {
                  flags.set(Flag::PropComputed)
                }

                let key_id =
                  serialize_prop_name(ctx, &key_value_prop.key, prop_id);
                let value_id =
                  serialize_expr(ctx, key_value_prop.value.as_ref(), prop_id);

                ctx.set_child(child_offset, key_id, 0);
                ctx.set_child(child_offset, value_id, 1);
              }
              Prop::Assign(assign_prop) => {
                let child_id = ctx.push_node(
                  AstNode::AssignmentPattern,
                  prop_id,
                  &assign_prop.span,
                );

                let offset = ctx.reserve_child_ids(2);

                let key_id = serialize_ident(ctx, &assign_prop.key, prop_id);
                let value_id =
                  serialize_expr(ctx, assign_prop.value.as_ref(), prop_id);

                ctx.set_child(offset, key_id, 0);
                ctx.set_child(offset, value_id, 1);

                ctx.set_child(child_offset, key_id, 0);
                ctx.set_child(child_offset, child_id, 1);
              }
              Prop::Getter(getter_prop) => {
                flags.set(Flag::PropGetter);

                let key_id =
                  serialize_prop_name(ctx, &getter_prop.key, prop_id);

                ctx.set_child(child_offset, key_id, 0);

                let value_id = serialize_expr(
                  ctx,
                  &Expr::Fn(FnExpr {
                    ident: None,
                    function: Box::new(Function {
                      params: vec![],
                      decorators: vec![],
                      span: getter_prop.span,
                      ctxt: SyntaxContext::empty(),
                      body: getter_prop.body.clone(),
                      is_generator: false,
                      is_async: false,
                      type_params: None, // FIXME
                      return_type: None,
                    }),
                  }),
                  prop_id,
                );

                ctx.set_child(child_offset, value_id, 1);
              }
              Prop::Setter(setter_prop) => {
                flags.set(Flag::PropSetter);

                let key_id =
                  serialize_prop_name(ctx, &setter_prop.key, prop_id);
                ctx.set_child(child_offset, key_id, 0);

                let param = Param::from(*setter_prop.param.clone());

                let value_id = serialize_expr(
                  ctx,
                  &Expr::Fn(FnExpr {
                    ident: None,
                    function: Box::new(Function {
                      params: vec![param],
                      decorators: vec![],
                      span: setter_prop.span,
                      ctxt: SyntaxContext::empty(),
                      body: setter_prop.body.clone(),
                      is_generator: false,
                      is_async: false,
                      type_params: None,
                      return_type: None,
                    }),
                  }),
                  prop_id,
                );

                ctx.set_child(child_offset, value_id, 1);
              }
              Prop::Method(method_prop) => {
                flags.set(Flag::PropMethod);

                let key_id =
                  serialize_prop_name(ctx, &method_prop.key, prop_id);
                ctx.set_child(child_offset, key_id, 0);

                let value_id = serialize_expr(
                  ctx,
                  &Expr::Fn(FnExpr {
                    ident: None,
                    function: method_prop.function.clone(),
                  }),
                  prop_id,
                );

                ctx.set_child(child_offset, value_id, 1);
              }
            }

            ctx.result[flag_offset] = flags.0;

            prop_id
          }
        };

        ctx.set_child(offset, child_id, i);
      }

      id
    }
    Expr::Fn(node) => {
      let fn_obj = node.function.as_ref();
      let id = ctx.push_node(AstNode::FnExpr, parent_id, &fn_obj.span);

      let flag_offset = ctx.reserve_flag();
      let mut flags = FlagValue::new();
      if fn_obj.is_async {
        flags.set(Flag::FnAsync)
      }
      if fn_obj.is_generator {
        flags.set(Flag::FnGenerator)
      }
      ctx.result[flag_offset] = flags.0;

      let offset = ctx.reserve_child_ids(2);
      if let Some(ident) = &node.ident {
        let ident_id = serialize_ident(ctx, ident, id);
        ctx.set_child(offset, ident_id, 0);
      }

      if let Some(type_params) = &fn_obj.type_params {
        // FIXME
        ctx.set_child(offset, 0, 1);
      }

      let param_offset = ctx.reserve_child_ids_with_count(fn_obj.params.len());

      for (i, param) in fn_obj.params.iter().enumerate() {
        let child_id = serialize_pat(ctx, &param.pat, id);
        ctx.set_child(param_offset, child_id, i);
      }

      let offset = ctx.reserve_child_ids(2);
      if let Some(return_type) = &fn_obj.return_type {
        // FIXME
        ctx.set_child(offset, 0, 1);
      }

      if let Some(block) = &fn_obj.body {
        let block_id = serialize_stmt(ctx, &Stmt::Block(block.clone()), id);
        ctx.set_child(offset, block_id, 1);
      }

      id
    }
    Expr::Unary(node) => {
      let id = ctx.push_node(AstNode::Unary, parent_id, &node.span);
      let mut flags = FlagValue::new();
      flags.set(match node.op {
        UnaryOp::Minus => Flag::UnaryMinus,
        UnaryOp::Plus => Flag::UnaryPlus,
        UnaryOp::Bang => Flag::UnaryBang,
        UnaryOp::Tilde => Flag::UnaryTilde,
        UnaryOp::TypeOf => Flag::UnaryTypeOf,
        UnaryOp::Void => Flag::UnaryVoid,
        UnaryOp::Delete => Flag::UnaryDelete,
      });
      ctx.result.push(flags.0);

      let offset = ctx.reserve_child_ids(1);

      let child_id = serialize_expr(ctx, &node.arg, id);
      ctx.set_child(offset, child_id, 0);

      id
    }
    Expr::Update(node) => {
      let id = ctx.push_node(AstNode::UpdateExpression, parent_id, &node.span);
      let mut flags = FlagValue::new();
      if node.prefix {
        flags.set(Flag::UpdatePrefix);
      }
      flags.set(match node.op {
        UpdateOp::PlusPlus => Flag::UpdatePlusPlus,
        UpdateOp::MinusMinus => Flag::UpdateMinusMinus,
      });
      ctx.result.push(flags.0);

      let offset = ctx.reserve_child_ids(1);

      let child_id = serialize_expr(ctx, node.arg.as_ref(), id);
      ctx.set_child(offset, child_id, 0);

      id
    }
    Expr::Bin(node) => {
      let (node_type, flag) = match node.op {
        BinaryOp::LogicalOr => (AstNode::LogicalExpression, Flag::LogicalOr),
        BinaryOp::LogicalAnd => (AstNode::LogicalExpression, Flag::LogicalAnd),
        BinaryOp::NullishCoalescing => {
          (AstNode::LogicalExpression, Flag::LogicalNullishCoalescin)
        }
        BinaryOp::EqEq => (AstNode::BinaryExpression, Flag::BinEqEq),
        BinaryOp::NotEq => (AstNode::BinaryExpression, Flag::BinNotEq),
        BinaryOp::EqEqEq => (AstNode::BinaryExpression, Flag::BinEqEqEq),
        BinaryOp::NotEqEq => (AstNode::BinaryExpression, Flag::BinNotEqEq),
        BinaryOp::Lt => (AstNode::BinaryExpression, Flag::BinLt),
        BinaryOp::LtEq => (AstNode::BinaryExpression, Flag::BinLtEq),
        BinaryOp::Gt => (AstNode::BinaryExpression, Flag::BinGt),
        BinaryOp::GtEq => (AstNode::BinaryExpression, Flag::BinGtEq),
        BinaryOp::LShift => (AstNode::BinaryExpression, Flag::BinLShift),
        BinaryOp::RShift => (AstNode::BinaryExpression, Flag::BinRShift),
        BinaryOp::ZeroFillRShift => {
          (AstNode::BinaryExpression, Flag::BinZeroFillRShift)
        }
        BinaryOp::Add => (AstNode::BinaryExpression, Flag::BinAdd),
        BinaryOp::Sub => (AstNode::BinaryExpression, Flag::BinSub),
        BinaryOp::Mul => (AstNode::BinaryExpression, Flag::BinMul),
        BinaryOp::Div => (AstNode::BinaryExpression, Flag::BinDiv),
        BinaryOp::Mod => (AstNode::BinaryExpression, Flag::BinMod),
        BinaryOp::BitOr => (AstNode::BinaryExpression, Flag::BinBitOr),
        BinaryOp::BitXor => (AstNode::BinaryExpression, Flag::BinBitXor),
        BinaryOp::BitAnd => (AstNode::BinaryExpression, Flag::BinBitAnd),
        BinaryOp::In => (AstNode::BinaryExpression, Flag::BinIn),
        BinaryOp::InstanceOf => {
          (AstNode::BinaryExpression, Flag::BinInstanceOf)
        }
        BinaryOp::Exp => (AstNode::BinaryExpression, Flag::BinExp),
      };

      let id = ctx.push_node(node_type, parent_id, &node.span);

      let mut flags = FlagValue::new();
      flags.set(flag);
      ctx.result.push(flags.0);

      let offset = ctx.reserve_child_ids(2);

      let left_id = serialize_expr(ctx, node.left.as_ref(), id);
      let right_id = serialize_expr(ctx, node.right.as_ref(), id);

      ctx.set_child(offset, left_id, 0);
      ctx.set_child(offset, right_id, 1);

      id
    }
    Expr::Assign(node) => {
      let id = ctx.push_node(AstNode::Assign, parent_id, &node.span);
      ctx.result.push(assign_op_to_flag(node.op));

      let offset = ctx.reserve_child_ids(2);

      let left_id = match &node.left {
        AssignTarget::Simple(simple_assign_target) => {
          match simple_assign_target {
            SimpleAssignTarget::Ident(binding_ident) => {
              serialize_ident(ctx, &binding_ident.id, id)
            }
            SimpleAssignTarget::Member(member_expr) => {
              serialize_expr(ctx, &Expr::Member(member_expr.clone()), id)
            }
            SimpleAssignTarget::SuperProp(super_prop_expr) => todo!(),
            SimpleAssignTarget::Paren(paren_expr) => todo!(),
            SimpleAssignTarget::OptChain(opt_chain_expr) => todo!(),
            SimpleAssignTarget::TsAs(ts_as_expr) => todo!(),
            SimpleAssignTarget::TsSatisfies(ts_satisfies_expr) => todo!(),
            SimpleAssignTarget::TsNonNull(ts_non_null_expr) => todo!(),
            SimpleAssignTarget::TsTypeAssertion(ts_type_assertion) => todo!(),
            SimpleAssignTarget::TsInstantiation(ts_instantiation) => todo!(),
            SimpleAssignTarget::Invalid(invalid) => todo!(),
          }
        }
        AssignTarget::Pat(target) => match target {
          AssignTargetPat::Array(array_pat) => {
            serialize_pat(ctx, &Pat::Array(array_pat.clone()), id)
          }
          AssignTargetPat::Object(object_pat) => {
            serialize_pat(ctx, &Pat::Object(object_pat.clone()), id)
          }
          AssignTargetPat::Invalid(invalid) => todo!(),
        },
      };

      let right_id = serialize_expr(ctx, node.right.as_ref(), id);

      ctx.set_child(offset, left_id, 0);
      ctx.set_child(offset, right_id, 1);

      id
    }
    Expr::Member(node) => {
      let id = ctx.push_node(AstNode::MemberExpression, parent_id, &node.span);
      let mut flags = FlagValue::new();
      if let MemberProp::Computed(_) = node.prop {
        flags.set(Flag::MemberComputed)
      }
      ctx.result.push(flags.0);

      let offset = ctx.reserve_child_ids(2);
      let obj_id = serialize_expr(ctx, node.obj.as_ref(), id);

      let prop_id = match &node.prop {
        MemberProp::Ident(ident_name) => {
          serialize_ident_name(ctx, ident_name, id)
        }
        MemberProp::PrivateName(private_name) => {
          let child_id =
            ctx.push_node(AstNode::PrivateIdentifier, id, &private_name.span);

          let str_id = ctx.str_table.insert(&private_name.name.as_str());
          append_usize(&mut ctx.result, str_id);

          child_id
        }
        MemberProp::Computed(computed_prop_name) => {
          serialize_expr(ctx, computed_prop_name.expr.as_ref(), id)
        }
      };

      ctx.set_child(offset, obj_id, 0);
      ctx.set_child(offset, prop_id, 1);

      id
    }
    Expr::SuperProp(node) => {
      let id = ctx.push_node(AstNode::MemberExpression, parent_id, &node.span);
      let mut flags = FlagValue::new();
      if let SuperProp::Computed(_) = node.prop {
        flags.set(Flag::MemberComputed)
      }
      ctx.result.push(flags.0);

      let offset = ctx.reserve_child_ids(2);

      let super_id = ctx.push_node(AstNode::Super, id, &node.obj.span);
      ctx.set_child(offset, super_id, 0);

      let child_id = match &node.prop {
        SuperProp::Ident(ident_name) => {
          serialize_ident_name(ctx, ident_name, id)
        }
        SuperProp::Computed(prop) => serialize_expr(ctx, &prop.expr, id),
      };
      ctx.set_child(offset, child_id, 1);

      id
    }
    Expr::Cond(node) => {
      let id = ctx.push_node(AstNode::Cond, parent_id, &node.span);
      let offset = ctx.reserve_child_ids(3);

      let test_id = serialize_expr(ctx, node.test.as_ref(), id);
      let cons_id = serialize_expr(ctx, node.cons.as_ref(), id);
      let alt_id = serialize_expr(ctx, node.alt.as_ref(), id);

      ctx.set_child(offset, test_id, 0);
      ctx.set_child(offset, cons_id, 1);
      ctx.set_child(offset, alt_id, 2);

      id
    }
    Expr::Call(node) => {
      let id = ctx.push_node(AstNode::Call, parent_id, &node.span);
      // Callee + type args
      let offset = ctx.reserve_child_ids(2);
      let arg_offset = ctx.reserve_child_ids_with_count(node.args.len());

      let callee_id = match &node.callee {
        Callee::Super(super_node) => {
          ctx.push_node(AstNode::Super, id, &super_node.span)
        }
        Callee::Import(import) => todo!(),
        Callee::Expr(expr) => serialize_expr(ctx, expr, id),
      };
      ctx.set_child(offset, callee_id, 0);

      if let Some(type_args) = &node.type_args {
        // FIXME
        // ctx.set_child(offset, expr_id, 1);
      }

      for (i, arg) in node.args.iter().enumerate() {
        let child_id = serialize_expr_or_spread(ctx, arg, id);
        ctx.set_child(arg_offset, child_id, i);
      }

      id
    }
    Expr::New(node) => {
      let id = ctx.push_node(AstNode::New.into(), parent_id, &node.span);
      let offset = ctx.reserve_child_ids(2);
      let child_offset = ctx.reserve_child_ids_with_count(
        node.args.as_ref().map_or(0, |args| args.len()),
      );

      let expr_id = serialize_expr(ctx, node.callee.as_ref(), id);
      ctx.set_child(offset, expr_id, 0);

      if let Some(args) = &node.args {
        for (i, arg) in args.iter().enumerate() {
          let arg_id = serialize_expr_or_spread(ctx, arg, id);
          ctx.set_child(child_offset, arg_id, i);
        }
      }

      if let Some(type_ann) = &node.type_args {
        // FIXME
        // ctx.set_child(offset, child_id, 1);
      }

      id
    }
    Expr::Seq(node) => {
      let id = ctx.push_node(AstNode::Seq, parent_id, &node.span);

      let offset = ctx.reserve_child_ids_with_count(node.exprs.len());

      for (i, expr) in node.exprs.iter().enumerate() {
        let child_id = serialize_expr(ctx, expr, id);
        ctx.set_child(offset, child_id, i);
      }

      id
    }
    Expr::Ident(node) => serialize_ident(ctx, node, parent_id),
    Expr::Lit(node) => serialize_lit(ctx, node, parent_id),
    Expr::Tpl(node) => {
      let id = ctx.push_node(AstNode::TemplateLiteral, parent_id, &node.span);

      let quasi_offset = ctx.reserve_child_ids_with_count(node.quasis.len());
      let expr_offset = ctx.reserve_child_ids_with_count(node.exprs.len());

      for (i, quasi) in node.quasis.iter().enumerate() {
        let child_id = ctx.push_node(AstNode::TemplateElement, id, &quasi.span);
        let mut flags = FlagValue::new();
        flags.set(Flag::TplTail);
        ctx.result.push(flags.0);

        let raw_str_id = ctx.str_table.insert(quasi.raw.as_str());
        append_usize(&mut ctx.result, raw_str_id);

        let cooked_str_id = if let Some(cooked) = &quasi.cooked {
          ctx.str_table.insert(cooked.as_str())
        } else {
          0
        };
        append_usize(&mut ctx.result, cooked_str_id);

        ctx.set_child(quasi_offset, child_id, i);
      }
      for (i, expr) in node.exprs.iter().enumerate() {
        let child_id = serialize_expr(ctx, expr, id);
        ctx.set_child(expr_offset, child_id, i);
      }

      id
    }
    Expr::TaggedTpl(tagged_tpl) => {
      let id = ctx.push_node(AstNode::TaggedTpl, parent_id, &tagged_tpl.span);

      id
    }
    Expr::Arrow(node) => {
      let id =
        ctx.push_node(AstNode::ArrowFunctionExpression, parent_id, &node.span);

      let mut flags = FlagValue::new();
      if node.is_async {
        flags.set(Flag::FnAsync);
      }
      if node.is_generator {
        flags.set(Flag::FnGenerator);
      }
      ctx.result.push(flags.0);

      let type_offset = ctx.reserve_child_ids(1);
      let param_offset = ctx.reserve_child_ids_with_count(node.params.len());
      let body_offset = ctx.reserve_child_ids(2);

      if let Some(type_params) = &node.type_params {
        // FIXME;
      }

      for (i, pat) in node.params.iter().enumerate() {
        let child_id = serialize_pat(ctx, pat, id);
        ctx.set_child(param_offset, child_id, i);
      }

      // FIXME

      let body_id = match node.body.as_ref() {
        BlockStmtOrExpr::BlockStmt(block_stmt) => {
          serialize_stmt(ctx, &Stmt::Block(block_stmt.clone()), id)
        }
        BlockStmtOrExpr::Expr(expr) => serialize_expr(ctx, expr.as_ref(), id),
      };
      ctx.set_child(body_offset, body_id, 0);

      if let Some(return_type) = &node.return_type {
        // FIXME
        // ctx.set_child(body_offset, body_id, 1);
      }

      id
    }
    Expr::Class(node) => {
      let id = ctx.push_node(AstNode::ClassExpr, parent_id, &node.class.span);

      // FIXME

      id
    }
    Expr::Yield(node) => {
      let id = ctx.push_node(AstNode::Yield, parent_id, &node.span);
      let offset = ctx.reserve_child_ids(1);

      if let Some(arg) = &node.arg {
        let child_id = serialize_expr(ctx, arg.as_ref(), id);
        ctx.set_child(offset, child_id, 0);
      }

      id
    }
    Expr::MetaProp(node) => {
      ctx.push_node(AstNode::MetaProp.into(), parent_id, &node.span)
    }
    Expr::Await(node) => {
      let id = ctx.push_node(AstNode::AwaitExpression, parent_id, &node.span);
      let offset = ctx.reserve_child_ids(1);

      let child_id = serialize_expr(ctx, node.arg.as_ref(), id);
      ctx.set_child(offset, child_id, 0);

      id
    }
    Expr::Paren(node) => {
      // Paren nodes are treated as a syntax only thing in TSEStree
      // and are never materialized to actual AST nodes.
      serialize_expr(ctx, &node.expr, parent_id)
    }
    Expr::JSXMember(node) => serialize_jsx_member_expr(ctx, &node, parent_id),
    Expr::JSXNamespacedName(node) => {
      serialize_jsx_namespaced_name(ctx, node, parent_id)
    }
    Expr::JSXEmpty(node) => serialize_jsx_empty_expr(ctx, node, parent_id),
    Expr::JSXElement(node) => serialize_jsx_element(ctx, &node, parent_id),
    Expr::JSXFragment(node) => serialize_jsx_fragment(ctx, &node, parent_id),
    Expr::TsTypeAssertion(node) => {
      let id = ctx.push_node(AstNode::TsTypeAssertion, parent_id, &node.span);

      id
    }
    Expr::TsConstAssertion(node) => {
      let id = ctx.push_node(AstNode::TsConstAssertion, parent_id, &node.span);
      let offset = ctx.reserve_child_ids(1);

      let expr_id = serialize_expr(ctx, node.expr.as_ref(), id);
      ctx.set_child(offset, expr_id, 0);

      id
    }
    Expr::TsNonNull(node) => {
      let id = ctx.push_node(AstNode::TsNonNull, parent_id, &node.span);
      let offset = ctx.reserve_child_ids(1);

      let child_id = serialize_expr(ctx, node.expr.as_ref(), id);
      ctx.set_child(offset, child_id, 0);

      id
    }
    Expr::TsAs(node) => {
      let id = ctx.push_node(AstNode::TsAs, parent_id, &node.span);
      let offset = ctx.reserve_child_ids(2);

      let expr_id = serialize_expr(ctx, node.expr.as_ref(), id);
      let type_id = serialize_ts_type(ctx, node.type_ann.as_ref(), id);

      ctx.set_child(offset, expr_id, 0);
      ctx.set_child(offset, type_id, 1);

      id
    }
    Expr::TsInstantiation(node) => {
      let id = ctx.push_node(AstNode::TsInstantiation, parent_id, &node.span);
      let offset = ctx.reserve_child_ids(2);
      let expr_id = serialize_expr(ctx, node.expr.as_ref(), id);
      // let expr_id = serialize_expr(ctx, ts_instantiation.type_args.as_ref(), id);
      ctx.set_child(offset, expr_id, 0);

      // FIXME
      id
    }
    Expr::TsSatisfies(node) => {
      let id = ctx.push_node(AstNode::TsSatisfies, parent_id, &node.span);
      let offset = ctx.reserve_child_ids(2);

      let expr_id = serialize_expr(ctx, node.expr.as_ref(), id);
      let type_id = serialize_ts_type(ctx, node.type_ann.as_ref(), id);

      ctx.set_child(offset, expr_id, 0);
      ctx.set_child(offset, type_id, 1);

      id
    }
    Expr::PrivateName(node) => {
      let id = ctx.push_node(AstNode::PrivateIdentifier, parent_id, &node.span);

      id
    }
    Expr::OptChain(node) => {
      let id = ctx.push_node(AstNode::OptChain, parent_id, &node.span);

      id
    }
    Expr::Invalid(invalid) => {
      // ctx.push_node(result, AstNode::Invalid.into(), &invalid.span);
      todo!()
    }
  }
}

fn serialize_jsx_element(
  ctx: &mut SerializeCtx,
  node: &JSXElement,
  parent_id: usize,
) -> usize {
  let id = ctx.push_node(AstNode::JSXElement, parent_id, &node.span);
  let offset = ctx.reserve_child_ids(2);
  let child_offset = ctx.reserve_child_ids_with_count(node.children.len());

  let opening_id = serialize_jsx_opening_element(ctx, &node.opening, id);
  ctx.set_child(offset, opening_id, 0);

  if let Some(closing) = &node.closing {
    let closing_id =
      ctx.push_node(AstNode::JSXClosingElement, id, &closing.span);
    ctx.set_child(offset, closing_id, 1);

    let closing_offset = ctx.reserve_child_ids(1);
    let child_id = serialize_jsx_element_name(ctx, &closing.name, id);
    ctx.set_child(closing_offset, child_id, 1);
  }

  serialize_jsx_children(ctx, &node.children, id, child_offset);

  id
}

fn serialize_jsx_fragment(
  ctx: &mut SerializeCtx,
  node: &JSXFragment,
  parent_id: usize,
) -> usize {
  let id = ctx.push_node(AstNode::JSXFragment, parent_id, &node.span);

  let offset = ctx.reserve_child_ids(2);
  let child_offset = ctx.reserve_child_ids_with_count(node.children.len());

  let opening_id =
    ctx.push_node(AstNode::JSXOpeningFragment, id, &node.opening.span);
  let closing_id =
    ctx.push_node(AstNode::JSXClosingFragment, id, &node.closing.span);

  ctx.set_child(offset, opening_id, 0);
  ctx.set_child(offset, closing_id, 1);

  serialize_jsx_children(ctx, &node.children, id, child_offset);

  id
}

fn serialize_jsx_children(
  ctx: &mut SerializeCtx,
  children: &Vec<JSXElementChild>,
  parent_id: usize,
  offset: usize,
) {
  for (i, child) in children.iter().enumerate() {
    let child_id = match child {
      JSXElementChild::JSXText(text) => {
        let id = ctx.push_node(AstNode::JSXText, parent_id, &text.span);

        let raw_id = ctx.str_table.insert(&text.raw.as_str());
        let value_id = ctx.str_table.insert(&text.value.as_str());

        append_usize(&mut ctx.result, raw_id);
        append_usize(&mut ctx.result, value_id);

        id
      }
      JSXElementChild::JSXExprContainer(container) => {
        serialize_jsx_container_expr(ctx, container, parent_id)
      }
      JSXElementChild::JSXElement(el) => {
        serialize_jsx_element(ctx, el, parent_id)
      }
      JSXElementChild::JSXFragment(frag) => {
        serialize_jsx_fragment(ctx, frag, parent_id)
      }
      // No parser supports this
      JSXElementChild::JSXSpreadChild(_) => unreachable!(),
    };
    ctx.set_child(offset, child_id, i);
  }
}

fn serialize_jsx_member_expr(
  ctx: &mut SerializeCtx,
  node: &JSXMemberExpr,
  parent_id: usize,
) -> usize {
  let id = ctx.push_node(AstNode::JSXMemberExpression, parent_id, &node.span);
  let offset = ctx.reserve_child_ids(2);

  let obj_id = match &node.obj {
    JSXObject::JSXMemberExpr(member) => {
      serialize_jsx_member_expr(ctx, member, id)
    }
    JSXObject::Ident(ident) => serialize_jsx_identifier(ctx, &ident, parent_id),
  };

  let prop_id = serialize_ident_name_as_jsx_identifier(ctx, &node.prop, id);

  ctx.set_child(offset, obj_id, 0);
  ctx.set_child(offset, prop_id, 1);

  id
}

fn serialize_jsx_element_name(
  ctx: &mut SerializeCtx,
  node: &JSXElementName,
  parent_id: usize,
) -> usize {
  match &node {
    JSXElementName::Ident(ident) => {
      serialize_jsx_identifier(ctx, &ident, parent_id)
    }
    JSXElementName::JSXMemberExpr(member) => {
      serialize_jsx_member_expr(ctx, &member, parent_id)
    }
    JSXElementName::JSXNamespacedName(ns) => {
      serialize_jsx_namespaced_name(ctx, &ns, parent_id)
    }
  }
}

fn serialize_jsx_opening_element(
  ctx: &mut SerializeCtx,
  node: &JSXOpeningElement,
  parent_id: usize,
) -> usize {
  let id = ctx.push_node(AstNode::JSXOpeningElement, parent_id, &node.span);

  let mut flags = FlagValue::new();
  if node.self_closing {
    flags.set(Flag::JSXSelfClosing);
  }
  ctx.result.push(flags.0);

  let offset = ctx.reserve_child_ids(1);
  let child_offset = ctx.reserve_child_ids_with_count(node.attrs.len());

  let name_id = serialize_jsx_element_name(ctx, &node.name, id);
  ctx.set_child(offset, name_id, 0);

  // FIXME: type args

  for (i, attr) in node.attrs.iter().enumerate() {
    let child_id = match attr {
      JSXAttrOrSpread::JSXAttr(jsxattr) => {
        let attr_id = ctx.push_node(AstNode::JSXAttribute, id, &jsxattr.span);
        let offset = ctx.reserve_child_ids(2);

        let name_id = match &jsxattr.name {
          JSXAttrName::Ident(name) => {
            serialize_ident_name_as_jsx_identifier(ctx, &name, attr_id)
          }
          JSXAttrName::JSXNamespacedName(node) => {
            serialize_jsx_namespaced_name(ctx, node, attr_id)
          }
        };
        ctx.set_child(offset, name_id, 0);

        if let Some(value) = &jsxattr.value {
          let value_id = match value {
            JSXAttrValue::Lit(lit) => serialize_lit(ctx, lit, attr_id),
            JSXAttrValue::JSXExprContainer(container) => {
              serialize_jsx_container_expr(ctx, container, attr_id)
            }
            JSXAttrValue::JSXElement(el) => {
              serialize_jsx_element(ctx, &el, attr_id)
            }
            JSXAttrValue::JSXFragment(frag) => {
              serialize_jsx_fragment(ctx, &frag, attr_id)
            }
          };

          ctx.set_child(offset, value_id, 1);
        }

        attr_id
      }
      JSXAttrOrSpread::SpreadElement(spread) => {
        let attr_id =
          ctx.push_node(AstNode::JSXSpreadAttribute, id, &spread.dot3_token);

        let offset = ctx.reserve_child_ids(1);
        let child_id = serialize_expr(ctx, &spread.expr, attr_id);
        ctx.set_child(offset, child_id, 0);

        attr_id
      }
    };

    ctx.set_child(child_offset, child_id, i);
  }

  id
}

fn serialize_jsx_container_expr(
  ctx: &mut SerializeCtx,
  node: &JSXExprContainer,
  parent_id: usize,
) -> usize {
  let id =
    ctx.push_node(AstNode::JSXExpressionContainer, parent_id, &node.span);

  let offset = ctx.reserve_child_ids(1);

  let child_id = match &node.expr {
    JSXExpr::JSXEmptyExpr(expr) => serialize_jsx_empty_expr(ctx, expr, id),
    JSXExpr::Expr(expr) => serialize_expr(ctx, &expr, id),
  };

  ctx.set_child(offset, child_id, 0);

  id
}

fn serialize_jsx_empty_expr(
  ctx: &mut SerializeCtx,
  node: &JSXEmptyExpr,
  parent_id: usize,
) -> usize {
  ctx.push_node(AstNode::JSXEmptyExpression, parent_id, &node.span)
}

fn serialize_jsx_namespaced_name(
  ctx: &mut SerializeCtx,
  node: &JSXNamespacedName,
  parent_id: usize,
) -> usize {
  let id = ctx.push_node(AstNode::JSXNamespacedName, parent_id, &node.span);

  let offset = ctx.reserve_child_ids(2);
  let ns_id = serialize_ident_name_as_jsx_identifier(ctx, &node.ns, id);
  let name_id = serialize_ident_name_as_jsx_identifier(ctx, &node.name, id);

  ctx.set_child(offset, ns_id, 0);
  ctx.set_child(offset, name_id, 1);

  id
}

fn serialize_ident_name_as_jsx_identifier(
  ctx: &mut SerializeCtx,
  node: &IdentName,
  parent_id: usize,
) -> usize {
  let id = ctx.push_node(AstNode::JSXIdentifier, parent_id, &node.span);

  let str_id = ctx.str_table.insert(node.sym.as_str());
  append_usize(&mut ctx.result, str_id);

  id
}

fn serialize_jsx_identifier(
  ctx: &mut SerializeCtx,
  node: &Ident,
  parent_id: usize,
) -> usize {
  let id = ctx.push_node(AstNode::JSXIdentifier, parent_id, &node.span);

  let str_id = ctx.str_table.insert(node.sym.as_str());
  append_usize(&mut ctx.result, str_id);

  id
}

fn serialize_pat(ctx: &mut SerializeCtx, pat: &Pat, parent_id: usize) -> usize {
  match pat {
    Pat::Ident(node) => serialize_ident(ctx, &node.id, parent_id),
    Pat::Array(node) => {
      let id = ctx.push_node(AstNode::ArrayPattern, parent_id, &node.span);

      // FIXME: Optional
      // FIXME: Type Ann
      let offset = ctx.reserve_child_ids_with_count(node.elems.len());

      for (i, elem) in node.elems.iter().enumerate() {
        if let Some(pat) = elem {
          let child_id = serialize_pat(ctx, pat, id);
          ctx.set_child(offset, child_id, i);
        }
      }

      id
    }
    Pat::Rest(node) => {
      let id = ctx.push_node(AstNode::RestElement, parent_id, &node.span);

      let offset = ctx.reserve_child_ids(2);
      if let Some(type_ann) = &node.type_ann {
        // FIXME
        // ctx.set_child(offset, type_id, 0);
      }

      let arg_id = serialize_pat(ctx, &node.arg, parent_id);
      ctx.set_child(offset, arg_id, 1);

      id
    }
    Pat::Object(node) => {
      let id = ctx.push_node(AstNode::ObjectPattern, parent_id, &node.span);

      // FIXME: Optional
      // FIXME: Type Ann
      let offset = ctx.reserve_child_ids_with_count(node.props.len());

      for (i, prop) in node.props.iter().enumerate() {
        let child_id = match prop {
          ObjectPatProp::KeyValue(key_value_prop) => {
            let child_id =
              ctx.push_node(AstNode::Property, id, &key_value_prop.span());
            let mut flags = FlagValue::new();
            if let PropName::Computed(_) = key_value_prop.key {
              flags.set(Flag::PropComputed)
            }
            ctx.result.push(flags.0);

            let child_offset = ctx.reserve_child_ids(2);

            let key_id =
              serialize_prop_name(ctx, &key_value_prop.key, child_id);
            let value_id =
              serialize_pat(ctx, key_value_prop.value.as_ref(), child_id);

            ctx.set_child(child_offset, key_id, 0);
            ctx.set_child(child_offset, value_id, 1);

            child_id
          }
          ObjectPatProp::Assign(assign_pat_prop) => {
            let child_id =
              ctx.push_node(AstNode::Property, id, &assign_pat_prop.span);
            ctx.result.push(0); // No flags

            let child_offset = ctx.reserve_child_ids(2);

            let ident_id =
              serialize_ident(ctx, &assign_pat_prop.key.id, parent_id);
            ctx.set_child(child_offset, ident_id, 0);

            if let Some(value) = &assign_pat_prop.value {
              let value_id = serialize_expr(ctx, value, child_id);
              ctx.set_child(child_offset, value_id, 1);
            }

            child_id
          }
          ObjectPatProp::Rest(rest_pat) => {
            serialize_pat(ctx, &Pat::Rest(rest_pat.clone()), parent_id)
          }
        };

        ctx.set_child(offset, child_id, i);
      }

      id
    }
    Pat::Assign(node) => {
      let id = ctx.push_node(AstNode::AssignmentPattern, parent_id, &node.span);

      let offset = ctx.reserve_child_ids(2);
      let left_id = serialize_pat(ctx, &node.left, id);
      let right_id = serialize_expr(ctx, &node.right, id);

      ctx.set_child(offset, left_id, 0);
      ctx.set_child(offset, right_id, 1);

      id
    }
    Pat::Invalid(node) => todo!(),
    Pat::Expr(node) => serialize_expr(ctx, node, parent_id),
  }
}

fn serialize_for_head(
  ctx: &mut SerializeCtx,
  for_head: &ForHead,
  parent_id: usize,
) -> usize {
  match for_head {
    ForHead::VarDecl(var_decl) => {
      serialize_decl(ctx, &Decl::Var(var_decl.clone()), parent_id)
    }
    ForHead::UsingDecl(using_decl) => todo!(),
    ForHead::Pat(pat) => serialize_pat(ctx, pat, parent_id),
  }
}

fn serialize_expr_or_spread(
  ctx: &mut SerializeCtx,
  arg: &ExprOrSpread,
  parent_id: usize,
) -> usize {
  if let Some(spread) = &arg.spread {
    serialize_spread(ctx, &arg.expr, spread, parent_id)
  } else {
    serialize_expr(ctx, arg.expr.as_ref(), parent_id)
  }
}

fn serialize_spread(
  ctx: &mut SerializeCtx,
  expr: &Expr,
  span: &Span,
  parent_id: usize,
) -> usize {
  let id = ctx.push_node(AstNode::Spread, parent_id, span);

  let child_offset = ctx.reserve_child_ids(1);
  let expr_id = serialize_expr(ctx, expr, id);

  ctx.set_child(child_offset, expr_id, 0);

  id
}

fn serialize_ident_name(
  ctx: &mut SerializeCtx,
  ident_name: &IdentName,
  parent_id: usize,
) -> usize {
  let id = ctx.push_node(AstNode::Identifier, parent_id, &ident_name.span);

  let str_id = ctx.str_table.insert(ident_name.sym.as_str());
  append_usize(&mut ctx.result, str_id);

  id
}

fn serialize_prop_name(
  ctx: &mut SerializeCtx,
  prop_name: &PropName,
  parent_id: usize,
) -> usize {
  match prop_name {
    PropName::Ident(ident_name) => {
      serialize_ident_name(ctx, ident_name, parent_id)
    }
    PropName::Str(str_prop) => {
      let child_id =
        ctx.push_node(AstNode::StringLiteral, parent_id, &str_prop.span);

      let str_id = ctx.str_table.insert(&str_prop.value.as_str());
      append_usize(&mut ctx.result, str_id);

      child_id
    }
    PropName::Num(number) => {
      serialize_lit(ctx, &Lit::Num(number.clone()), parent_id)
    }
    PropName::Computed(node) => serialize_expr(ctx, &node.expr, parent_id),
    PropName::BigInt(big_int) => {
      serialize_lit(ctx, &Lit::BigInt(big_int.clone()), parent_id)
    }
  }
}

fn serialize_ident(
  ctx: &mut SerializeCtx,
  ident: &Ident,
  parent_id: usize,
) -> usize {
  let id = ctx.push_node(AstNode::Identifier, parent_id, &ident.span);

  let str_id = ctx.str_table.insert(ident.sym.as_str());
  append_usize(&mut ctx.result, str_id);

  id
}

fn serialize_lit(ctx: &mut SerializeCtx, lit: &Lit, parent_id: usize) -> usize {
  match lit {
    Lit::Str(node) => {
      let id = ctx.push_node(AstNode::StringLiteral, parent_id, &node.span);
      let str_id = ctx.str_table.insert(&node.value.as_str());
      append_usize(&mut ctx.result, str_id);

      id
    }
    Lit::Bool(lit_bool) => {
      let id = ctx.push_node(AstNode::Bool, parent_id, &lit_bool.span);

      let value: u8 = if lit_bool.value { 1 } else { 0 };
      ctx.result.push(value);

      id
    }
    Lit::Null(node) => ctx.push_node(AstNode::Null, parent_id, &node.span),
    Lit::Num(node) => {
      let id = ctx.push_node(AstNode::NumericLiteral, parent_id, &node.span);

      let value = node.raw.as_ref().unwrap();
      let str_id = ctx.str_table.insert(&value.as_str());
      append_usize(&mut ctx.result, str_id);

      id
    }
    Lit::BigInt(node) => {
      let id = ctx.push_node(AstNode::BigIntLiteral, parent_id, &node.span);

      let str_id = ctx.str_table.insert(&node.value.to_string());
      append_usize(&mut ctx.result, str_id);

      id
    }
    Lit::Regex(node) => {
      let id = ctx.push_node(AstNode::RegExpLiteral, parent_id, &node.span);

      let pattern_id = ctx.str_table.insert(&node.exp.as_str());
      let flag_id = ctx.str_table.insert(&node.flags.as_str());

      append_usize(&mut ctx.result, pattern_id);
      append_usize(&mut ctx.result, flag_id);

      id
    }
    Lit::JSXText(jsxtext) => {
      ctx.push_node(AstNode::JSXText, parent_id, &jsxtext.span)
    }
  }
}

fn serialize_ts_type(
  ctx: &mut SerializeCtx,
  ts_type: &TsType,
  parent_id: usize,
) -> usize {
  todo!();
}