mirror of
https://github.com/facebook/sapling.git
synced 2024-10-12 01:39:21 +03:00
3210553e4c
Summary: With listserverbookmarks in hg, we have to wait for an hg release to be able to run our monitoring, which isn't ideal. Furthermore, the listserverbookmarks method command is unlikely to be used by anyone else, so we might as well provide it directly for review-bookmarks (like we do in e.g. the commit cloud backfiller for, though if we change our minds later we can always just revert this diff and the corresponding fbpkg change). Reviewed By: farnz Differential Revision: D15778340 fbshipit-source-id: d8fd6c9e1d33a420f95bdb4fadb582b864c51d5f
95 lines
2.7 KiB
Python
95 lines
2.7 KiB
Python
# checkserverbookmark.py - check whether the bookmark is where we expect
|
|
# it to be on a server
|
|
#
|
|
# Copyright 2019-present 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.
|
|
from __future__ import absolute_import
|
|
|
|
from edenscm.mercurial import error, hg, util
|
|
from edenscm.mercurial.commands import command
|
|
from edenscm.mercurial.i18n import _
|
|
from edenscm.mercurial.node import hex
|
|
|
|
|
|
@util.timed(annotation="creating a peer took")
|
|
def getremote(ui, path):
|
|
remote = hg.peer(ui, {}, path)
|
|
return remote
|
|
|
|
|
|
@util.timed(annotation="running lookup took")
|
|
def runlookup(ui, remote, name):
|
|
return remote.lookup(name)
|
|
|
|
|
|
@util.timed(annotation="running listkeys took")
|
|
def runlistkeys(ui, remote):
|
|
return remote.listkeys("bookmarks")
|
|
|
|
|
|
def verifyexisting(ui, remote, name, hash):
|
|
location = hex(runlookup(ui, remote, name))
|
|
if location.strip() != hash.strip():
|
|
ui.warn(
|
|
_(
|
|
"hg server does not have an expected bookmark location. "
|
|
+ "book: %s, server: %s; expected %s\n"
|
|
)
|
|
% (name, location, hash)
|
|
)
|
|
return 1
|
|
ui.warn(
|
|
_("hg server has expected bookmark location. book: %s, hash: %s\n")
|
|
% (name, hash)
|
|
)
|
|
return 0
|
|
|
|
|
|
def verifydeleted(ui, remote, name):
|
|
serverkeys = runlistkeys(ui, remote)
|
|
if name in serverkeys:
|
|
ui.warn(
|
|
_("hg server has bookmark, which is expected to have been deleted: %s\n")
|
|
% (name,)
|
|
)
|
|
return 1
|
|
ui.warn(_("hg server expectedly does not have a bookmark: %s\n") % (name,))
|
|
return 0
|
|
|
|
|
|
@command(
|
|
"^checkserverbookmark",
|
|
[
|
|
("", "path", "", _("hg server remotepath (ssh)"), ""),
|
|
("", "name", "", _("bookmark name to check"), ""),
|
|
("", "hash", "", _("hash to verify against the bookmark"), ""),
|
|
(
|
|
"",
|
|
"deleted",
|
|
False,
|
|
_("bookmark is expected to not exist, cannot be used with `--hash`"),
|
|
),
|
|
],
|
|
_("[OPTION]..."),
|
|
norepo=True,
|
|
)
|
|
def checkserverbookmark(ui, **opts):
|
|
"""Verify whether the bookmark on hg server points to a given hash"""
|
|
name = opts["name"]
|
|
path = opts["path"]
|
|
hash = opts["hash"]
|
|
deleted = opts["deleted"]
|
|
if hash and deleted:
|
|
raise error.Abort("can't use `--hash` and `--deleted`")
|
|
|
|
if not (hash or deleted):
|
|
raise error.Abort("either `--hash` or `--deleted` should be used")
|
|
|
|
remote = getremote(ui, path)
|
|
if deleted:
|
|
return verifydeleted(ui, remote, name)
|
|
else:
|
|
return verifyexisting(ui, remote, name, hash)
|