// TODO: there's no indication of error in case worker.js is fails to dynamically import

import fs from 'fs';
import os from 'os';
import path from 'path';
import {$} from 'bun';

const randomDir = os.tmpdir();

const wasmDir = path.join(randomDir, 'whisper-web');

if (fs.existsSync(wasmDir)) {
	fs.rmSync(wasmDir, {recursive: true});
}

fs.mkdirSync(wasmDir, {recursive: true});

const cwd = path.join(wasmDir, 'build-em');

await $`git clone https://github.com/ggerganov/whisper.cpp ${wasmDir}`;
await $`git checkout v1.7.5`.cwd(wasmDir);

fs.mkdirSync(cwd);

const cmakeListsFile = path.join(
	wasmDir,
	'examples',
	'whisper.wasm',
	'CMakeLists.txt',
);

const file = fs.readFileSync(cmakeListsFile, 'utf8');

// Disable Node.JS target, compile with assertions to get stack trace
const lines = file.split('\n').map((line) => {
	if (line.includes('-s FORCE_FILESYSTEM=1 \\')) {
		// output ES6 module so we can import it dynamically without injeting the script tag
		return `-s FORCE_FILESYSTEM=1 -s ENVIRONMENT='web,worker' -s EXPORT_ES6=1 -s MODULARIZE=1 -s EXPORT_NAME=\\"createModule\\" \\`;
	}

	if (line.includes('-s EXPORTED_RUNTIME_METHODS')) {
		return `-s EXPORTED_RUNTIME_METHODS=\\"['print', 'printErr', 'ccall', 'cwrap', 'HEAPU8']\\" \\`;
	}

	return line;
});

fs.writeFileSync(
	cmakeListsFile,
	[
		...lines,
		// Generate a .d.ts file
		`set(CMAKE_CXX_FLAGS "$\{CMAKE_CXX_FLAGS\} -lembind --emit-tsd $\{CMAKE_CURRENT_BINARY_DIR\}/interface.d.ts")`,
	].join('\n'),
);

const emscriptenFilePath = path.join(
	wasmDir,
	'examples',
	'whisper.wasm',
	'emscripten.cpp',
);
//now get our version
const modifiedVersion = fs.readFileSync('./emscripten.cpp', 'utf8');
fs.writeFileSync(emscriptenFilePath, modifiedVersion);

// brew install emscripten if necessary
await $`emcmake cmake ..`.cwd(cwd);
await $`make -j`.cwd(cwd);

const mainJsFile = path.join(cwd, 'bin', 'whisper.wasm', 'main.js');

let content = fs
	.readFileSync(mainJsFile, 'utf8')
	.replace('libmain.js', './main.js');

// Write the modified content directly to the destination
fs.writeFileSync(path.join(__dirname, 'main.js'), content, 'utf8');

const dTsFile = path.join(
	wasmDir,
	'build-em',
	'examples',
	'whisper.wasm',
	'interface.d.ts',
);

fs.copyFileSync(dTsFile, path.join(__dirname, 'main.d.ts'));

fs.rmSync(wasmDir, {recursive: true});