fileindexapi.py
110 lines
| 3.4 KiB
| text/x-python
|
PythonLexer
Pulkit Goyal
|
r37204 | # Infinite push | ||
| # | ||||
| # Copyright 2016 Facebook, Inc. | ||||
| # | ||||
| # This software may be used and distributed according to the terms of the | ||||
| # GNU General Public License version 2 or any later version. | ||||
| """ | ||||
| [infinitepush] | ||||
| # Server-side option. Used only if indextype=disk. | ||||
| # Filesystem path to the index store | ||||
| indexpath = PATH | ||||
| """ | ||||
| import os | ||||
Matt Harbison
|
r37829 | from mercurial import util | ||
|
Pulkit Goyal
|
r37221 | from mercurial.utils import stringutil | ||
|
Pulkit Goyal
|
r37204 | |||
| from . import indexapi | ||||
Augie Fackler
|
r43346 | |||
|
Pulkit Goyal
|
r37204 | class fileindexapi(indexapi.indexapi): | ||
| def __init__(self, repo): | ||||
| super(fileindexapi, self).__init__() | ||||
| self._repo = repo | ||||
|
Augie Fackler
|
r43347 | root = repo.ui.config(b'infinitepush', b'indexpath') | ||
|
Pulkit Goyal
|
r37204 | if not root: | ||
|
Augie Fackler
|
r43347 | root = os.path.join(b'scratchbranches', b'index') | ||
|
Pulkit Goyal
|
r37204 | |||
|
Augie Fackler
|
r43347 | self._nodemap = os.path.join(root, b'nodemap') | ||
| self._bookmarkmap = os.path.join(root, b'bookmarkmap') | ||||
| self._metadatamap = os.path.join(root, b'nodemetadatamap') | ||||
|
Pulkit Goyal
|
r37204 | self._lock = None | ||
| def __enter__(self): | ||||
| self._lock = self._repo.wlock() | ||||
| return self | ||||
| def __exit__(self, exc_type, exc_val, exc_tb): | ||||
| if self._lock: | ||||
| self._lock.__exit__(exc_type, exc_val, exc_tb) | ||||
| def addbundle(self, bundleid, nodesctx): | ||||
| for node in nodesctx: | ||||
| nodepath = os.path.join(self._nodemap, node.hex()) | ||||
| self._write(nodepath, bundleid) | ||||
| def addbookmark(self, bookmark, node): | ||||
| bookmarkpath = os.path.join(self._bookmarkmap, bookmark) | ||||
| self._write(bookmarkpath, node) | ||||
| def addmanybookmarks(self, bookmarks): | ||||
| for bookmark, node in bookmarks.items(): | ||||
| self.addbookmark(bookmark, node) | ||||
| def deletebookmarks(self, patterns): | ||||
| for pattern in patterns: | ||||
| for bookmark, _ in self._listbookmarks(pattern): | ||||
| bookmarkpath = os.path.join(self._bookmarkmap, bookmark) | ||||
| self._delete(bookmarkpath) | ||||
| def getbundle(self, node): | ||||
| nodepath = os.path.join(self._nodemap, node) | ||||
| return self._read(nodepath) | ||||
| def getnode(self, bookmark): | ||||
| bookmarkpath = os.path.join(self._bookmarkmap, bookmark) | ||||
| return self._read(bookmarkpath) | ||||
| def getbookmarks(self, query): | ||||
| return dict(self._listbookmarks(query)) | ||||
| def saveoptionaljsonmetadata(self, node, jsonmetadata): | ||||
| vfs = self._repo.vfs | ||||
| vfs.write(os.path.join(self._metadatamap, node), jsonmetadata) | ||||
| def _listbookmarks(self, pattern): | ||||
|
Augie Fackler
|
r43347 | if pattern.endswith(b'*'): | ||
| pattern = b're:^' + pattern[:-1] + b'.*' | ||||
|
Pulkit Goyal
|
r37221 | kind, pat, matcher = stringutil.stringmatcher(pattern) | ||
|
Pulkit Goyal
|
r37204 | prefixlen = len(self._bookmarkmap) + 1 | ||
| for dirpath, _, books in self._repo.vfs.walk(self._bookmarkmap): | ||||
| for book in books: | ||||
| bookmark = os.path.join(dirpath, book)[prefixlen:] | ||||
|
Matt Harbison
|
r37829 | bookmark = util.pconvert(bookmark) | ||
|
Pulkit Goyal
|
r37204 | if not matcher(bookmark): | ||
| continue | ||||
| yield bookmark, self._read(os.path.join(dirpath, book)) | ||||
| def _write(self, path, value): | ||||
| vfs = self._repo.vfs | ||||
| dirname = vfs.dirname(path) | ||||
| if not vfs.exists(dirname): | ||||
| vfs.makedirs(dirname) | ||||
| vfs.write(path, value) | ||||
| def _read(self, path): | ||||
| vfs = self._repo.vfs | ||||
| if not vfs.exists(path): | ||||
| return None | ||||
| return vfs.read(path) | ||||
| def _delete(self, path): | ||||
| vfs = self._repo.vfs | ||||
| if not vfs.exists(path): | ||||
| return | ||||
| return vfs.unlink(path) | ||||