From 89a2884adeb5ecd52bd69ec18d869d9159e459bf Mon Sep 17 00:00:00 2001 From: Philip Feairheller Date: Mon, 8 Jul 2024 15:16:41 -0700 Subject: [PATCH] Update to latest Hio build. (#815) New command line command for adding location records for an AID Signed-off-by: pfeairheller --- setup.py | 2 +- .../app/cli/commands/location/__init__.py | 0 src/keri/app/cli/commands/location/add.py | 123 ++++++++++++++++++ 3 files changed, 124 insertions(+), 1 deletion(-) create mode 100644 src/keri/app/cli/commands/location/__init__.py create mode 100644 src/keri/app/cli/commands/location/add.py diff --git a/setup.py b/setup.py index cf891f233..dea228df3 100644 --- a/setup.py +++ b/setup.py @@ -83,7 +83,7 @@ 'cbor2>=5.6.2', 'multidict>=6.0.5', 'ordered-set>=4.1.0', - 'hio>=0.6.12', + 'hio>=0.6.13', 'multicommand>=1.0.0', 'jsonschema>=4.21.1', 'falcon>=3.1.3', diff --git a/src/keri/app/cli/commands/location/__init__.py b/src/keri/app/cli/commands/location/__init__.py new file mode 100644 index 000000000..e69de29bb diff --git a/src/keri/app/cli/commands/location/add.py b/src/keri/app/cli/commands/location/add.py new file mode 100644 index 000000000..d549c0ded --- /dev/null +++ b/src/keri/app/cli/commands/location/add.py @@ -0,0 +1,123 @@ +# -*- encoding: utf-8 -*- +""" +KERI +keri.kli.commands module + +""" +import argparse +from urllib.parse import urlparse + +from hio import help +from hio.base import doing + +from keri import kering +from keri.app import habbing, grouping, indirecting, forwarding +from keri.app.agenting import WitnessPublisher +from keri.app.cli.common import existing +from keri.app.notifying import Notifier +from keri.core import parsing +from keri.peer import exchanging + +logger = help.ogler.getLogger() + +parser = argparse.ArgumentParser(description='Add new endpoint location record.') +parser.set_defaults(handler=lambda args: add_loc(args), + transferable=True) +parser.add_argument('--name', '-n', help='keystore name and file location of KERI keystore', required=True) +parser.add_argument('--base', '-b', help='additional optional prefix to file location of KERI keystore', + required=False, default="") +parser.add_argument('--alias', '-a', help='human readable alias for the new identifier prefix', required=True) +parser.add_argument('--passcode', '-p', help='21 character encryption passcode for keystore (is not saved)', + dest="bran", default=None) # passcode => bran +parser.add_argument("--url", "-u", help="Location URL", + required=True) +parser.add_argument("--eid", "-e", help="qualified base64 of AID to associate a location with, defaults to alias aid ", + required=False, default=None) +parser.add_argument("--time", help="timestamp for the end auth", required=False, default=None) + + +def add_loc(args): + """ Command line tool for adding location scheme records + + """ + ld = LocationDoer(name=args.name, + base=args.base, + alias=args.alias, + bran=args.bran, + url=args.url, + eid=args.eid, + timestamp=args.time) + return [ld] + + +class LocationDoer(doing.DoDoer): + + def __init__(self, name, base, alias, bran, url, eid, timestamp=None): + self.url = url + self.eid = eid + self.timestamp = timestamp + + self.hby = existing.setupHby(name=name, base=base, bran=bran) + self.hab = self.hby.habByName(alias) + self.witpub = WitnessPublisher(hby=self.hby) + self.postman = forwarding.Poster(hby=self.hby) + notifier = Notifier(self.hby) + mux = grouping.Multiplexor(self.hby, notifier=notifier) + exc = exchanging.Exchanger(hby=self.hby, handlers=[]) + grouping.loadHandlers(exc, mux) + + mbx = indirecting.MailboxDirector(hby=self.hby, topics=["/receipt", "/multisig", "/replay"], exc=exc) + + if self.hab is None: + raise kering.ConfigurationError(f"unknown alias={alias}") + + self.toRemove = [self.witpub, self.postman, mbx] + + super(LocationDoer, self).__init__(doers=self.toRemove + [doing.doify(self.roleDo)]) + + def roleDo(self, tymth, tock=0.0): + """ Export any end reply messages previous saved for the provided AID + + Parameters: + tymth (function): injected function wrapper closure returned by .tymen() of + Tymist instance. Calling tymth() returns associated Tymist .tyme. + tock (float): injected initial tock value + + Returns: doifiable Doist compatible generator method + + """ + # enter context + self.wind(tymth) + self.tock = tock + _ = (yield self.tock) + + up = urlparse(self.url) + eid = self.eid if self.eid is not None else self.hab.pre + + msg = self.hab.makeLocScheme(url=self.url, eid=eid, scheme=up.scheme) + parsing.Parser().parse(ims=bytes(msg), kvy=self.hab.kvy, rvy=self.hab.rvy) + + if isinstance(self.hab, habbing.GroupHab): + smids = self.hab.db.signingMembers(pre=self.hab.pre) + smids.remove(self.hab.mhab.pre) + + for recp in smids: # this goes to other participants only as a signaling mechanism + exn, atc = grouping.multisigRpyExn(ghab=self.hab, rpy=msg) + self.postman.send(src=self.hab.mhab.pre, + dest=recp, + topic="multisig", + serder=exn, + attachment=atc) + + while not self.hab.loadLocScheme(scheme=up.scheme, eid=eid): + yield self.tock + + self.witpub.msgs.append(dict(pre=self.hab.pre, msg=bytes(msg))) + + while not self.witpub.cues: + yield self.tock + + print(f"Location {self.url} added for aid {eid} with scheme {up.scheme}") + + self.remove(self.toRemove) + return