docz/packages/playgrodd-core/src/Entry.ts

import * as fs from 'fs'
import * as path from 'path'
import * as t from 'babel-types'
import { parse } from 'babylon'

import * as paths from './config/paths'
import { traverseAndAssign } from './utils/traverse'

export const convertToAst = (entry: string): t.File =>
  parse(fs.readFileSync(entry, 'utf-8'), {
    sourceType: 'module',
    plugins: ['jsx'],
  })

const getNameFromDoc = traverseAndAssign<any, string>(
  path => path.isCallExpression() && path.node.callee.name === 'doc',
  path => path.node.arguments[0].value
)

export interface IEntryConstructor {
  file: string
  src: string
}

export class Entry {
  public name: string
  public filepath: string
  public route: string

  constructor({ src, file }: IEntryConstructor) {
    const ast = convertToAst(file)
    const name = getNameFromDoc(ast) || ''
    const srcPath = path.resolve(paths.root, src)
    const filepath = path.relative(path.relative(paths.root, src), file)
    const dir = path.relative(srcPath, path.parse(file).dir)
    const route = path.join('/', dir, name)

    this.name = name
    this.route = route
    this.filepath = filepath
  }
}