102 lines
3.8 KiB
Python
102 lines
3.8 KiB
Python
|
import re
|
||
|
|
import os
|
||
|
|
import subprocess
|
||
|
|
import argparse
|
||
|
|
import shutil
|
||
|
|
import time
|
||
|
|
|
||
|
|
import gitlab_api
|
||
|
|
|
||
|
|
SUBMODULE_PATTERN = re.compile(r"\[submodule \"([^\"]+)\"]")
|
||
|
|
PATH_PATTERN = re.compile(r"path\s+=\s+(\S+)")
|
||
|
|
URL_PATTERN = re.compile(r"url\s+=\s+(\S+)")
|
||
|
|
|
||
|
|
SUBMODULE_ARCHIVE_TEMP_FOLDER = "submodule_archive"
|
||
|
|
|
||
|
|
|
||
|
|
class SubModule(object):
|
||
|
|
# We don't need to support recursive submodule clone now
|
||
|
|
|
||
|
|
GIT_LS_TREE_OUTPUT_PATTERN = re.compile(r"\d+\s+commit\s+([0-9a-f]+)\s+")
|
||
|
|
|
||
|
|
def __init__(self, gitlab_inst, path, url):
|
||
|
|
self.path = path
|
||
|
|
self.gitlab_inst = gitlab_inst
|
||
|
|
self.project_id = self._get_project_id(url)
|
||
|
|
self.commit_id = self._get_commit_id(path)
|
||
|
|
|
||
|
|
def _get_commit_id(self, path):
|
||
|
|
output = subprocess.check_output(["git", "ls-tree", "HEAD", path])
|
||
|
|
# example output: 160000 commit d88a262fbdf35e5abb372280eb08008749c3faa0 components/esp_wifi/lib
|
||
|
|
match = self.GIT_LS_TREE_OUTPUT_PATTERN.search(output)
|
||
|
|
return match.group(1)
|
||
|
|
|
||
|
|
def _get_project_id(self, url):
|
||
|
|
base_name = os.path.basename(url)
|
||
|
|
project_id = self.gitlab_inst.get_project_id(os.path.splitext(base_name)[0], # remove .git
|
||
|
|
namespace="espressif")
|
||
|
|
return project_id
|
||
|
|
|
||
|
|
def download_archive(self):
|
||
|
|
print("Update submodule: {}: {}".format(self.path, self.commit_id))
|
||
|
|
path_name = self.gitlab_inst.download_archive(self.commit_id, SUBMODULE_ARCHIVE_TEMP_FOLDER,
|
||
|
|
self.project_id)
|
||
|
|
renamed_path = os.path.join(os.path.dirname(path_name), os.path.basename(self.path))
|
||
|
|
os.rename(path_name, renamed_path)
|
||
|
|
shutil.rmtree(self.path, ignore_errors=True)
|
||
|
|
shutil.move(renamed_path, os.path.dirname(self.path))
|
||
|
|
|
||
|
|
|
||
|
|
def update_submodule(git_module_file, submodules_to_update):
|
||
|
|
gitlab_inst = gitlab_api.Gitlab()
|
||
|
|
submodules = []
|
||
|
|
with open(git_module_file, "r") as f:
|
||
|
|
data = f.read()
|
||
|
|
match = SUBMODULE_PATTERN.search(data)
|
||
|
|
while True:
|
||
|
|
next_match = SUBMODULE_PATTERN.search(data, pos=match.end())
|
||
|
|
if next_match:
|
||
|
|
end_pos = next_match.start()
|
||
|
|
else:
|
||
|
|
end_pos = len(data)
|
||
|
|
path_match = PATH_PATTERN.search(data, pos=match.end(), endpos=end_pos)
|
||
|
|
url_match = URL_PATTERN.search(data, pos=match.end(), endpos=end_pos)
|
||
|
|
path = path_match.group(1)
|
||
|
|
url = url_match.group(1)
|
||
|
|
|
||
|
|
filter_result = True
|
||
|
|
if submodules_to_update:
|
||
|
|
if path not in submodules_to_update:
|
||
|
|
filter_result = False
|
||
|
|
if filter_result:
|
||
|
|
submodules.append(SubModule(gitlab_inst, path, url))
|
||
|
|
|
||
|
|
match = next_match
|
||
|
|
if not match:
|
||
|
|
break
|
||
|
|
|
||
|
|
shutil.rmtree(SUBMODULE_ARCHIVE_TEMP_FOLDER, ignore_errors=True)
|
||
|
|
|
||
|
|
for submodule in submodules:
|
||
|
|
submodule.download_archive()
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
start_time = time.time()
|
||
|
|
parser = argparse.ArgumentParser()
|
||
|
|
parser.add_argument("--repo_path", "-p", default=".", help="repo path")
|
||
|
|
parser.add_argument("--submodule", "-s", default="all",
|
||
|
|
help="Submodules to update. By default update all submodules. "
|
||
|
|
"For multiple submodules, separate them with `;`. "
|
||
|
|
"`all` and `none` are special values that indicates we fetch all / none submodules")
|
||
|
|
args = parser.parse_args()
|
||
|
|
if args.submodule == "none":
|
||
|
|
print("don't need to update submodules")
|
||
|
|
exit(0)
|
||
|
|
if args.submodule == "all":
|
||
|
|
_submodules = []
|
||
|
|
else:
|
||
|
|
_submodules = args.submodule.split(";")
|
||
|
|
update_submodule(os.path.join(args.repo_path, ".gitmodules"), _submodules)
|
||
|
|
print("total time spent on update submodule: {:.02f}s".format(time.time() - start_time))
|