sexp-template/src/parser.js

const Node = require('./node')
const tokenTypes = require('./tokenTypes')

module.exports = class Parser {
  constructor(tokenStream) {
    this.tokenStream = tokenStream
  }

  parse() {
    let tree = []
    while (this.tokenStream.peek().type !== tokenTypes.EOF) {
      tree.push(this.expr())
    }
    return tree
  }

  expr() {
    this.tokenStream.eat(tokenTypes.OPAREN)

    let node = new Node()

    if (this.tokenStream.peek().type === tokenTypes.LITERAL) {
      node.type = 'functionCall'
      node.functionName = this.tokenStream.eat(tokenTypes.LITERAL).value
    }

    node.args = []
    node.subtree = []

    while (
      this.tokenStream.peek().type != tokenTypes.CPAREN &&
      this.tokenStream.peek().type !== tokenTypes.EOF
    ) {
      if (this.tokenStream.peek().type === tokenTypes.ATTRIBUTE) {
        node.args.push(this.attribute())
      } else if (this.tokenStream.peek().type === tokenTypes.OPAREN) {
        node.subtree.push(this.expr())
      } else if (this.tokenStream.peek().type === tokenTypes.LITERAL) {
        node.subtree.push(this.identifier())
      } else if (this.tokenStream.peek().type === tokenTypes.QUOTE) {
        node.subtree.push(this.quotedString())
      } else if (this.tokenStream.peek().type === tokenTypes.KEYWORD) {
        node = this.keyword()
      }
    }

    this.tokenStream.eat(tokenTypes.CPAREN)

    return node
  }

  attribute() {
    let attributeNode = new Node()
    attributeNode.attributeName = this.tokenStream.eat(
      tokenTypes.ATTRIBUTE,
    ).value
    if (this.tokenStream.peek().type === tokenTypes.QUOTE) {
      attributeNode.attributeValue = this.quotedString()
    } else if (this.tokenStream.peek().type === tokenTypes.LITERAL) {
      attributeNode.attributeValue = this.identifier()
    }

    return attributeNode
  }

  identifier() {
    const identifier = this.tokenStream.eat(tokenTypes.LITERAL)
    return new Node({
      type: 'identifier',
      name: identifier.value,
    })
  }

  quotedString() {
    return new Node({
      type: 'string',
      content: this.string(),
    })
  }

  keyword() {
    const keyword = this.tokenStream.eat(tokenTypes.KEYWORD)

    if (keyword.value === 'each') {
      const subject = this.identifier()
      const symbol = this.tokenStream.eat(tokenTypes.SYMBOL)
      const body = this.expr()
      return new Node({
        type: 'each',
        symbol: symbol,
        body: body,
        subject: subject,
      })
    }
  }

  string() {
    this.tokenStream.eat(tokenTypes.QUOTE)
    let stringValue = this.tokenStream.eat(tokenTypes.LITERAL).value
    this.tokenStream.eat(tokenTypes.QUOTE)
    return stringValue
  }
}