limit-meter

import type { Register } from 'claude-code'

// Drawn like limit-meter by Alyan Khattak (pane, band, status, toast), as modscode read it from the mod's code:
// https://modscode.com/gallery/alyan-khattak-limit-meter/
// 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 band above the prompt; what else goes there is drawn under it
  on('ui.render', { component: 'AbovePrompt' }, async ($, e, next) => {
    const { Box, Text } = $.ui.resolve(e)
    const shortLabel = '…'
    const pct = '…'
    const untilReset = '…'
    const pct2 = '…'
    const shortLabel2 = '…'
    const pct3 = '…'
    const untilReset2 = '…'
    const pct4 = '…'
    const shortLabel3 = '…'
    const pct5 = '…'
    const untilReset3 = '…'
    const pct6 = '…'
    return (
      <Box flexDirection="column">
        <Box flexDirection="row" paddingX={1} gap={2}>
          <Box flexDirection="row">
            <Text dimColor>
              {shortLabel}
              {" "}
            </Text>
            <Text>
              ████████░░░░░░░░
              {" "}
            </Text>
            <Text bold>
              {pct}
            </Text>
            <Text dimColor>
              {" · "}
              {untilReset}
            </Text>
            <Text dimColor>
              {" · today +"}
              {pct2}
            </Text>
          </Box>
          <Box flexDirection="row">
            <Text dimColor>
              {shortLabel2}
              {" "}
            </Text>
            <Text>
              ████████░░░░░░░░
              {" "}
            </Text>
            <Text bold>
              {pct3}
            </Text>
            <Text dimColor>
              {" · "}
              {untilReset2}
            </Text>
            <Text dimColor>
              {" · today +"}
              {pct4}
            </Text>
          </Box>
          <Box flexDirection="row">
            <Text dimColor>
              {shortLabel3}
              {" "}
            </Text>
            <Text>
              ████████░░░░░░░░
              {" "}
            </Text>
            <Text bold>
              {pct5}
            </Text>
            <Text dimColor>
              {" · "}
              {untilReset3}
            </Text>
            <Text dimColor>
              {" · today +"}
              {pct6}
            </Text>
          </Box>
        </Box>
        {await next(e)}
      </Box>
    )
  })

  // the pane: open it with $.ui.open({ id: 'my-pane', title: "Limits" })
  on('ui.render', { component: 'Pane', requestId: 'my-pane' }, async ($, e) => {
    const { Box, Text, Button } = $.ui.resolve(e)
    const label = '…'
    const pct = '…'
    const untilReset = '…'
    const when = '…'
    const pct2 = '…'
    const label2 = '…'
    const pct3 = '…'
    const untilReset2 = '…'
    const when2 = '…'
    const pct4 = '…'
    const label3 = '…'
    const pct5 = '…'
    const untilReset3 = '…'
    const when3 = '…'
    const pct6 = '…'
    const cost = '…'
    return (
      <Box flexDirection="column">
        <Box flexDirection="column" marginBottom={1}>
          <Box flexDirection="row" justifyContent="space-between">
            <Text bold>
              {label}
              {" limit"}
            </Text>
            <Text bold>
              {pct}
              {" used"}
            </Text>
          </Box>
          <Text>
            ████████░░░░░░░░
          </Text>
          <Text dimColor>
            {"Resets in "}
            {untilReset}
            {" · "}
            {when}
          </Text>
          <Text dimColor>
            Used today: +
            {pct2}
            {" of the week"}
          </Text>
        </Box>
        <Box flexDirection="column" marginBottom={1}>
          <Box flexDirection="row" justifyContent="space-between">
            <Text bold>
              {label2}
              {" limit"}
            </Text>
            <Text bold>
              {pct3}
              {" used"}
            </Text>
          </Box>
          <Text>
            ████████░░░░░░░░
          </Text>
          <Text dimColor>
            {"Resets in "}
            {untilReset2}
            {" · "}
            {when2}
          </Text>
          <Text dimColor>
            Used today: +
            {pct4}
            {" of the week"}
          </Text>
        </Box>
        <Box flexDirection="column" marginBottom={1}>
          <Box flexDirection="row" justifyContent="space-between">
            <Text bold>
              {label3}
              {" limit"}
            </Text>
            <Text bold>
              {pct5}
              {" used"}
            </Text>
          </Box>
          <Text>
            ████████░░░░░░░░
          </Text>
          <Text dimColor>
            {"Resets in "}
            {untilReset3}
            {" · "}
            {when3}
          </Text>
          <Text dimColor>
            Used today: +
            {pct6}
            {" of the week"}
          </Text>
        </Box>
        <Text>
          This session: $
          {cost}
        </Text>
        <Text dimColor>
          {"Updated "}
          never
          {" · from Claude Code's latest response"}
        </Text>
        <Box flexDirection="row" marginTop={1} gap={1}>
          <Button label="Refresh" hotkey="r" />
          <Button label="Close" hotkey="c" role="dismiss" />
        </Box>
      </Box>
    )
  })

  on('session.start', async ($, e, next) => {
    const label = '…'
    const reset = '…'
    await $.ui.toast(`Limit Meter: ${label} limit reached${reset}`)
    return next(e)
  })
}