init

wangzhibo

2023-10-31 9da39812105bfc387db3b3b7c4896983e03d247d

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
import { createEventSignal } from '@solid-primitives/event-listener'
import { makePersisted } from '@solid-primitives/storage'
import { createReconnectingWS } from '@solid-primitives/websocket'
import ky from 'ky'
import { createMemo, createSignal } from 'solid-js'
 
export const [selectedEndpoint, setSelectedEndpoint] = makePersisted(
  createSignal(''),
  {
    name: 'selectedEndpoint',
    storage: localStorage,
  },
)
 
export const [endpointList, setEndpointList] = makePersisted(
  createSignal<
    {
      id: string
      url: string
      secret: string
    }[]
  >([]),
  { name: 'endpointList', storage: localStorage },
)
 
export const useRequest = () => {
  const e = endpoint()
 
  if (!e) {
    return ky.create({})
  }
 
  const headers = new Headers()
 
  if (e.secret) {
    headers.set('Authorization', `Bearer ${e.secret}`)
  }
 
  return ky.create({
    prefixUrl: e.url,
    headers,
  })
}
 
export const endpoint = () =>
  endpointList().find(({ id }) => id === selectedEndpoint())
 
export const secret = () => endpoint()?.secret
 
export const wsEndpointURL = () =>
  new URL(endpoint()?.url ?? '').origin.replace('http', 'ws')
 
export const useWsRequest = <T>(
  path: string,
  queries: Record<string, string> = {},
) => {
  const queryParams = new URLSearchParams(queries)
  queryParams.set('token', secret() ?? '')
 
  const ws = createReconnectingWS(
    `${wsEndpointURL()}/${path}?${queryParams.toString()}`,
  )
 
  const event = createEventSignal<{
    message: MessageEvent
  }>(ws, 'message')
 
  return createMemo<T | null>(() => {
    const e = event()
 
    if (!e) {
      return null
    }
 
    return JSON.parse(event()?.data)
  })
}