Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

misc Orchestration cleanup #9720

Merged
merged 5 commits into from
Jul 16, 2024
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion packages/orchestration/package.json
Original file line number Diff line number Diff line change
Expand Up @@ -92,6 +92,6 @@
"access": "public"
},
"typeCoverage": {
"atLeast": 98.04
"atLeast": 98.09
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -13,8 +13,9 @@ import { orchestrationFns } from './sendAnywhereFlows.js';
* @import {TimerService} from '@agoric/time';
* @import {LocalChain} from '@agoric/vats/src/localchain.js';
* @import {NameHub} from '@agoric/vats';
* @import {Remote} from '@agoric/vow';
* @import {Remote, Vow} from '@agoric/vow';
* @import {Zone} from '@agoric/zone';
* @import {VBankAssetDetail} from '@agoric/vats/tools/board-utils.js';
* @import {CosmosChainInfo, IBCConnectionInfo} from '../cosmos-api';
* @import {CosmosInterchainService} from '../exos/cosmos-interchain-service.js';
* @import {OrchestrationTools} from '../utils/start-helper.js';
Expand Down Expand Up @@ -60,6 +61,7 @@ const contract = async (
);

// TODO should be a provided helper
/** @type {(brand: Brand) => Vow<VBankAssetDetail>} */
const findBrandInVBank = vowTools.retriable(
zone,
'findBrandInVBank',
Expand Down
11 changes: 7 additions & 4 deletions packages/orchestration/src/examples/sendAnywhereFlows.js
Original file line number Diff line number Diff line change
@@ -1,7 +1,10 @@
import { M, mustMatch } from '@endo/patterns';

/**
* @import {Orchestrator, OrchestrationAccount} from '../types.js';
* @import {GuestOf} from '@agoric/async-flow';
* @import {VBankAssetDetail} from '@agoric/vats/tools/board-utils.js';
* @import {ZoeTools} from '../utils/zoe-tools.js';
* @import {Orchestrator, OrchestrationAccount, LocalAccountMethods, OrchestrationAccountI} from '../types.js';
*/

const { entries } = Object;
Expand All @@ -13,9 +16,9 @@ export const orchestrationFns = harden({
/**
* @param {Orchestrator} orch
* @param {object} ctx
* @param {{ account: OrchestrationAccount<any> }} ctx.contractState
* @param {any} ctx.localTransfer
* @param {any} ctx.findBrandInVBank
* @param {{ account?: OrchestrationAccountI & LocalAccountMethods }} ctx.contractState
* @param {GuestOf<ZoeTools['localTransfer']>} ctx.localTransfer
* @param {(brand: Brand) => Promise<VBankAssetDetail>} ctx.findBrandInVBank
* @param {ZCFSeat} seat
* @param {{ chainName: string; destAddr: string }} offerArgs
*/
Expand Down
Original file line number Diff line number Diff line change
@@ -1,7 +1,4 @@
/** @file Use-object for the owner of a staking account */
import { Fail } from '@endo/errors';
import { decodeBase64 } from '@endo/base64';
import { E } from '@endo/far';
import { toRequestQueryJson } from '@agoric/cosmic-proto';
import {
QueryBalanceRequest,
Expand All @@ -22,12 +19,14 @@ import { makeTracer } from '@agoric/internal';
import { Shape as NetworkShape } from '@agoric/network';
import { M } from '@agoric/vat-data';
import { VowShape } from '@agoric/vow';
import { decodeBase64 } from '@endo/base64';
import { Fail } from '@endo/errors';
import { E } from '@endo/far';
import {
AmountArgShape,
ChainAddressShape,
ChainAmountShape,
CoinShape,
DelegationShape,
DenomAmountShape,
} from '../typeGuards.js';
import { maxClockSkew, tryDecodeResponse } from '../utils/cosmos.js';
import { orchestrationAccountMethods } from '../utils/orchestrationAccount.js';
Expand Down Expand Up @@ -78,9 +77,9 @@ export const IcaAccountHolderI = M.interface('IcaAccountHolder', {
AmountArgShape,
).returns(VowShape),
withdrawReward: M.call(ChainAddressShape).returns(
Vow$(M.arrayOf(ChainAmountShape)),
Vow$(M.arrayOf(DenomAmountShape)),
),
withdrawRewards: M.call().returns(Vow$(M.arrayOf(ChainAmountShape))),
withdrawRewards: M.call().returns(Vow$(M.arrayOf(DenomAmountShape))),
undelegate: M.call(M.arrayOf(DelegationShape)).returns(VowShape),
});

Expand Down Expand Up @@ -130,7 +129,7 @@ export const prepareCosmosOrchestrationAccountKit = (
withdrawRewardWatcher: M.interface('withdrawRewardWatcher', {
onFulfilled: M.call(M.string())
.optional(M.arrayOf(M.undefined())) // empty context
.returns(M.arrayOf(CoinShape)),
.returns(M.arrayOf(DenomAmountShape)),
}),
holder: IcaAccountHolderI,
invitationMakers: M.interface('invitationMakers', {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,6 @@ import { VowShape } from '@agoric/vow';
import { E } from '@endo/far';
import {
ChainAddressShape,
ChainAmountShape,
DenomAmountShape,
DenomShape,
IBCTransferOptionsShape,
Expand Down Expand Up @@ -105,7 +104,7 @@ export const prepareLocalOrchestrationAccountKit = (
.optional({
destination: ChainAddressShape,
opts: M.or(M.undefined(), IBCTransferOptionsShape),
amount: ChainAmountShape,
amount: DenomAmountShape,
})
.returns(Vow$(M.record())),
}),
Expand Down
41 changes: 33 additions & 8 deletions packages/orchestration/src/facade.js
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
import { assertAllDefined } from '@agoric/internal';

/**
* @import {AsyncFlowTools, GuestInterface, HostArgs} from '@agoric/async-flow';
* @import {AsyncFlowTools, GuestInterface, HostArgs, HostOf} from '@agoric/async-flow';
* @import {Zone} from '@agoric/zone';
* @import {Vow, VowTools} from '@agoric/vow';
* @import {TimerService} from '@agoric/time';
Expand All @@ -14,6 +14,19 @@ import { assertAllDefined } from '@agoric/internal';
* @import {Chain, ChainInfo, CosmosChainInfo, IBCConnectionInfo, OrchestrationAccount, Orchestrator} from './types.js';
*/

/**
* For a given guest passed to orchestrate(), return the host-side form.
*
* @template {(orc: Orchestrator, ctx: any, ...args: any[]) => Promise<any>} GF
* @typedef {GF extends (
* orc: Orchestrator,
* ctx: any,
* ...args: infer GA
* ) => Promise<infer GR>
* ? (...args: HostArgs<GA>) => Vow<GR>
* : never} HostForGuest
Comment on lines +20 to +27
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

this is at the edge of my tsc-fu; an example in the form of a type test would help me.

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

worthwhile. probably another PR.

*/

/**
* @param {{
* zone: Zone;
Expand Down Expand Up @@ -91,15 +104,27 @@ export const makeOrchestrationFacade = ({
*
* NOTE multiple calls to this with the same guestFn name will fail
*
* @param {{ [durableName: string]: (...args: any[]) => any }} guestFns
* @param {any} hostCtx
* @template HC - host context
* @template {{
* [durableName: string]: (
* orc: Orchestrator,
* ctx: GuestInterface<HC>,
* ...args: any[]
* ) => Promise<any>;
* }} GFM
* guest fn map
* @param {GFM} guestFns
* @param {HC} hostCtx
* @returns {{ [N in keyof GFM]: HostForGuest<GFM[N]> }}
*/
const orchestrateAll = (guestFns, hostCtx) =>
Object.fromEntries(
Object.entries(guestFns).map(([name, guestFn]) => [
name,
orchestrate(name, hostCtx, guestFn),
]),
/** @type {{ [N in keyof GFM]: HostForGuest<GFM[N]> }} */ (
Object.fromEntries(
Object.entries(guestFns).map(([name, guestFn]) => [
name,
orchestrate(name, hostCtx, guestFn),
]),
)
);

return harden({
Expand Down
9 changes: 2 additions & 7 deletions packages/orchestration/src/typeGuards.js
Original file line number Diff line number Diff line change
Expand Up @@ -35,13 +35,6 @@ export const Proto3Shape = {
value: M.string(),
};

// XXX same as ChainAmountShape and DenomAmount type
export const CoinShape = { value: M.bigint(), denom: M.string() };
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

👍


export const ChainAmountShape = harden({ denom: M.string(), value: M.nat() });

export const AmountArgShape = M.or(AmountShape, ChainAmountShape);

// FIXME missing `delegatorAddress` from the type
/** @type {TypedPattern<Delegation>} */
export const DelegationShape = harden({
Expand Down Expand Up @@ -111,6 +104,8 @@ export const BrandInfoShape = M.any();
/** @type {TypedPattern<DenomAmount>} */
export const DenomAmountShape = { denom: DenomShape, value: M.bigint() };

export const AmountArgShape = M.or(AmountShape, DenomAmountShape);

/** @see {Chain} */
export const ChainFacadeI = M.interface('ChainFacade', {
getChainInfo: M.call().returns(VowShape),
Expand Down
10 changes: 7 additions & 3 deletions packages/orchestration/src/utils/orchestrationAccount.js
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,11 @@ import { M } from '@endo/patterns';
import { Shape as NetworkShape } from '@agoric/network';
import { VowShape } from '@agoric/vow';
import { TopicsRecordShape } from '@agoric/zoe/src/contractSupport/topics.js';
import { AmountArgShape, ChainAddressShape, CoinShape } from '../typeGuards.js';
import {
AmountArgShape,
ChainAddressShape,
DenomAmountShape,
} from '../typeGuards.js';

/** @import {OrchestrationAccountI} from '../orchestration-api.js'; */

Expand All @@ -11,8 +15,8 @@ const { Vow$ } = NetworkShape; // TODO #9611
/** @see {OrchestrationAccountI} */
export const orchestrationAccountMethods = {
getAddress: M.call().returns(ChainAddressShape),
getBalance: M.call(M.any()).returns(Vow$(CoinShape)),
getBalances: M.call().returns(Vow$(M.arrayOf(CoinShape))),
getBalance: M.call(M.any()).returns(Vow$(DenomAmountShape)),
getBalances: M.call().returns(Vow$(M.arrayOf(DenomAmountShape))),
send: M.call(ChainAddressShape, AmountArgShape).returns(VowShape),
transfer: M.call(AmountArgShape, ChainAddressShape)
.optional(M.record())
Expand Down
1 change: 1 addition & 0 deletions packages/orchestration/src/utils/zoe-tools.js
Original file line number Diff line number Diff line change
Expand Up @@ -77,3 +77,4 @@ export const makeZoeTools = (zone, { zcf, vowTools }) => {
localTransfer,
});
};
/** @typedef {ReturnType<typeof makeZoeTools>} ZoeTools */
Loading