local make_prompt = require("99.ops.make-prompt")
local CleanUp = require("99.ops.clean-up")
local QFixHelpers = require("99.ops.qfix-helpers")

local make_observer = CleanUp.make_observer

--- @param context _99.Prompt
--- @param response string
local function create_search_locations(context, response)
  local qf_list = QFixHelpers.create_qfix_entries(response)
  context.logger:set_area("search"):debug("qf_list created", "qf_list", qf_list)
  context.data = {
    type = "search",
    qfix_items = qf_list,
    response = response,
  }

  if #qf_list > 0 then
    require("99").open_qfix_for_request(context)
  else
    vim.notify("No search results found", vim.log.levels.INFO)
  end
end

--- @param context _99.Prompt
---@param opts _99.ops.SearchOpts
local function search(context, opts)
  opts = opts or {}

  local logger = context.logger:set_area("search")
  logger:debug("search", "with opts", opts.additional_prompt)

  local prompt, refs =
    make_prompt(context, context._99.prompts.prompts.semantic_search(), opts)

  context:add_prompt_content(prompt)
  context:add_references(refs)

  --- TODO: part of the context request clean up there needs to be a refactoring of
  --- make observer... it really should just be within the context observer creation.
  --- same with cleanup.. that should just be clean_ups from context, instead of a
  --- once cleanup function wrapper.
  ---
  --- i think an interface, CleanUpI could be something that is worth it :)
  context:start_request(make_observer(context, function(status, response)
    if status == "cancelled" then
      logger:debug("request cancelled for search")
    elseif status == "failed" then
      logger:error(
        "request failed for search",
        "error response",
        response or "no response provided"
      )
    elseif status == "success" then
      create_search_locations(context, response)
      context._99:sync()
    end
  end))
end
return search