require 'cli/ui'

module CLI
  module UI
    module ANSI
      ESC = "\x1b"

      # ANSI escape sequences (like \x1b[31m) have zero width.
      # when calculating the padding width, we must exclude them.
      # This also implements a basic version of utf8 character width calculation like
      # we could get for real from something like utf8proc.
      #
      def self.printing_width(str)
        zwj = false
        strip_codes(str).codepoints.reduce(0) do |acc, cp|
          if zwj
            zwj = false
            next acc
          end
          case cp
          when 0x200d # zero-width joiner
            zwj = true
            acc
          else
            acc + 1
          end
        end
      end

      # Strips ANSI codes from a str
      #
      # ==== Attributes
      #
      # - +str+ - The string from which to strip codes
      #
      def self.strip_codes(str)
        str.gsub(/\x1b\[[\d;]+[A-z]|\r/, '')
      end

      # Returns an ANSI control sequence
      #
      # ==== Attributes
      #
      # - +args+ - Argument to pass to the ANSI control sequence
      # - +cmd+ - ANSI control sequence Command
      #
      def self.control(args, cmd)
        ESC + "[" + args + cmd
      end

      # https://en.wikipedia.org/wiki/ANSI_escape_code#graphics
      def self.sgr(params)
        control(params.to_s, 'm')
      end

      # Cursor Movement

      # Move the cursor up n lines
      #
      # ==== Attributes
      #
      # * +n+ - number of lines by which to move the cursor up
      #
      def self.cursor_up(n = 1)
        return '' if n.zero?
        control(n.to_s, 'A')
      end

      # Move the cursor down n lines
      #
      # ==== Attributes
      #
      # * +n+ - number of lines by which to move the cursor down
      #
      def self.cursor_down(n = 1)
        return '' if n.zero?
        control(n.to_s, 'B')
      end

      # Move the cursor forward n columns
      #
      # ==== Attributes
      #
      # * +n+ - number of columns by which to move the cursor forward
      #
      def self.cursor_forward(n = 1)
        return '' if n.zero?
        control(n.to_s, 'C')
      end

      # Move the cursor back n columns
      #
      # ==== Attributes
      #
      # * +n+ - number of columns by which to move the cursor back
      #
      def self.cursor_back(n = 1)
        return '' if n.zero?
        control(n.to_s, 'D')
      end

      # Move the cursor to a specific column
      #
      # ==== Attributes
      #
      # * +n+ - The column to move to
      #
      def self.cursor_horizontal_absolute(n = 1)
        control(n.to_s, 'G')
      end

      # Show the cursor
      #
      def self.show_cursor
        control('', "?25h")
      end

      # Hide the cursor
      #
      def self.hide_cursor
        control('', "?25l")
      end

      # Save the cursor position
      #
      def self.cursor_save
        control('', 's')
      end

      # Restore the saved cursor position
      #
      def self.cursor_restore
        control('', 'u')
      end

      # Move to the next line
      #
      def self.next_line
        cursor_down + control('1', 'G')
      end

      # Move to the previous line
      #
      def self.previous_line
        cursor_up + control('1', 'G')
      end

      def self.clear_to_end_of_line
        control('', 'K')
      end
    end
  end
end