branch-usage

import type { Register } from 'claude-code'

// Drawn like branch-usage by gpr (pane, band, toast), as modscode read it from the mod's code:
// https://modscode.com/gallery/gpr-branch-usage/
// 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, Link } = $.ui.resolve(e)
    const branch = '…'
    const url = '…'
    const number = '…'
    return (
      <Box flexDirection="column">
        <Box>
          <Text dimColor>
            0
            {" · "}
            {branch}
            {" "}
          </Text>
          <Link href={`${url}`}>
            <Text color="blue" underline>
              (#
              {number}
              )
            </Text>
          </Link>
        </Box>
        {await next(e)}
      </Box>
    )
  })

  // the pane: open it with $.ui.open({ id: 'my-pane', title: "Branch usage" })
  on('ui.render', { component: 'Pane', requestId: 'my-pane' }, async ($, e) => {
    const { Box, Text, Link } = $.ui.resolve(e)
    const branch = '…'
    const url = '…'
    const number = '…'
    const length = '…'
    const formatUsd = '…'
    const shortModel = '…'
    const sharePct = '…'
    const formatUsd2 = '…'
    const tokenLine = '…'
    const sharePct2 = '…'
    const formatUsd3 = '…'
    const tokenLine2 = '…'
    const sharePct3 = '…'
    const formatUsd4 = '…'
    const tokenLine3 = '…'
    const agent = '…'
    const requests = '…'
    const sharePct4 = '…'
    const formatUsd5 = '…'
    const tokenLine4 = '…'
    const sharePct5 = '…'
    const formatUsd6 = '…'
    const tokenLine5 = '…'
    const sharePct6 = '…'
    const formatUsd7 = '…'
    const tokenLine6 = '…'
    const sessionId = '…'
    const relativeTime = '…'
    const value = '…'
    const formatUsd8 = '…'
    const value2 = '…'
    const formatUsd9 = '…'
    const value3 = '…'
    const formatUsd10 = '…'
    return (
      <Box flexDirection="column">
        <Box flexDirection="row" justifyContent="space-between">
          <Box flexShrink={1}>
            <Box flexShrink={1}>
              <Text wrap="truncate-end" bold>
                {branch}
              </Text>
            </Box>
            <Box flexShrink={0}>
              <Text>
                {" "}
              </Text>
              <Link href={`${url}`}>
                <Text color="blue" underline>
                  (#
                  {number}
                  )
                </Text>
              </Link>
            </Box>
          </Box>
          <Box flexShrink={0} marginLeft={1}>
            <Text bold>
              0
            </Text>
          </Box>
        </Box>
        <Text dimColor>
          {"  "}
          {length}
          {" session"}
        </Text>
        <Text dimColor>
          {" this session, engine total "}
          {formatUsd}
        </Text>
        <Box marginTop={1}>
          <Text bold>
            By model
          </Text>
        </Box>
        <Box flexDirection="column">
          <Box flexDirection="row" justifyContent="space-between">
            <Box flexShrink={1}>
              <Text wrap="truncate-end">
                {`  - ${shortModel} ~`}
              </Text>
            </Box>
            <Box flexShrink={0} marginLeft={1}>
              <Text dimColor>
                {sharePct}
                {" "}
              </Text>
              <Text>
                {formatUsd2}
              </Text>
            </Box>
          </Box>
          <Text dimColor wrap="truncate-end">
            {" "}
            {tokenLine}
          </Text>
        </Box>
        <Box flexDirection="column">
          <Box flexDirection="row" justifyContent="space-between">
            <Box flexShrink={1}>
              <Text wrap="truncate-end">
                {`  - ${shortModel} ~`}
              </Text>
            </Box>
            <Box flexShrink={0} marginLeft={1}>
              <Text dimColor>
                {sharePct2}
                {" "}
              </Text>
              <Text>
                {formatUsd3}
              </Text>
            </Box>
          </Box>
          <Text dimColor wrap="truncate-end">
            {" "}
            {tokenLine2}
          </Text>
        </Box>
        <Box flexDirection="column">
          <Box flexDirection="row" justifyContent="space-between">
            <Box flexShrink={1}>
              <Text wrap="truncate-end">
                {`  - ${shortModel} ~`}
              </Text>
            </Box>
            <Box flexShrink={0} marginLeft={1}>
              <Text dimColor>
                {sharePct3}
                {" "}
              </Text>
              <Text>
                {formatUsd4}
              </Text>
            </Box>
          </Box>
          <Text dimColor wrap="truncate-end">
            {" "}
            {tokenLine3}
          </Text>
        </Box>
        <Box marginTop={1}>
          <Text bold>
            By agent
          </Text>
        </Box>
        <Box flexDirection="column">
          <Box flexDirection="row" justifyContent="space-between">
            <Box flexShrink={1}>
              <Text wrap="truncate-end">
                {`  - ${agent} · ${requests} req`}
              </Text>
            </Box>
            <Box flexShrink={0} marginLeft={1}>
              <Text dimColor>
                {sharePct4}
                {" "}
              </Text>
              <Text>
                {formatUsd5}
              </Text>
            </Box>
          </Box>
          <Text dimColor wrap="truncate-end">
            {" "}
            {tokenLine4}
          </Text>
        </Box>
        <Box flexDirection="column">
          <Box flexDirection="row" justifyContent="space-between">
            <Box flexShrink={1}>
              <Text wrap="truncate-end">
                {`  - ${agent} · ${requests} req`}
              </Text>
            </Box>
            <Box flexShrink={0} marginLeft={1}>
              <Text dimColor>
                {sharePct5}
                {" "}
              </Text>
              <Text>
                {formatUsd6}
              </Text>
            </Box>
          </Box>
          <Text dimColor wrap="truncate-end">
            {" "}
            {tokenLine5}
          </Text>
        </Box>
        <Box flexDirection="column">
          <Box flexDirection="row" justifyContent="space-between">
            <Box flexShrink={1}>
              <Text wrap="truncate-end">
                {`  - ${agent} · ${requests} req`}
              </Text>
            </Box>
            <Box flexShrink={0} marginLeft={1}>
              <Text dimColor>
                {sharePct6}
                {" "}
              </Text>
              <Text>
                {formatUsd7}
              </Text>
            </Box>
          </Box>
          <Text dimColor wrap="truncate-end">
            {" "}
            {tokenLine6}
          </Text>
        </Box>
        <Box marginTop={1}>
          <Text bold>
            By session
          </Text>
        </Box>
        <Box flexDirection="row" justifyContent="space-between">
          <Box flexShrink={1}>
            <Text wrap="truncate-end">
              {`  - ● ${sessionId}  ${relativeTime}`}
            </Text>
          </Box>
          <Box flexShrink={0} marginLeft={1}>
            <Text dimColor>
              {value}
              {" "}
            </Text>
            <Text>
              {formatUsd8}
            </Text>
          </Box>
        </Box>
        <Box flexDirection="row" justifyContent="space-between">
          <Box flexShrink={1}>
            <Text wrap="truncate-end">
              {`  - ● ${sessionId}  ${relativeTime}`}
            </Text>
          </Box>
          <Box flexShrink={0} marginLeft={1}>
            <Text dimColor>
              {value2}
              {" "}
            </Text>
            <Text>
              {formatUsd9}
            </Text>
          </Box>
        </Box>
        <Box flexDirection="row" justifyContent="space-between">
          <Box flexShrink={1}>
            <Text wrap="truncate-end">
              {`  - ● ${sessionId}  ${relativeTime}`}
            </Text>
          </Box>
          <Box flexShrink={0} marginLeft={1}>
            <Text dimColor>
              {value3}
              {" "}
            </Text>
            <Text>
              {formatUsd10}
            </Text>
          </Box>
        </Box>
      </Box>
    )
  })

  on('session.start', async ($, e, next) => {
    const pr = '…'
    await $.ui.toast(`branch-usage: PR #${pr} comment created`)
    return next(e)
  })
}