diff options
Diffstat (limited to 'git.py')
| -rw-r--r-- | git.py | 226 |
1 files changed, 226 insertions, 0 deletions
@@ -0,0 +1,226 @@ +# +# RTEMS Tools Project (http://www.rtems.org/) +# Copyright 2010-2016, 2023 Chris Johns (chrisj@rtems.org) +# All rights reserved. +# +# This file is part of the RTEMS Tools package in 'rtems-tools'. +# +# Permission to use, copy, modify, and/or distribute this software for any +# purpose with or without fee is hereby granted, provided that the above +# copyright notice and this permission notice appear in all copies. +# +# THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES +# WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF +# MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR +# ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES +# WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN +# ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF +# OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. + +# +# Provide some basic access to the git command. +# + +from __future__ import print_function + +import os +import os.path + + +class repo: + """An object to manage a git repo.""" + def _git_exit_code(self, ec): + if ec: + raise self.ctx.fatal('git command failed (%s): %d' % + (self.git, ec)) + + def _run(self, args, check=False): + import waflib + if os.path.exists(self.path): + cwd = self.path + else: + cwd = None + cmd = [self.git] + args + exit_code = 0 + try: + output = self.ctx.cmd_and_log(cmd, + cwd=cwd, + output=waflib.Context.STDOUT, + quiet=waflib.Context.BOTH) + except waflib.Errors.WafError as e: + exit_code = e.returncode + output = e.stderr + if check: + self._git_exit_code(exit_code) + return exit_code, output + + def __init__(self, ctx, path): + self.ctx = ctx + self.path = path + self.git = 'git' + + def git_version(self): + ec, output = self._run(['--version'], True) + gvs = output.split() + if len(gvs) < 3: + raise self.ctx.fatal('invalid version string from git: %s' % + (output)) + vs = gvs[2].split('.') + if len(vs) not in [3, 4]: + raise self.ctx.fatal('invalid version number from git: %s' % + (gvs[2])) + return tuple(map(int, vs)) + + def clone(self, url, path): + ec, output = self._run(['clone', url, path], check=True) + + def fetch(self): + ec, output = self._run(['fetch'], check=True) + + def merge(self): + ec, output = self._run(['merge'], check=True) + + def pull(self): + ec, output = self._run(['pull'], check=True) + + def reset(self, args): + if type(args) == str: + args = [args] + ec, output = self._run(['reset'] + args, check=True) + + def branch(self): + ec, output = self._run(['branch']) + if ec == 0: + for b in output.split(os.linesep): + if b[0] == '*': + return b[2:] + return None + + def checkout(self, branch='master'): + ec, output = self._run(['checkout', branch], check=True) + + def submodule(self, module): + ec, output = self._run(['submodule', 'update', '--init', module], + check=True) + + def submodule_foreach(self, args=[]): + if type(args) == str: + args = [args.split(args)] + ec, output = self._run( + ['submodule', 'foreach', '--recursive', self.git] + args, + check=True) + + def submodules(self): + smodules = {} + ec, output = self._run(['submodule'], check=True) + if ec == 0: + for l in output.split(os.linesep): + ms = l.split() + if len(ms) == 3: + smodules[ms[1]] = (ms[0], ms[2][1:-1]) + return smodules + + def clean(self, args=[]): + if type(args) == str: + args = [args] + ec, output = self._run(['clean'] + args, check=True) + + def status(self, submodules_always_clean=False): + _status = {} + if os.path.exists(self.path): + if submodules_always_clean: + submodules = self.submodules() + else: + submodules = {} + ec, output = self._run(['status']) + if ec == 0: + state = 'none' + for l in output.split(os.linesep): + if l.startswith('# '): + l = l[2:] + if l.startswith('On branch '): + _status['branch'] = l[len('On branch '):] + elif l.startswith('Changes to be committed:'): + state = 'staged' + elif l.startswith('Changes not staged for commit:'): + state = 'unstaged' + elif l.startswith('Untracked files:'): + state = 'untracked' + elif l.startswith('HEAD detached'): + state = 'detached' + elif state != 'none' and len(l.strip()) != 0: + if l[0].isspace(): + l = l.strip() + if l[0] != '(': + if ':' in l: + l = l.split(':')[1] + if len(l.strip()) > 0: + l = l.strip() + ls = l.split() + if state != 'unstaged' or ls[ + 0] not in submodules: + if state not in _status: + _status[state] = [l] + else: + _status[state] += [l] + return _status + + def dirty(self): + _status = self.status() + _status.pop('untracked', None) + _status.pop('detached', None) + return not (len(_status) == 1 and 'branch' in _status) + + def valid(self): + if os.path.exists(self.path): + ec, output = self._run(['status']) + return ec == 0 + return False + + def remotes(self): + _remotes = {} + ec, output = self._run(['config', '--list']) + if ec == 0: + for l in output.split(os.linesep): + if l.startswith('remote'): + ls = l.split('=') + if len(ls) >= 2: + rs = ls[0].split('.') + if len(rs) == 3: + r_name = rs[1] + r_type = rs[2] + if r_name not in _remotes: + _remotes[r_name] = {} + if r_type not in _remotes[r_name]: + _remotes[r_name][r_type] = [] + _remotes[r_name][r_type] = '='.join(ls[1:]) + return _remotes + + def email(self): + _email = None + _name = None + ec, output = self._run(['config', '--list']) + if ec == 0: + for l in output.split(os.linesep): + if l.startswith('user.email'): + ls = l.split('=') + if len(ls) >= 2: + _email = ls[1] + elif l.startswith('user.name'): + ls = l.split('=') + if len(ls) >= 2: + _name = ls[1] + if _email is not None: + if _name is not None: + _email = '%s <%s>' % (_name, _email) + return _email + return None + + def head(self): + hash = '' + ec, output = self._run(['log', '-n', '1']) + if ec == 0: + l1 = output.split(os.linesep)[0] + if l1.startswith('commit '): + hash = l1[len('commit '):] + return hash |