guardrails

import type { Register } from 'claude-code'

// Drawn like guardrails by Michael Goldenberg (pane, status, toast), as modscode read it from the mod's code:
// https://modscode.com/gallery/mishgoldenberg-guardrails/
// Written by modscode from that reading, not copied from the mod: use it as you like.
// Each value below stands for one the mod works out at run time; put in your own.
export const register: Register = on => {
  // the pane: open it with $.ui.open({ id: 'my-pane', title: "Guardrails" })
  on('ui.render', { component: 'Pane', requestId: 'my-pane' }, async ($, e) => {
    const { Box, Text, Button, Input } = $.ui.resolve(e)
    const status = '…'
    const i = '…'
    const title = '…'
    const explain = '…'
    const explain2 = '…'
    const explain3 = '…'
    const value = '…'
    const pattern = '…'
    const pattern2 = '…'
    const pattern3 = '…'
    const length = '…'
    const tool = '…'
    const rule = '…'
    const what = '…'
    const tool2 = '…'
    const rule2 = '…'
    const what2 = '…'
    const tool3 = '…'
    const rule3 = '…'
    const what3 = '…'
    return (
      <Box flexDirection="column" gap={1}>
        <Box gap={1}>
          <Text color="claude">
            ●
          </Text>
          <Text bold>
            guardrails
          </Text>
          <Text dimColor wrap="truncate-end">
            {status}
          </Text>
        </Box>
        <Box flexDirection="column">
          <Box gap={1}>
            <Text dimColor>
              Presets
            </Text>
          </Box>
          <Box gap={1} flexWrap="wrap">
            <Button hotkey={`${i}`} variant="primary" label={`${title}`} />
            <Button hotkey={`${i}`} variant="primary" label={`${title}`} />
            <Button hotkey={`${i}`} variant="primary" label={`${title}`} />
          </Box>
        </Box>
        <Box flexDirection="column">
          <Box gap={1}>
            <Text dimColor>
              Rules
            </Text>
          </Box>
          <Box flexDirection="column">
            <Box gap={1}>
              <Text color="claude">
                ●
              </Text>
              <Button plain label={`${title}`} />
            </Box>
            <Box paddingLeft={2}>
              <Text dimColor wrap="truncate-end">
                {explain}
              </Text>
            </Box>
          </Box>
          <Box flexDirection="column">
            <Box gap={1}>
              <Text color="claude">
                ●
              </Text>
              <Button plain label={`${title}`} />
            </Box>
            <Box paddingLeft={2}>
              <Text dimColor wrap="truncate-end">
                {explain2}
              </Text>
            </Box>
          </Box>
          <Box flexDirection="column">
            <Box gap={1}>
              <Text color="claude">
                ●
              </Text>
              <Button plain label={`${title}`} />
            </Box>
            <Box paddingLeft={2}>
              <Text dimColor wrap="truncate-end">
                {explain3}
              </Text>
            </Box>
          </Box>
        </Box>
        <Box flexDirection="column">
          <Box gap={1}>
            <Text dimColor>
              Your own block patterns (regex on shell commands)
            </Text>
          </Box>
          <Box flexDirection="column">
            <Text dimColor>
              {value}
            </Text>
            <Box gap={1}>
              <Text color="claude">
                ●
              </Text>
              <Text wrap="truncate-end">
                /
                {pattern}
                /
              </Text>
              <Button plain label="remove" />
            </Box>
            <Box gap={1}>
              <Text color="claude">
                ●
              </Text>
              <Text wrap="truncate-end">
                /
                {pattern2}
                /
              </Text>
              <Button plain label="remove" />
            </Box>
            <Box gap={1}>
              <Text color="claude">
                ●
              </Text>
              <Text wrap="truncate-end">
                /
                {pattern3}
                /
              </Text>
              <Button plain label="remove" />
            </Box>
            <Input placeholder="regex, e.g. docker\s+system\s+prune" submitLabel="Add" />
          </Box>
        </Box>
        <Box flexDirection="column">
          <Box gap={1}>
            <Text dimColor>
              {`Blocked this session (${length})`}
            </Text>
          </Box>
          <Box flexDirection="column">
            <Text dimColor>
              Nothing blocked yet. When a rule stops a command, it shows up here with the rule that caught it.
            </Text>
            <Text wrap="truncate-end">
              <Text color="error">
                ✗
              </Text>
              {" "}
              {tool}
              {" "}
              <Text dimColor>
                {rule}
                {": "}
                {what}
              </Text>
            </Text>
            <Text wrap="truncate-end">
              <Text color="error">
                ✗
              </Text>
              {" "}
              {tool2}
              {" "}
              <Text dimColor>
                {rule2}
                {": "}
                {what2}
              </Text>
            </Text>
            <Text wrap="truncate-end">
              <Text color="error">
                ✗
              </Text>
              {" "}
              {tool3}
              {" "}
              <Text dimColor>
                {rule3}
                {": "}
                {what3}
              </Text>
            </Text>
          </Box>
        </Box>
        <Text dimColor>
          Rules are best-effort pattern checks, a seatbelt, not a sandbox. Saved for all your sessions.
        </Text>
      </Box>
    )
  })

  on('session.start', async ($, e, next) => {
    const count = '…'
    const tool = '…'
    const rule = '…'
    await $.ui.status(`guard ${count} rules`)
    await $.ui.toast(`Blocked ${tool}: ${rule}`)
    await $.ui.toast(`That is not a valid regular expression.`)
    return next(e)
  })
}