firescript/src/fs-elements/VariableDeclarator.js

Firescript transpiler

const FirescriptElement = require('./FirescriptElement')

/**
 * VariableDeclarator
 *
 * @class VariableDeclarator
 * @extends FirescriptElement
 *
 * interface VariableDeclarator {
 *   type: 'VariableDeclarator';
 *   id: Identifier | BindingPattern;
 *   init: Expression | null;
 * }
*/
class VariableDeclarator extends FirescriptElement {
  constructor (ast) {
    super(ast)

    this.id = this.createElement(ast.id)
    this.init = ast.init ? this.createElement(ast.init) : null
    this.fsTyping = ast.fsTyping ? this.createElement(ast.fsTyping) : null
  }

  toFSString (ctx) {
    if (!this.init) {
      return this.renderElement(
        this.id.toFSString(ctx)
      )
    }

    const init = this.init.toFSString(ctx)
    const initSpacing = init.startsWith('\n') ? '' : ' '
    const hasTyping = this.fsTyping && this.fsTyping.name !== 'any'
    const fsTyping = hasTyping ? this.fsTyping.toFSString(ctx) : ''
    const typeSpacing = hasTyping ? ' ' : ''

    return this.renderElement(
      fsTyping +
      typeSpacing +
      this.id.toFSString(ctx) +
      ' =' +
      initSpacing +
      init
    )
  }
}

module.exports = VariableDeclarator