-
-
Notifications
You must be signed in to change notification settings - Fork 2.4k
/
index.ts
92 lines (77 loc) · 2.14 KB
/
index.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
/* this implementation is original ported from https://github.com/logaretm/vue-use-web by Abdelrahman Awad */
import type { Ref, ShallowRef } from 'vue-demi'
import { ref, shallowRef } from 'vue-demi'
import { isFunction, isString, tryOnScopeDispose } from '@vueuse/shared'
import type { ConfigurableWindow } from '../_configurable'
import { defaultWindow } from '../_configurable'
export interface UseWebWorkerReturn<Data = any> {
data: Ref<Data>
post: typeof Worker.prototype['postMessage']
terminate: () => void
worker: ShallowRef<Worker | undefined>
}
type WorkerFn = (...args: unknown[]) => Worker
/**
* Simple Web Workers registration and communication.
*
* @see https://vueuse.org/useWebWorker
* @param url
* @param workerOptions
* @param options
*/
export function useWebWorker<T = any>(
url: string,
workerOptions?: WorkerOptions,
options?: ConfigurableWindow,
): UseWebWorkerReturn<T>
/**
* Simple Web Workers registration and communication.
*
* @see https://vueuse.org/useWebWorker
* @param worker
*/
export function useWebWorker<T = any>(
worker: Worker | WorkerFn
): UseWebWorkerReturn<T>
export function useWebWorker<Data = any>(
arg0: string | WorkerFn | Worker,
workerOptions?: WorkerOptions,
options?: ConfigurableWindow,
): UseWebWorkerReturn<Data> {
const {
window = defaultWindow,
} = options ?? {}
const data: Ref<any> = ref(null)
const worker = shallowRef<Worker>()
const post: typeof Worker.prototype['postMessage'] = function post(val: any) {
if (!worker.value)
return
worker.value.postMessage(val)
}
const terminate: typeof Worker.prototype['terminate'] = function terminate() {
if (!worker.value)
return
worker.value.terminate()
}
if (window) {
if (isString(arg0))
worker.value = new Worker(arg0, workerOptions)
else if (isFunction(arg0))
worker.value = arg0()
else
worker.value = arg0
worker.value!.onmessage = (e: MessageEvent) => {
data.value = e.data
}
tryOnScopeDispose(() => {
if (worker.value)
worker.value.terminate()
})
}
return {
data,
post,
terminate,
worker,
}
}