/* global Buffer */
import {
  getSource as getSourceImpl,
  load as reactLoad,
  resolve,
  transformSource as reactTransformSource,
} from 'react-server-dom-webpack/node-loader'

export {resolve}

import babel from '@babel/core'

const babelOptions = {
  babelrc: false,
  ignore: [/\/(build|node_modules)\//],
  plugins: [
    '@babel/plugin-syntax-import-meta',
    '@babel/plugin-transform-react-jsx',
  ],
  sourceMaps: process.env.NODE_ENV === 'development' ? 'inline' : false,
}

async function babelLoad(url, context, defaultLoad) {
  const result = await defaultLoad(url, context, defaultLoad)
  if (result.format === 'module') {
    const opt = Object.assign({filename: url}, babelOptions)
    const newResult = await babel.transformAsync(result.source, opt)
    if (!newResult) {
      if (typeof result.source === 'string') {
        return result
      }
      return {
        source: Buffer.from(result.source).toString('utf8'),
        format: 'module',
      }
    }
    return {source: newResult.code, format: 'module'}
  }
  return defaultLoad(url, context, defaultLoad)
}

export async function load(url, context, defaultLoad) {
  return await reactLoad(url, context, (u, c) => {
    return babelLoad(u, c, defaultLoad)
  })
}

async function babelTransformSource(source, context, defaultTransformSource) {
  const {format} = context
  if (format === 'module') {
    const opt = Object.assign({filename: context.url}, babelOptions)
    const newResult = await babel.transformAsync(source, opt)
    if (!newResult) {
      if (typeof source === 'string') {
        return {source}
      }
      return {
        source: Buffer.from(source).toString('utf8'),
      }
    }
    return {source: newResult.code}
  }
  return defaultTransformSource(source, context, defaultTransformSource)
}

async function transformSourceImpl(source, context, defaultTransformSource) {
  return await reactTransformSource(source, context, (s, c) => {
    return babelTransformSource(s, c, defaultTransformSource)
  })
}

export const transformSource =
  process.version < 'v16' ? transformSourceImpl : undefined
export const getSource = process.version < 'v16' ? getSourceImpl : undefined