usage-band

import type { Register } from 'claude-code'

// Drawn like usage-band by Wilson Kinyua (pane, band, toast), as modscode read it from the mod's code:
// https://modscode.com/gallery/wilsonkinyua-usage-band/
// 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, Button } = $.ui.resolve(e)
    const glyph = '…'
    const percent = '…'
    const label = '…'
    const detail = '…'
    const glyph2 = '…'
    const percent2 = '…'
    const glyph3 = '…'
    const percent3 = '…'
    const glyph4 = '…'
    const percent4 = '…'
    const usd = '…'
    const branch = '…'
    const changes = '…'
    return (
      <Box flexDirection="column">
        <Box flexDirection="row" flexWrap="wrap" alignItems="center" columnGap={3} paddingX={1}>
          <Box flexDirection="row" alignItems="center" gap={1}>
            <Text>
              {glyph}
            </Text>
            <Text bold>
              {percent}
              %
            </Text>
            <Text dimColor>
              {`${label} · ${detail}`}
            </Text>
          </Box>
          <Box flexDirection="row" alignItems="center" gap={1}>
            <Text>
              {glyph2}
            </Text>
            <Text bold>
              {percent2}
              %
            </Text>
            <Text dimColor>
              {`${label} · ${detail}`}
            </Text>
          </Box>
          <Box flexDirection="row" alignItems="center" gap={1}>
            <Text>
              {glyph3}
            </Text>
            <Text bold>
              {percent3}
              %
            </Text>
            <Text dimColor>
              {`${label} · ${detail}`}
            </Text>
          </Box>
          <Box flexDirection="row" alignItems="center" gap={1}>
            <Text>
              {glyph4}
            </Text>
            <Text bold>
              {percent4}
              %
            </Text>
            <Text dimColor>
              {"ctx · "}
            </Text>
          </Box>
          <Box flexDirection="row" alignItems="center" gap={1}>
            <Text bold color="#3fb950">
              {usd}
            </Text>
            <Text dimColor>
              {`${usd} today  ${usd} mo`}
            </Text>
          </Box>
          <Box flexDirection="row" alignItems="center" gap={1}>
            <Text color="#58a6ff">
              {"⎇ "}
              {branch}
            </Text>
            <Text color="#d29922">
              ±
              {changes}
            </Text>
          </Box>
          <Button label="details" plain />
        </Box>
        {await next(e)}
      </Box>
    )
  })

  // the pane: open it with $.ui.open({ id: 'my-pane', title: "Usage" })
  on('ui.render', { component: 'Pane', requestId: 'my-pane' }, async ($, e) => {
    const { Box, Text } = $.ui.resolve(e)
    const totalTokens = '…'
    const maxTokens = '…'
    const percentage = '…'
    const name = '…'
    const tokens = '…'
    const name2 = '…'
    const tokens2 = '…'
    const name3 = '…'
    const tokens3 = '…'
    const day = '…'
    const usd = '…'
    const day2 = '…'
    const usd2 = '…'
    const day3 = '…'
    const usd3 = '…'
    return (
      <Box flexDirection="column" gap={1}>
        <Box flexDirection="column">
          <Text bold>
            Context
          </Text>
          <Text dimColor>
            {totalTokens}
            {" / "}
            {maxTokens}
            {" tokens ("}
            {percentage}
            %)
          </Text>
          <Text>
            <Text>
              {"■ "}
            </Text>
            {name}
            {": "}
            {tokens}
          </Text>
          <Text>
            <Text>
              {"■ "}
            </Text>
            {name2}
            {": "}
            {tokens2}
          </Text>
          <Text>
            <Text>
              {"■ "}
            </Text>
            {name3}
            {": "}
            {tokens3}
          </Text>
        </Box>
        <Box flexDirection="column">
          <Text bold>
            Spend (sessions with this mod loaded)
          </Text>
          <Text>
            {day}
            {" "}
            {usd}
          </Text>
          <Text>
            {day2}
            {" "}
            {usd2}
          </Text>
          <Text>
            {day3}
            {" "}
            {usd3}
          </Text>
        </Box>
      </Box>
    )
  })

  on('session.start', async ($, e, next) => {
    const LABELS = '…'
    const percentUsed = '…'
    const countdown = '…'
    await $.ui.toast(`${LABELS} limit at ${percentUsed}% · resets in ${countdown}`)
    return next(e)
  })
}