Medplum

// SPDX-FileCopyrightText: Copyright Orangebot, Inc. and Medplum contributors // SPDX-License-Identifier: Apache-2.0 import { Button, LoadingOverlay } from '@mantine/core'; import { showNotification } from '@mantine/notifications'; import { capitalize, getReferenceString, isOk, normalizeErrorString } from '@medplum/core'; import type { MedplumClient, WithId } from '@medplum/core'; import type { Binary, Bot, Bundle, BundleEntry, Practitioner, Resource } from '@medplum/fhirtypes'; import { Document, useMedplum, useMedplumProfile } from '@medplum/react'; import { IconCircleCheck, IconCircleOff } from '@tabler/icons-react'; import { useCallback, useState } from 'react'; import type { JSX } from 'react'; import { useNavigate, useParams } from 'react-router'; import coreData from '../../data/core/appointment-service-types.json'; type UploadFunction = (medplum: MedplumClient, profile: Practitioner) => Promise<void>; export function UploadDataPage(): JSX.Element { const medplum = useMedplum(); const profile = useMedplumProfile(); const navigate = useNavigate(); const [pageDisabled, setPageDisabled] = useState<boolean>(false); const { dataType } = useParams(); const dataTypeDisplay = dataType ? capitalize(dataType) : ''; const buttonDisabled = dataType === 'bots' && checkBotsUploaded(medplum); const handleUpload = useCallback(() => { if (!profile) { return; } setPageDisabled(true); let uploadFunction: UploadFunction; switch (dataType) { case 'core': uploadFunction = uploadCoreData; break; case 'bots': uploadFunction = uploadExampleBots; break; case 'example': uploadFunction = uploadExampleData; break; default: throw new Error(`Invalid upload type: ${dataType}`); } uploadFunction(medplum, profile as Practitioner) .then(() => navigate('/')) .catch((error) => { showNotification({ color: 'red', icon: <IconCircleOff />, title: 'Error', message: normalizeErrorString(error), autoClose: false, }); }) .finally(() => setPageDisabled(false)); }, [medplum, profile, dataType, navigate]); return ( <Document> <LoadingOverlay visible={pageDisabled} /> <Button disabled={buttonDisabled} onClick={handleUpload}> Upload {dataTypeDisplay} data </Button> </Document> ); } async function uploadCoreData(medplum: MedplumClient): Promise<void> { const batch = coreData as Bundle; const result = await medplum.executeBatch(batch); if (result.entry?.every((entry) => entry.response?.outcome && isOk(entry.response?.outcome))) { await setTimeout( () => showNotification({ icon: <IconCircleCheck />, title: 'Success', message: 'Uploaded Core Data', }), 1000 ); } else { throw new Error('Error uploading core data'); } } const EXAMPLE_BOTS_JSON = '../../data/core/example-bots.json'; async function uploadExampleBots(medplum: MedplumClient, profile: Practitioner): Promise<void> { let exampleBotData: Bundle; try { exampleBotData = await import(/* @vite-ignore */ EXAMPLE_BOTS_JSON); } catch (err) { console.log(err); if (err instanceof TypeError && err.message.includes('Failed to fetch')) { throw new Error('Error loading bot data. Run `npm run build:bots` and try again.'); } throw err; } let transactionString = JSON.stringify(exampleBotData); const botEntries: BundleEntry[] = (exampleBotData as Bundle).entry?.filter((e: any) => (e.resource as Resource)?.resourceType === 'Bot') || []; const botNames = botEntries.map((e) => (e.resource as Bot).name ?? ''); const botIds: Record<string, string> = {}; for (const botName of botNames) { let existingBot = await medplum.searchOne('Bot', { name: botName }); // Create a new Bot if it doesn't already exist if (!existingBot) { const projectId = profile.meta?.project; const createBotUrl = new URL('admin/projects/' + (projectId as string) + '/bot', medplum.getBaseUrl()); existingBot = (await medplum.post(createBotUrl, { name: botName, })) as WithId<Bot>; } botIds[botName] = existingBot.id as string; // Replace the Bot id placeholder in the bundle transactionString = transactionString .replaceAll(`$bot-${botName}-reference`, getReferenceString(existingBot)) .replaceAll(`$bot-${botName}-id`, existingBot.id as string); } // Execute the transaction to upload / update the bot const transaction = JSON.parse(transactionString); await medplum.executeBatch(transaction); // Deploy the new bots for (const entry of botEntries) { const botName = (entry?.resource as Bot)?.name as string; const distUrl = (entry.resource as Bot).executableCode?.url; const distBinaryEntry = exampleBotData.entry?.find((e: any) => e.fullUrl === distUrl) as | BundleEntry<Binary> | undefined; if (!distBinaryEntry) { throw new Error('Error finding Bundle entry with fullUrl: ' + distUrl); } if (!distBinaryEntry.resource?.data) { throw new Error('Could not find encoded code for bot: ' + botName); } // Decode the base64 encoded code and deploy const code = atob(distBinaryEntry.resource.data); await medplum.post(medplum.fhirUrl('Bot', botIds[botName], '$deploy'), { code }); } showNotification({ icon: <IconCircleCheck />, title: 'Success', message: 'Deployed Example Bots', }); } function checkBotsUploaded(medplum: MedplumClient): boolean { const bots = medplum.searchResources('Bot').read(); const exampleBots = bots.filter( (bot) => bot.name && ['book-appointment', 'cancel-appointment', 'set-availability', 'block-availability', 'example-data'].includes( bot.name ) ); if (exampleBots.length === 5) { return true; } return false; } async function uploadExampleData(medplum: MedplumClient, profile: Practitioner): Promise<void> { // Unlike the other bundles, which source data from JSON files, this example data bundle is // provided by a bot to handle dynamic content. const batch: Bundle = await medplum.executeBot({ system: 'http://example.com', value: 'example-data' }, profile); const result = await medplum.executeBatch(batch); if (result.entry?.every((entry) => entry.response?.outcome && isOk(entry.response?.outcome))) { setTimeout( () => showNotification({ icon: <IconCircleCheck />, title: 'Success', message: 'Uploaded Example Data', }), 1000 ); } else { throw new Error('Error uploading example data'); } }