firescript/src/fs-elements/ClassBody.js

Firescript transpiler

const FirescriptElement = require('./FirescriptElement')

/**
 * ClassBody
 *
 * @class ClassBody
 * @extends FirescriptElement
 *
 * interface ClassBody {
 *   type: 'ClassBody';
 *   body: MethodDefinition[];
 * }
 */
class ClassBody extends FirescriptElement {
  constructor (ast) {
    super(ast)

    this.body = this.createElementList(ast.body)
  }

  toFSString (ctx) {
    if (this.body.length === 0) {
      return '\n'
    }

    return this.renderElement(
      ctx.indent(1) +
      ctx.each(this.body, (item) => item.trim(), '\n' + ctx.indent()) +
      ctx.indent(-1)
    )
  }
}

module.exports = ClassBody