require "rbconfig"

module Textbringer
  module Utils
    module_function

    def message(msg, log: true, sit_for: nil, sleep_for: nil)
      str = msg.to_s
      if log && Buffer.current&.name != "*Messages*"
        buffer = Buffer["*Messages*"] ||
          Buffer.new_buffer("*Messages*", undo_limit: 0).tap { |b|
            b[:top_of_window] = b.new_mark
        }
        buffer.read_only = false
        begin
          buffer.end_of_buffer
          buffer.insert(str + "\n")
          if buffer.current_line > 1000
            buffer.beginning_of_buffer
            10.times do
              buffer.forward_line
            end
            buffer.delete_region(buffer.point_min, buffer.point)
            buffer.end_of_buffer
          end
        ensure
          buffer.read_only = true
        end
      end
      Window.echo_area.show(str)
      if sit_for
        sit_for(sit_for)
        Window.echo_area.clear_message
      end
      if sleep_for
        sleep_for(sleep_for)
        Window.echo_area.clear_message
      end
    end

    def sit_for(secs, no_redisplay = false)
      Window.redisplay unless no_redisplay
      Controller.current.wait_input((secs * 1000).to_i)
    end

    def sleep_for(secs)
      sleep(secs)
    end

    def background
      Thread.start do
        begin
          yield
        rescue Exception => e
          foreground do
            raise e
          end
        end
      end
    end

    def foreground(&block)
      Controller.current.next_tick(&block)
    end

    alias next_tick foreground

    def foreground!
      if Thread.current == Thread.main
        return yield
      end
      q = Queue.new
      foreground do
        begin
          result = yield
          q.push([:ok, result])
        rescue Exception => e
          q.push([:error, e])
        end
      end
      status, value = q.pop
      if status == :error
        raise value
      else
        value
      end
    end

    alias next_tick! foreground!

    def read_event
      Controller.current.read_event
    end

    def read_char
      event = Controller.current.read_event
      if !event.is_a?(String)
        raise EditorError, "Non character event: #{event.inspect}"
      end
      event
    end

    def received_keyboard_quit?
      Controller.current.received_keyboard_quit?
    end

    def show_exception(e)
      if e.is_a?(SystemExit) || e.is_a?(SignalException)
        raise
      end
      if !e.is_a?(Quit) && Buffer.current&.name != "*Backtrace*"
        buffer = Buffer.find_or_new("*Backtrace*", undo_limit: 0)
        if !buffer.mode.is_a?(BacktraceMode)
          buffer.apply_mode(BacktraceMode)
        end
        buffer.read_only = false
        begin
          buffer.delete_region(buffer.point_min, buffer.point_max)
          buffer.insert("#{e.class}: #{e}\n")
          if e.backtrace
            e.backtrace.each do |line|
              buffer.insert(line + "\n")
            end
          end
          buffer.beginning_of_buffer
        ensure
          buffer.read_only = true
        end
      end
      message(e.to_s.chomp)
      Window.beep
    end

    COMPLETION = {
      original_buffer: nil,
      completions_window: nil
    }

    def read_from_minibuffer(prompt, completion_proc: nil, default: nil,
                             initial_value: nil, completion_ignore_case: false,
                             keymap: MINIBUFFER_LOCAL_MAP)
      if Window.echo_area.active?
        raise EditorError,
          "Command attempted to use minibuffer while in minibuffer"
      end
      old_buffer = Buffer.current
      old_window = Window.current
      old_completion_proc = Buffer.minibuffer[:completion_proc]
      old_completion_ignore_case = Buffer.minibuffer[:completion_ignore_case]
      old_current_prefix_arg = Controller.current.current_prefix_arg
      old_minibuffer_map = Buffer.minibuffer.keymap
      Buffer.minibuffer.keymap = keymap
      Buffer.minibuffer[:completion_proc] = completion_proc
      Buffer.minibuffer[:completion_ignore_case] = completion_ignore_case
      Window.echo_area.active = true
      begin
        Window.current = Window.echo_area
        Buffer.minibuffer.clear
        Buffer.minibuffer.insert(initial_value) if initial_value
        if default
          prompt = prompt.sub(/:/, " (default #{default}):")
        end
        Window.echo_area.prompt = prompt
        Window.echo_area.redisplay
        Window.update
        recursive_edit
        s = Buffer.minibuffer.to_s
        if default && s.empty?
          default
        else
          s
        end
      ensure
        Window.echo_area.clear
        Window.echo_area.redisplay
        Window.update
        Window.echo_area.active = false
        Window.current = old_window
        # Just in case old_window has been deleted by resize,
        # in which case Window.current is set to the first window.
        Window.current.buffer = Buffer.current = old_buffer
        Buffer.minibuffer[:completion_ignore_case] = old_completion_ignore_case
        Buffer.minibuffer[:completion_proc] = old_completion_proc
        Buffer.minibuffer.keymap = old_minibuffer_map
        Buffer.minibuffer.disable_input_method
        Controller.current.current_prefix_arg = old_current_prefix_arg
        if COMPLETION[:original_buffer]
          COMPLETION[:completions_window].buffer = COMPLETION[:original_buffer]
          COMPLETION[:completions_window] = nil
          COMPLETION[:original_buffer] = nil
        end
      end
    end

    def read_file_name(prompt, default: nil)
      f = ->(s) {
        s = File.expand_path(s) if s.start_with?("~")
        Dir.glob(s + "*").map { |file|
          if File.directory?(file) && !file.end_with?(?/)
            file + "/"
          else
            file
          end
        }
      }
      initial_value = default&.sub(%r"\A#{Regexp.quote(Dir.pwd)}/", "")
      ignore_case = CONFIG[:read_file_name_completion_ignore_case]
      file = read_from_minibuffer(prompt, completion_proc: f,
                                  initial_value: initial_value,
                                  completion_ignore_case: ignore_case)
      File.expand_path(file)
    end

    def complete_for_minibuffer(s, candidates)
      candidates.select { |i| i.start_with?(s) }
    end

    def read_buffer(prompt, default: Buffer.other.name)
      f = ->(s) { complete_for_minibuffer(s, Buffer.names) }
      read_from_minibuffer(prompt, completion_proc: f, default: default)
    end

    def read_command_name(prompt)
      f = ->(s) {
        complete_for_minibuffer(s.tr("-", "_"), Commands.list.map(&:to_s))
      }
      read_from_minibuffer(prompt, completion_proc: f)
    end

    def read_encoding(prompt, **opts)
      encoding_names = (Encoding.list.map(&:name) + Encoding.aliases.keys).
        map(&:downcase).uniq
      f = ->(s) {
        complete_for_minibuffer(s.downcase, encoding_names)
      }
      read_from_minibuffer(prompt, completion_proc: f, **opts)
    end

    def yes_or_no?(prompt)
      loop {
        s = read_from_minibuffer(prompt + " (yes or no) ")
        case s
        when "yes"
          return true
        when "no"
          return false
        else
          message("Please answer yes or no.", sit_for: 2)
        end
      }
    end

    define_keymap :Y_OR_N_MAP
    Y_OR_N_MAP.define_key(?y, :self_insert_and_exit_minibuffer)
    Y_OR_N_MAP.define_key(?n, :self_insert_and_exit_minibuffer)
    Y_OR_N_MAP.define_key(?\C-g, :abort_recursive_edit)
    Y_OR_N_MAP.handle_undefined_key do |key|
      :exit_recursive_edit
    end

    def self_insert_and_exit_minibuffer
      self_insert
      exit_recursive_edit
    end

    def y_or_n?(prompt)
      new_prompt = prompt + " (y or n) "
      prompt_modified = false
      loop do
        s = read_from_minibuffer(new_prompt, keymap: Y_OR_N_MAP)
        case s
        when ?y
          break true
        when ?n
          break false
        else
          unless prompt_modified
            new_prompt.prepend("Answer y or n. ")
            prompt_modified = true
          end
        end
      end
    end

    def read_single_char(prompt, chars)
      map = Keymap.new
      chars.each do |c|
        map.define_key(c, :self_insert_and_exit_minibuffer)
      end
      map.define_key(?\C-g, :abort_recursive_edit)
      char_options = chars.join(?/)
      map.handle_undefined_key do |key|
        -> { message("Invalid key.  Type C-g to quit.", sit_for: 2) }
      end
      read_from_minibuffer(prompt + " (#{char_options}) ", keymap: map)
    end

    def read_key_sequence(prompt)
      buffer = Buffer.current
      key_sequence = []
      map = Keymap.new
      map.define_key("\C-g", :abort_recursive_edit)
      map.handle_undefined_key do |key|
        -> {
          key_sequence.push(key)
          cmd = buffer.keymap&.lookup(key_sequence) ||
            GLOBAL_MAP.lookup(key_sequence)
          if !cmd.is_a?(Keymap)
            exit_recursive_edit
          end
          Buffer.current.clear
          keys = Keymap.key_sequence_string(key_sequence)
          Buffer.current.insert("#{keys}-")
        }
      end
      read_from_minibuffer(prompt, keymap: map)
      if buffer.keymap&.lookup(key_sequence) ||
          GLOBAL_MAP.lookup(key_sequence)
        key_sequence
      else
        keys = Keymap.key_sequence_string(key_sequence)
        raise EditorError, "#{keys} is undefined"
      end
    end

    HOOKS = Hash.new { |h, k| h[k] = [] }

    def add_hook(name, func = nil, &block)
      HOOKS[name].unshift(func || block)
    end

    def remove_hook(name, func)
      HOOKS[name].delete(func)
    end

    def run_hooks(name, remove_on_error: false)
      HOOKS[name].delete_if do |func|
        begin
          case func
          when Symbol
            send(func)
          else
            func.call
          end
          false
        rescue Exception => e
          raise if e.is_a?(SystemExit)
          if remove_on_error
            true
          else
            raise
          end
        end
      end
    end

    def set_transient_map(map)
      old_overriding_map = Controller.current.overriding_map
      hook = -> {
        Controller.current.overriding_map = old_overriding_map
        remove_hook(:pre_command_hook, hook)
      }
      add_hook(:pre_command_hook, hook)
      Controller.current.overriding_map = map
    end

    def ruby_install_name
      RbConfig::CONFIG["ruby_install_name"]
    end

    [
      :beginning_of_buffer?,
      :end_of_buffer?,
      :beginning_of_line?,
      :end_of_line?,
      :insert,
      :gsub
    ].each do |name|
      define_method(name) do |*args, &block|
        Buffer.current.send(name, *args, &block)
      end
    end
  end
end