import { Hono } from 'hono'
import {
  pollWork,
  ackWork,
  stopWork,
  heartbeatWork,
} from '../../services/work-dispatch'
import { apiKeyAuth, acceptCliHeaders } from '../../auth/middleware'
import { updatePollTime } from '../../services/environment'

const app = new Hono()

/** GET /v1/environments/:id/work/poll — Long-poll for work */
app.get('/:id/work/poll', acceptCliHeaders, apiKeyAuth, async c => {
  const envId = c.req.param('id')!
  updatePollTime(envId)
  const result = await pollWork(envId)
  if (!result) {
    // Return 204 No Content so the client's axios parses it as null
    return c.body(null, 204)
  }
  return c.json(result, 200)
})

/** POST /v1/environments/:id/work/:workId/ack — Acknowledge work */
app.post('/:id/work/:workId/ack', acceptCliHeaders, apiKeyAuth, async c => {
  const workId = c.req.param('workId')!
  ackWork(workId)
  return c.json({ status: 'ok' }, 200)
})

/** POST /v1/environments/:id/work/:workId/stop — Stop work */
app.post('/:id/work/:workId/stop', acceptCliHeaders, apiKeyAuth, async c => {
  const workId = c.req.param('workId')!
  stopWork(workId)
  return c.json({ status: 'ok' }, 200)
})

/** POST /v1/environments/:id/work/:workId/heartbeat — Heartbeat */
app.post(
  '/:id/work/:workId/heartbeat',
  acceptCliHeaders,
  apiKeyAuth,
  async c => {
    const workId = c.req.param('workId')!
    const result = heartbeatWork(workId)
    // Work item was stopped (e.g. session archived via Web UI) — return 410
    // so the worker treats it as a fatal heartbeat and tears down the child
    // process, releasing capacity.
    if (result.state === 'completed') {
      return c.json(
        { error: { type: 'gone', message: 'Work item has been stopped' } },
        410,
      )
    }
    return c.json(result, 200)
  },
)

export default app