56 lines
1.5 KiB
JavaScript
56 lines
1.5 KiB
JavaScript
|
/**
|
||
|
|
* Scripts to be done before starting the server for the first time.
|
||
|
|
*/
|
||
|
|
const fs = require('fs');
|
||
|
|
const crypto = require('crypto');
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Returns the MD5 hash of the given data.
|
||
|
|
* @param {Buffer} data Input data
|
||
|
|
* @returns {string} MD5 hash of the input data
|
||
|
|
*/
|
||
|
|
function getMd5Hash(data) {
|
||
|
|
return crypto
|
||
|
|
.createHash('md5')
|
||
|
|
.update(data)
|
||
|
|
.digest('hex');
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Copies the WASM binaries from the sillytavern-transformers package to the dist folder.
|
||
|
|
*/
|
||
|
|
function copyWasmFiles() {
|
||
|
|
if (!fs.existsSync('./dist')) {
|
||
|
|
fs.mkdirSync('./dist');
|
||
|
|
}
|
||
|
|
|
||
|
|
const listDir = fs.readdirSync('./node_modules/sillytavern-transformers/dist');
|
||
|
|
|
||
|
|
for (const file of listDir) {
|
||
|
|
if (file.endsWith('.wasm')) {
|
||
|
|
const sourcePath = `./node_modules/sillytavern-transformers/dist/${file}`;
|
||
|
|
const targetPath = `./dist/${file}`;
|
||
|
|
|
||
|
|
// Don't copy if the file already exists and is the same checksum
|
||
|
|
if (fs.existsSync(targetPath)) {
|
||
|
|
const sourceChecksum = getMd5Hash(fs.readFileSync(sourcePath));
|
||
|
|
const targetChecksum = getMd5Hash(fs.readFileSync(targetPath));
|
||
|
|
|
||
|
|
if (sourceChecksum === targetChecksum) {
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
fs.copyFileSync(sourcePath, targetPath);
|
||
|
|
console.log(`${file} successfully copied to ./dist/${file}`);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
try {
|
||
|
|
// 1. Copy transformers WASM binaries from node_modules
|
||
|
|
copyWasmFiles();
|
||
|
|
} catch (error) {
|
||
|
|
console.error(error);
|
||
|
|
}
|