refactor(k8s): updated story book

pull/12062/head
stevensbkang 2024-08-01 08:46:06 +12:00
parent 1491b2c31b
commit bb8a7d5409
No known key found for this signature in database
1 changed files with 76 additions and 73 deletions

View File

@ -8,119 +8,119 @@
* - Please do NOT serve this file on production. * - Please do NOT serve this file on production.
*/ */
const INTEGRITY_CHECKSUM = 'c5f7f8e188b673ea4e677df7ea3c5a39'; const INTEGRITY_CHECKSUM = 'c5f7f8e188b673ea4e677df7ea3c5a39'
const IS_MOCKED_RESPONSE = Symbol('isMockedResponse'); const IS_MOCKED_RESPONSE = Symbol('isMockedResponse')
const activeClientIds = new Set(); const activeClientIds = new Set()
self.addEventListener('install', function () { self.addEventListener('install', function () {
self.skipWaiting(); self.skipWaiting()
}); })
self.addEventListener('activate', function (event) { self.addEventListener('activate', function (event) {
event.waitUntil(self.clients.claim()); event.waitUntil(self.clients.claim())
}); })
self.addEventListener('message', async function (event) { self.addEventListener('message', async function (event) {
const clientId = event.source.id; const clientId = event.source.id
if (!clientId || !self.clients) { if (!clientId || !self.clients) {
return; return
} }
const client = await self.clients.get(clientId); const client = await self.clients.get(clientId)
if (!client) { if (!client) {
return; return
} }
const allClients = await self.clients.matchAll({ const allClients = await self.clients.matchAll({
type: 'window', type: 'window',
}); })
switch (event.data) { switch (event.data) {
case 'KEEPALIVE_REQUEST': { case 'KEEPALIVE_REQUEST': {
sendToClient(client, { sendToClient(client, {
type: 'KEEPALIVE_RESPONSE', type: 'KEEPALIVE_RESPONSE',
}); })
break; break
} }
case 'INTEGRITY_CHECK_REQUEST': { case 'INTEGRITY_CHECK_REQUEST': {
sendToClient(client, { sendToClient(client, {
type: 'INTEGRITY_CHECK_RESPONSE', type: 'INTEGRITY_CHECK_RESPONSE',
payload: INTEGRITY_CHECKSUM, payload: INTEGRITY_CHECKSUM,
}); })
break; break
} }
case 'MOCK_ACTIVATE': { case 'MOCK_ACTIVATE': {
activeClientIds.add(clientId); activeClientIds.add(clientId)
sendToClient(client, { sendToClient(client, {
type: 'MOCKING_ENABLED', type: 'MOCKING_ENABLED',
payload: true, payload: true,
}); })
break; break
} }
case 'MOCK_DEACTIVATE': { case 'MOCK_DEACTIVATE': {
activeClientIds.delete(clientId); activeClientIds.delete(clientId)
break; break
} }
case 'CLIENT_CLOSED': { case 'CLIENT_CLOSED': {
activeClientIds.delete(clientId); activeClientIds.delete(clientId)
const remainingClients = allClients.filter((client) => { const remainingClients = allClients.filter((client) => {
return client.id !== clientId; return client.id !== clientId
}); })
// Unregister itself when there are no more clients // Unregister itself when there are no more clients
if (remainingClients.length === 0) { if (remainingClients.length === 0) {
self.registration.unregister(); self.registration.unregister()
} }
break; break
} }
} }
}); })
self.addEventListener('fetch', function (event) { self.addEventListener('fetch', function (event) {
const { request } = event; const { request } = event
// Bypass navigation requests. // Bypass navigation requests.
if (request.mode === 'navigate') { if (request.mode === 'navigate') {
return; return
} }
// Opening the DevTools triggers the "only-if-cached" request // Opening the DevTools triggers the "only-if-cached" request
// that cannot be handled by the worker. Bypass such requests. // that cannot be handled by the worker. Bypass such requests.
if (request.cache === 'only-if-cached' && request.mode !== 'same-origin') { if (request.cache === 'only-if-cached' && request.mode !== 'same-origin') {
return; return
} }
// Bypass all requests when there are no active clients. // Bypass all requests when there are no active clients.
// Prevents the self-unregistered worked from handling requests // Prevents the self-unregistered worked from handling requests
// after it's been deleted (still remains active until the next reload). // after it's been deleted (still remains active until the next reload).
if (activeClientIds.size === 0) { if (activeClientIds.size === 0) {
return; return
} }
// Generate unique request ID. // Generate unique request ID.
const requestId = crypto.randomUUID(); const requestId = crypto.randomUUID()
event.respondWith(handleRequest(event, requestId)); event.respondWith(handleRequest(event, requestId))
}); })
async function handleRequest(event, requestId) { async function handleRequest(event, requestId) {
const client = await resolveMainClient(event); const client = await resolveMainClient(event)
const response = await getResponse(event, client, requestId); const response = await getResponse(event, client, requestId)
// Send back the response clone for the "response:*" life-cycle events. // Send back the response clone for the "response:*" life-cycle events.
// Ensure MSW is active and ready to handle the message, otherwise // Ensure MSW is active and ready to handle the message, otherwise
// this message will pend indefinitely. // this message will pend indefinitely.
if (client && activeClientIds.has(client.id)) { if (client && activeClientIds.has(client.id)) {
(async function () { ;(async function () {
const responseClone = response.clone(); const responseClone = response.clone()
sendToClient( sendToClient(
client, client,
@ -136,12 +136,12 @@ async function handleRequest(event, requestId) {
headers: Object.fromEntries(responseClone.headers.entries()), headers: Object.fromEntries(responseClone.headers.entries()),
}, },
}, },
[responseClone.body] [responseClone.body],
); )
})(); })()
} }
return response; return response
} }
// Resolve the main client for the given event. // Resolve the main client for the given event.
@ -149,49 +149,49 @@ async function handleRequest(event, requestId) {
// that registered the worker. It's with the latter the worker should // that registered the worker. It's with the latter the worker should
// communicate with during the response resolving phase. // communicate with during the response resolving phase.
async function resolveMainClient(event) { async function resolveMainClient(event) {
const client = await self.clients.get(event.clientId); const client = await self.clients.get(event.clientId)
if (client?.frameType === 'top-level') { if (client?.frameType === 'top-level') {
return client; return client
} }
const allClients = await self.clients.matchAll({ const allClients = await self.clients.matchAll({
type: 'window', type: 'window',
}); })
return allClients return allClients
.filter((client) => { .filter((client) => {
// Get only those clients that are currently visible. // Get only those clients that are currently visible.
return client.visibilityState === 'visible'; return client.visibilityState === 'visible'
}) })
.find((client) => { .find((client) => {
// Find the client ID that's recorded in the // Find the client ID that's recorded in the
// set of clients that have registered the worker. // set of clients that have registered the worker.
return activeClientIds.has(client.id); return activeClientIds.has(client.id)
}); })
} }
async function getResponse(event, client, requestId) { async function getResponse(event, client, requestId) {
const { request } = event; const { request } = event
// Clone the request because it might've been already used // Clone the request because it might've been already used
// (i.e. its body has been read and sent to the client). // (i.e. its body has been read and sent to the client).
const requestClone = request.clone(); const requestClone = request.clone()
function passthrough() { function passthrough() {
const headers = Object.fromEntries(requestClone.headers.entries()); const headers = Object.fromEntries(requestClone.headers.entries())
// Remove internal MSW request header so the passthrough request // Remove internal MSW request header so the passthrough request
// complies with any potential CORS preflight checks on the server. // complies with any potential CORS preflight checks on the server.
// Some servers forbid unknown request headers. // Some servers forbid unknown request headers.
delete headers['x-msw-intention']; delete headers['x-msw-intention']
return fetch(requestClone, { headers }); return fetch(requestClone, { headers })
} }
// Bypass mocking when the client is not active. // Bypass mocking when the client is not active.
if (!client) { if (!client) {
return passthrough(); return passthrough()
} }
// Bypass initial page load requests (i.e. static assets). // Bypass initial page load requests (i.e. static assets).
@ -199,18 +199,18 @@ async function getResponse(event, client, requestId) {
// means that MSW hasn't dispatched the "MOCK_ACTIVATE" event yet // means that MSW hasn't dispatched the "MOCK_ACTIVATE" event yet
// and is not ready to handle requests. // and is not ready to handle requests.
if (!activeClientIds.has(client.id)) { if (!activeClientIds.has(client.id)) {
return passthrough(); return passthrough()
} }
// Bypass requests with the explicit bypass header. // Bypass requests with the explicit bypass header.
// Such requests can be issued by "ctx.fetch()". // Such requests can be issued by "ctx.fetch()".
const mswIntention = request.headers.get('x-msw-intention'); const mswIntention = request.headers.get('x-msw-intention')
if (['bypass', 'passthrough'].includes(mswIntention)) { if (['bypass', 'passthrough'].includes(mswIntention)) {
return passthrough(); return passthrough()
} }
// Notify the client that a request has been intercepted. // Notify the client that a request has been intercepted.
const requestBuffer = await request.arrayBuffer(); const requestBuffer = await request.arrayBuffer()
const clientMessage = await sendToClient( const clientMessage = await sendToClient(
client, client,
{ {
@ -232,36 +232,39 @@ async function getResponse(event, client, requestId) {
keepalive: request.keepalive, keepalive: request.keepalive,
}, },
}, },
[requestBuffer] [requestBuffer],
); )
switch (clientMessage.type) { switch (clientMessage.type) {
case 'MOCK_RESPONSE': { case 'MOCK_RESPONSE': {
return respondWithMock(clientMessage.data); return respondWithMock(clientMessage.data)
} }
case 'MOCK_NOT_FOUND': { case 'MOCK_NOT_FOUND': {
return passthrough(); return passthrough()
} }
} }
return passthrough(); return passthrough()
} }
function sendToClient(client, message, transferrables = []) { function sendToClient(client, message, transferrables = []) {
return new Promise((resolve, reject) => { return new Promise((resolve, reject) => {
const channel = new MessageChannel(); const channel = new MessageChannel()
channel.port1.onmessage = (event) => { channel.port1.onmessage = (event) => {
if (event.data && event.data.error) { if (event.data && event.data.error) {
return reject(event.data.error); return reject(event.data.error)
} }
resolve(event.data); resolve(event.data)
}; }
client.postMessage(message, [channel.port2].concat(transferrables.filter(Boolean))); client.postMessage(
}); message,
[channel.port2].concat(transferrables.filter(Boolean)),
)
})
} }
async function respondWithMock(response) { async function respondWithMock(response) {
@ -270,15 +273,15 @@ async function respondWithMock(response) {
// instance will have status code set to 0. Since it's not possible to create // instance will have status code set to 0. Since it's not possible to create
// a Response instance with status code 0, handle that use-case separately. // a Response instance with status code 0, handle that use-case separately.
if (response.status === 0) { if (response.status === 0) {
return Response.error(); return Response.error()
} }
const mockedResponse = new Response(response.body, response); const mockedResponse = new Response(response.body, response)
Reflect.defineProperty(mockedResponse, IS_MOCKED_RESPONSE, { Reflect.defineProperty(mockedResponse, IS_MOCKED_RESPONSE, {
value: true, value: true,
enumerable: true, enumerable: true,
}); })
return mockedResponse; return mockedResponse
} }