class Command::ChatQuery < Command
  store_accessor :data, :query, :params

  def title
    "Chat query '#{query}'"
  end

  def execute
    response = chat.ask query
    Rails.logger.info "***\n#{response.content}\n***"
    response = replace_names_with_ids(JSON.parse(response.content))
    Command::Result::ChatResponse.new(JSON.pretty_generate(response))
  end

  private
    def chat
      chat = RubyLLM.chat
      chat.with_instructions(prompt)
    end

    def prompt
      <<~PROMPT
        You are a helpful assistant that translates natural language into commands that Fizzy understand, and that
        can answer questions about the Fizzy system and the information it contains.

        Fizzy supports the following commands:

        - Assign users to cards: /assign user
        - Close cards: /close optional reason
        - Tag cards: /tag tag-name
        - Search cards: /search. See how this works below:

        asks for a certain set of cards, you can use the /search command to filter. It supports the following
        conditions:

          - assignment_status: can be "unassigned"
          - indexed_by: can be "newest", "oldest", "latest", "stalled", "closed"
          - engagement_status: can be "considering" or "doing"
          - card_ids: a list of card ids
          - assignee_ids: a list of assignee names
          - creator_id: the name of a person
          - collection_ids: a list of collection names. Cards are contained in collections. Don't use unless mentioning
              specific collections.
          - tag_ids: a list of tag names.
          - terms: a list of terms to search for. Use this option to refine searches based on further keyword-based
             queries.

        Please combine commands to satisfy what the user needs. E.g: search with keywords and filters and then apply
        as many commands as needed.

        The output will be in JSON. It will contain a list of commands. Each command will be a JSON object like:

        { command: "/close" }

        For the case of the /search command, it will also contain the additional params. E.g:

        { command: "/search", indexed_by: "closed", collection_ids: [ "Writebook", "Design" ] }

        Notice that only /search commands carry additional JSON params. For /tag, /close and /assign just append the
        param to the string command.

        Make sure to place into double quotes the strings in JSON values and that you generate valid JSON. I want a
        JSON list like [{}, {}...]
      PROMPT
    end

    def replace_names_with_ids(commands)
      commands.each do |command|
        if command["command"] == "/search"
          command["assignee_ids"] = command["assignee_ids"]&.filter_map { |name| assignee_from(name).id }
          command["creator_id"] = assignee_from(command["creator_id"]).id if command["creator_id"]
          command["collection_ids"] = command["collection_ids"]&.filter_map { |name| Collection.where("lower(name) = ?", name.downcase).first&.id }
          command["tag_ids"] = command["tag_ids"]&.filter_map { |name| ::Tag.find_by_title(name)&.id }
          command.compact!
        end
      end
    end

    def assignee_from(string)
      string_without_at = string.delete_prefix("@")
      User.all.find { |user| user.mentionable_handles.include?(string_without_at) }
    end

end