forked from dfinity/agent-js
-
Notifications
You must be signed in to change notification settings - Fork 0
/
management.ts
30 lines (27 loc) · 1005 Bytes
/
management.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
import { Actor, ActorSubclass, CallConfig } from '../actor';
import { Principal } from '@dfinity/principal';
import managementCanisterIdl from './management_idl';
import _SERVICE from './management_service';
export type ManagementCanisterRecord = _SERVICE;
/**
* Create a management canister actor
* @param config
*/
export function getManagementCanister(config: CallConfig): ActorSubclass<ManagementCanisterRecord> {
function transform(_methodName: string, args: unknown[], _callConfig: CallConfig) {
const first = args[0] as any;
let effectiveCanisterId = Principal.fromHex('');
if (first && typeof first === 'object' && first.canister_id) {
effectiveCanisterId = Principal.from(first.canister_id as unknown);
}
return { effectiveCanisterId };
}
return Actor.createActor<ManagementCanisterRecord>(managementCanisterIdl, {
...config,
canisterId: Principal.fromHex(''),
...{
callTransform: transform,
queryTransform: transform,
},
});
}