orca/mobile/src/source-control/use-mobile-commit-message-generation.ts

import { useCallback, type MutableRefObject } from 'react'
import type { RpcClient } from '../transport/rpc-client'
import { triggerError, triggerSuccess } from '../platform/haptics'
import { cancelMobileCommitMessage, requestMobileCommitMessage } from './mobile-commit-message-ai'

type Params = {
  client: RpcClient | null
  worktreeId: string
  generatingMessage: boolean
  mountedRef: MutableRefObject<boolean>
  busyActionRef: MutableRefObject<string | null>
  setGeneratingMessage: (next: boolean) => void
  setCommitMessage: (next: string) => void
  setActionError: (next: string | null) => void
}

// AI commit-message generation + cancellation. Split from the runners hook to
// keep each file under the line limit; behavior is unchanged.
export function useMobileCommitMessageGeneration(params: Params) {
  const {
    client,
    worktreeId,
    generatingMessage,
    mountedRef,
    busyActionRef,
    setGeneratingMessage,
    setCommitMessage,
    setActionError
  } = params

  // AI-generate a commit message from the staged diff. Matches desktop: the
  // button is always available; a missing model surfaces as a toast.
  const generateCommitMessage = useCallback(async () => {
    if (!client || generatingMessage || busyActionRef.current) {
      return
    }
    setGeneratingMessage(true)
    setActionError(null)
    try {
      const result = await requestMobileCommitMessage(client, worktreeId)
      if (!mountedRef.current) {
        return
      }
      if (result.success) {
        setCommitMessage(result.message)
        triggerSuccess()
      } else if (!result.canceled) {
        triggerError()
        setActionError(result.error)
      }
    } catch (err) {
      // Why: a transport drop rejects the RPC; without this the error haptic +
      // message are skipped and the rejection escapes the void-called handler.
      if (mountedRef.current) {
        triggerError()
        setActionError(err instanceof Error ? err.message : 'Failed to generate commit message')
      }
    } finally {
      if (mountedRef.current) {
        setGeneratingMessage(false)
      }
    }
  }, [
    busyActionRef,
    client,
    generatingMessage,
    mountedRef,
    setActionError,
    setCommitMessage,
    setGeneratingMessage,
    worktreeId
  ])

  const cancelGenerateCommitMessage = useCallback(() => {
    if (client) {
      void cancelMobileCommitMessage(client, worktreeId)
    }
  }, [client, worktreeId])

  return { generateCommitMessage, cancelGenerateCommitMessage }
}