pages235/scripts/esbuildPlugins.mjs
Vitaly 976f6cab2b add mesher HMR!
fix: fix all colored glass rendering!
2024-04-20 15:28:31 +03:00

342 lines
11 KiB
JavaScript

//@ts-check
import { polyfillNode } from 'esbuild-plugin-polyfill-node'
import { join, dirname, basename } from 'path'
import * as fs from 'fs'
import { filesize } from 'filesize'
import MCProtocol from 'minecraft-protocol'
import MCData from 'minecraft-data'
import { throttle } from 'lodash-es'
const { supportedVersions } = MCProtocol
const prod = process.argv.includes('--prod')
let connectedClients = []
const writeToClients = (data) => {
connectedClients.forEach((res) => {
res.write(`data: ${JSON.stringify(data)}\n\n`)
res.flush()
})
}
export const startWatchingHmr = () => {
const eventsPerFile = {
'mesher.js': 'mesher',
// 'dist/webglRendererWorker.js': 'webglRendererWorker',
}
for (const name of Object.keys(eventsPerFile)) {
const file = join('dist', name);
if (!fs.existsSync(file)) console.warn(`[missing worker] File ${name} does not exist`)
fs.watchFile(file, () => {
writeToClients({ replace: { type: eventsPerFile[name] } })
})
}
}
/** @type {import('esbuild').Plugin[]} */
const plugins = [
{
name: 'strict-aliases',
setup (build) {
build.onResolve({
filter: /^minecraft-protocol$/,
}, async ({ kind, resolveDir }) => {
return {
path: (await build.resolve('minecraft-protocol/src/index.js', { kind, resolveDir })).path,
}
})
build.onLoad({
filter: /minecraft-data[\/\\]data.js$/,
}, (args) => {
const version = supportedVersions.at(-1)
if (!version) throw new Error('unreachable')
const data = MCData(version)
const defaultVersionsObj = {
// default protocol data, needed for auto-version
[version]: {
version: data.version,
// protocol: JSON.parse(fs.readFileSync(join(args.path, '..', 'minecraft-data/data/pc/1.20/protocol.json'), 'utf8')),
protocol: data.protocol,
}
}
return {
contents: `window.mcData ??= ${JSON.stringify(defaultVersionsObj)};module.exports = { pc: window.mcData }`,
loader: 'js',
}
})
build.onResolve({
filter: /^minecraft-assets$/,
}, () => {
throw new Error('hit banned package')
})
build.onLoad({
filter: /^prismarine-auth/,
}, () => {
return {
contents: 'module.exports = {}',
}
})
build.onResolve({
filter: /^three$/,
}, async ({ kind, resolveDir }) => {
return {
path: (await build.resolve('three/src/Three.js', { kind, resolveDir })).path,
}
})
}
},
{
name: 'data-assets',
setup (build) {
build.onResolve({
filter: /.*/,
}, async ({ path, ...rest }) => {
if (['.woff', '.woff2', '.ttf'].some(ext => path.endsWith(ext)) || path.startsWith('extra-textures/')) {
return {
path,
namespace: 'assets',
external: true,
}
}
})
const removeNodeModulesSourcemaps = (map) => {
const doNotRemove = ['prismarine', 'mineflayer', 'flying-squid', '@jspm/core', 'minecraft']
map.sourcesContent.forEach((_, i) => {
if (map.sources[i].includes('node_modules') && !doNotRemove.some(x => map.sources[i].includes(x))) {
map.sourcesContent[i] = null
}
})
}
build.onEnd(async ({ metafile, outputFiles }) => {
// write outputFiles
//@ts-ignore
for (const file of outputFiles) {
let contents = file.text
if (file.path.endsWith('.map') && file.text) {
const map = JSON.parse(file.text)
removeNodeModulesSourcemaps(map)
contents = JSON.stringify(map)
}
await fs.promises.writeFile(file.path, contents)
}
if (!prod) return
// const deps = Object.entries(metafile.inputs).sort(([, a], [, b]) => b.bytes - a.bytes).map(([x, { bytes }]) => [x, filesize(bytes)]).slice(0, 5)
//@ts-ignore
const sizeByExt = {}
//@ts-ignore
Object.entries(metafile.inputs).sort(([, a], [, b]) => b.bytes - a.bytes).forEach(([x, { bytes }]) => {
const ext = x.slice(x.lastIndexOf('.'))
sizeByExt[ext] ??= 0
sizeByExt[ext] += bytes
})
console.log('Input size by ext:')
console.log(Object.fromEntries(Object.entries(sizeByExt).map(x => [x[0], filesize(x[1])])))
})
},
},
{
name: 'prevent-incorrect-linking',
setup (build) {
build.onResolve({
filter: /.+/,
}, async ({ resolveDir, path, importer, kind, pluginData }) => {
if (pluginData?.__internal) return
// not ideal as packages can have different version, on the other hand we should not have multiple versions of the same package of developing deps
const packageName = path.startsWith('@') ? path.split('/', 2).join('/') : path.split('/', 1)[0]
const localPackageToReuse = join('node_modules', packageName)
if (!resolveDir.startsWith(process.cwd()) && ['./', '../'].every(x => !path.startsWith(x)) && fs.existsSync(localPackageToReuse)) {
const redirected = await build.resolve(path, { kind: 'import-statement', resolveDir: process.cwd(), pluginData: { __internal: true }, })
return redirected
}
// disallow imports from outside the root directory to ensure modules are resolved from node_modules of this workspace
// if ([resolveDir, path].some(x => x.includes('node_modules')) && !resolveDir.startsWith(process.cwd())) {
// // why? ensure workspace dependency versions are used (we have overrides and need to dedupe so it doesn't grow in size)
// throw new Error(`Restricted package import from outside the root directory: ${resolveDir}`)
// }
return undefined
})
}
},
{
name: 'watch-notify',
setup (build) {
let count = 0
let time
let prevHash
build.onStart(() => {
time = Date.now()
})
build.onEnd(({ errors, outputFiles: _outputFiles, metafile, warnings }) => {
/** @type {import('esbuild').OutputFile[]} */
// @ts-ignore
const outputFiles = _outputFiles
const elapsed = Date.now() - time
outputFiles.find(outputFile => outputFile.path)
if (errors.length) {
writeToClients({ errors: errors.map(error => error.text) })
return
}
// write metafile to disk if needed to analyze
fs.writeFileSync('dist/meta.json', JSON.stringify(metafile, null, 2))
/** @type {import('esbuild').OutputFile} */
//@ts-ignore
const outputFile = outputFiles.find(x => x.path.endsWith('.js'))
if (outputFile.hash === prevHash) {
// todo also check workers and css
console.log('Ignoring reload as contents the same')
return
}
prevHash = outputFile.hash
let outputText = outputFile.text
//@ts-ignore
if (['inline', 'both'].includes(build.initialOptions.sourcemap)) {
outputText = outputText.slice(0, outputText.indexOf('//# sourceMappingURL=data:application/json;base64,'))
}
console.log(`Done in ${elapsed}ms. Size: ${filesize(outputText.length)} (${build.initialOptions.minify ? 'minified' : 'without minify'})`)
if (count++ === 0) {
return
}
writeToClients({ update: { time: elapsed } })
connectedClients.length = 0
})
}
},
{
name: 'esbuild-readdir',
setup (build) {
build.onResolve({
filter: /^esbuild-readdir:.+$/,
}, ({ resolveDir, path }) => {
return {
namespace: 'esbuild-readdir',
path,
pluginData: {
resolveDir: join(resolveDir, path.replace(/^esbuild-readdir:/, ''))
},
}
})
build.onLoad({
filter: /.+/,
namespace: 'esbuild-readdir',
}, async ({ pluginData }) => {
const { resolveDir } = pluginData
const files = await fs.promises.readdir(resolveDir)
return {
contents: `module.exports = ${JSON.stringify(files)}`,
resolveDir,
loader: 'js',
}
})
}
},
{
name: 'esbuild-import-glob',
setup (build) {
build.onResolve({
filter: /^esbuild-import-glob\(path:(.+),skipFiles:(.+)\)+$/,
}, ({ resolveDir, path }) => {
return {
namespace: 'esbuild-import-glob',
path,
pluginData: {
resolveDir
},
}
})
build.onLoad({
filter: /.+/,
namespace: 'esbuild-import-glob',
}, async ({ pluginData, path }) => {
const { resolveDir } = pluginData
//@ts-ignore
const [, userPath, skipFiles] = /^esbuild-import-glob\(path:(.+),skipFiles:(.+)\)+$/g.exec(path)
const files = (await fs.promises.readdir(join(resolveDir, userPath))).filter(f => !skipFiles.includes(f))
return {
contents: `module.exports = { ${files.map(f => `'${f}': require('./${join(userPath, f)}')`).join(',')} }`,
resolveDir,
loader: 'js',
}
})
}
},
{
name: 'fix-dynamic-require',
setup (build) {
build.onResolve({
filter: /1\.14\/chunk/,
}, async ({ resolveDir, path }) => {
if (!resolveDir.includes('prismarine-provider-anvil')) return
return {
namespace: 'fix-dynamic-require',
path,
pluginData: {
resolvedPath: `${join(resolveDir, path)}.js`,
resolveDir
},
}
})
build.onLoad({
filter: /.+/,
namespace: 'fix-dynamic-require',
}, async ({ pluginData: { resolvedPath, resolveDir } }) => {
const resolvedFile = await fs.promises.readFile(resolvedPath, 'utf8')
return {
contents: resolvedFile.replace("require(`prismarine-chunk/src/pc/common/BitArray${noSpan ? 'NoSpan' : ''}`)", "noSpan ? require(`prismarine-chunk/src/pc/common/BitArray`) : require(`prismarine-chunk/src/pc/common/BitArrayNoSpan`)"),
resolveDir,
loader: 'js',
}
})
}
},
{
name: 'react-displayname',
setup (build) {
build.onLoad({
filter: /.tsx$/,
}, async ({ path }) => {
let file = await fs.promises.readFile(path, 'utf8')
const fileName = basename(path, '.tsx')
let replaced = false
const varName = `__${fileName}_COMPONENT`
file = file.replace(/export default /, () => {
replaced = true
return `const ${varName} = `
})
if (replaced) {
file += `;${varName}.displayName = '${fileName}';export default ${varName};`
}
return {
contents: file,
loader: 'tsx',
}
})
}
},
polyfillNode({
polyfills: {
fs: false,
dns: false,
crypto: false,
events: false,
http: false,
stream: false,
buffer: false,
perf_hooks: false,
net: false,
assert: false,
},
})
]
export { plugins, connectedClients as clients }