Something went wrong. Try again.
a tool for shared writing and social publishing
Something went wrong. Try again.
5.7 kB · 165 lines
TSX
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166import { BundledLanguage, bundledLanguagesInfo, bundledThemesInfo, codeToHtml,} from "shiki";import { useEntity, useReplicache } from "src/replicache";import "katex/dist/katex.min.css";import { BlockProps } from "./Block";import { useCallback, useLayoutEffect, useMemo, useState } from "react";import { useUIState } from "src/useUIState";import { BaseTextareaBlock } from "./BaseTextareaBlock";import { useEntitySetContext } from "components/EntitySetProvider";import { flushSync } from "react-dom";import { elementId } from "src/utils/elementId";import { LAST_USED_CODE_LANGUAGE_KEY } from "src/utils/codeLanguageStorage";
export function CodeBlock(props: BlockProps) { let { rep, rootEntity } = useReplicache(); let content = useEntity(props.entityID, "block/code"); let lang = useEntity(props.entityID, "block/code-language")?.data.value || "plaintext";
let theme = useEntity(rootEntity, "theme/code-theme")?.data.value || "github-light"; let focusedBlock = useUIState( (s) => s.focusedEntity?.entityID === props.entityID, ); let entity_set = useEntitySetContext(); let { permissions } = entity_set; const [html, setHTML] = useState<string | null>(null);
useLayoutEffect(() => { if (!content) return; void codeToHtml(content.data.value, { lang, theme, structure: "classic", }).then((h) => { setHTML(h.replaceAll("<br>", "\n")); }); }, [content, lang, theme]);
const onClick = useCallback((e: React.MouseEvent<HTMLElement>) => { let selection = window.getSelection(); if (!selection || selection.rangeCount === 0) return; let range = selection.getRangeAt(0); if (!range) return; let length = range.toString().length; range.setStart(e.currentTarget, 0); let end = range.toString().length; let start = end - length;
flushSync(() => { useUIState.getState().setSelectedBlock(props); useUIState.getState().setFocusedBlock({ entityType: "block", entityID: props.value, parent: props.parent, }); }); let el = document.getElementById( elementId.block(props.entityID).input, ) as HTMLTextAreaElement; if (!el) return; el.focus(); el.setSelectionRange(start, end); }, []); return ( <div className="codeBlock w-full flex flex-col rounded-md gap-0.5 "> {permissions.write && ( <div className="text-sm text-tertiary flex justify-between"> <div className="flex gap-1"> Theme:{" "} <select className="codeBlockLang text-left bg-transparent pr-1 sm:max-w-none max-w-24" onClick={(e) => { e.preventDefault(); e.stopPropagation(); }} value={theme} onChange={async (e) => { await rep?.mutate.assertFact({ attribute: "theme/code-theme", entity: rootEntity, data: { type: "string", value: e.target.value }, }); }} > {bundledThemesInfo.map((t) => ( <option key={t.id} value={t.id}> {t.displayName} </option> ))} </select> </div> <select className="codeBlockLang text-right bg-transparent pr-1 sm:max-w-none max-w-24" onClick={(e) => { e.preventDefault(); e.stopPropagation(); }} value={lang} onChange={async (e) => { localStorage.setItem(LAST_USED_CODE_LANGUAGE_KEY, e.target.value); await rep?.mutate.assertFact({ attribute: "block/code-language", entity: props.entityID, data: { type: "string", value: e.target.value }, }); }} > <option value="plaintext">Plaintext</option> {bundledLanguagesInfo.map((l) => ( <option key={l.id} value={l.id}> {l.name} </option> ))} </select> </div> )} <div className="w-full min-h-[42px] rounded-md border-border-light outline-border-light selected-outline"> {focusedBlock && permissions.write ? ( <BaseTextareaBlock data-editable-block data-entityid={props.entityID} id={elementId.block(props.entityID).input} block={props} rep={rep} permissionSet={entity_set.set} spellCheck={false} autoCapitalize="none" autoCorrect="off" className="codeBlockEditor whitespace-nowrap! overflow-auto! font-mono p-2" value={content?.data.value} onChange={async (e) => { // Update the entity with the new value await rep?.mutate.assertFact({ attribute: "block/code", entity: props.entityID, data: { type: "string", value: e.target.value }, }); }} /> ) : !html ? ( <pre onClick={onClick} onMouseDown={(e) => e.stopPropagation()} className="codeBlockRendered overflow-auto! font-mono p-2 w-full h-full" > {content?.data.value} </pre> ) : ( <div onMouseDown={(e) => e.stopPropagation()} onClick={onClick} data-lang={lang} className="contents" dangerouslySetInnerHTML={{ __html: html || "" }} /> )} </div> </div> );}