import { UserConfigSettings, runTwoSlash } from "shiki-twoslash"

import type { TwoSlashReturn } from "@typescript/twoslash"

/**

* Keeps a cache of the JSON responses to a twoslash call in node_modules/.cache/twoslash

* which should keep CI times down (e.g. the epub vs the handbook etc) - but also during

* dev time, where it can be super useful.

*/

export const cachedTwoslashCall = (

code: string,

lang: string,

settings: UserConfigSettings

): TwoSlashReturn | undefined => {

// @ts-expect-error

const isWebWorker = typeof self !== "undefined" && typeof self.WorkerGlobalScope !== "undefined"

const isBrowser =

isWebWorker ||

(typeof window !== "undefined" && typeof window.document !== "undefined" && typeof fetch !== "undefined")

if (isBrowser) {

// Not in Node, run un-cached

return runTwoSlash(code, lang, settings)

}

const { createHash } = require("crypto")

const { readFileSync, existsSync, mkdirSync, writeFileSync } = require("fs")

const { join } = require("path")

const shikiVersion = require('@typescript/twoslash/package.json').version

const tsVersion = require('typescript/package.json').version

const shasum = createHash("sha1")

const codeSha = shasum.update(`${code}-${shikiVersion}-${tsVersion}`).digest("hex")

const getNmCache = () => {

if (__dirname.includes("node_modules")) {

return join(__dirname.split("node_modules")[0], "node_modules", ".cache", "twoslash")

} else {

return join(__dirname, "..", "..", ".cache", "twoslash");

}

};

const getPnpCache = () => {

try {

const pnp = require("pnpapi");

return join(pnp.getPackageInformation(pnp.topLevel).packageLocation, "node_modules", ".cache", "twoslash");

} catch (error) {

return getNmCache()

}

};

const cacheRoot = process.versions.pnp

? getPnpCache()

: getNmCache();

const cachePath = join(cacheRoot, `${codeSha}.json`)

if (existsSync(cachePath)) {

if (process.env.debug)

console.log(`Using cached twoslash results from ${cachePath}`)

return JSON.parse(readFileSync(cachePath, "utf8"))

} else {

const results = runTwoSlash(code, lang, settings)

if (!existsSync(cacheRoot)) mkdirSync(cacheRoot, { recursive: true })

writeFileSync(cachePath, JSON.stringify(results), "utf8")

return results

}

}