Codebase list ruby-gitlab / scrub-obsolete/main lib / gitlab / shell.rb
scrub-obsolete/main

Tree @scrub-obsolete/main (Download .tar.gz)

shell.rb @scrub-obsolete/mainraw · history · blame

# frozen_string_literal: true

require 'gitlab'
require 'gitlab/help'
require 'gitlab/cli_helpers'
require 'gitlab/shell_history'
require 'readline'
require 'shellwords'

class Gitlab::Shell
  extend Gitlab::CLI::Helpers

  class << self
    attr_reader :arguments, :command

    def start
      trap('INT') { quit_shell } # capture ctrl-c
      setup

      while (buffer = Readline.readline('gitlab> '))
        begin
          parse_input buffer

          @arguments.map! { |arg| symbolize_keys(yaml_load(arg)) }

          case buffer
          when nil, ''
            next
          when 'exit'
            quit_shell
          when /^\bhelp\b+/
            puts help(arguments[0]) { |out| out.gsub!(/Gitlab\./, 'gitlab> ') }
          else
            history << buffer

            data = execute command, arguments
            output_table command, arguments, data
          end
        rescue StandardError => e
          puts e.message
        end
      end

      quit_shell # save history if user presses ctrl-d
    end

    def parse_input(buffer)
      buf = Shellwords.shellwords(buffer)

      @command = buf.shift
      @arguments = buf.count.positive? ? buf : []
    end

    def setup
      history.load

      Readline.completion_proc = completion
      Readline.completion_append_character = ' '
    end

    # Gets called when user hits TAB key to do completion
    def completion
      proc { |str| actions.map(&:to_s).grep(/^#{Regexp.escape(str)}/) }
    end

    # Execute a given command with arguements
    def execute(cmd = command, args = arguments)
      raise "Unknown command: #{cmd}. See the 'help' for a list of valid commands." unless actions.include?(cmd.to_sym)

      confirm_command(cmd)
      gitlab_helper(cmd, args)
    end

    def quit_shell
      history.save
      exit
    end

    def history
      @history ||= History.new
    end
  end
end