116 lines
3.2 KiB
Python
116 lines
3.2 KiB
Python
# ContentDB
|
|
# Copyright (C) 2018-21 rubenwardy
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
|
|
import contextlib, git, gitdb, os, shutil, tempfile
|
|
from urllib.parse import urlsplit
|
|
from git import GitCommandError
|
|
|
|
from app.tasks import TaskError
|
|
from app.utils import randomString
|
|
|
|
def generateGitURL(urlstr):
|
|
scheme, netloc, path, query, frag = urlsplit(urlstr)
|
|
|
|
return "http://:@" + netloc + path + query
|
|
|
|
|
|
@contextlib.contextmanager
|
|
def get_temp_dir():
|
|
temp = os.path.join(tempfile.gettempdir(), randomString(10))
|
|
yield temp
|
|
shutil.rmtree(temp)
|
|
|
|
|
|
# Clones a repo from an unvalidated URL.
|
|
# Returns a tuple of path and repo on sucess.
|
|
# Throws `TaskError` on failure.
|
|
# Caller is responsible for deleting returned directory.
|
|
@contextlib.contextmanager
|
|
def clone_repo(urlstr, ref=None, recursive=False):
|
|
gitDir = os.path.join(tempfile.gettempdir(), randomString(10))
|
|
|
|
err = None
|
|
try:
|
|
gitUrl = generateGitURL(urlstr)
|
|
print("Cloning from " + gitUrl)
|
|
|
|
if ref is None:
|
|
repo = git.Repo.clone_from(gitUrl, gitDir,
|
|
progress=None, env=None, depth=1, recursive=recursive, kill_after_timeout=15)
|
|
else:
|
|
assert ref != ""
|
|
|
|
repo = git.Repo.init(gitDir)
|
|
origin = repo.create_remote("origin", url=gitUrl)
|
|
assert origin.exists()
|
|
origin.fetch()
|
|
repo.git.checkout(ref)
|
|
|
|
repo.git.submodule('update', '--init')
|
|
|
|
yield repo
|
|
shutil.rmtree(gitDir)
|
|
return
|
|
|
|
except GitCommandError as e:
|
|
# This is needed to stop the backtrace being weird
|
|
err = e.stderr
|
|
|
|
except gitdb.exc.BadName as e:
|
|
err = "Unable to find the reference " + (ref or "?") + "\n" + e.stderr
|
|
|
|
raise TaskError(err.replace("stderr: ", "") \
|
|
.replace("Cloning into '" + gitDir + "'...", "") \
|
|
.strip())
|
|
|
|
|
|
def get_latest_commit(git_url, ref_name=None):
|
|
git_url = generateGitURL(git_url)
|
|
|
|
if ref_name:
|
|
ref_name = "refs/heads/" + ref_name
|
|
else:
|
|
ref_name = "HEAD"
|
|
|
|
g = git.cmd.Git()
|
|
|
|
remote_refs = {}
|
|
for ref in g.ls_remote(git_url).split('\n'):
|
|
hash_ref_list = ref.split('\t')
|
|
remote_refs[hash_ref_list[1]] = hash_ref_list[0]
|
|
|
|
return remote_refs.get(ref_name)
|
|
|
|
|
|
def get_latest_tag(git_url):
|
|
with get_temp_dir() as git_dir:
|
|
repo = git.Repo.init(git_dir)
|
|
origin = repo.create_remote("origin", url=git_url)
|
|
origin.fetch()
|
|
|
|
refs = repo.git.for_each_ref(sort="creatordate", format="%(objectname)\t%(refname)").split("\n")
|
|
refs = [ref for ref in refs if "refs/tags/" in ref]
|
|
if len(refs) == 0:
|
|
return None, None
|
|
|
|
last_ref = refs[-1]
|
|
hash_ref_list = last_ref.split('\t')
|
|
|
|
tag = hash_ref_list[1].replace("refs/tags/", "")
|
|
commit_hash = repo.git.rev_parse(tag + "^{}")
|
|
return tag, commit_hash
|