feature: prepare to resolve json schema (#77)

Author: levivilet

packages/json-worker/src/parts/JsonCompletion/JsonCompletion.ts

@@ -17,6 +17,7 @@ export const jsonCompletion = async (
     return []
   }
   const { node, schema } = parsed
+
   if (node.type === TokenType.String) {
     const options = schema?.properties?.type?.enum || []
     return options.map(EnumToCompletionOption.enumToCompletionOption)

packages/json-worker/src/parts/JsonCompletionProperty/JsonCompletionProperty.ts

@@ -1,11 +1,45 @@
 import { AstNode } from '../AstNode/AstNode.ts'
 import type { CompletionItem } from '../CompletionItem/CompletionItem.ts'
 import * as PropertyKeyToCompletionOption from '../PropertyKeyToCompletionOption/PropertyKeyToCompletionOption.ts'
+import * as ResolveSchemaRef from '../ResolveSchemaRef/ResolveSchemaRef.ts'
+import type { JsonSchema } from '../JsonSchema/JsonSchema.ts'
+
+const getSchemaProperties = (schema: JsonSchema): JsonSchema['properties'] => {
+  if (schema.properties) {
+    return schema.properties
+  }
+
+  if (schema.$ref) {
+    const resolved = ResolveSchemaRef.resolveSchemaRef(schema, schema.$ref)
+    return resolved.properties || {}
+  }
+
+  if (schema.allOf) {
+    const properties: { [key: string]: JsonSchema } = {}
+    for (const subSchema of schema.allOf) {
+      const subProperties = getSchemaProperties(subSchema)
+      Object.assign(properties, subProperties)
+    }
+    return properties
+  }
+
+  if (schema.anyOf) {
+    const properties: { [key: string]: JsonSchema } = {}
+    for (const subSchema of schema.anyOf) {
+      const subProperties = getSchemaProperties(subSchema)
+      Object.assign(properties, subProperties)
+    }
+    return properties
+  }
+
+  return {}
+}
 
 export const jsonCompletionProperty = (
-  schema: any,
+  schema: JsonSchema,
   node: AstNode,
 ): readonly CompletionItem[] => {
-  const keys = Object.keys(schema?.properties || {})
+  const properties = getSchemaProperties(schema)
+  const keys = Object.keys(properties || {})
   return keys.map(PropertyKeyToCompletionOption.propertyKeyToCompletionOption)
 }

packages/json-worker/src/parts/JsonSchema/JsonSchema.ts

@@ -0,0 +1,14 @@
+export interface JsonSchema {
+  readonly $ref?: string
+  readonly properties?: {
+    readonly [key: string]: JsonSchema
+  }
+  readonly definitions?: {
+    readonly [key: string]: JsonSchema
+  }
+  readonly allOf?: readonly JsonSchema[]
+  readonly anyOf?: readonly JsonSchema[]
+  readonly description?: string
+  readonly type?: string
+  readonly enum?: readonly string[]
+}

packages/json-worker/src/parts/ResolveSchema/ResolveSchema.ts

@@ -0,0 +1,3 @@
+export const resolveSchema = (schema: any): any => {
+  return schema
+}

packages/json-worker/src/parts/ResolveSchemaRef/ResolveSchemaRef.ts

@@ -0,0 +1,18 @@
+import type { JsonSchema } from '../JsonSchema/JsonSchema.ts'
+
+const definitionPrefix = '#/definitions/'
+
+export const resolveSchemaRef = (
+  schema: JsonSchema,
+  ref: string,
+): JsonSchema => {
+  if (!ref.startsWith(definitionPrefix)) {
+    return {}
+  }
+  const path = ref.slice(definitionPrefix.length)
+  const resolved = schema.definitions?.[path]
+  if (!resolved) {
+    return {}
+  }
+  return resolved
+}

packages/json-worker/test/JsonCompletionProperty.test.ts

@@ -0,0 +1,41 @@
+import { jest, test, expect } from '@jest/globals'
+import * as JsonCompletionProperty from '../src/parts/JsonCompletionProperty/JsonCompletionProperty.ts'
+
+test.skip('handles schema references', () => {
+  const schema = {
+    allOf: [
+      {
+        $ref: '#/definitions/compilerOptionsDefinition',
+      },
+    ],
+    definitions: {
+      compilerOptionsDefinition: {
+        properties: {
+          compilerOptions: {
+            type: 'object',
+            properties: {
+              target: {
+                type: 'string',
+              },
+            },
+          },
+        },
+      },
+    },
+  }
+
+  const node = {
+    type: 0,
+    offset: 0,
+    length: 0,
+    childCount: 0,
+  }
+
+  const result = JsonCompletionProperty.jsonCompletionProperty(schema, node)
+  expect(result).toEqual([
+    {
+      kind: 10,
+      label: 'compilerOptions',
+    },
+  ])
+})