Mike Frysinger | f601376 | 2019-06-13 02:30:51 -0400 | [diff] [blame] | 1 | # -*- coding:utf-8 -*- |
Shawn O. Pearce | d237b69 | 2009-04-17 18:49:50 -0700 | [diff] [blame] | 2 | # |
| 3 | # Copyright (C) 2009 The Android Open Source Project |
| 4 | # |
| 5 | # Licensed under the Apache License, Version 2.0 (the "License"); |
| 6 | # you may not use this file except in compliance with the License. |
| 7 | # You may obtain a copy of the License at |
| 8 | # |
| 9 | # http://www.apache.org/licenses/LICENSE-2.0 |
| 10 | # |
| 11 | # Unless required by applicable law or agreed to in writing, software |
| 12 | # distributed under the License is distributed on an "AS IS" BASIS, |
| 13 | # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 14 | # See the License for the specific language governing permissions and |
| 15 | # limitations under the License. |
| 16 | |
| 17 | import os |
Mike Frysinger | 8a11f6f | 2019-08-27 00:26:15 -0400 | [diff] [blame] | 18 | from repo_trace import Trace |
Renaud Paquay | bed8b62 | 2018-09-27 10:46:58 -0700 | [diff] [blame] | 19 | import platform_utils |
Shawn O. Pearce | d237b69 | 2009-04-17 18:49:50 -0700 | [diff] [blame] | 20 | |
David Pursehouse | 54a4e60 | 2020-02-12 14:31:05 +0900 | [diff] [blame] | 21 | HEAD = 'HEAD' |
Zac Livingston | 9ead97b | 2017-06-13 08:29:04 -0600 | [diff] [blame] | 22 | R_CHANGES = 'refs/changes/' |
David Pursehouse | 54a4e60 | 2020-02-12 14:31:05 +0900 | [diff] [blame] | 23 | R_HEADS = 'refs/heads/' |
| 24 | R_TAGS = 'refs/tags/' |
| 25 | R_PUB = 'refs/published/' |
Mike Frysinger | 21b7fbe | 2020-02-26 23:53:36 -0500 | [diff] [blame^] | 26 | R_WORKTREE = 'refs/worktree/' |
| 27 | R_WORKTREE_M = R_WORKTREE + 'm/' |
David Pursehouse | 54a4e60 | 2020-02-12 14:31:05 +0900 | [diff] [blame] | 28 | R_M = 'refs/remotes/m/' |
Shawn O. Pearce | d237b69 | 2009-04-17 18:49:50 -0700 | [diff] [blame] | 29 | |
| 30 | |
| 31 | class GitRefs(object): |
| 32 | def __init__(self, gitdir): |
| 33 | self._gitdir = gitdir |
| 34 | self._phyref = None |
| 35 | self._symref = None |
| 36 | self._mtime = {} |
| 37 | |
| 38 | @property |
| 39 | def all(self): |
Shawn O. Pearce | 0f3dd23 | 2009-04-17 20:32:44 -0700 | [diff] [blame] | 40 | self._EnsureLoaded() |
Shawn O. Pearce | d237b69 | 2009-04-17 18:49:50 -0700 | [diff] [blame] | 41 | return self._phyref |
| 42 | |
| 43 | def get(self, name): |
| 44 | try: |
| 45 | return self.all[name] |
| 46 | except KeyError: |
| 47 | return '' |
| 48 | |
Shawn O. Pearce | fbcde47 | 2009-04-17 20:58:02 -0700 | [diff] [blame] | 49 | def deleted(self, name): |
| 50 | if self._phyref is not None: |
| 51 | if name in self._phyref: |
| 52 | del self._phyref[name] |
| 53 | |
| 54 | if name in self._symref: |
| 55 | del self._symref[name] |
| 56 | |
| 57 | if name in self._mtime: |
| 58 | del self._mtime[name] |
| 59 | |
Shawn O. Pearce | 0f3dd23 | 2009-04-17 20:32:44 -0700 | [diff] [blame] | 60 | def symref(self, name): |
| 61 | try: |
| 62 | self._EnsureLoaded() |
| 63 | return self._symref[name] |
| 64 | except KeyError: |
| 65 | return '' |
| 66 | |
| 67 | def _EnsureLoaded(self): |
| 68 | if self._phyref is None or self._NeedUpdate(): |
| 69 | self._LoadAll() |
| 70 | |
Shawn O. Pearce | d237b69 | 2009-04-17 18:49:50 -0700 | [diff] [blame] | 71 | def _NeedUpdate(self): |
Shawn O. Pearce | ad3193a | 2009-04-18 09:54:51 -0700 | [diff] [blame] | 72 | Trace(': scan refs %s', self._gitdir) |
| 73 | |
Chirayu Desai | 217ea7d | 2013-03-01 19:14:38 +0530 | [diff] [blame] | 74 | for name, mtime in self._mtime.items(): |
Shawn O. Pearce | d237b69 | 2009-04-17 18:49:50 -0700 | [diff] [blame] | 75 | try: |
| 76 | if mtime != os.path.getmtime(os.path.join(self._gitdir, name)): |
| 77 | return True |
| 78 | except OSError: |
| 79 | return True |
| 80 | return False |
| 81 | |
| 82 | def _LoadAll(self): |
Shawn O. Pearce | ad3193a | 2009-04-18 09:54:51 -0700 | [diff] [blame] | 83 | Trace(': load refs %s', self._gitdir) |
| 84 | |
Shawn O. Pearce | d237b69 | 2009-04-17 18:49:50 -0700 | [diff] [blame] | 85 | self._phyref = {} |
| 86 | self._symref = {} |
| 87 | self._mtime = {} |
| 88 | |
| 89 | self._ReadPackedRefs() |
| 90 | self._ReadLoose('refs/') |
| 91 | self._ReadLoose1(os.path.join(self._gitdir, HEAD), HEAD) |
| 92 | |
| 93 | scan = self._symref |
| 94 | attempts = 0 |
| 95 | while scan and attempts < 5: |
| 96 | scan_next = {} |
Chirayu Desai | 217ea7d | 2013-03-01 19:14:38 +0530 | [diff] [blame] | 97 | for name, dest in scan.items(): |
Shawn O. Pearce | d237b69 | 2009-04-17 18:49:50 -0700 | [diff] [blame] | 98 | if dest in self._phyref: |
| 99 | self._phyref[name] = self._phyref[dest] |
| 100 | else: |
| 101 | scan_next[name] = dest |
| 102 | scan = scan_next |
| 103 | attempts += 1 |
| 104 | |
| 105 | def _ReadPackedRefs(self): |
| 106 | path = os.path.join(self._gitdir, 'packed-refs') |
| 107 | try: |
Chirayu Desai | 0eb35cb | 2013-11-19 18:46:29 +0530 | [diff] [blame] | 108 | fd = open(path, 'r') |
Shawn O. Pearce | d237b69 | 2009-04-17 18:49:50 -0700 | [diff] [blame] | 109 | mtime = os.path.getmtime(path) |
| 110 | except IOError: |
| 111 | return |
| 112 | except OSError: |
| 113 | return |
| 114 | try: |
| 115 | for line in fd: |
Chirayu Desai | 217ea7d | 2013-03-01 19:14:38 +0530 | [diff] [blame] | 116 | line = str(line) |
Shawn O. Pearce | d237b69 | 2009-04-17 18:49:50 -0700 | [diff] [blame] | 117 | if line[0] == '#': |
| 118 | continue |
| 119 | if line[0] == '^': |
| 120 | continue |
| 121 | |
| 122 | line = line[:-1] |
| 123 | p = line.split(' ') |
David Pursehouse | 8a68ff9 | 2012-09-24 12:15:13 +0900 | [diff] [blame] | 124 | ref_id = p[0] |
Shawn O. Pearce | d237b69 | 2009-04-17 18:49:50 -0700 | [diff] [blame] | 125 | name = p[1] |
| 126 | |
David Pursehouse | 8a68ff9 | 2012-09-24 12:15:13 +0900 | [diff] [blame] | 127 | self._phyref[name] = ref_id |
Shawn O. Pearce | d237b69 | 2009-04-17 18:49:50 -0700 | [diff] [blame] | 128 | finally: |
| 129 | fd.close() |
| 130 | self._mtime['packed-refs'] = mtime |
| 131 | |
| 132 | def _ReadLoose(self, prefix): |
| 133 | base = os.path.join(self._gitdir, prefix) |
Renaud Paquay | bed8b62 | 2018-09-27 10:46:58 -0700 | [diff] [blame] | 134 | for name in platform_utils.listdir(base): |
Shawn O. Pearce | d237b69 | 2009-04-17 18:49:50 -0700 | [diff] [blame] | 135 | p = os.path.join(base, name) |
Renaud Paquay | bed8b62 | 2018-09-27 10:46:58 -0700 | [diff] [blame] | 136 | if platform_utils.isdir(p): |
Shawn O. Pearce | d237b69 | 2009-04-17 18:49:50 -0700 | [diff] [blame] | 137 | self._mtime[prefix] = os.path.getmtime(base) |
| 138 | self._ReadLoose(prefix + name + '/') |
| 139 | elif name.endswith('.lock'): |
| 140 | pass |
| 141 | else: |
| 142 | self._ReadLoose1(p, prefix + name) |
| 143 | |
| 144 | def _ReadLoose1(self, path, name): |
| 145 | try: |
Mike Frysinger | 3164d40 | 2019-11-11 05:40:22 -0500 | [diff] [blame] | 146 | with open(path) as fd: |
Shawn O. Pearce | cc14fa9 | 2011-11-29 12:32:56 -0800 | [diff] [blame] | 147 | mtime = os.path.getmtime(path) |
David Pursehouse | 8a68ff9 | 2012-09-24 12:15:13 +0900 | [diff] [blame] | 148 | ref_id = fd.readline() |
Mike Frysinger | 3164d40 | 2019-11-11 05:40:22 -0500 | [diff] [blame] | 149 | except (IOError, OSError): |
| 150 | return |
Shawn O. Pearce | d237b69 | 2009-04-17 18:49:50 -0700 | [diff] [blame] | 151 | |
Chirayu Desai | 217ea7d | 2013-03-01 19:14:38 +0530 | [diff] [blame] | 152 | try: |
| 153 | ref_id = ref_id.decode() |
| 154 | except AttributeError: |
| 155 | pass |
David Pursehouse | 8a68ff9 | 2012-09-24 12:15:13 +0900 | [diff] [blame] | 156 | if not ref_id: |
Shawn O. Pearce | d237b69 | 2009-04-17 18:49:50 -0700 | [diff] [blame] | 157 | return |
David Pursehouse | 8a68ff9 | 2012-09-24 12:15:13 +0900 | [diff] [blame] | 158 | ref_id = ref_id[:-1] |
Shawn O. Pearce | d237b69 | 2009-04-17 18:49:50 -0700 | [diff] [blame] | 159 | |
David Pursehouse | 8a68ff9 | 2012-09-24 12:15:13 +0900 | [diff] [blame] | 160 | if ref_id.startswith('ref: '): |
| 161 | self._symref[name] = ref_id[5:] |
Shawn O. Pearce | d237b69 | 2009-04-17 18:49:50 -0700 | [diff] [blame] | 162 | else: |
David Pursehouse | 8a68ff9 | 2012-09-24 12:15:13 +0900 | [diff] [blame] | 163 | self._phyref[name] = ref_id |
Shawn O. Pearce | d237b69 | 2009-04-17 18:49:50 -0700 | [diff] [blame] | 164 | self._mtime[name] = mtime |