feat(extension): metadata autocompletion

extensions/vscode/package.json

@@ -102,29 +102,53 @@
           "when": "view == tutorialkit-lessons-tree && viewItem == part"
         }
       ]
-    }
+    },
+    "languages": [
+      {
+        "id": "markdown",
+        "extensions": [
+          ".md"
+        ]
+      },
+      {
+        "id": "mdx",
+        "extensions": [
+          ".mdx"
+        ]
+      }
+    ]
   },
   "scripts": {
     "__esbuild-base": "esbuild ./src/extension.ts --bundle --outfile=dist/extension.js --external:vscode --format=cjs --platform=node",
     "__dev": "pnpm run esbuild-base -- --sourcemap --watch",
     "__vscode:prepublish": "pnpm run esbuild-base -- --minify",
     "__build": "vsce package",
-    "dev": "node build.mjs --watch",
-    "build": "pnpm run check-types && node build.mjs",
+    "dev": "node scripts/build.mjs --watch",
+    "build": "pnpm run check-types && node scripts/build.mjs",
     "check-types": "tsc --noEmit",
     "vscode:prepublish": "pnpm run package",
-    "package": "pnpm run check-types && node build.mjs --production"
+    "package": "pnpm run check-types && node scripts/build.mjs --production"
   },
   "dependencies": {
+    "@volar/language-core": "2.3.4",
+    "@volar/language-server": "2.3.4",
+    "@volar/language-service": "2.3.4",
+    "@volar/vscode": "2.3.4",
     "case-anything": "^3.1.0",
-    "gray-matter": "^4.0.3"
+    "gray-matter": "^4.0.3",
+    "volar-service-yaml": "volar-2.3",
+    "vscode-languageclient": "^9.0.1",
+    "vscode-uri": "^3.0.8",
+    "yaml-language-server": "1.15.0"
   },
   "devDependencies": {
-    "@types/mocha": "^10.0.6",
+    "@tutorialkit/types": "workspace:*",
     "@types/node": "20.14.11",
     "@types/vscode": "^1.80.0",
+    "chokidar": "3.6.0",
     "esbuild": "^0.21.5",
     "execa": "^9.2.0",
-    "typescript": "^5.4.5"
+    "typescript": "^5.4.5",
+    "zod-to-json-schema": "3.23.1"
   }
 }

extensions/vscode/scripts/build.mjs

@@ -0,0 +1,117 @@
+import { watch } from 'chokidar';
+import * as esbuild from 'esbuild';
+import { execa } from 'execa';
+import fs from 'node:fs';
+import { createRequire } from 'node:module';
+import { join, dirname } from 'path';
+import { Worker } from 'node:worker_threads';
+import { fileURLToPath } from 'node:url';
+
+const __dirname = dirname(fileURLToPath(import.meta.url));
+const require = createRequire(import.meta.url);
+const production = process.argv.includes('--production');
+const isWatch = process.argv.includes('--watch');
+
+async function main() {
+  const ctx = await esbuild.context({
+    entryPoints: {
+      extension: 'src/extension.ts',
+      server: './src/language-server/index.ts',
+    },
+    bundle: true,
+    format: 'cjs',
+    minify: production,
+    sourcemap: !production,
+    sourcesContent: false,
+    tsconfig: './tsconfig.json',
+    platform: 'node',
+    outdir: 'dist',
+    define: { 'process.env.NODE_ENV': production ? '"production"' : '"development"' },
+    external: ['vscode'],
+    plugins: [esbuildUMD2ESMPlugin],
+  });
+
+  if (isWatch) {
+    const buildMetadataSchemaDebounced = debounce(buildMetadataSchema, 100);
+    const dependencyPath = dirname(require.resolve('@tutorialkit/types'));
+
+    watch(dependencyPath).on('all', (eventName, path) => {
+      if (eventName !== 'change' && eventName !== 'add' && eventName !== 'unlink') {
+        return;
+      }
+
+      buildMetadataSchemaDebounced();
+    });
+
+    await Promise.all([
+      ctx.watch(),
+      execa('tsc', ['--noEmit', '--watch', '--preserveWatchOutput', '--project', 'tsconfig.json'], {
+        stdio: 'inherit',
+        preferLocal: true,
+      }),
+    ]);
+  } else {
+    await ctx.rebuild();
+    await ctx.dispose();
+
+    await buildMetadataSchema();
+
+    if (production) {
+      // rename name in package json to match extension name on store:
+      const pkgJSON = JSON.parse(fs.readFileSync('./package.json', { encoding: 'utf8' }));
+
+      pkgJSON.name = 'tutorialkit';
+
+      fs.writeFileSync('./package.json', JSON.stringify(pkgJSON, undefined, 2), 'utf8');
+    }
+  }
+}
+
+async function buildMetadataSchema() {
+  const schema = await new Promise((resolve) => {
+    const worker = new Worker(join(__dirname, './load-schema-worker.mjs'));
+    worker.on('message', (value) => resolve(value));
+  });
+
+  fs.mkdirSync('./dist', { recursive: true });
+  fs.writeFileSync('./dist/schema.json', JSON.stringify(schema, undefined, 2), 'utf-8');
+
+  console.log('Updated schema.json');
+}
+
+/**
+ * @type {import('esbuild').Plugin}
+ */
+const esbuildUMD2ESMPlugin = {
+  name: 'umd2esm',
+  setup(build) {
+    build.onResolve({ filter: /^(vscode-.*-languageservice|jsonc-parser)/ }, (args) => {
+      const pathUmdMay = require.resolve(args.path, { paths: [args.resolveDir] });
+      const pathEsm = pathUmdMay.replace('/umd/', '/esm/').replace('\\umd\\', '\\esm\\');
+
+      return { path: pathEsm };
+    });
+  },
+};
+
+main().catch((error) => {
+  console.error(error);
+  process.exit(1);
+});
+
+/**
+ * Debounce the provided function.
+ *
+ * @param {Function} fn Function to debounce
+ * @param {number} duration Duration of the debounce
+ * @returns {Function} Debounced function
+ */
+function debounce(fn, duration) {
+  let timeoutId = 0;
+
+  return function () {
+    clearTimeout(timeoutId);
+
+    timeoutId = setTimeout(fn.bind(this), duration, ...arguments);
+  };
+}

extensions/vscode/scripts/load-schema-worker.mjs

@@ -0,0 +1,7 @@
+import { parentPort } from 'node:worker_threads';
+import { zodToJsonSchema } from 'zod-to-json-schema';
+import { chapterSchema, lessonSchema, partSchema, tutorialSchema } from '@tutorialkit/types';
+
+const schema = tutorialSchema.strict().or(partSchema.strict()).or(chapterSchema.strict()).or(lessonSchema.strict());
+
+parentPort.postMessage(zodToJsonSchema(schema));

extensions/vscode/src/extension.ts

@@ -1,16 +1,49 @@
+import * as serverProtocol from '@volar/language-server/protocol';
+import { createLabsInfo } from '@volar/vscode';
 import * as vscode from 'vscode';
 import { useCommands } from './commands';
 import { useLessonTree } from './views/lessonsTree';
+import * as lsp from 'vscode-languageclient/node';
 
 export let extContext: vscode.ExtensionContext;
 
-export function activate(context: vscode.ExtensionContext) {
+let client: lsp.BaseLanguageClient;
+
+export async function activate(context: vscode.ExtensionContext) {
   extContext = context;
 
   useCommands();
   useLessonTree();
+
+  const serverModule = vscode.Uri.joinPath(context.extensionUri, 'dist', 'server.js');
+  const runOptions = { execArgv: <string[]>[] };
+  const debugOptions = { execArgv: ['--nolazy', '--inspect=' + 6009] };
+  const serverOptions: lsp.ServerOptions = {
+    run: {
+      module: serverModule.fsPath,
+      transport: lsp.TransportKind.ipc,
+      options: runOptions,
+    },
+    debug: {
+      module: serverModule.fsPath,
+      transport: lsp.TransportKind.ipc,
+      options: debugOptions,
+    },
+  };
+  const clientOptions: lsp.LanguageClientOptions = {
+    documentSelector: [{ language: 'markdown' }, { language: 'mdx' }],
+    initializationOptions: {},
+  };
+  client = new lsp.LanguageClient('tutorialkit-language-server', 'TutorialKit', serverOptions, clientOptions);
+
+  await client.start();
+
+  const labsInfo = createLabsInfo(serverProtocol);
+  labsInfo.addLanguageClient(client);
+
+  return labsInfo.extensionExports;
 }
 
 export function deactivate() {
-  // do nothing
+  return client?.stop();
 }

extensions/vscode/src/language-server/index.ts

@@ -0,0 +1,54 @@
+import { createConnection, createServer, createSimpleProject } from '@volar/language-server/node';
+import { create as createYamlService } from 'volar-service-yaml';
+import { SchemaPriority } from 'yaml-language-server';
+import { frontmatterPlugin } from './languagePlugin';
+import { readSchema } from './schema';
+
+const connection = createConnection();
+const server = createServer(connection);
+
+connection.listen();
+
+connection.onInitialize((params) => {
+  const yamlService = createYamlService({
+    getLanguageSettings(_context) {
+      const schema = readSchema();
+
+      return {
+        completion: true,
+        validate: true,
+        hover: true,
+        format: true,
+        yamlVersion: '1.2',
+        isKubernetes: false,
+        schemas: [
+          {
+            uri: 'https://tutorialkit.dev/reference/configuration/',
+            schema,
+            fileMatch: [
+              '**/*',
+
+              // TODO: those don't work
+              'src/content/*.md',
+              'src/content/**/*.md',
+              'src/content/**/*.mdx',
+            ],
+            priority: SchemaPriority.Settings,
+          },
+        ],
+      };
+    },
+  });
+
+  delete yamlService.capabilities.codeLensProvider;
+
+  return server.initialize(
+    params,
+    createSimpleProject([frontmatterPlugin(connection.console.debug.bind(connection.console.debug))]),
+    [yamlService],
+  );
+});
+
+connection.onInitialized(server.initialized);
+
+connection.onShutdown(server.shutdown);