forked from external-repos/squoosh
130 lines
4.0 KiB
JavaScript
130 lines
4.0 KiB
JavaScript
/**
|
|
* Copyright 2020 Google Inc. All Rights Reserved.
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
import { promisify } from 'util';
|
|
import * as path from 'path';
|
|
import { promises as fsp } from 'fs';
|
|
|
|
import glob from 'glob';
|
|
|
|
const globP = promisify(glob);
|
|
|
|
export default function () {
|
|
let previousWorkerContent;
|
|
|
|
return {
|
|
name: 'image-worker-plugin',
|
|
async buildStart() {
|
|
const featuresWorkerBase = path.join(
|
|
process.cwd(),
|
|
'src',
|
|
'features-worker',
|
|
);
|
|
const featuresWorkerBridgeBase = path.join(
|
|
process.cwd(),
|
|
'src',
|
|
'client',
|
|
'lazy-app',
|
|
'worker-bridge',
|
|
);
|
|
|
|
const tsImports = (
|
|
await globP('src/features/*/**/worker/*.ts', {
|
|
absolute: true,
|
|
})
|
|
)
|
|
.filter((tsFile) => !tsFile.endsWith('.d.ts'))
|
|
.map((tsFile) => tsFile.slice(0, -'.ts'.length));
|
|
|
|
const featuresWorkerTsNames = tsImports.map((tsImport) => [
|
|
path.relative(featuresWorkerBase, tsImport),
|
|
path.basename(tsImport),
|
|
]);
|
|
|
|
const featuresWorkerBridgeTsNames = tsImports.map((tsImport) => [
|
|
path.relative(featuresWorkerBridgeBase, tsImport),
|
|
path.basename(tsImport),
|
|
]);
|
|
|
|
const workerFile = [
|
|
`// This file is autogenerated by lib/image-worker-plugin.js`,
|
|
`import { expose } from 'comlink';`,
|
|
`import { timed } from './util';`,
|
|
featuresWorkerTsNames.map(
|
|
([path, name]) => `import ${name} from './${path}';`,
|
|
),
|
|
`const exports = {`,
|
|
featuresWorkerTsNames.map(([_, name]) => [
|
|
` ${name}(`,
|
|
` ...args: Parameters<typeof ${name}>`,
|
|
` ): ReturnType<typeof ${name}> {`,
|
|
` return timed('${name}', () => ${name}(...args));`,
|
|
` },`,
|
|
]),
|
|
`};`,
|
|
`export type ProcessorWorkerApi = typeof exports;`,
|
|
`expose(exports, self);`,
|
|
]
|
|
.flat(Infinity)
|
|
.join('\n');
|
|
|
|
// If nothing's changed, avoid touching the file to avoid infinite rebuilding in watch mode
|
|
if (previousWorkerContent === workerFile) return;
|
|
previousWorkerContent = workerFile;
|
|
|
|
const workerTsConfig = {
|
|
extends: '../../generic-tsconfig.json',
|
|
compilerOptions: {
|
|
lib: ['webworker', 'esnext'],
|
|
},
|
|
references: featuresWorkerTsNames.map(([tsImport]) => ({
|
|
path: path.dirname(tsImport),
|
|
})),
|
|
};
|
|
|
|
const bridgeMeta = [
|
|
`// This file is autogenerated by lib/image-worker-plugin.js`,
|
|
featuresWorkerBridgeTsNames.map(
|
|
([path, name]) => `import type ${name} from '${path}';`,
|
|
),
|
|
`export const methodNames = ${JSON.stringify(
|
|
featuresWorkerBridgeTsNames.map(([_, name]) => name),
|
|
null,
|
|
' ',
|
|
)} as const;`,
|
|
`export interface BridgeMethods {`,
|
|
featuresWorkerBridgeTsNames.map(([_, name]) => [
|
|
` ${name}(`,
|
|
` signal: AbortSignal,`,
|
|
` ...args: Parameters<typeof ${name}>`,
|
|
` ): Promise<ReturnType<typeof ${name}>>;`,
|
|
]),
|
|
`}`,
|
|
]
|
|
.flat(Infinity)
|
|
.join('\n');
|
|
|
|
await Promise.all([
|
|
fsp.writeFile(
|
|
path.join(featuresWorkerBase, 'tsconfig.json'),
|
|
JSON.stringify(workerTsConfig, null, ' '),
|
|
),
|
|
fsp.writeFile(path.join(featuresWorkerBase, 'index.ts'), workerFile),
|
|
fsp.writeFile(
|
|
path.join(featuresWorkerBridgeBase, 'meta.ts'),
|
|
bridgeMeta,
|
|
),
|
|
]);
|
|
},
|
|
};
|
|
}
|