enso/app/gui2/stories/MockFSWrapper.vue
somebody1234 9b7e3d0f16
E2E tests (#8239)
- Closes #8179

# Important Notes
- ⚠️ These tests are currently *not run* on any CI workflow.
- There is some unused code for mocking the PM. This has been intentionally kept, as this may be useful in the future.
Note that this may be useful for testing the dashboard, however the dashboard is currently only tested in cloud mode
- that is, without the backend switcher, and with only the remote backend available. As such, currently it uses HTTP API mocks, and no PM mock.
2023-11-27 15:48:37 +00:00

97 lines
2.9 KiB
Vue

<script setup lang="ts">
import { useProjectStore } from '@/stores/project'
import { mockFsDirectoryHandle } from '@/util/convert/fsAccess'
import { MockWebSocket, type WebSocketHandler } from '@/util/net'
import { mockDataWSHandler } from 'shared/dataServer/mock'
import { type Path as LSPath } from 'shared/languageServerTypes'
import { watchEffect } from 'vue'
const projectStore = useProjectStore()
interface FileTree {
[name: string]: FileTree | string | ArrayBuffer
}
const props = defineProps<{
files: FileTree | undefined
directory: FileSystemDirectoryHandle | undefined
/** The path of the root directory. */
prefix?: string[] | undefined
}>()
let resolveDataWsHandler: ((handler: WebSocketHandler) => void) | undefined
let dataWsHandler: Promise<WebSocketHandler> = new Promise((resolve) => {
resolveDataWsHandler = resolve
})
function setDataWsHandler(handler: WebSocketHandler) {
if (resolveDataWsHandler) {
resolveDataWsHandler(handler)
resolveDataWsHandler = undefined
} else {
dataWsHandler = Promise.resolve(handler)
}
}
MockWebSocket.addMock('data', async (data, send) => {
;(await dataWsHandler)(data, send)
})
watchEffect(async (onCleanup) => {
let maybeDirectory = props.files ? mockFsDirectoryHandle(props.files, '(root)') : props.directory
if (!maybeDirectory) return
const prefixLength = props.prefix?.length ?? 0
const directory = maybeDirectory
const ls = await projectStore.lsRpcConnection
const maybeProjectRoot = (await projectStore.contentRoots).find((root) => root.type === 'Project')
?.id
if (!maybeProjectRoot) return
const projectRoot = maybeProjectRoot
async function walkFiles(
dir: FileSystemDirectoryHandle,
segments: string[],
cb: (path: LSPath) => void,
) {
for await (const [name, dirOrFile] of dir.entries()) {
const newSegments = [...segments, name]
if (dirOrFile.kind === 'directory') walkFiles(dirOrFile, newSegments, cb)
else {
cb({
rootId: projectRoot,
segments: newSegments.slice(prefixLength),
})
}
}
}
walkFiles(directory, props.prefix ?? [], (path) =>
ls.emit('file/event', [{ kind: 'Added', path }]),
)
onCleanup(() => {
walkFiles(directory, props.prefix ?? [], (path) =>
ls.emit('file/event', [{ kind: 'Removed', path }]),
)
})
setDataWsHandler(
mockDataWSHandler(async (segments) => {
segments = segments.slice(prefixLength)
if (!segments.length) return
let file
try {
let dir = directory
for (const segment of segments.slice(0, -1)) {
dir = await dir.getDirectoryHandle(segment)
}
const fileHandle = await dir.getFileHandle(segments.at(-1)!)
file = await fileHandle.getFile()
} catch {
return
}
return await file?.arrayBuffer()
}),
)
})
</script>
<template>
<slot></slot>
</template>