Merge branch 'master' of https://github.com/rafamel/yaml-import

rafamel · May 1, 2019 · 2281e6b · 2281e6b
2 parents 184ed8d + 9cf1f3c
commit 2281e6b
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/get-schema.ts b/src/get-schema.ts
@@ -5,7 +5,11 @@ import namify from 'namify';
 import { IOpts } from './types';
 import read from './read';
 
-export default function getSchema(dir: string, options: IOpts = {}) {
+export default function getSchema(
+  dir: string,
+  options: IOpts = {},
+  schemas: yaml.Schema[] = [yaml.DEFAULT_SAFE_SCHEMA]
+) {
   if (!options.hasOwnProperty('ext') || !Array.isArray(options.ext)) {
     options.ext = ['.yml', '.yaml'];
   }
@@ -110,6 +114,6 @@ export default function getSchema(dir: string, options: IOpts = {}) {
       }
     })
   ];
-  const schema = yaml.Schema.create(types);
+  const schema = yaml.Schema.create(schemas, types);
   return schema;
 }