cache-caffeine

  on('ui.render', { component: 'AbovePrompt' }, async ($, e, next) => {
    if ((e.surface !== 'terminal' && e.surface !== 'desktop') || e.props.hasSurvey) {
      return next(e)
    }
    const { Box, Button, Input, Text } = $.ui.resolve(e)
    const now = await $.clock.now()
    c.bandId = e.requestId

    // the wake row: near the limit an offer, once booked a countdown
    const block: Block | null = blocking(c.limits)
    const isOffered = !c.armed && block !== null && (isNear(c.limits) || c.isHit) && c.dismissed !== block.resetsAt
    const armed = c.armed
    const wakeRow = armed ? (
      <Box key="caffeine-wake-row" flexDirection="row" columnGap={2}>
        <Text key="wake-text" color="cyan" wrap="truncate-end">
          {`caffeine · Claude continues at ${clock(armed.at, now)} (in ${span(armed.at - now)})`}
        </Text>
        <Button key="caffeine-wake-cancel" plain dimColor hotkey="n" onPress={() => void disarm($, c).then(() => tell($, 'wake cancelled.'))}>
          cancel
        </Button>
      </Box>
    ) : isOffered ? (
      <Box key="caffeine-wake-row" flexDirection="row" columnGap={2}>
        <Text key="wake-text" color={block!.percent >= 100 ? 'red' : 'yellow'} wrap="truncate-end">
          {`${block!.label} limit ${block!.percent}% · resets ${clock(block!.resetsAt, now)} (in ${span(block!.resetsAt - now)})`}
        </Text>
        <Button key="caffeine-wake" plain hotkey="u" onPress={() => void armAtReset($, c)}>
          {`wake Claude at ${clock(wakeAfter(block!.resetsAt), now)}`}
        </Button>
        <Button
          key="caffeine-wake-dismiss"
          plain
          dimColor
          hotkey="n"
          onPress={() => {
            c.dismissed = block!.resetsAt
            $.ui.invalidate('ui.render')
          }}
        >
          not now
        </Button>
      </Box>
    ) : null

    // the switch row, under its cup
    const line = describe(c, now)
    const warm = cup(c, now)
    const row = !c.isBanded ? null : c.isEditing ? (
      <Box key="caffeine" flexDirection="row" columnGap={2}>
        <Input
          key="caffeine-message"
          label="caffeine says: "
          value={c.message}
          placeholder={DEFAULT_MESSAGE}
          submitLabel="save"
          autoFocus
          onSubmit={text => void setMessage($, c, text).then(saved => tell($, `the poke now says: ${saved}`))}
        />
        <Button
          key="caffeine-cancel"
          plain
          dimColor
          hotkey="q"
          onPress={() => {
            c.isEditing = false
            $.ui.invalidate('ui.render')
          }}
        >
          cancel
        </Button>
      </Box>
    ) : (
      <Box key="caffeine" flexDirection="column">
        <Text key="steam" color={warm?.color} dimColor={!warm}>
          {` ${warm?.steam ?? '   '}`}
        </Text>
        <Box key="caffeine-row" flexDirection="row" columnGap={2}>
          <Box key="caffeine-cup" flexDirection="row" columnGap={1} flexShrink={0}>
            <Text key="cup" color={warm?.color} dimColor={!warm}>
              c[_]
            </Text>
            <Text key="degrees" color={warm?.color} dimColor={!warm}>
              {warm ? `${warm.degrees}°F` : '--°F'}
            </Text>
          </Box>
          <Box key="caffeine-line" flexShrink={1} minWidth={0}>
            <Text key="text" color={line.color} dimColor={line.isDim} wrap="truncate-end">
              {line.text}
            </Text>
          </Box>
          <Box key="caffeine-buttons" flexDirection="row" columnGap={2} flexShrink={0}>
          <Button
            key="caffeine-toggle"
            plain
            hotkey="t"
            onPress={() => {
              if (c.isOn) c.isAutoDeclined = true
              void switchOn($, c, !c.isOn)
            }}
          >
            {c.isOn ? 'turn off' : 'turn on'}
          </Button>
          <Button key="caffeine-edit" plain dimColor hotkey="e" onPress={() => void editInBand($, c)}>
            message
          </Button>
          </Box>
        </Box>
      </Box>
    )

    if (!wakeRow && !row) {
      return next(e)
    }
    const below = await next(e)
    return (
      <Box flexDirection="column">
        {wakeRow}
        {row}
        {below}
      </Box>
    )
  })