Process commands with Gitaly if a flag is passed from the app

lib/gitlab_access_status.rb

 require 'json'
 class GitAccessStatus
-  attr_reader :message, :repository_path
-  def initialize(status, message, repository_path)
+  attr_reader :message, :repository_path, :gitaly_address
+  def initialize(status, message, repository_path, gitaly_address = nil)
     @status = status
     @message = message
     @repository_path = repository_path
+    @gitaly_address = gitaly_address
   end
   def self.create_from_json(json)
     values = JSON.parse(json)
-    self.new(values["status"], values["message"], values["repository_path"])
+    self.new(values["status"], values["message"], values["repository_path"], values["gitaly_address"])
   end
   def allowed?

lib/gitlab_shell.rb

@@ -13,7 +13,7 @@ class GitlabShell
   API_COMMANDS = %w(2fa_recovery_codes)
   GL_PROTOCOL = 'ssh'.freeze
-  attr_accessor :key_id, :repo_name, :command, :git_access
+  attr_accessor :key_id, :repo_name, :command, :git_access, :gitaly_address
   attr_reader :repo_path
   def initialize(key_id)
@@ -98,6 +98,7 @@ class GitlabShell
     raise AccessDeniedError, status.message unless status.allowed?
     self.repo_path = status.repository_path
+    self.gitaly_address = status.gitaly_address
   end
   def process_cmd(args)
@@ -126,9 +127,14 @@ class GitlabShell
         $logger.info "gitlab-shell: Processing LFS authentication for #{log_username}."
         lfs_authenticate
       end
-    else
+    end
+
+    if gitaly_address.nil?
       $logger.info "gitlab-shell: executing git command <#{@command} #{repo_path}> for #{log_username}."
       exec_cmd(@command, repo_path)
+    else
+      $logger.info "gitlab-shell: processing with gitaly <#{gitaly_address}> command: <#{@command}> repo: <#{repo_path}> for <#{log_username}>."
+      process_cmd_with_gitaly(@command, repo_path)
     end
   end
@@ -165,6 +171,10 @@ class GitlabShell
     Kernel::exec(env, *args, unsetenv_others: true)
   end
+  def process_cmd_with_gitaly(cmd, repo_path)
+    # To be implemented
+  end
+
   def api
     GitlabNet.new
   end