test: staking-combinations

packages/orchestration/src/examples/staking-combinations.contract.js

@@ -0,0 +1,138 @@
+/**
+ * @file This contract demonstrates the continuing invitation pattern with async
+ *   flows.
+ *
+ *   The primary offer result is a power for invitation makers that can perform
+ *   actions with an ICA account.
+ */
+import { AmountShape } from '@agoric/ertp';
+import { VowShape } from '@agoric/vow';
+import { M } from '@endo/patterns';
+import { prepareCombineInvitationMakers } from '../exos/combine-invitation-makers.js';
+import { CosmosOrchestrationInvitationMakersInterface } from '../exos/cosmos-orchestration-account.js';
+import { withOrchestration } from '../utils/start-helper.js';
+import * as flows from './staking-combinations.flows.js';
+
+/**
+ * @import {GuestInterface} from '@agoric/async-flow';
+ * @import {Delegation} from '@agoric/cosmic-proto/cosmos/staking/v1beta1/staking.js';
+ * @import {ContinuingOfferResult} from '@agoric/smart-wallet/src/types.js';
+ * @import {TimerService} from '@agoric/time';
+ * @import {LocalChain} from '@agoric/vats/src/localchain.js';
+ * @import {NameHub} from '@agoric/vats';
+ * @import {Vow} from '@agoric/vow';
+ * @import {Remote} from '@agoric/internal';
+ * @import {Zone} from '@agoric/zone';
+ * @import {CosmosInterchainService} from '../exos/cosmos-interchain-service.js';
+ * @import {OrchestrationTools} from '../utils/start-helper.js';
+ * @import {CosmosOrchestrationAccount} from '../exos/cosmos-orchestration-account.js';
+ */
+
+const emptyOfferShape = harden({
+  // Nothing to give; the funds are deposited offline
+  give: {},
+  want: {}, // UNTIL https://github.com/Agoric/agoric-sdk/issues/2230
+  exit: M.any(),
+});
+
+/**
+ * Orchestration contract to be wrapped by withOrchestration for Zoe.
+ *
+ * @param {ZCF} zcf
+ * @param {{
+ *   agoricNames: Remote<NameHub>;
+ *   localchain: Remote<LocalChain>;
+ *   orchestrationService: Remote<CosmosInterchainService>;
+ *   storageNode: Remote<StorageNode>;
+ *   marshaller: Marshaller;
+ *   timerService: Remote<TimerService>;
+ * }} privateArgs
+ * @param {Zone} zone
+ * @param {OrchestrationTools} tools
+ */
+const contract = async (
+  zcf,
+  privateArgs,
+  zone,
+  { orchestrateAll, vowTools },
+) => {
+  const ExtraInvitationMakerInterface = M.interface('', {
+    DepositAndDelegate: M.call(M.array()).returns(VowShape),
+    UndelegateAndTransfer: M.call(M.array()).returns(VowShape),
+  });
+  /** @type {any} XXX async membrane */
+  const makeExtraInvitationMaker = zone.exoClass(
+    'ContinuingInvitationExampleInvitationMakers',
+    ExtraInvitationMakerInterface,
+    /** @param {GuestInterface<CosmosOrchestrationAccount>} account */
+    account => {
+      return { account };
+    },
+    {
+      DepositAndDelegate() {
+        const { account } = this.state;
+
+        const invP = zcf.makeInvitation(
+          (seat, validatorAddr, amountArg) =>
+            // eslint-disable-next-line no-use-before-define -- defined by orchestrateAll, necessarily after this
+            orchFns.depositAndDelegate(account, seat, validatorAddr, amountArg),
+          'Deposit and delegate',
+          undefined,
+          {
+            give: {
+              Stake: AmountShape,
+            },
+          },
+        );
+
+        return vowTools.watch(invP);
+      },
+      /**
+       * @param {Omit<Delegation, 'delegatorAddress'>[]} delegations
+       */
+      UndelegateAndTransfer(delegations) {
+        const { account } = this.state;
+
+        const invP = zcf.makeInvitation(
+          // eslint-disable-next-line no-use-before-define -- defined by orchestrateAll, necessarily after this
+          () => orchFns.undelegateAndTransfer(account, delegations),
+          'Undelegate and transfer',
+          undefined,
+          emptyOfferShape,
+        );
+
+        return vowTools.watch(invP);
+      },
+    },
+  );
+
+  /** @type {any} XXX async membrane */
+  const makeCombineInvitationMakers = prepareCombineInvitationMakers(
+    zone,
+    CosmosOrchestrationInvitationMakersInterface,
+    ExtraInvitationMakerInterface,
+  );
+
+  const orchFns = orchestrateAll(flows, {
+    makeCombineInvitationMakers,
+    makeExtraInvitationMaker,
+    flows,
+    zcf,
+  });
+
+  const publicFacet = zone.exo('publicFacet', undefined, {
+    makeAccount() {
+      return zcf.makeInvitation(
+        orchFns.makeAccount,
+        'Make an ICA account',
+        undefined,
+        emptyOfferShape,
+      );
+    },
+  });
+
+  return harden({ publicFacet });
+};
+
+export const start = withOrchestration(contract);
+harden(start);

packages/orchestration/src/examples/staking-combinations.flows.js

@@ -0,0 +1,83 @@
+/**
+ * @import {GuestInterface} from '@agoric/async-flow';
+ * @import {Orchestrator, OrchestrationFlow, OrchestrationAccount, OrchestrationAccountI, StakingAccountActions, AmountArg, CosmosValidatorAddress} from '../types.js'
+ * @import {ContinuingOfferResult, InvitationMakers} from '@agoric/smart-wallet/src/types.js';
+ * @import {MakeCombineInvitationMakers} from '../exos/combine-invitation-makers.js';
+ * @import {Delegation} from '@agoric/cosmic-proto/cosmos/staking/v1beta1/staking.js';
+ * @import {CosmosOrchestrationAccount} from '../exos/cosmos-orchestration-account.js';
+ */
+
+/**
+ * @satisfies {OrchestrationFlow}
+ * @param {Orchestrator} orch
+ * @param {{
+ *   makeCombineInvitationMakers: MakeCombineInvitationMakers;
+ *   makeExtraInvitationMaker: (account: any) => InvitationMakers;
+ * }} ctx
+ * @param {ZCFSeat} _seat
+ * @param {{ chainName: string }} offerArgs
+ * @returns {Promise<ContinuingOfferResult>}
+ */
+export const makeAccount = async (orch, ctx, _seat, { chainName }) => {
+  const chain = await orch.getChain(chainName);
+  const account = await chain.makeAccount();
+
+  const extraMakers = ctx.makeExtraInvitationMaker(account);
+
+  /** @type {ContinuingOfferResult} */
+  const result = await account.asContinuingOffer();
+
+  return {
+    ...result,
+    invitationMakers: ctx.makeCombineInvitationMakers(
+      extraMakers,
+      result.invitationMakers,
+    ),
+  };
+};
+harden(makeAccount);
+
+/**
+ * @satisfies {OrchestrationFlow}
+ * @param {Orchestrator} orch
+ * @param {object} ctx
+ * @param {GuestInterface<CosmosOrchestrationAccount>} account
+ * @param {ZCFSeat} seat
+ * @param {CosmosValidatorAddress} validator
+ * @param {AmountArg} amount
+ * @returns {Promise<string>}
+ */
+export const depositAndDelegate = async (
+  orch,
+  ctx,
+  account,
+  seat,
+  validator,
+  amount,
+) => {
+  console.log('depositAndDelegate', account, seat, validator, amount);
+  // TODO deposit the amount
+  await account.delegate(validator, amount);
+  return 'guest depositAndDelegate complete';
+};
+harden(depositAndDelegate);
+
+/**
+ * @satisfies {OrchestrationFlow}
+ * @param {Orchestrator} orch
+ * @param {object} ctx
+ * @param {GuestInterface<CosmosOrchestrationAccount>} account
+ * @param {Omit<Delegation, 'delegatorAddress'>[]} delegations
+ * @returns {Promise<string>}
+ */
+export const undelegateAndTransfer = async (
+  orch,
+  ctx,
+  account,
+  delegations,
+) => {
+  await account.undelegate(delegations);
+  // TODO transfer something
+  return 'guest undelegateAndTransfer complete';
+};
+harden(undelegateAndTransfer);

packages/orchestration/test/examples/snapshots/staking-combinations.test.ts.md

@@ -0,0 +1,116 @@
+# Snapshot report for `test/examples/staking-combinations.test.ts`
+
+The actual snapshot is saved in `staking-combinations.test.ts.snap`.
+
+Generated by [AVA](https://avajs.dev).
+
+## start
+
+> contract baggage after start
+
+    {
+      'Durable Publish Kit_kindHandle': 'Alleged: kind',
+      Recorder_kindHandle: 'Alleged: kind',
+      asyncFlow: {
+        AdminAsyncFlow_kindHandle: 'Alleged: kind',
+        AdminAsyncFlow_singleton: 'Alleged: AdminAsyncFlow',
+        Bijection_kindHandle: 'Alleged: kind',
+        FunctionUnwrapper_kindHandle: 'Alleged: kind',
+        FunctionUnwrapper_singleton: 'Alleged: FunctionUnwrapper',
+        LogStore_kindHandle: 'Alleged: kind',
+        StateUnwrapper_kindHandle: 'Alleged: kind',
+        asyncFuncEagerWakers: [],
+        asyncFuncFailures: {},
+        flowForOutcomeVow: {},
+        unwrapMap: 'Alleged: weakMapStore',
+      },
+      contract: {
+        CombinedInvitationMakers_kindHandle: 'Alleged: kind',
+        ContinuingInvitationExampleInvitationMakers_kindHandle: 'Alleged: kind',
+        orchestration: {
+          depositAndDelegate: {
+            asyncFlow_kindHandle: 'Alleged: kind',
+            endowments: {
+              0: {
+                flows: {
+                  depositAndDelegate_kindHandle: 'Alleged: kind',
+                  depositAndDelegate_singleton: 'Alleged: depositAndDelegate',
+                  makeAccount_kindHandle: 'Alleged: kind',
+                  makeAccount_singleton: 'Alleged: makeAccount',
+                  undelegateAndTransfer_kindHandle: 'Alleged: kind',
+                  undelegateAndTransfer_singleton: 'Alleged: undelegateAndTransfer',
+                },
+                makeCombineInvitationMakers_kindHandle: 'Alleged: kind',
+                makeCombineInvitationMakers_singleton: 'Alleged: makeCombineInvitationMakers',
+                makeExtraInvitationMaker_kindHandle: 'Alleged: kind',
+                makeExtraInvitationMaker_singleton: 'Alleged: makeExtraInvitationMaker',
+              },
+            },
+          },
+          makeAccount: {
+            asyncFlow_kindHandle: 'Alleged: kind',
+            endowments: {
+              0: {
+                flows: {
+                  depositAndDelegate_kindHandle: 'Alleged: kind',
+                  depositAndDelegate_singleton: 'Alleged: depositAndDelegate',
+                  makeAccount_kindHandle: 'Alleged: kind',
+                  makeAccount_singleton: 'Alleged: makeAccount',
+                  undelegateAndTransfer_kindHandle: 'Alleged: kind',
+                  undelegateAndTransfer_singleton: 'Alleged: undelegateAndTransfer',
+                },
+                makeCombineInvitationMakers_kindHandle: 'Alleged: kind',
+                makeCombineInvitationMakers_singleton: 'Alleged: makeCombineInvitationMakers',
+                makeExtraInvitationMaker_kindHandle: 'Alleged: kind',
+                makeExtraInvitationMaker_singleton: 'Alleged: makeExtraInvitationMaker',
+              },
+            },
+          },
+          undelegateAndTransfer: {
+            asyncFlow_kindHandle: 'Alleged: kind',
+            endowments: {
+              0: {
+                flows: {
+                  depositAndDelegate_kindHandle: 'Alleged: kind',
+                  depositAndDelegate_singleton: 'Alleged: depositAndDelegate',
+                  makeAccount_kindHandle: 'Alleged: kind',
+                  makeAccount_singleton: 'Alleged: makeAccount',
+                  undelegateAndTransfer_kindHandle: 'Alleged: kind',
+                  undelegateAndTransfer_singleton: 'Alleged: undelegateAndTransfer',
+                },
+                makeCombineInvitationMakers_kindHandle: 'Alleged: kind',
+                makeCombineInvitationMakers_singleton: 'Alleged: makeCombineInvitationMakers',
+                makeExtraInvitationMaker_kindHandle: 'Alleged: kind',
+                makeExtraInvitationMaker_singleton: 'Alleged: makeExtraInvitationMaker',
+              },
+            },
+          },
+        },
+        publicFacet_kindHandle: 'Alleged: kind',
+        publicFacet_singleton: 'Alleged: publicFacet',
+      },
+      orchestration: {
+        'Cosmos Orchestration Account Holder_kindHandle': 'Alleged: kind',
+        'Local Orchestration Account Kit_kindHandle': 'Alleged: kind',
+        LocalChainFacade_kindHandle: 'Alleged: kind',
+        Orchestrator_kindHandle: 'Alleged: kind',
+        RemoteChainFacade_kindHandle: 'Alleged: kind',
+        chainName: {
+          osmosis: 'Alleged: RemoteChainFacade public',
+        },
+        ibcTools: {
+          IBCTransferSenderKit_kindHandle: 'Alleged: kind',
+          ibcResultWatcher_kindHandle: 'Alleged: kind',
+          ibcResultWatcher_singleton: 'Alleged: ibcResultWatcher',
+        },
+        packetTools: {
+          PacketToolsKit_kindHandle: 'Alleged: kind',
+        },
+      },
+      vows: {
+        PromiseWatcher_kindHandle: 'Alleged: kind',
+        VowInternalsKit_kindHandle: 'Alleged: kind',
+        WatchUtils_kindHandle: 'Alleged: kind',
+      },
+      zoe: {},
+    }