1""" 2BitBake 'Fetch' git submodules implementation 3 4Inherits from and extends the Git fetcher to retrieve submodules of a git repository 5after cloning. 6 7SRC_URI = "gitsm://<see Git fetcher for syntax>" 8 9See the Git fetcher, git://, for usage documentation. 10 11NOTE: Switching a SRC_URI from "git://" to "gitsm://" requires a clean of your recipe. 12 13""" 14 15# Copyright (C) 2013 Richard Purdie 16# 17# SPDX-License-Identifier: GPL-2.0-only 18# 19 20import os 21import bb 22import copy 23import shutil 24import tempfile 25from bb.fetch2.git import Git 26from bb.fetch2 import runfetchcmd 27from bb.fetch2 import logger 28from bb.fetch2 import Fetch 29 30class GitSM(Git): 31 def supports(self, ud, d): 32 """ 33 Check to see if a given url can be fetched with git. 34 """ 35 return ud.type in ['gitsm'] 36 37 def process_submodules(self, ud, workdir, function, d): 38 """ 39 Iterate over all of the submodules in this repository and execute 40 the 'function' for each of them. 41 """ 42 43 submodules = [] 44 paths = {} 45 revision = {} 46 uris = {} 47 subrevision = {} 48 49 def parse_gitmodules(gitmodules): 50 modules = {} 51 module = "" 52 for line in gitmodules.splitlines(): 53 if line.startswith('[submodule'): 54 module = line.split('"')[1] 55 modules[module] = {} 56 elif module and line.strip().startswith('path'): 57 path = line.split('=')[1].strip() 58 modules[module]['path'] = path 59 elif module and line.strip().startswith('url'): 60 url = line.split('=')[1].strip() 61 modules[module]['url'] = url 62 return modules 63 64 # Collect the defined submodules, and their attributes 65 for name in ud.names: 66 try: 67 gitmodules = runfetchcmd("%s show %s:.gitmodules" % (ud.basecmd, ud.revisions[name]), d, quiet=True, workdir=workdir) 68 except: 69 # No submodules to update 70 continue 71 72 for m, md in parse_gitmodules(gitmodules).items(): 73 try: 74 module_hash = runfetchcmd("%s ls-tree -z -d %s %s" % (ud.basecmd, ud.revisions[name], md['path']), d, quiet=True, workdir=workdir) 75 except: 76 # If the command fails, we don't have a valid file to check. If it doesn't 77 # fail -- it still might be a failure, see next check... 78 module_hash = "" 79 80 if not module_hash: 81 logger.debug("submodule %s is defined, but is not initialized in the repository. Skipping", m) 82 continue 83 84 submodules.append(m) 85 paths[m] = md['path'] 86 revision[m] = ud.revisions[name] 87 uris[m] = md['url'] 88 subrevision[m] = module_hash.split()[2] 89 90 # Convert relative to absolute uri based on parent uri 91 if uris[m].startswith('..') or uris[m].startswith('./'): 92 newud = copy.copy(ud) 93 newud.path = os.path.realpath(os.path.join(newud.path, uris[m])) 94 uris[m] = Git._get_repo_url(self, newud) 95 96 for module in submodules: 97 # Translate the module url into a SRC_URI 98 99 if "://" in uris[module]: 100 # Properly formated URL already 101 proto = uris[module].split(':', 1)[0] 102 url = uris[module].replace('%s:' % proto, 'gitsm:', 1) 103 else: 104 if ":" in uris[module]: 105 # Most likely an SSH style reference 106 proto = "ssh" 107 if ":/" in uris[module]: 108 # Absolute reference, easy to convert.. 109 url = "gitsm://" + uris[module].replace(':/', '/', 1) 110 else: 111 # Relative reference, no way to know if this is right! 112 logger.warning("Submodule included by %s refers to relative ssh reference %s. References may fail if not absolute." % (ud.url, uris[module])) 113 url = "gitsm://" + uris[module].replace(':', '/', 1) 114 else: 115 # This has to be a file reference 116 proto = "file" 117 url = "gitsm://" + uris[module] 118 if url.endswith("{}{}".format(ud.host, ud.path)): 119 raise bb.fetch2.FetchError("Submodule refers to the parent repository. This will cause deadlock situation in current version of Bitbake." \ 120 "Consider using git fetcher instead.") 121 122 url += ';protocol=%s' % proto 123 url += ";name=%s" % module 124 url += ";subpath=%s" % module 125 126 ld = d.createCopy() 127 # Not necessary to set SRC_URI, since we're passing the URI to 128 # Fetch. 129 #ld.setVar('SRC_URI', url) 130 ld.setVar('SRCREV_%s' % module, subrevision[module]) 131 132 # Workaround for issues with SRCPV/SRCREV_FORMAT errors 133 # error refer to 'multiple' repositories. Only the repository 134 # in the original SRC_URI actually matters... 135 ld.setVar('SRCPV', d.getVar('SRCPV')) 136 ld.setVar('SRCREV_FORMAT', module) 137 138 function(ud, url, module, paths[module], workdir, ld) 139 140 return submodules != [] 141 142 def need_update(self, ud, d): 143 if Git.need_update(self, ud, d): 144 return True 145 146 need_update_list = [] 147 def need_update_submodule(ud, url, module, modpath, workdir, d): 148 url += ";bareclone=1;nobranch=1" 149 150 try: 151 newfetch = Fetch([url], d, cache=False) 152 new_ud = newfetch.ud[url] 153 if new_ud.method.need_update(new_ud, d): 154 need_update_list.append(modpath) 155 except Exception as e: 156 logger.error('gitsm: submodule update check failed: %s %s' % (type(e).__name__, str(e))) 157 need_update_result = True 158 159 # If we're using a shallow mirror tarball it needs to be unpacked 160 # temporarily so that we can examine the .gitmodules file 161 if ud.shallow and os.path.exists(ud.fullshallow) and not os.path.exists(ud.clonedir): 162 tmpdir = tempfile.mkdtemp(dir=d.getVar("DL_DIR")) 163 runfetchcmd("tar -xzf %s" % ud.fullshallow, d, workdir=tmpdir) 164 self.process_submodules(ud, tmpdir, need_update_submodule, d) 165 shutil.rmtree(tmpdir) 166 else: 167 self.process_submodules(ud, ud.clonedir, need_update_submodule, d) 168 169 if need_update_list: 170 logger.debug('gitsm: Submodules requiring update: %s' % (' '.join(need_update_list))) 171 return True 172 173 return False 174 175 def download(self, ud, d): 176 def download_submodule(ud, url, module, modpath, workdir, d): 177 url += ";bareclone=1;nobranch=1" 178 179 # Is the following still needed? 180 #url += ";nocheckout=1" 181 182 try: 183 newfetch = Fetch([url], d, cache=False) 184 newfetch.download() 185 except Exception as e: 186 logger.error('gitsm: submodule download failed: %s %s' % (type(e).__name__, str(e))) 187 raise 188 189 Git.download(self, ud, d) 190 191 # If we're using a shallow mirror tarball it needs to be unpacked 192 # temporarily so that we can examine the .gitmodules file 193 if ud.shallow and os.path.exists(ud.fullshallow) and self.need_update(ud, d): 194 tmpdir = tempfile.mkdtemp(dir=d.getVar("DL_DIR")) 195 runfetchcmd("tar -xzf %s" % ud.fullshallow, d, workdir=tmpdir) 196 self.process_submodules(ud, tmpdir, download_submodule, d) 197 shutil.rmtree(tmpdir) 198 else: 199 self.process_submodules(ud, ud.clonedir, download_submodule, d) 200 201 def unpack(self, ud, destdir, d): 202 def unpack_submodules(ud, url, module, modpath, workdir, d): 203 url += ";bareclone=1;nobranch=1" 204 205 # Figure out where we clone over the bare submodules... 206 if ud.bareclone: 207 repo_conf = ud.destdir 208 else: 209 repo_conf = os.path.join(ud.destdir, '.git') 210 211 try: 212 newfetch = Fetch([url], d, cache=False) 213 newfetch.unpack(root=os.path.dirname(os.path.join(repo_conf, 'modules', module))) 214 except Exception as e: 215 logger.error('gitsm: submodule unpack failed: %s %s' % (type(e).__name__, str(e))) 216 raise 217 218 local_path = newfetch.localpath(url) 219 220 # Correct the submodule references to the local download version... 221 runfetchcmd("%(basecmd)s config submodule.%(module)s.url %(url)s" % {'basecmd': ud.basecmd, 'module': module, 'url' : local_path}, d, workdir=ud.destdir) 222 223 if ud.shallow: 224 runfetchcmd("%(basecmd)s config submodule.%(module)s.shallow true" % {'basecmd': ud.basecmd, 'module': module}, d, workdir=ud.destdir) 225 226 # Ensure the submodule repository is NOT set to bare, since we're checking it out... 227 try: 228 runfetchcmd("%s config core.bare false" % (ud.basecmd), d, quiet=True, workdir=os.path.join(repo_conf, 'modules', module)) 229 except: 230 logger.error("Unable to set git config core.bare to false for %s" % os.path.join(repo_conf, 'modules', module)) 231 raise 232 233 Git.unpack(self, ud, destdir, d) 234 235 ret = self.process_submodules(ud, ud.destdir, unpack_submodules, d) 236 237 if not ud.bareclone and ret: 238 # All submodules should already be downloaded and configured in the tree. This simply sets 239 # up the configuration and checks out the files. The main project config should remain 240 # unmodified, and no download from the internet should occur. 241 runfetchcmd("%s submodule update --recursive --no-fetch" % (ud.basecmd), d, quiet=True, workdir=ud.destdir) 242 243 def implicit_urldata(self, ud, d): 244 import shutil, subprocess, tempfile 245 246 urldata = [] 247 def add_submodule(ud, url, module, modpath, workdir, d): 248 url += ";bareclone=1;nobranch=1" 249 newfetch = Fetch([url], d, cache=False) 250 urldata.extend(newfetch.expanded_urldata()) 251 252 # If we're using a shallow mirror tarball it needs to be unpacked 253 # temporarily so that we can examine the .gitmodules file 254 if ud.shallow and os.path.exists(ud.fullshallow) and ud.method.need_update(ud, d): 255 tmpdir = tempfile.mkdtemp(dir=d.getVar("DL_DIR")) 256 subprocess.check_call("tar -xzf %s" % ud.fullshallow, cwd=tmpdir, shell=True) 257 self.process_submodules(ud, tmpdir, add_submodule, d) 258 shutil.rmtree(tmpdir) 259 else: 260 self.process_submodules(ud, ud.clonedir, add_submodule, d) 261 262 return urldata 263