123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123
import runtime from 'https://esm.sh/react-refresh@0.8.3/runtime'import events from './events.ts'import util, { hashShort } from './util.ts'
interface Callback {    (...args: any[]): void}
// react-refresh// @link https://github.com/facebook/react/issues/16604#issuecomment-528663101runtime.injectIntoGlobalHook(window)Object.assign(window, {    $RefreshReg$: () => { },    $RefreshSig$: () => (type: any) => type})export const performReactRefresh = util.debounce(runtime.performReactRefresh, 30)export const RefreshRuntime = runtime
class Module {    private _id: string    private _isLocked: boolean = false    private _isAccepted: boolean = false    private _acceptCallbacks: Callback[] = []
    get id() {        return this._id    }
    constructor(id: string) {        this._id = id    }
    lock(): void {        this._isLocked = true    }
    accept(callback?: () => void): void {        if (this._isLocked) {            return        }        if (!this._isAccepted) {            sendMessage({ id: this.id, type: 'hotAccept' })            this._isAccepted = true        }        if (callback) {            this._acceptCallbacks.push(callback)        }    }
    async applyUpdate(updateUrl: string) {        try {            const module = await import(updateUrl + '?t=' + Date.now())            this._acceptCallbacks.forEach(cb => cb(module))        } catch (e) {            location.reload()        }    }}
const { location } = window as anyconst { protocol, host } = locationconst modules: Map<string, Module> = new Map()const messageQueue: any[] = []const socket = new WebSocket((protocol === 'https:' ? 'wss' : 'ws') + '://' + host + '/_hmr', /*  'aleph-hmr' */)
socket.addEventListener('open', () => {    messageQueue.forEach(msg => socket.send(JSON.stringify(msg)))    messageQueue.splice(0, messageQueue.length)    console.log('[HMR] listening for file changes...')})
socket.addEventListener('close', () => {    location.reload()})
socket.addEventListener('message', ({ data: rawData }: { data?: string }) => {    if (rawData) {        try {            const { type, moduleId, hash, updateUrl } = JSON.parse(rawData)            switch (type) {                case 'add':                    events.emit('add-module', { id: moduleId, hash })                    break                case 'update':                    const mod = modules.get(moduleId)                    if (mod) {                        mod.applyUpdate(updateUrl)                    }                    break                case 'remove':                    if (modules.has(moduleId)) {                        modules.delete(moduleId)                        events.emit('remove-module', moduleId)                    }                    break            }            console.log(`[HMR]${hash ? ' [' + hash.slice(0, hashShort) + ']' : ''} ${type} module '${moduleId}'`)        } catch (err) {            console.warn(err)        }    }})
function sendMessage(msg: any) {    if (socket.readyState !== socket.OPEN) {        messageQueue.push(msg)    } else {        socket.send(JSON.stringify(msg))    }}
export function createHotContext(id: string) {    if (modules.has(id)) {        const mod = modules.get(id)!        mod.lock()        return mod    }
    const mod = new Module(id)    modules.set(id, mod)    return mod}