firescript/src/fs-elements/Program.js

Firescript transpiler

const FirescriptElement = require('./FirescriptElement')

/**
 * Program
 *
 * @class Program
 * @extends FirescriptElement
 *
 * interface Program {
 *   type: 'Program';
 *   sourceType: 'module';
 *   body: ModuleItem[];
 * }
 */
class Program extends FirescriptElement {
  constructor (ast) {
    super(ast)

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

  toFSString (ctx) {
    const body = this.body.map((item, index, arr) => {
      const str = item.toFSString(ctx)
      if (/^Import/.test(item.type)) {
        const next = arr[index + 1]
        if (next && !/^Import/.test(next.type)) {
          return str + '\n'
        }
      }

      return str
    }).join('\n').trim()

    return this.renderElement(
      body + '\n'
    )
  }
}

module.exports = Program