summaryrefslogtreecommitdiff
path: root/backends/gitlab-ce
diff options
context:
space:
mode:
Diffstat (limited to 'backends/gitlab-ce')
-rwxr-xr-xbackends/gitlab-ce221
1 files changed, 127 insertions, 94 deletions
diff --git a/backends/gitlab-ce b/backends/gitlab-ce
index c91effc..84cb933 100755
--- a/backends/gitlab-ce
+++ b/backends/gitlab-ce
@@ -19,33 +19,20 @@ class GitLabCE < GitMirrorBackend
return @obj
end
end
+
def initialize()
@connections = {}
- @cache = {}
+ @projects = {}
+ @config = {}
+ end
- # API docs suck, look at `lib/api/projects.rb` instead.
- @vars = [
- :builds_enabled, # create | create-user | edit
- :container_registry_enabled, # XXX # create | | edit
- :default_branch, # | create-user | edit
- :description, # create | create-user | edit
- #:import_url, # XXX # create | create-user |
- :issues_enabled, # create | create-user | edit
- :lfs_enabled, # create | create-user | edit
- :merge_requests_enabled, # create | create-user | edit
- :name, # create | create-user | edit
- #:namespace_id, # create | |
- :only_allow_merge_if_build_succeeds, # create | create-user | edit
- #:path, # create | | edit
- :public, # create | create-user | edit
- :public_builds, # create | create-user | edit
- :request_access_enabled, # create | create-user | edit
- :shared_runners_enabled, # create | create-user | edit
- :snippets_enabled, # create | create-user | edit
- :visibility_level, # create | create-user | edit
- :wiki_enabled, # create | create-user | edit
- :only_allow_merge_if_all_discussions_are_resolved # create | create-user | edit
- ]
+ def connection(uri)
+ key=URI(uri.scheme+":")
+ key.host = uri.host
+ key.port = uri.port
+
+ @connections[key] ||= Net::HTTP::start(uri.host, uri.port, :use_ssl => uri.scheme == 'https')
+ return @connections[key]
end
def finish
@@ -56,51 +43,110 @@ class GitLabCE < GitMirrorBackend
super
end
- def _connection(uri)
- key=URI(uri.scheme+":")
- key.host = uri.host
- key.port = uri.port
+ def config
+ return @config
+ end
- @connections[key] ||= Net::HTTP::start(uri.host, uri.port, :use_ssl => uri.scheme == 'https')
- return @connections[key]
+ # Project
+ def project(project_id)
+ unless @projects.has_key?(project_id)
+ @projects[project_id] = Project.new(self, project_id)
+ end
+ return @projects[project_id]
end
+ class Project
+ def initialize(gl, project_id)
+ @gl = gl
+ @project_id = project_id
+ @cache = {}
+ end
+
+ def info
+ unless @cache.has_key?(:info)
+ req = Net::HTTP::Get.new(@gl.config['apiurl'] + "projects/" + CGI::escape(@project_id))
+ req.add_field("PRIVATE-TOKEN", @gl.config['apikey'])
+ res = @gl.connection(req.uri).request(req)
+ case res.code
+ when "200"
+ @cache[:info] = JSON::parse(res.body)
+ when "404"
+ @cache[:info] = nil
+ else
+ raise Error.new(res)
+ end
+ end
+ return @cache[:info]
+ end
+
+ def info=(i)
+ @cache[:info] = i
+ end
+
+ def get_meta
+ return self.info.select{|k,v| @vars.include?(k.to_sym)}
+ end
+
+ def set_meta(map)
+ mirror = map["mirror"]
+ map.delete("mirror")
+
+ illegal = map.select{|k,v| not @vars.include?(k.to_sym)}
+ if illegal.count > 0
+ raise Error.new(illegal)
+ end
- def _info(project_id)
- @cache[project_id] ||= {}
- unless @cache[project_id].has_key?(:info)
- req = Net::HTTP::Get.new(@api_uri + "projects/" + CGI::escape(project_id))
- req.add_field("PRIVATE-TOKEN", @api_key)
- con = _connection(req.uri)
- res = con.request(req)
- case res.code
- when "200"
- @cache[project_id][:info] = JSON::parse(res.body)
- when "404"
- @cache[project_id][:info] = nil
+ if info == nil
+ # create
+ req = Net::HTTP::Put.new(@gl.config['apiurl'] + "projects")
+ req.add_field("PRIVATE-TOKEN", @gl.config['apikey'])
+ req.add_field("Content-Type", "application/json")
+ map["path"] = @project_id
+ if not mirror.nil?
+ map["import_url"] = mirror
+ end
+ req.body = JSON::dump(map)
+ res = @gl.connection(req.uri).request(req)
+ if res.code != "201"
+ raise Error.new(res)
+ end
+ info = JSON::parse(res.body)
else
- raise Error.new(res)
+ # update
+ req = Net::HTTP::Put.new(@gl.config['apiurl'] + "projects/" + CGI::escape(info["id"].to_s))
+ req.add_field("PRIVATE-TOKEN", @gl.config['apikey'])
+ req.add_field("Content-Type", "application/json")
+ req.body = JSON::dump(map)
+ res = @gl.connection(req.uri).request(req)
+ if res.code != "200"
+ raise Error.new(res)
+ end
+ info = JSON::parse(res.body)
end
+ return self.get_meta
+ end
+
+ def repo_mode
+ return "passive"
end
- return @cache[project_id][:info]
end
+ # commands
def cmd_config(*args)
args.each do |arg|
key, val = arg.split('=', 2)
case key
when "apiurl"
- @api_uri = URI(val)
- unless @api_uri.path.end_with?("/")
- @api_uri.path += "/"
+ val = URI(val)
+ unless val.path.end_with?("/")
+ val.path += "/"
end
- when "apikey"
- @api_key = val
end
+ @config[key] = val
end
end
def cmd_get_meta(project_id)
- return _info(project_id).select{|k,v| @vars.include?(k.to_sym)}
+ return self.project(project_id).get_meta()
end
def cmd_set_meta(project_id, *pairs)
@@ -109,58 +155,45 @@ class GitLabCE < GitMirrorBackend
key, val = arg.split('=', 2)
map[key] = val
end
-
- mirror = map["mirror"]
- map.delete("mirror")
-
- illegal = map.select{|k,v| not @vars.include?(k.to_sym)}
- if illegal.count > 0
- raise Error.new(illegal)
- end
-
- info = self._info(project_id)
- if info == nil
- # create
- req = Net::HTTP::Put.new(@api_uri + "projects"))
- req.add_field("PRIVATE-TOKEN", @api_key)
- req.add_field("Content-Type", "application/json")
- map["path"] = project_id
- if not mirror.nil?
- map["import_url"] = mirror
- end
- req.body = JSON::dump(map)
- con = _connection(req.uri)
- res = con.request(req)
- if res.code != "201"
- raise Error.new(res)
- end
- @cache[project_id][:info] = JSON::parse(res.body)
- else
- # update
- req = Net::HTTP::Put.new(@api_uri + "projects/" + CGI::escape(info["id"].to_s))
- req.add_field("PRIVATE-TOKEN", @api_key)
- req.add_field("Content-Type", "application/json")
- req.body = JSON::dump(map)
- con = _connection(req.uri)
- res = con.request(req)
- if res.code != "200"
- raise Error.new(res)
- end
- @cache[project_id][:info] = JSON::parse(res.body)
- end
- return get_meta
+ return self.project(project_id).set_meta(map)
end
def cmd_push_url(project_id)
- return _info(project_id)["ssh_url_to_repo"]
+ return self.project(project_id).info["ssh_url_to_repo"]
end
def cmd_pull_url(project_id)
- return _info(project_id)["http_url_to_repo"]
+ return self.project(project_id).info["http_url_to_repo"]
end
- def cmd_repo_mode
- return "passive"
+ def cmd_repo_mode(project_id)
+ return self.project(projecT_id).repo_mode()
+ end
+
+ def vars
+ # API docs suck, look at `lib/api/projects.rb` instead.
+ return [
+ :builds_enabled, # create | create-user | edit
+ :container_registry_enabled, # XXX # create | | edit
+ :default_branch, # | create-user | edit
+ :description, # create | create-user | edit
+ #:import_url, # XXX # create | create-user |
+ :issues_enabled, # create | create-user | edit
+ :lfs_enabled, # create | create-user | edit
+ :merge_requests_enabled, # create | create-user | edit
+ :name, # create | create-user | edit
+ #:namespace_id, # create | |
+ :only_allow_merge_if_build_succeeds, # create | create-user | edit
+ #:path, # create | | edit
+ :public, # create | create-user | edit
+ :public_builds, # create | create-user | edit
+ :request_access_enabled, # create | create-user | edit
+ :shared_runners_enabled, # create | create-user | edit
+ :snippets_enabled, # create | create-user | edit
+ :visibility_level, # create | create-user | edit
+ :wiki_enabled, # create | create-user | edit
+ :only_allow_merge_if_all_discussions_are_resolved # create | create-user | edit
+ ]
end
end