Tldraw/packages/editor/src/lib/components/ShapeIndicator.tsx

91 wiersze
2.4 KiB
TypeScript

import { TLShape, TLShapeId } from '@tldraw/tlschema'
import classNames from 'classnames'
import * as React from 'react'
import {
// @ts-expect-error 'private' export
useStateTracking,
useValue,
} from 'signia-react'
import { useEditor } from '../..'
import type { Editor } from '../editor/Editor'
import { ShapeUtil } from '../editor/shapeutils/ShapeUtil'
import { useEditorComponents } from '../hooks/useEditorComponents'
import { OptionalErrorBoundary } from './ErrorBoundary'
class ShapeWithPropsEquality {
constructor(public shape: TLShape | undefined) {}
equals(other: ShapeWithPropsEquality) {
return this.shape?.props === other?.shape?.props
}
}
// need an extra layer of indirection here to allow hooks to be used inside the indicator render
const EvenInnererIndicator = ({ shape, util }: { shape: TLShape; util: ShapeUtil<any> }) => {
return useStateTracking('Indicator:' + shape.type, () => util.indicator(shape))
}
export const InnerIndicator = ({ editor, id }: { editor: Editor; id: TLShapeId }) => {
const shape = useValue('shape', () => new ShapeWithPropsEquality(editor.store.get(id)), [
editor,
id,
])
const { ShapeIndicatorErrorFallback } = useEditorComponents()
if (!shape.shape) return null
return (
<OptionalErrorBoundary
fallback={
ShapeIndicatorErrorFallback
? (error) => <ShapeIndicatorErrorFallback error={error} />
: null
}
onError={(error) =>
editor.annotateError(error, { origin: 'react.shapeIndicator', willCrashApp: false })
}
>
<EvenInnererIndicator
key={shape.shape.id}
shape={shape.shape}
util={editor.getShapeUtil(shape.shape)}
/>
</OptionalErrorBoundary>
)
}
export type TLShapeIndicatorComponent = (props: {
id: TLShapeId
color?: string | undefined
opacity?: number
className?: string
}) => JSX.Element | null
const _ShapeIndicator: TLShapeIndicatorComponent = ({ id, className, color, opacity }) => {
const editor = useEditor()
const transform = useValue(
'transform',
() => {
const pageTransform = editor.getPageTransformById(id)
if (!pageTransform) return ''
return pageTransform.toCssString()
},
[editor, id]
)
return (
<svg className={classNames('tl-overlays__item', className)}>
<g
className="tl-shape-indicator"
transform={transform}
stroke={color ?? 'var(--color-selected)'}
opacity={opacity}
>
<InnerIndicator editor={editor} id={id} />
</g>
</svg>
)
}
export const ShapeIndicator = React.memo(_ShapeIndicator)