node-acorn 5.4.1+ds1-1 / usr / lib / nodejs / acorn / tokencontext.js

"use strict";
var $$state$$ = require("./state"), $$tokentype$$ = require("./tokentype"), $$whitespace$$ = require("./whitespace");

class TokContext {
  constructor(token, isExpr, preserveSpace, override, generator) {
    this.token = token
    this.isExpr = !!isExpr
    this.preserveSpace = !!preserveSpace
    this.override = override
    this.generator = !!generator
  }
}

const types = {
  b_stat: new TokContext("{", false),
  b_expr: new TokContext("{", true),
  b_tmpl: new TokContext("${", false),
  p_stat: new TokContext("(", false),
  p_expr: new TokContext("(", true),
  q_tmpl: new TokContext("`", true, true, p => p.tryReadTemplateToken()),
  f_stat: new TokContext("function", false),
  f_expr: new TokContext("function", true),
  f_expr_gen: new TokContext("function", true, false, null, true),
  f_gen: new TokContext("function", false, false, null, true)
};

const pp = $$state$$.Parser.prototype

pp.initialContext = function() {
  return [types.b_stat]
}

pp.braceIsBlock = function(prevType) {
  let parent = this.curContext()
  if (parent === types.f_expr || parent === types.f_stat)
    return true
  if (prevType === $$tokentype$$.types.colon && (parent === types.b_stat || parent === types.b_expr))
    return !parent.isExpr

  // The check for `tt.name && exprAllowed` detects whether we are
  // after a `yield` or `of` construct. See the `updateContext` for
  // `tt.name`.
  if (prevType === $$tokentype$$.types._return || prevType == $$tokentype$$.types.name && this.exprAllowed)
    return $$whitespace$$.lineBreak.test(this.input.slice(this.lastTokEnd, this.start));
  if (prevType === $$tokentype$$.types._else || prevType === $$tokentype$$.types.semi || prevType === $$tokentype$$.types.eof || prevType === $$tokentype$$.types.parenR || prevType == $$tokentype$$.types.arrow)
    return true
  if (prevType == $$tokentype$$.types.braceL)
    return parent === types.b_stat
  if (prevType == $$tokentype$$.types._var || prevType == $$tokentype$$.types.name)
    return false
  return !this.exprAllowed
}

pp.inGeneratorContext = function() {
  for (let i = this.context.length - 1; i >= 1; i--) {
    let context = this.context[i]
    if (context.token === "function")
      return context.generator
  }
  return false
}

pp.updateContext = function(prevType) {
  let update, type = this.type
  if (type.keyword && prevType == $$tokentype$$.types.dot)
    this.exprAllowed = false
  else if (update = type.updateContext)
    update.call(this, prevType)
  else
    this.exprAllowed = type.beforeExpr
}

// Token-specific context update code

$$tokentype$$.types.parenR.updateContext = $$tokentype$$.types.braceR.updateContext = function() {
  if (this.context.length == 1) {
    this.exprAllowed = true
    return
  }
  let out = this.context.pop()
  if (out === types.b_stat && this.curContext().token === "function") {
    out = this.context.pop()
  }
  this.exprAllowed = !out.isExpr
}

$$tokentype$$.types.braceL.updateContext = function(prevType) {
  this.context.push(this.braceIsBlock(prevType) ? types.b_stat : types.b_expr)
  this.exprAllowed = true
}

$$tokentype$$.types.dollarBraceL.updateContext = function() {
  this.context.push(types.b_tmpl)
  this.exprAllowed = true
}

$$tokentype$$.types.parenL.updateContext = function(prevType) {
  let statementParens = prevType === $$tokentype$$.types._if || prevType === $$tokentype$$.types._for || prevType === $$tokentype$$.types._with || prevType === $$tokentype$$.types._while
  this.context.push(statementParens ? types.p_stat : types.p_expr)
  this.exprAllowed = true
}

$$tokentype$$.types.incDec.updateContext = function() {
  // tokExprAllowed stays unchanged
}

$$tokentype$$.types._function.updateContext = $$tokentype$$.types._class.updateContext = function(prevType) {
  if (prevType.beforeExpr && prevType !== $$tokentype$$.types.semi && prevType !== $$tokentype$$.types._else &&
      !((prevType === $$tokentype$$.types.colon || prevType === $$tokentype$$.types.braceL) && this.curContext() === types.b_stat))
    this.context.push(types.f_expr)
  else
    this.context.push(types.f_stat)
  this.exprAllowed = false
}

$$tokentype$$.types.backQuote.updateContext = function() {
  if (this.curContext() === types.q_tmpl)
    this.context.pop()
  else
    this.context.push(types.q_tmpl)
  this.exprAllowed = false
}

$$tokentype$$.types.star.updateContext = function(prevType) {
  if (prevType == $$tokentype$$.types._function) {
    let index = this.context.length - 1
    if (this.context[index] === types.f_expr)
      this.context[index] = types.f_expr_gen
    else
      this.context[index] = types.f_gen
  }
  this.exprAllowed = true
}

$$tokentype$$.types.name.updateContext = function(prevType) {
  let allowed = false
  if (this.options.ecmaVersion >= 6) {
    if (this.value == "of" && !this.exprAllowed ||
        this.value == "yield" && this.inGeneratorContext())
      allowed = true
  }
  this.exprAllowed = allowed
}
exports.TokContext = TokContext, exports.types = types;