import { rootFolderBase } from './fixtures/file-tree-base' import { rootFolderLimit } from './fixtures/file-tree-limit' import FileTreeRoot from '../js/features/file-tree/components/file-tree-root' import FileTreeError from '../js/features/file-tree/components/file-tree-error' import useFetchMock from './hooks/use-fetch-mock' import { ScopeDecorator } from './decorators/scope' import { useScope } from './hooks/use-scope' import { useIdeContext } from '@/shared/context/ide-context' import { bsVersionDecorator } from '../../.storybook/utils/with-bootstrap-switcher' const MOCK_DELAY = 2000 const DEFAULT_PROJECT = { _id: '123abc', name: 'Some Project', rootDocId: '5e74f1a7ce17ae0041dfd056', rootFolder: rootFolderBase, } function defaultSetupMocks(fetchMock, socket) { fetchMock .post( /\/project\/\w+\/(file|doc|folder)\/\w+\/rename/, (path, req) => { const body = JSON.parse(req.body) const entityId = path.match(/([^/]+)\/rename$/)[1] socket.emitToClient('reciveEntityRename', entityId, body.name) return 204 }, { delay: MOCK_DELAY, } ) .post( /\/project\/\w+\/folder/, (_path, req) => { const body = JSON.parse(req.body) const newFolder = { folders: [], fileRefs: [], docs: [], _id: Math.random().toString(16).replace(/0\./, 'random-test-id-'), name: body.name, } socket.emitToClient('reciveNewFolder', body.parent_folder_id, newFolder) return newFolder }, { delay: MOCK_DELAY, } ) .delete( /\/project\/\w+\/(file|doc|folder)\/\w+/, path => { const entityId = path.match(/[^/]+$/)[0] socket.emitToClient('removeEntity', entityId) return 204 }, { delay: MOCK_DELAY, } ) .post(/\/project\/\w+\/(file|doc|folder)\/\w+\/move/, (path, req) => { const body = JSON.parse(req.body) const entityId = path.match(/([^/]+)\/move/)[1] socket.emitToClient('reciveEntityMove', entityId, body.folder_id) return 204 }) } export const FullTree = args => { const { socket } = useIdeContext() useFetchMock(fetchMock => defaultSetupMocks(fetchMock, socket)) useScope({ project: DEFAULT_PROJECT, permissionsLevel: 'owner', }) return } export const ReadOnly = args => { useScope({ project: DEFAULT_PROJECT, permissionsLevel: 'readOnly', }) return } export const Disconnected = args => { useScope({ project: DEFAULT_PROJECT, permissionsLevel: 'owner', }) return } Disconnected.args = { isConnected: false } export const NetworkErrors = args => { useFetchMock(fetchMock => { fetchMock .post(/\/project\/\w+\/folder/, 500, { delay: MOCK_DELAY, }) .post(/\/project\/\w+\/(file|doc|folder)\/\w+\/rename/, 500, { delay: MOCK_DELAY, }) .post(/\/project\/\w+\/(file|doc|folder)\/\w+\/move/, 500, { delay: MOCK_DELAY, }) .delete(/\/project\/\w+\/(file|doc|folder)\/\w+/, 500, { delay: MOCK_DELAY, }) }) useScope({ project: DEFAULT_PROJECT, permissionsLevel: 'owner', }) return } export const FallbackError = args => { useScope({ project: DEFAULT_PROJECT, }) return } export const FilesLimit = args => { const { socket } = useIdeContext() useFetchMock(fetchMock => defaultSetupMocks(fetchMock, socket)) useScope({ project: { ...DEFAULT_PROJECT, rootFolder: rootFolderLimit, }, permissionsLevel: 'owner', }) return } export default { title: 'Editor / File Tree', component: FileTreeRoot, args: { setStartedFreeTrial: () => { console.log('started free trial') }, refProviders: {}, setRefProviderEnabled: provider => { console.log(`ref provider ${provider} enabled`) }, isConnected: true, }, argTypes: { onInit: { action: 'onInit' }, onSelect: { action: 'onSelect' }, ...bsVersionDecorator.argTypes, }, decorators: [ ScopeDecorator, Story => ( <>
), ], }