tldraw/state/commands/style.ts

import Command from './command'
import history from '../history'
import { Data, ShapeStyles } from 'types'
import {
  getDocumentBranch,
  getPage,
  getSelectedIds,
  setToArray,
} from 'utils/utils'
import { getShapeUtils } from 'state/shape-utils'
import { current } from 'immer'

export default function styleCommand(
  data: Data,
  styles: Partial<ShapeStyles>
): void {
  const cData = current(data)
  const page = getPage(cData)
  const { currentPageId } = cData

  const selectedIds = setToArray(getSelectedIds(data))

  const shapesToStyle = selectedIds
    .flatMap((id) => getDocumentBranch(data, id))
    .map((id) => page.shapes[id])

  history.execute(
    data,
    new Command({
      name: 'style_shapes',
      category: 'canvas',
      manualSelection: true,
      do(data) {
        const { shapes } = getPage(data, currentPageId)

        for (const { id } of shapesToStyle) {
          const shape = shapes[id]
          getShapeUtils(shape).applyStyles(shape, styles)
        }
      },
      undo(data) {
        const { shapes } = getPage(data, currentPageId)

        for (const { id, style } of shapesToStyle) {
          const shape = shapes[id]
          getShapeUtils(shape).applyStyles(shape, style)
        }
      },
    })
  )
}