assemblyscript/src/extra/ast.ts

import {
  Token,
  Range,

  Node,
  NodeKind,

  Source,

  TypeNode,
  TypeParameter,

  Expression,
  IdentifierExpression,
  LiteralExpression,
  LiteralKind,
  FloatLiteralExpression,
  IntegerLiteralExpression,
  StringLiteralExpression,
  RegexpLiteralExpression,
  ArrayLiteralExpression,
  AssertionExpression,
  AssertionKind,
  BinaryExpression,
  CallExpression,
  CommaExpression,
  ElementAccessExpression,
  NewExpression,
  ParenthesizedExpression,
  PropertyAccessExpression,
  TernaryExpression,
  UnaryPostfixExpression,
  UnaryExpression,
  UnaryPrefixExpression,

  Statement,
  BlockStatement,
  BreakStatement,
  ContinueStatement,
  DoStatement,
  EmptyStatement,
  ExportImportStatement,
  ExportStatement,
  ExpressionStatement,
  ForStatement,
  IfStatement,
  ImportStatement,
  ReturnStatement,
  SwitchStatement,
  ThrowStatement,
  TryStatement,
  VariableStatement,
  WhileStatement,

  DeclarationStatement,
  ClassDeclaration,
  EnumDeclaration,
  EnumValueDeclaration,
  FieldDeclaration,
  FunctionDeclaration,
  ImportDeclaration,
  InterfaceDeclaration,
  MethodDeclaration,
  NamespaceDeclaration,
  TypeDeclaration,
  VariableDeclaration,

  Decorator,
  Modifier,
  ModifierKind,
  Parameter,
  ParameterKind,
  ExportMember,
  SwitchCase,

  hasModifier
} from "../ast";

import {
  CharCode
} from "../util/charcode";

export function serializeNode(node: Node, sb: string[]): void {
  switch (node.kind) {

    case NodeKind.SOURCE:
      serializeSource(<Source>node, sb);
      break;

    // types

    case NodeKind.TYPE:
      serializeTypeNode(<TypeNode>node, sb);
      break;

    case NodeKind.TYPEPARAMETER:
      serializeTypeParameter(<TypeParameter>node, sb);
      break;

    // expressions

    case NodeKind.FALSE:
    case NodeKind.NULL:
    case NodeKind.SUPER:
    case NodeKind.THIS:
    case NodeKind.TRUE:
    case NodeKind.CONSTRUCTOR:
    case NodeKind.IDENTIFIER:
      serializeIdentifierExpression(<IdentifierExpression>node, sb);
      break;

    case NodeKind.ASSERTION:
      serializeAssertionExpression(<AssertionExpression>node, sb);
      break;

    case NodeKind.BINARY:
      serializeBinaryExpression(<BinaryExpression>node, sb);
      break;

    case NodeKind.CALL:
      serializeCallExpression(<CallExpression>node, sb);
      break;

    case NodeKind.ELEMENTACCESS:
      serializeElementAccessExpression(<ElementAccessExpression>node, sb);
      break;

    case NodeKind.LITERAL:
      serializeLiteralExpression(<LiteralExpression>node, sb);
      break;

    case NodeKind.NEW:
      serializeNewExpression(<NewExpression>node, sb);
      break;

    case NodeKind.PARENTHESIZED:
      serializeParenthesizedExpression(<ParenthesizedExpression>node, sb);
      break;

    case NodeKind.PROPERTYACCESS:
      serializePropertyAccessExpression(<PropertyAccessExpression>node, sb);
      break;

    case NodeKind.TERNARY:
      serializeTernaryExpression(<TernaryExpression>node, sb);
      break;

    case NodeKind.UNARYPOSTFIX:
      serializeUnaryPostfixExpression(<UnaryPostfixExpression>node, sb);
      break;

    case NodeKind.UNARYPREFIX:
      serializeUnaryPrefixExpression(<UnaryPrefixExpression>node, sb);
      break;

    // statements

    case NodeKind.BLOCK:
      serializeBlockStatement(<BlockStatement>node, sb);
      break;

    case NodeKind.BREAK:
      serializeBreakStatement(<BreakStatement>node, sb);
      break;

    case NodeKind.CONTINUE:
      serializeContinueStatement(<ContinueStatement>node, sb);
      break;

    case NodeKind.DO:
      serializeDoStatement(<DoStatement>node, sb);
      break;

    case NodeKind.EMPTY:
      serializeEmptyStatement(<EmptyStatement>node, sb);
      break;

    case NodeKind.EXPORT:
      serializeExportStatement(<ExportStatement>node, sb);
      break;

    case NodeKind.EXPORTIMPORT:
      serializeExportImportStatement(<ExportImportStatement>node, sb);
      break;

    case NodeKind.EXPRESSION:
      serializeExpressionStatement(<ExpressionStatement>node, sb);
      break;

    case NodeKind.FOR:
      serializeForStatement(<ForStatement>node, sb);
      break;

    case NodeKind.IF:
      serializeIfStatement(<IfStatement>node, sb);
      break;

    case NodeKind.IMPORT:
      serializeImportStatement(<ImportStatement>node, sb);
      break;

    case NodeKind.RETURN:
      serializeReturnStatement(<ReturnStatement>node, sb);
      break;

    case NodeKind.SWITCH:
      serializeSwitchStatement(<SwitchStatement>node, sb);
      break;

    case NodeKind.THROW:
      serializeThrowStatement(<ThrowStatement>node, sb);
      break;

    case NodeKind.TRY:
      serializeTryStatement(<TryStatement>node, sb);

    case NodeKind.VARIABLE:
      serializeVariableStatement(<VariableStatement>node, sb);
      break;

    case NodeKind.WHILE:
      serializeWhileStatement(<WhileStatement>node, sb);
      break;

    // declaration statements

    case NodeKind.CLASSDECLARATION:
      serializeClassDeclaration(<ClassDeclaration>node, sb);
      break;

    case NodeKind.ENUMDECLARATION:
      serializeEnumDeclaration(<EnumDeclaration>node, sb);
      break;

    case NodeKind.ENUMVALUEDECLARATION:
      serializeEnumValueDeclaration(<EnumValueDeclaration>node, sb);
      break;

    case NodeKind.FIELDDECLARATION:
      serializeFieldDeclaration(<FieldDeclaration>node, sb);
      break;

    case NodeKind.FUNCTIONDECLARATION:
      serializeFunctionDeclaration(<FunctionDeclaration>node, sb);
      break;

    case NodeKind.IMPORTDECLARATION:
      serializeImportDeclaration(<ImportDeclaration>node, sb);
      break;

    case NodeKind.INTERFACEDECLARATION:
      serializeInterfaceDeclaration(<InterfaceDeclaration>node, sb);
      break;

    case NodeKind.METHODDECLARATION:
      serializeMethodDeclaration(<MethodDeclaration>node, sb);
      break;

    case NodeKind.NAMESPACEDECLARATION:
      serializeNamespaceDeclaration(<NamespaceDeclaration>node, sb);
      break;

    case NodeKind.TYPEDECLARATION:
      serializeTypeDeclaration(<TypeDeclaration>node, sb);
      break;

    case NodeKind.VARIABLEDECLARATION:
      serializeVariableDeclaration(<VariableDeclaration>node, sb);
      break;

    // other

    case NodeKind.DECORATOR:
      serializeDecorator(<Decorator>node, sb);
      break;

    case NodeKind.EXPORTMEMBER:
      serializeExportMember(<ExportMember>node, sb);
      break;

    case NodeKind.MODIFIER:
      serializeModifier(<Modifier>node, sb);
      break;

    case NodeKind.PARAMETER:
      serializeParameter(<Parameter>node, sb);
      break;

    case NodeKind.SWITCHCASE:
      serializeSwitchCase(<SwitchCase>node, sb);
      break;

    default:
      assert(false);
      break;
  }
}

export function serializeSource(source: Source, sb: string[]): void {
  for (var i = 0, k = source.statements.length; i < k; ++i) {
    serializeStatement(source.statements[i], sb);
    if (builderEndsWith(sb, CharCode.CLOSEBRACE))
      sb.push("\n");
    else
      sb.push(";\n");
  }
}

// types

export function serializeTypeNode(node: TypeNode, sb: string[]): void {
  serializeIdentifierExpression(<IdentifierExpression>node.name, sb);
  if (node.typeArguments.length) {
    sb.push("<");
    for (var i = 0, k = node.typeArguments.length; i < k; ++i) {
      if (i > 0)
        sb.push(", ");
        serializeTypeNode(node.typeArguments[i], sb);
    }
    sb.push(">");
  }
  if (node.isNullable)
    sb.push(" | null");
}

export function serializeTypeParameter(node: TypeParameter, sb: string[]): void {
  serializeIdentifierExpression(node.name, sb);
  if (node.extendsType) {
    sb.push(" extends ");
    serializeTypeNode(node.extendsType, sb);
  }
}

// expressions

export function serializeExpression(node: Expression, sb: string[]): void {
  serializeNode(node, sb);
}

export function serializeIdentifierExpression(node: IdentifierExpression, sb: string[]): void {
  sb.push(node.text);
}

export function serializeArrayLiteralExpression(node: ArrayLiteralExpression, sb: string[]): void {
  sb.push("[");
  for (var i = 0, k = node.elementExpressions.length; i < k; ++i) {
    if (i > 0)
      sb.push(", ");
    if (node.elementExpressions[i])
      serializeExpression(<Expression>node.elementExpressions[i], sb);
  }
  sb.push("]");
}

export function serializeAssertionExpression(node: AssertionExpression, sb: string[]): void {
  if (node.assertionKind == AssertionKind.PREFIX) {
    sb.push("<");
    serializeTypeNode(node.toType, sb);
    sb.push(">");
    serializeExpression(node.expression, sb);
  } else {
    serializeExpression(node.expression, sb);
    sb.push(" as ");
    serializeTypeNode(node.toType, sb);
  }
}

export function serializeBinaryExpression(node: BinaryExpression, sb: string[]): void {
  serializeExpression(node.left, sb);
  sb.push(" ");
  sb.push(operatorToString(node.operator));
  sb.push(" ");
  serializeExpression(node.right, sb);
}

export function serializeCallExpression(node: CallExpression, sb: string[]): void {
  serializeExpression(node.expression, sb);
  var k: i32;
  if (node.typeArguments && (k = node.typeArguments.length)) {
    sb.push("<");
    for (var i = 0; i < k; ++i) {
      if (i > 0)
        sb.push(", ");
      serializeTypeNode(node.typeArguments[i], sb);
    }
    sb.push(">(");
  } else
    sb.push("(");
  for (i = 0, k = node.arguments.length; i < k; ++i) {
    if (i > 0)
      sb.push(", ");
      serializeExpression(node.arguments[i], sb);
  }
  sb.push(")");
}

export function serializeCommaExpression(node: CommaExpression, sb: string[]): void {
  serializeExpression(node.expressions[0], sb);
  for (var i = 1, k = node.expressions.length; i < k; ++i) {
    sb.push(",");
    serializeExpression(node.expressions[i], sb);
  }
}

export function serializeElementAccessExpression(node: ElementAccessExpression, sb: string[]): void {
  serializeExpression(node.expression, sb);
  sb.push("[");
  serializeExpression(node.elementExpression, sb);
  sb.push("]");
}

export function serializeLiteralExpression(node: LiteralExpression, sb: string[]): void {
  switch (node.literalKind) {

    case LiteralKind.FLOAT:
      serializeFloatLiteralExpression(<FloatLiteralExpression>node, sb);
      break;

    case LiteralKind.INTEGER:
      serializeIntegerLiteralExpression(<IntegerLiteralExpression>node, sb);
      break;

    case LiteralKind.STRING:
      serializeStringLiteralExpression(<StringLiteralExpression>node, sb);
      break;

    case LiteralKind.REGEXP:
      serializeRegexpLiteralExpression(<RegexpLiteralExpression>node, sb);
      break;

    case LiteralKind.ARRAY:
      serializeArrayLiteralExpression(<ArrayLiteralExpression>node, sb);
      break;

    // case LiteralKind.OBJECT:
    //   serializeObjectLiteralExpression(<ObjectLiteralExpression>node, sb);
    //   break;

    default:
      assert(false);
      break;
  }
}

export function serializeFloatLiteralExpression(node: FloatLiteralExpression, sb: string[]): void {
  sb.push(node.value.toString(10));
}

export function serializeIntegerLiteralExpression(node: IntegerLiteralExpression, sb: string[]): void {
  sb.push(node.value.toString());
}

export function serializeStringLiteral(str: string, sb: string[], singleQuoted: bool = false): void {
  var off = 0;
  sb.push(singleQuoted ? "'" : "\"");
  for (var i = 0, k = str.length; i < k;) {
    switch (str.charCodeAt(i)) {

      case CharCode.NULL:
        if (i > off)
          sb.push(str.substring(off, off = i + 1));
        sb.push("\\0");
        off = ++i;
        break;

      case CharCode.BACKSPACE:
        if (i > off)
          sb.push(str.substring(off, i));
        off = ++i;
        sb.push("\\b");
        break;

      case CharCode.TAB:
        if (i > off)
          sb.push(str.substring(off, i));
        off = ++i;
        sb.push("\\t");
        break;

      case CharCode.LINEFEED:
        if (i > off)
          sb.push(str.substring(off, i));
        off = ++i;
        sb.push("\\n");
        break;

      case CharCode.VERTICALTAB:
        if (i > off)
          sb.push(str.substring(off, i));
        off = ++i;
        sb.push("\\v");
        break;

      case CharCode.FORMFEED:
        if (i > off)
          sb.push(str.substring(off, i));
        off = ++i;
        sb.push("\\f");
        break;

      case CharCode.CARRIAGERETURN:
        if (i > off)
          sb.push(str.substring(off, i));
        sb.push("\\r");
        off = ++i;
        break;

      case CharCode.DOUBLEQUOTE:
        if (!singleQuoted) {
          if (i > off)
            sb.push(str.substring(off, i));
          sb.push("\\\"");
          off = ++i;
        } else
          ++i;
        break;

      case CharCode.SINGLEQUOTE:
        if (singleQuoted) {
          if (i > off)
            sb.push(str.substring(off, i));
          sb.push("\\'");
          off = ++i;
        } else
          ++i;
        break;

      case CharCode.BACKSLASH:
        if (i > off)
          sb.push(str.substring(off, i));
        sb.push("\\\\");
        off = ++i;
        break;

      default:
        ++i;
        break;
    }
  }
  if (i > off)
    sb.push(str.substring(off, i));
  sb.push(singleQuoted ? "'" : "\"");
}

export function serializeStringLiteralExpression(node: StringLiteralExpression, sb: string[]): void {
  serializeStringLiteral(node.value, sb);
}

export function serializeRegexpLiteralExpression(node: RegexpLiteralExpression, sb: string[]): void {
  sb.push("/");
  sb.push(node.pattern);
  sb.push("/");
  sb.push(node.patternFlags);
}

export function serializeNewExpression(node: NewExpression, sb: string[]): void {
  sb.push("new ");
  serializeCallExpression(node, sb);
}

export function serializeParenthesizedExpression(node: ParenthesizedExpression, sb: string[]): void {
  sb.push("(");
  serializeExpression(node.expression, sb);
  sb.push(")");
}

export function serializePropertyAccessExpression(node: PropertyAccessExpression, sb: string[]): void {
  serializeExpression(node.expression, sb);
  sb.push(".");
  serializeIdentifierExpression(node.property, sb);
}

export function serializeTernaryExpression(node: TernaryExpression, sb: string[]): void {
  serializeExpression(node.condition, sb);
  sb.push(" ? ");
  serializeExpression(node.ifThen, sb);
  sb.push(" : ");
  serializeExpression(node.ifElse, sb);
}

export function serializeUnaryExpression(node: UnaryExpression, sb: string[]): void {
  switch (node.kind) {

    case NodeKind.UNARYPOSTFIX:
      serializeUnaryPostfixExpression(<UnaryPostfixExpression>node, sb);
      break;

    case NodeKind.UNARYPREFIX:
      serializeUnaryPrefixExpression(<UnaryPrefixExpression>node, sb);
      break;

    default:
      assert(false);
      break;
  }
}

export function operatorToString(token: Token): string {
  switch (token) {
    case Token.DELETE: return "delete";
    case Token.IN: return "in";
    case Token.INSTANCEOF: return "instanceof";
    case Token.NEW: return "new";
    case Token.TYPEOF: return "typeof";
    case Token.VOID: return "void";
    case Token.YIELD: return "yield";
    case Token.DOT_DOT_DOT: return "...";
    case Token.COMMA: return ",";
    case Token.LESSTHAN: return "<";
    case Token.GREATERTHAN: return ">";
    case Token.LESSTHAN_EQUALS: return "<=";
    case Token.GREATERTHAN_EQUALS: return ">=";
    case Token.EQUALS_EQUALS: return "==";
    case Token.EXCLAMATION_EQUALS: return "!=";
    case Token.EQUALS_EQUALS_EQUALS: return "===";
    case Token.EXCLAMATION_EQUALS_EQUALS: return "!==";
    case Token.PLUS: return "+";
    case Token.MINUS: return "-";
    case Token.ASTERISK_ASTERISK: return "**";
    case Token.ASTERISK: return "*";
    case Token.SLASH: return "/";
    case Token.PERCENT: return "%";
    case Token.PLUS_PLUS: return "++";
    case Token.MINUS_MINUS: return "--";
    case Token.LESSTHAN_LESSTHAN: return "<<";
    case Token.GREATERTHAN_GREATERTHAN: return ">>";
    case Token.GREATERTHAN_GREATERTHAN_GREATERTHAN: return ">>>";
    case Token.AMPERSAND: return "&";
    case Token.BAR: return "|";
    case Token.CARET: return "^";
    case Token.EXCLAMATION: return "!";
    case Token.TILDE: return "~";
    case Token.AMPERSAND_AMPERSAND: return "&&";
    case Token.BAR_BAR: return "||";
    case Token.EQUALS: return "=";
    case Token.PLUS_EQUALS: return "+=";
    case Token.MINUS_EQUALS: return "-=";
    case Token.ASTERISK_EQUALS: return "*=";
    case Token.ASTERISK_ASTERISK_EQUALS: return "**=";
    case Token.SLASH_EQUALS: return "/=";
    case Token.PERCENT_EQUALS: return "%=";
    case Token.LESSTHAN_LESSTHAN_EQUALS: return "<<=";
    case Token.GREATERTHAN_GREATERTHAN_EQUALS: return ">>=";
    case Token.GREATERTHAN_GREATERTHAN_GREATERTHAN_EQUALS: return ">>>=";
    case Token.AMPERSAND_EQUALS: return "&=";
    case Token.BAR_EQUALS: return "|=";
    case Token.CARET_EQUALS: return "^=";
    default: assert(false); return "";
  }
}

export function serializeUnaryPostfixExpression(node: UnaryPostfixExpression, sb: string[]): void {
  serializeExpression(node.operand, sb);
  sb.push(operatorToString(node.operator));
}

export function serializeUnaryPrefixExpression(node: UnaryPrefixExpression, sb: string[]): void {
  sb.push(operatorToString(node.operator));
  serializeExpression(node.operand, sb);
}

// statements

export function serializeStatement(node: Statement, sb: string[]): void {
  serializeNode(node, sb);
}

export function serializeBlockStatement(node: BlockStatement, sb: string[]): void {
  sb.push("{\n");
  for (var i = 0, k = node.statements.length; i < k; ++i) {
    serializeStatement(node.statements[i], sb);
    if (builderEndsWith(sb, CharCode.CLOSEBRACE))
      sb.push("\n");
    else
      sb.push(";\n");
  }
  sb.push("}");
}

export function serializeBreakStatement(node: BreakStatement, sb: string[]): void {
  if (node.label) {
    sb.push("break ");
    serializeIdentifierExpression(node.label, sb);
  } else
    sb.push("break");
}

export function serializeContinueStatement(node: ContinueStatement, sb: string[]): void {
  if (node.label) {
    sb.push("continue ");
    serializeIdentifierExpression(node.label, sb);
  } else
    sb.push("continue");
}

export function serializeClassDeclaration(node: ClassDeclaration, sb: string[]): void {
  if (node.decorators)
    for (var i = 0, k = node.decorators.length; i < k; ++i) {
      serializeDecorator(node.decorators[i], sb);
      sb.push("\n");
    }
  if (node.modifiers)
    for (i = 0, k = node.modifiers.length; i < k; ++i) {
      serializeModifier(node.modifiers[i], sb);
      sb.push(" ");
    }
  sb.push("class ");
  sb.push(node.name.text);
  if (node.typeParameters.length) {
    sb.push("<");
    for (i = 0, k = node.typeParameters.length; i < k; ++i) {
      if (i > 0)
        sb.push(", ");
      serializeTypeParameter(node.typeParameters[i], sb);
    }
    sb.push(">");
  }
  if (node.extendsType) {
    sb.push(" extends ");
    serializeTypeNode(node.extendsType, sb);
  }
  if (node.implementsTypes.length) {
    sb.push(" implements ");
    for (i = 0, k = node.implementsTypes.length; i < k; ++i) {
      if (i > 0)
        sb.push(", ");
      serializeTypeNode(node.implementsTypes[i], sb);
    }
  }
  sb.push(" {\n");
  for (i = 0, k = node.members.length; i < k; ++i) {
    serializeStatement(node.members[i], sb);
    if (builderEndsWith(sb, CharCode.CLOSEBRACE))
      sb.push("\n");
    else
      sb.push(";\n");
  }
  sb.push("}");
}

export function serializeDoStatement(node: DoStatement, sb: string[]): void {
  sb.push("do ");
  serializeStatement(node.statement, sb);
  if (node.statement.kind == NodeKind.BLOCK)
    sb.push(" while (");
  else
    sb.push(";\nwhile (");
  serializeExpression(node.condition, sb);
  sb.push(")");
}

export function serializeEmptyStatement(node: EmptyStatement, sb: string[]): void {
}

export function serializeEnumDeclaration(node: EnumDeclaration, sb: string[]): void {
  if (node.modifiers)
    for (var i = 0, k = node.modifiers.length; i < k; ++i) {
      serializeModifier(node.modifiers[i], sb);
      sb.push(" ");
    }
  sb.push("enum ");
  serializeIdentifierExpression(node.name, sb);
  sb.push(" {\n");
  for (i = 0, k = node.values.length; i < k; ++i) {
    if (i > 0)
      sb.push(",\n");
    serializeEnumValueDeclaration(node.values[i], sb);
  }
  sb.push("\n}");
}

export function serializeEnumValueDeclaration(node: EnumValueDeclaration, sb: string[]): void {
  serializeIdentifierExpression(node.name, sb);
  if (node.value) {
    sb.push(" = ");
    serializeExpression(node.value, sb);
  }
}

export function serializeExportImportStatement(node: ExportImportStatement, sb: string[]): void {
  sb.push("export import ");
  serializeIdentifierExpression(node.externalName, sb);
  sb.push(" = ");
  serializeIdentifierExpression(node.name, sb);
}

export function serializeExportMember(node: ExportMember, sb: string[]): void {
  serializeIdentifierExpression(node.name, sb);
  if (node.externalName.text != node.name.text) {
    sb.push(" as ");
    serializeIdentifierExpression(node.externalName, sb);
  }
}

export function serializeExportStatement(node: ExportStatement, sb: string[]): void {
  if (node.modifiers)
    for (var i = 0, k = node.modifiers.length; i < k; ++i) {
      serializeModifier(node.modifiers[i], sb);
      sb.push(" ");
    }
  sb.push("export {\n");
  for (i = 0, k = node.members.length; i < k; ++i) {
    if (i > 0)
      sb.push(",\n");
      serializeExportMember(node.members[i], sb);
  }
  if (node.path) {
    sb.push("\n} from ");
    serializeStringLiteralExpression(node.path, sb);
  } else
    sb.push("\n}");
}

export function serializeExpressionStatement(node: ExpressionStatement, sb: string[]): void {
  serializeExpression(node.expression, sb);
}

export function serializeFieldDeclaration(node: FieldDeclaration, sb: string[]): void {
  if (node.decorators)
    for (var i = 0, k = node.decorators.length; i < k; ++i) {
      serializeDecorator(node.decorators[i], sb);
      sb.push("\n");
    }
  if (node.modifiers)
    for (i = 0, k = node.modifiers.length; i < k; ++i) {
      serializeModifier(node.modifiers[i], sb);
      sb.push(" ");
    }
    serializeIdentifierExpression(node.name, sb);
  if (node.type) {
    sb.push(": ");
    serializeTypeNode(node.type, sb);
  }
  if (node.initializer) {
    sb.push(" = ");
    serializeExpression(node.initializer, sb);
  }
}

export function serializeForStatement(node: ForStatement, sb: string[]): void {
  sb.push("for (");
  if (node.initializer)
    serializeStatement(node.initializer, sb);
  if (node.condition) {
    sb.push("; ");
    serializeExpression(node.condition, sb);
  } else
    sb.push(";");
  if (node.incrementor) {
    sb.push("; ");
    serializeExpression(node.incrementor, sb);
  } else
    sb.push(";");
  sb.push(") ");
  serializeStatement(node.statement, sb);
}

export function serializeFunctionDeclaration(node: FunctionDeclaration, sb: string[]): void {
  if (node.decorators)
    for (var i = 0, k = node.decorators.length; i < k; ++i) {
      serializeDecorator(node.decorators[i], sb);
      sb.push("\n");
    }
  if (node.modifiers)
    for (i = 0, k = node.modifiers.length; i < k; ++i) {
      serializeModifier(node.modifiers[i], sb);
      sb.push(" ");
    }
  sb.push("function ");
  serializeFunctionCommon(node, sb);
}

function serializeFunctionCommon(node: FunctionDeclaration, sb: string[]): void {
  serializeIdentifierExpression(node.name, sb);
  if (node.typeParameters.length) {
    sb.push("<");
    for (var i = 0, k = node.typeParameters.length; i < k; ++i) {
      if (i > 0)
        sb.push(", ");
      serializeTypeParameter(node.typeParameters[i], sb);
    }
    sb.push(">");
  }
  sb.push("(");
  for (i = 0, k = node.parameters.length; i < k; ++i) {
    if (i > 0)
      sb.push(", ");
    serializeParameter(node.parameters[i], sb);
  }
  if (node.returnType && !hasModifier(ModifierKind.SET, node.modifiers)) {
    sb.push("): ");
    serializeTypeNode(node.returnType, sb);
  } else
    sb.push(")");
  if (node.statements) {
    sb.push(" {\n");
    for (i = 0, k = node.statements.length; i < k; ++i) {
      serializeStatement(node.statements[i], sb);
      if (builderEndsWith(sb, CharCode.CLOSEBRACE))
        sb.push("\n");
      else
        sb.push(";\n");
    }
    sb.push("}");
  }
}

export function serializeIfStatement(node: IfStatement, sb: string[]): void {
  sb.push("if (");
  serializeExpression(node.condition, sb);
  sb.push(") ");
  serializeStatement(node.ifTrue, sb);
  if (node.ifTrue.kind != NodeKind.BLOCK)
    sb.push(";\n");
  if (node.ifFalse) {
    if (node.ifTrue.kind == NodeKind.BLOCK)
      sb.push(" else ");
    else
      sb.push("else ");
    serializeStatement(node.ifFalse, sb);
  }
}

export function serializeImportDeclaration(node: ImportDeclaration, sb: string[]): void {
  serializeIdentifierExpression(node.externalName, sb);
  if (node.externalName.text != node.name.text) {
    sb.push(" as ");
    serializeIdentifierExpression(node.name, sb);
  }
}

export function serializeImportStatement(node: ImportStatement, sb: string[]): void {
  sb.push("import ");
  if (node.declarations) {
    sb.push("{\n");
    for (var i: i32 = 0, k: i32 = node.declarations.length; i < k; ++i) {
      if (i > 0)
        sb.push(",\n");
        serializeImportDeclaration(node.declarations[i], sb);
    }
    sb.push("\n} from ");
  } else if (node.namespaceName) {
    sb.push("* as ");
    serializeIdentifierExpression(node.namespaceName, sb);
    sb.push(" from ");
  }
  serializeStringLiteralExpression(node.path, sb);
}

export function serializeInterfaceDeclaration(node: InterfaceDeclaration, sb: string[]): void {
  if (node.decorators)
    for (var i = 0, k = node.decorators.length; i < k; ++i) {
      serializeDecorator(node.decorators[i], sb);
      sb.push("\n");
    }
  if (node.modifiers)
    for (i = 0, k = node.modifiers.length; i < k; ++i) {
      serializeModifier(node.modifiers[i], sb);
      sb.push(" ");
    }
  sb.push("interface ");
  serializeIdentifierExpression(node.name, sb);
  if (node.typeParameters.length) {
    sb.push("<");
    for (i = 0, k =  node.typeParameters.length; i < k; ++i) {
      if (i > 0)
        sb.push(", ");
      serializeTypeParameter(node.typeParameters[i], sb);
    }
    sb.push(">");
  }
  if (node.extendsType) {
    sb.push(" extends ");
    serializeTypeNode(node.extendsType, sb);
  }
  sb.push(" {\n");
  for (i = 0, k = node.members.length; i < k; ++i) {
    serializeStatement(node.members[i], sb);
    if (builderEndsWith(sb, CharCode.CLOSEBRACE))
      sb.push("\n");
    else
      sb.push(";\n");
  }
  sb.push("}");
}

export function serializeMethodDeclaration(node: MethodDeclaration, sb: string[]): void {
  if (node.decorators)
    for (var i = 0, k = node.decorators.length; i < k; ++i) {
      serializeDecorator(node.decorators[i], sb);
      sb.push("\n");
    }
  if (node.modifiers)
    for (i = 0, k = node.modifiers.length; i < k; ++i) {
      serializeModifier(node.modifiers[i], sb);
      sb.push(" ");
    }
  serializeFunctionCommon(node, sb);
}

export function serializeNamespaceDeclaration(node: NamespaceDeclaration, sb: string[]): void {
  if (node.decorators)
    for (var i = 0, k = node.decorators.length; i < k; ++i) {
      serializeDecorator(node.decorators[i], sb);
      sb.push("\n");
    }
  if (node.modifiers)
    for (i = 0, k = node.modifiers.length; i < k; ++i) {
      serializeModifier(node.modifiers[i], sb);
      sb.push(" ");
    }
  sb.push("namespace ");
  serializeIdentifierExpression(node.name, sb);
  sb.push(" {\n");
  for (i = 0, k = node.members.length; i < k; ++i) {
    serializeStatement(node.members[i], sb);
    if (builderEndsWith(sb, CharCode.CLOSEBRACE))
      sb.push("\n");
    else
      sb.push(";\n");
  }
  sb.push("}");
}

export function serializeReturnStatement(node: ReturnStatement, sb: string[]): void {
  if (node.value) {
    sb.push("return ");
    serializeExpression(node.value, sb);
  } else
    sb.push("return");
}

export function serializeSwitchCase(node: SwitchCase, sb: string[]): void {
  if (node.label) {
    sb.push("case ");
    serializeExpression(node.label, sb);
    sb.push(":\n");
  } else
    sb.push("default:\n");
  for (var i = 0, k = node.statements.length; i < k; ++i) {
    if (i > 0)
      sb.push("\n");
    serializeStatement(node.statements[i], sb);
    if (builderEndsWith(sb, CharCode.CLOSEBRACE))
      sb.push("\n");
    else
      sb.push(";\n");
  }
}

export function serializeSwitchStatement(node: SwitchStatement, sb: string[]): void {
  sb.push("switch (");
  serializeExpression(node.condition, sb);
  sb.push(") {\n");
  for (var i = 0, k = node.cases.length; i < k; ++i) {
    serializeSwitchCase(node.cases[i], sb);
    sb.push("\n");
  }
  sb.push("}");
}

export function serializeThrowStatement(node: ThrowStatement, sb: string[]): void {
  sb.push("throw ");
  serializeExpression(node.value, sb);
  sb.push(";");
}

export function serializeTryStatement(node: TryStatement, sb: string[]): void {
  sb.push("try {\n");
  for (var i = 0, k = node.statements.length; i < k; ++i) {
    serializeStatement(node.statements[i], sb);
    sb.push(";\n");
  }
  if (node.catchVariable) {
    sb.push("} catch (");
    serializeIdentifierExpression(node.catchVariable, sb);
    sb.push(") {\n");
    if (node.catchStatements)
      for (i = 0, k = node.catchStatements.length; i < k; ++i) {
        serializeStatement(node.catchStatements[i], sb);
        sb.push(";\n");
      }
  }
  if (node.finallyStatements) {
    sb.push("} finally {\n");
    for (i = 0, k = node.finallyStatements.length; i < k; ++i) {
      serializeStatement(node.finallyStatements[i], sb);
      sb.push(";\n");
    }
  }
  sb.push("}");
}

export function serializeTypeDeclaration(node: TypeDeclaration, sb: string[]): void {
  if (node.decorators)
    for (var i = 0, k = node.decorators.length; i < k; ++i) {
      serializeDecorator(node.decorators[i], sb);
      sb.push("\n");
    }
  if (node.modifiers)
    for (i = 0, k = node.modifiers.length; i < k; ++i) {
      serializeModifier(node.modifiers[i], sb);
      sb.push(" ");
    }
  sb.push("type ");
  serializeIdentifierExpression(node.name, sb);
  sb.push(" = ");
  serializeTypeNode(node.alias, sb);
}

export function serializeVariableDeclaration(node: VariableDeclaration, sb: string[]): void {
  serializeIdentifierExpression(node.name, sb);
  if (node.type) {
    sb.push(": ");
    serializeTypeNode(node.type, sb);
  }
  if (node.initializer) {
    sb.push(" = ");
    serializeExpression(node.initializer, sb);
  }
}

export function serializeVariableStatement(node: VariableStatement, sb: string[]): void {
  var isConst = false;
  if (node.decorators)
    for (var i = 0, k = node.decorators.length; i < k; ++i) {
      serializeDecorator(node.decorators[i], sb);
      sb.push("\n");
    }
  if (node.modifiers)
    for (i = 0, k = node.modifiers.length; i < k; ++i) {
      serializeModifier(node.modifiers[i], sb);
      sb.push(" ");
      if (node.modifiers[i].modifierKind == ModifierKind.CONST)
        isConst = true;
    }
  if (!isConst)
    sb.push("var ");
  for (i = 0, k = node.declarations.length; i < k; ++i) {
    if (i > 0)
      sb.push(", ");
      serializeVariableDeclaration(node.declarations[i], sb);
  }
}

export function serializeWhileStatement(node: WhileStatement, sb: string[]): void {
  sb.push("while (");
  serializeExpression(node.condition, sb);
  sb.push(") ");
  serializeStatement(node.statement, sb);
}

// other

export function serializeDecorator(node: Decorator, sb: string[]): void {
  sb.push("@");
  serializeExpression(node.name, sb);
  if (node.arguments) {
    sb.push("(");
    for (var i = 0, k = node.arguments.length; i < k; ++i) {
      if (i > 0)
        sb.push(", ");
      serializeExpression(node.arguments[i], sb);
    }
    sb.push(")");
  }
}

export function modifierToString(node: Modifier): string {
  switch (node.modifierKind) {
    case ModifierKind.ABSTRACT: return "abstract";
    case ModifierKind.ASYNC: return "async";
    case ModifierKind.CONST: return "const";
    case ModifierKind.DECLARE: return "declare";
    case ModifierKind.EXPORT: return "export";
    case ModifierKind.GET: return "get";
    case ModifierKind.IMPORT: return "import";
    case ModifierKind.PRIVATE: return "private";
    case ModifierKind.PROTECTED: return "protected";
    case ModifierKind.PUBLIC: return "public";
    case ModifierKind.READONLY: return "readonly";
    case ModifierKind.SET: return "set";
    case ModifierKind.STATIC: return "static";
    default: assert(false); return "";
  }
}

export function serializeModifier(node: Modifier, sb: string[]): void {
  sb.push(modifierToString(node));
}

export function serializeParameter(node: Parameter, sb: string[]): void {
  if (node.parameterKind == ParameterKind.REST)
    sb.push("...");
  serializeIdentifierExpression(node.name, sb);
  if (node.type) {
    if (node.parameterKind == ParameterKind.OPTIONAL && !node.initializer)
      sb.push("?: ");
    else
      sb.push(": ");
    serializeTypeNode(node.type, sb);
  }
  if (node.initializer) {
    sb.push(" = ");
    serializeExpression(node.initializer, sb);
  }
}

// helpers

function builderEndsWith(sb: string[], code: CharCode): bool {
  if (sb.length) {
    var last = sb[sb.length - 1];
    return last.length > 0 ? last.charCodeAt(last.length - 1) == code : false;
  }
  return false;
}