main.js

const glslx = require('glslx');

module.exports = (options = {}) => ({
  name: 'glslx',
  setup(build) {
    const writeTypeDeclarations = !!(options && options.writeTypeDeclarations);
    const path = require('path');
    const fs = require('fs');

    build.onLoad({ filter: /\.glslx$/ }, async (args) => {
      const contents = await fs.promises.readFile(args.path, 'utf8');
      const input = [{ name: args.path, contents }];
      const errors = [];
      const warnings = [];
      const watchFiles = [];
      const cache = Object.create(null);
      cache[args.path] = contents;

      const fileAccess = (filePath, relativeTo) => {
        const name = path.join(path.dirname(relativeTo), filePath);
        let contents = cache[name];
        if (contents === undefined) {
          watchFiles.push(name);
          try {
            contents = fs.readFileSync(name, 'utf8');
          } catch {
            return null;
          }
          cache[name] = contents;
        }
        return { name, contents };
      };

      for (const { kind, text, range } of glslx.compileIDE(input, { fileAccess }).diagnostics) {
        const message = { text };

        if (range) {
          const lineText = cache[range.source].split(/\r|\n|\r\n/g)[range.start.line];
          message.location = {
            file: range.source,
            line: range.start.line + 1,
            column: range.start.column,
            length: range.end.line === range.start.line ? range.end.column - range.start.column : 0,
            lineText,
          };
        }

        if (kind === 'error') errors.push(message);
        if (kind === 'warning') warnings.push(message);
      }

      if (errors.length > 0) return { errors, warnings, watchFiles };

      const json = JSON.parse(glslx.compile(input, {
        format: 'json',
        fileAccess,
        prettyPrint: true,
        disableRewriting: true,
        renaming: 'none',
      }).output);

      let js = '';
      let ts = '';
      for (const shader of json.shaders) {
        js += `export var ${shader.name} = ${JSON.stringify(shader.contents)};\n`;
        ts += `export var ${shader.name}: string;\n`;
      }
      if (writeTypeDeclarations) {
        await fs.promises.writeFile(args.path + '.d.ts', ts);
      }
      return { contents: js, warnings, watchFiles };
    });
  },
});