Support inline $id references (#411)

* support inline  references

* lint

Author: sinclairzx81

index.d.ts

@@ -1,6 +1,10 @@
 import { Options as AjvOptions } from "ajv"
 declare namespace build {
   interface BaseSchema {
+    /**
+     * Schema id
+     */
+    $id?: string
     /**
      * Schema title
      */

index.js

@@ -52,11 +52,14 @@ function mergeLocation (source, dest) {
 
 const arrayItemsReferenceSerializersMap = new Map()
 const objectReferenceSerializersMap = new Map()
+const schemaReferenceMap = new Map()
+
 let ajvInstance = null
 
 function build (schema, options) {
   arrayItemsReferenceSerializersMap.clear()
   objectReferenceSerializersMap.clear()
+  schemaReferenceMap.clear()
 
   options = options || {}
 
@@ -174,6 +177,7 @@ function build (schema, options) {
 
   arrayItemsReferenceSerializersMap.clear()
   objectReferenceSerializersMap.clear()
+  schemaReferenceMap.clear()
 
   return (Function.apply(null, dependenciesName).apply(null, dependencies))
 }
@@ -612,8 +616,18 @@ function refFinder (ref, location) {
   // Split file from walk
   ref = ref.split('#')
 
-  // If external file
-  if (ref[0]) {
+  // Check schemaReferenceMap for $id entry
+  if (ref[0] && schemaReferenceMap.has(ref[0])) {
+    schema = schemaReferenceMap.get(ref[0])
+    root = schemaReferenceMap.get(ref[0])
+    if (schema.$ref) {
+      return refFinder(schema.$ref, {
+        schema: schema,
+        root: root,
+        externalSchema: externalSchema
+      })
+    }
+  } else if (ref[0]) { // If external file
     schema = externalSchema[ref[0]]
     root = externalSchema[ref[0]]
 
@@ -921,7 +935,9 @@ function toJSON (variableName) {
 
 function buildObject (location, code, name) {
   const schema = location.schema
-
+  if (schema.$id !== undefined) {
+    schemaReferenceMap.set(schema.$id, schema)
+  }
   code += `
     function ${name} (input) {
   `
@@ -971,6 +987,9 @@ function buildObject (location, code, name) {
 
 function buildArray (location, code, name, key = null) {
   let schema = location.schema
+  if (schema.$id !== undefined) {
+    schemaReferenceMap.set(schema.$id, schema)
+  }
   code += `
     function ${name} (obj) {
   `

test/recursion.test.js

@@ -178,3 +178,68 @@ test('can stringify recursive references in object types (issue #365)', t => {
   const value = stringify(data)
   t.equal(value, '{"category":{"parent":{"parent":{"parent":{"parent":{}}}}}}')
 })
+
+test('can stringify recursive inline $id references (issue #410)', t => {
+  t.plan(1)
+  const schema = {
+    $id: 'Node',
+    type: 'object',
+    properties: {
+      id: {
+        type: 'string'
+      },
+      nodes: {
+        type: 'array',
+        items: {
+          $ref: 'Node'
+        }
+      }
+    },
+    required: [
+      'id',
+      'nodes'
+    ]
+  }
+
+  const stringify = build(schema)
+  const data = {
+    id: '0',
+    nodes: [
+      {
+        id: '1',
+        nodes: [{
+          id: '2',
+          nodes: [
+            { id: '3', nodes: [] },
+            { id: '4', nodes: [] },
+            { id: '5', nodes: [] }
+          ]
+        }]
+      },
+      {
+        id: '6',
+        nodes: [{
+          id: '7',
+          nodes: [
+            { id: '8', nodes: [] },
+            { id: '9', nodes: [] },
+            { id: '10', nodes: [] }
+          ]
+        }]
+      },
+      {
+        id: '11',
+        nodes: [{
+          id: '12',
+          nodes: [
+            { id: '13', nodes: [] },
+            { id: '14', nodes: [] },
+            { id: '15', nodes: [] }
+          ]
+        }]
+      }
+    ]
+  }
+  const value = stringify(data)
+  t.equal(value, '{"id":"0","nodes":[{"id":"1","nodes":[{"id":"2","nodes":[{"id":"3","nodes":[]},{"id":"4","nodes":[]},{"id":"5","nodes":[]}]}]},{"id":"6","nodes":[{"id":"7","nodes":[{"id":"8","nodes":[]},{"id":"9","nodes":[]},{"id":"10","nodes":[]}]}]},{"id":"11","nodes":[{"id":"12","nodes":[{"id":"13","nodes":[]},{"id":"14","nodes":[]},{"id":"15","nodes":[]}]}]}]}')
+})