BigW Consortium Gitlab

workhorse.rb 896 Bytes
Newer Older
1 2 3 4 5
require 'base64'
require 'json'

module Gitlab
  class Workhorse
Jacob Vosmaer committed
6
    SEND_DATA_HEADER = 'Gitlab-Workhorse-Send-Data'
7

Jacob Vosmaer committed
8
    class << self
9
      def send_git_blob(repository, blob)
10
        params = {
11 12 13 14 15
          'RepoPath' => repository.path_to_repo,
          'BlobId' => blob.id,
        }

        [
16 17
          SEND_DATA_HEADER,
          "git-blob:#{encode(params)}",
18 19
        ]
      end
20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37

      def send_git_archive(project, ref, format)
        format ||= 'tar.gz'
        format.downcase!
        params = project.repository.archive_metadata(ref, Gitlab.config.gitlab.repository_downloads_path, format)
        raise "Repository or ref not found" if params.empty?

        [
          SEND_DATA_HEADER,
          "git-archive:#{encode(params)}",
        ]
      end
      
      protected
      
      def encode(hash)
        Base64.urlsafe_encode64(JSON.dump(hash))
      end
38 39
    end
  end
Jacob Vosmaer committed
40
end