feat(core/load): passes public functions if kpo scripts file exports …

…a function
rafamel · May 3, 2019 · 04a93b8 · 04a93b8
1 parent 0a420d4
commit 04a93b8
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/core/load.ts b/src/core/load.ts
@@ -8,6 +8,7 @@ import { IOfType, IPackageOptions, TCoreOptions } from '~/types';
 import options from './options';
 import { absolute } from '~/utils/file';
 import { diff } from 'semver';
+import * as _public from '../public';
 
 export default async function load(
   paths: IPaths,
@@ -116,5 +117,6 @@ export async function requireLocal(
     Object.assign(local.errors, errors);
   }
 
-  return errors.open.throws(() => require(file));
+  const scripts = errors.open.throws(() => require(file));
+  return typeof scripts === 'function' ? scripts(_public) : scripts;
 }