# HG changeset patch # User Pulkit Goyal <7895pulkit@gmail.com> # Date 1507147313 -19800 # Node ID 744d1c874a59d5ebfcd8ee83fd1275674426bfc0 # Parent 54bb94b715adf35c77f243e2943d333833405797 remotenames: add functions to read remotenames data from .hg/remotenames/ This patch functions which can be used to read remotenames data from .hg/remotenames/. The logic for the function which reads the remotenames file is taken from the remotenames extension. Previously reviewed as D940. Differential Revision: https://phab.mercurial-scm.org/D1550 diff -r 54bb94b715ad -r 744d1c874a59 mercurial/remotenames.py --- a/mercurial/remotenames.py Fri Nov 10 22:54:59 2017 +0530 +++ b/mercurial/remotenames.py Thu Oct 05 01:31:53 2017 +0530 @@ -17,6 +17,46 @@ # directory name in .hg/ in which remotenames files will be present remotenamedir = 'remotenames' +def readremotenamefile(repo, filename): + """ + reads a file from .hg/remotenames/ directory and yields it's content + filename: the file to be read + yield a tuple (node, remotepath, name) + """ + + vfs = vfsmod.vfs(repo.vfs.join(remotenamedir)) + if not vfs.exists(filename): + return + f = vfs(filename) + lineno = 0 + for line in f: + line = line.strip() + if not line: + continue + # contains the version number + if lineno == 0: + lineno += 1 + try: + node, remote, rname = line.split('\0') + yield node, remote, rname + except ValueError: + pass + + f.close() + +def readremotenames(repo): + """ + read the details about the remotenames stored in .hg/remotenames/ and + yields a tuple (node, remotepath, name). It does not yields information + about whether an entry yielded is branch or bookmark. To get that + information, call the respective functions. + """ + + for bmentry in readremotenamefile(repo, 'bookmarks'): + yield bmentry + for branchentry in readremotenamefile(repo, 'branches'): + yield branchentry + def writeremotenamefile(repo, remotepath, names, nametype): vfs = vfsmod.vfs(repo.vfs.join(remotenamedir)) f = vfs(nametype, 'w', atomictemp=True)