import type { PluginInput } from "@opencode-ai/plugin"
import {
  findNearestMessageWithFields,
  findNearestMessageWithFieldsFromSDK,
} from "../../features/hook-message-injector"
import { getMessageDir, isSqliteBackend } from "../../shared"
import type { ModelInfo } from "./types"

export async function resolveRecentModelForSession(
  ctx: PluginInput,
  sessionID: string
): Promise<ModelInfo | undefined> {
  try {
    const messagesResp = await ctx.client.session.messages({ path: { id: sessionID } })
    const messages = (messagesResp.data ?? []) as Array<{
      info?: { model?: ModelInfo; modelID?: string; providerID?: string }
    }>

    for (let i = messages.length - 1; i >= 0; i--) {
      const info = messages[i].info
      const model = info?.model
      if (model?.providerID && model?.modelID) {
        return { providerID: model.providerID, modelID: model.modelID }
      }

      if (info?.providerID && info?.modelID) {
        return { providerID: info.providerID, modelID: info.modelID }
      }
    }
  } catch {
    // ignore - fallback to message storage
  }

  let currentMessage = null
  if (isSqliteBackend()) {
    currentMessage = await findNearestMessageWithFieldsFromSDK(ctx.client, sessionID)
  } else {
    const messageDir = getMessageDir(sessionID)
    currentMessage = messageDir ? findNearestMessageWithFields(messageDir) : null
  }
  const model = currentMessage?.model
  if (!model?.providerID || !model?.modelID) {
    return undefined
  }
  return { providerID: model.providerID, modelID: model.modelID }
}