import type { createOpencodeClient } from "@opencode-ai/sdk"
import type { MessageData } from "./types"
import { findMessagesWithThinkingBlocks, stripThinkingParts } from "./storage"
import { isSqliteBackend } from "../../shared/opencode-storage-detection"
import { stripThinkingPartsAsync } from "./storage/thinking-strip"
import { THINKING_TYPES } from "./constants"
import { log } from "../../shared/logger"
import { normalizeSDKResponse } from "../../shared"

type Client = ReturnType<typeof createOpencodeClient>

export async function recoverThinkingDisabledViolation(
  client: Client,
  sessionID: string,
  _failedAssistantMsg: MessageData
): Promise<boolean> {
  if (isSqliteBackend()) {
    return recoverThinkingDisabledViolationFromSDK(client, sessionID)
  }

  const messagesWithThinking = findMessagesWithThinkingBlocks(sessionID)
  if (messagesWithThinking.length === 0) {
    return false
  }

  let anySuccess = false
  for (const messageID of messagesWithThinking) {
    if (stripThinkingParts(messageID)) {
      anySuccess = true
    }
  }

  return anySuccess
}

async function recoverThinkingDisabledViolationFromSDK(
  client: Client,
  sessionID: string
): Promise<boolean> {
  try {
    const response = await client.session.messages({ path: { id: sessionID } })
    const messages = normalizeSDKResponse(response, [] as MessageData[], { preferResponseOnMissingData: true })

    const messageIDsWithThinking: string[] = []
    for (const msg of messages) {
      if (msg.info?.role !== "assistant") continue
      if (!msg.info?.id) continue
      if (!msg.parts) continue

      const hasThinking = msg.parts.some((part) => THINKING_TYPES.has(part.type))
      if (hasThinking) {
        messageIDsWithThinking.push(msg.info.id)
      }
    }

    if (messageIDsWithThinking.length === 0) {
      return false
    }

    let anySuccess = false
    for (const messageID of messageIDsWithThinking) {
      if (await stripThinkingPartsAsync(client, sessionID, messageID)) {
        anySuccess = true
      }
    }

    return anySuccess
  } catch (error) {
    log("[session-recovery] recoverThinkingDisabledViolationFromSDK failed", {
      sessionID,
      error: String(error),
    })
    return false
  }
}