obsidian-typst/compiler.worker.ts

//@ts-ignore
import wasmBin from './pkg/obsidian_typst_bg.wasm'
import * as typst from './pkg'

import { CompileCommand, WorkerRequest } from "types";

typst.initSync(wasmBin);

let canUseSharedArrayBuffer = new Boolean(false);

// let buffer: Int32Array;
let decoder = new TextDecoder()

function requestData(path: string): string {
    if (!canUseSharedArrayBuffer) {
        throw "Cannot read files on mobile"
    }
    // @ts-expect-error
    let buffer = new Int32Array(new SharedArrayBuffer(4, { maxByteLength: 1e8 }))
    buffer[0] = 0;
    postMessage({ buffer, path })
    const res = Atomics.wait(buffer, 0, 0);
    if (buffer[0] == 0) {
        return decoder.decode(Uint8Array.from(buffer.slice(1)))
    }

    throw buffer[0]
}

let compiler = new typst.SystemWorld("", requestData)
const compileAttempts = 5;

// function compile(data: CompileCommand) {
//     for (let i = 1; i <= compileAttempts; i += 1) {
//         try {
//             return compiler.compile(data.source, data.path, data.pixel_per_pt, data.fill, data.size, data.display)
//         } catch (error) {
//             if (i < compileAttempts && ((error.name == "RuntimeError" && error.message == "unreachable") || (error.name == "Uncaught Error"))) {
//                 console.warn("Typst compiler crashed, attempting to restart: ", i);
//                 console.log(data);

//                 compiler.free()
//                 compiler = new typst.SystemWorld("", requestData)
//             } else {
//                 console.log("name", error.name);
//                 console.log("message", error.message);

//                 throw error;
//             }
//         }
//     }
// }


onmessage = (ev: MessageEvent<CompileCommand | true>) => {
    if (ev.data == true) {
        canUseSharedArrayBuffer = ev.data
    } else if ("source" in ev.data) {
        const data: CompileCommand = ev.data;
        postMessage(compiler.compile(data.source, data.path, data.pixel_per_pt, data.fill, data.size, data.display))

        // postMessage(compile(ev.data))
    } else {
        throw ev;
    }
}

console.log("Typst compiler worker loaded!");