# frozen_string_literal: true

module AIRefactor
  class RunConfiguration
    def self.add_new_option(key)
      self.class.define_method(key) { instance_variable_get("@#{key}") }
      self.class.define_method("#{key}=") { |v| instance_variable_set("@#{key}", v) }
    end

    attr_reader :refactor,
      :description,
      :input_file_paths,
      :output_file_path,
      :output_template_path,
      :context_file_paths,
      :context_file_paths_from_gems,
      :context_text,
      :review_prompt,
      :prompt,
      :prompt_file_path,
      :overwrite,
      :diff,
      :verbose,
      :debug

    def set!(hash)
      hash.each do |key, value|
        raise StandardError, "Invalid option: #{key}" unless respond_to?("#{key}=")
        send("#{key}=", value)
      end
    end

    attr_writer :refactor, :description

    # @deprecated
    def [](key)
      send(key)
    end

    def input_file_paths=(paths)
      @input_file_paths ||= []
      paths = [paths] unless paths.is_a?(Array)
      @input_file_paths.concat(paths)
    end

    attr_writer :output_file_path

    attr_writer :output_template_path

    def context_file_paths=(paths)
      @context_file_paths ||= []
      paths = [paths] unless paths.is_a?(Array)
      @context_file_paths.concat(paths)
    end

    # A hash is passed in, where the keys are gem names that should be in the bundle and the path is a path inside the gem
    # install location. We resolve the absolute path of each and then add to @context_file_paths
    def context_file_paths_from_gems=(paths)
      @context_file_paths ||= []
      @context_file_paths_from_gems ||= {}
      @context_file_paths_from_gems.merge!(paths)

      paths.each do |gem_name, paths|
        paths = [paths] unless paths.is_a?(Array)
        paths.each do |path|
          gem_spec = Gem::Specification.find_by_name(gem_name.to_s)
          raise "Gem #{gem_name} not found" unless gem_spec
          gem_path = gem_spec.gem_dir
          full_path = File.join(gem_path, path)
          @context_file_paths << full_path
        end
      end
    end

    def context_text=(text)
      @context_text ||= ""
      @context_text += text
    end

    attr_writer :review_prompt
    attr_writer :prompt
    attr_writer :prompt_file_path

    def rspec_run_command
      @rspec_run_command || "bundle exec rspec __FILE__"
    end

    def minitest_run_command
      @minitest_run_command || "ruby __FILE__"
    end

    attr_writer :rspec_run_command
    attr_writer :minitest_run_command

    def ai_max_attempts
      @ai_max_attempts || 3
    end

    def ai_max_attempts=(value)
      @ai_max_attempts = value
    end

    def ai_model
      @ai_model || "gpt-4-turbo"
    end

    def ai_model=(value)
      @ai_model = value
    end

    def ai_platform
      if ai_model&.start_with?("claude")
        "anthropic"
      else
        "openai"
      end
    end

    def ai_temperature
      @ai_temperature || 0.7
    end

    attr_writer :ai_temperature

    def ai_max_tokens
      @ai_max_tokens || 1500
    end

    attr_writer :ai_max_tokens

    def ai_timeout
      @ai_timeout || 60
    end

    attr_writer :ai_timeout

    def overwrite
      @overwrite || "a"
    end

    attr_writer :overwrite

    attr_writer :diff

    attr_writer :verbose

    attr_writer :debug

    def to_options
      instance_variables.each_with_object({}) do |var, hash|
        hash[var.to_s.delete("@").to_sym] = instance_variable_get(var)
      end
    end
  end
end