feat: add a way to add custom json schemas

Author: Loïc Mangeonjean

src/features/jsonContribution.ts

@@ -1,8 +1,13 @@
 import 'monaco-editor/esm/vs/language/json/monaco.contribution'
 import * as monaco from 'monaco-editor'
 import { getJsonSchemas, onDidChangeJsonSchema } from 'vscode/monaco'
+import { Disposable } from 'vscode'
 import { registerWorkerLoader } from '../worker'
 
+type Unpacked<T> = T extends (infer U)[] ? U : T
+type Schema = Unpacked<NonNullable<monaco.languages.json.DiagnosticsOptions['schemas']>>
+
+const customSchemas: Schema[] = []
 function updateDiagnosticsOptions () {
   monaco.languages.json.jsonDefaults.setDiagnosticsOptions({
     comments: 'ignore',
@@ -16,7 +21,8 @@ function updateDiagnosticsOptions () {
       {
         uri: 'https://json-schema.org/draft/2019-09/schema',
         fileMatch: ['*.schema.json']
-      }
+      },
+      ...customSchemas
     ]
   })
 }
@@ -26,3 +32,15 @@ onDidChangeJsonSchema(updateDiagnosticsOptions)
 
 const workerLoader = async () => (await import(/* webpackChunkName: "MonacoJsonWorker" */'monaco-editor/esm/vs/language/json/json.worker?worker')).default
 registerWorkerLoader('json', workerLoader)
+
+export function addJsonSchema (schema: Unpacked<NonNullable<monaco.languages.json.DiagnosticsOptions['schemas']>>): Disposable {
+  customSchemas.push(schema)
+  updateDiagnosticsOptions()
+  return new Disposable(() => {
+    const index = customSchemas.indexOf(schema)
+    if (index >= 0) {
+      customSchemas.splice(index, 1)
+      updateDiagnosticsOptions()
+    }
+  })
+}