From b20c4848651c01cab69aa187cdf877585df6df0d Mon Sep 17 00:00:00 2001
From: Pedro Nauck <pedronauck@gmail.com>
Date: Wed, 20 Jun 2018 19:58:21 -0300
Subject: [PATCH] chore(docz-core): add modifyBabelRc method as option on
 project configuration

---
 packages/docz-core/src/commands/args.ts     | 2 ++
 packages/docz-core/src/utils/babelrc.ts     | 3 +--
 packages/docz-core/src/utils/load-config.ts | 2 ++
 3 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/packages/docz-core/src/commands/args.ts b/packages/docz-core/src/commands/args.ts
index 5ae203df7..155ba2dfc 100644
--- a/packages/docz-core/src/commands/args.ts
+++ b/packages/docz-core/src/commands/args.ts
@@ -1,4 +1,5 @@
 import { Plugin } from '../Plugin'
+import { BabelRC } from '../utils/babelrc'
 
 export interface Argv {
   /* io args */
@@ -34,6 +35,7 @@ export interface Config extends Argv {
   hastPlugins: any[]
   themeConfig: ThemeConfig
   modifyBundlerConfig<C>(config: C, dev: boolean): C
+  modifyBabelRc(babelrc: BabelRC): BabelRC
 }
 
 export const args = (yargs: any) => {
diff --git a/packages/docz-core/src/utils/babelrc.ts b/packages/docz-core/src/utils/babelrc.ts
index 5b894e29a..835125c6c 100644
--- a/packages/docz-core/src/utils/babelrc.ts
+++ b/packages/docz-core/src/utils/babelrc.ts
@@ -23,6 +23,5 @@ export const babelrc = (args: Config): BabelRC => {
   })
 
   const reduce = Plugin.reduceFromPlugins<BabelRC>(args.plugins)
-
-  return reduce('modifyBabelRc', config)
+  return reduce('modifyBabelRc', args.modifyBabelRc(config))
 }
diff --git a/packages/docz-core/src/utils/load-config.ts b/packages/docz-core/src/utils/load-config.ts
index 7b9eb4b69..a2a7b054f 100644
--- a/packages/docz-core/src/utils/load-config.ts
+++ b/packages/docz-core/src/utils/load-config.ts
@@ -4,6 +4,7 @@ import * as paths from '../config/paths'
 import { Config } from '../commands/args'
 import { Plugin } from '../Plugin'
 import { omit } from './helpers'
+import { BabelRC } from './babelrc'
 
 const toOmit = ['_', '$0', 'version', 'help']
 
@@ -17,6 +18,7 @@ export const loadConfig = (args: Config): Config => {
     hastPlugins: [],
     themeConfig: {},
     modifyBundlerConfig: (config: any) => config,
+    modifyBabelRc: (babelrc: BabelRC) => babelrc,
   })
 
   const reduce = Plugin.reduceFromPlugins<Config>(config.plugins)