1
1
import { getViewingKey } from './rpc/getViewingKey' ;
2
2
import { InitOutput } from '@webzjs/webz-keys' ;
3
3
import { initialiseWasm } from './utils/initialiseWasm' ;
4
- import { OnRpcRequestHandler , OnUserInputHandler , UserInputEventType } from '@metamask/snaps-sdk' ;
5
- import { setBirthdayBlock , SetBirthdayBlockParams } from './rpc/setBirthdayBlock' ;
4
+ import {
5
+ OnRpcRequestHandler ,
6
+ OnUserInputHandler ,
7
+ UserInputEventType ,
8
+ } from '@metamask/snaps-sdk' ;
9
+ import { setBirthdayBlock } from './rpc/setBirthdayBlock' ;
10
+ import { getSnapState } from './rpc/getSnapState' ;
11
+ import { SetBirthdayBlockParams , SnapState } from './types' ;
12
+ import { setSnapState } from './rpc/setSnapState' ;
6
13
7
14
let wasm : InitOutput ;
8
15
@@ -24,24 +31,29 @@ export const onRpcRequest: OnRpcRequestHandler = async ({ request }) => {
24
31
case 'getViewingKey' :
25
32
return await getViewingKey ( ) ;
26
33
case 'setBirthdayBlock' :
27
- const params = request . params as SetBirthdayBlockParams ;
28
- return await setBirthdayBlock ( params ) ;
34
+ const setBirthdayBlockParams = request . params as SetBirthdayBlockParams ;
35
+ return await setBirthdayBlock ( setBirthdayBlockParams ) ;
36
+ case 'getSnapStete' :
37
+ return await getSnapState ( ) ;
38
+ case 'setSnapStete' :
39
+ const setSnapStateParams = request . params as unknown as SnapState ;
40
+ return await setSnapState ( setSnapStateParams ) ;
29
41
default :
30
42
throw new Error ( 'Method not found.' ) ;
31
43
}
32
44
} ;
33
45
34
- export const onUserInput : OnUserInputHandler = async ( { id, event, context } ) => {
46
+ export const onUserInput : OnUserInputHandler = async ( { id, event } ) => {
35
47
if ( event . type === UserInputEventType . FormSubmitEvent ) {
36
48
switch ( event . name ) {
37
- case " birthday-block-form" :
49
+ case ' birthday-block-form' :
38
50
await snap . request ( {
39
- method : " snap_resolveInterface" ,
51
+ method : ' snap_resolveInterface' ,
40
52
params : {
41
53
id,
42
- value : event . value
43
- }
44
- } )
54
+ value : event . value ,
55
+ } ,
56
+ } ) ;
45
57
46
58
default :
47
59
break ;
0 commit comments