diffusion/package_manager.move at main · yue1823/diffusion · GitHub

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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
module all_or_nothing::package_manager {

    use std::option::{none, Option};
    use aptos_framework::object;
    use aptos_framework::object::{DeleteRef, ExtendRef, TransferRef};

    const Seed :vector<u8> = b"all or nothing";
    const Limit_Seed:vector<u8> =b"limit";

    friend all_or_nothing::liquidity;
    friend all_or_nothing::limit;
    friend all_or_nothing::badges;
    friend all_or_nothing::whitelist;
    friend all_or_nothing::account;
    friend all_or_nothing::deploy_token;
    friend all_or_nothing::stake;
    friend all_or_nothing::uma;

    #[resource_group_member(group = aptos_framework::object::ObjectGroup)]
    struct Cap_of_obj has key,store{
        tran_cap:TransferRef,
        exten_cap:ExtendRef,
        del_cap:Option<DeleteRef>
    }

    public(friend) fun get_signer():signer acquires Cap_of_obj {
        let extendref = &borrow_global<Cap_of_obj>(object::create_object_address(&@all_or_nothing,Seed)).exten_cap;
        let sig = object::generate_signer_for_extending(extendref);
        return sig
    }
    public(friend) fun get_limit_signer():signer acquires Cap_of_obj {
        let extendref = &borrow_global<Cap_of_obj>(object::create_object_address(&@all_or_nothing,Limit_Seed)).exten_cap;
        let sig = object::generate_signer_for_extending(extendref);
        return sig
    }

     fun init_module(caller:&signer){
        let conf = &object::create_named_object(caller,Seed);
        let obj_signer = object::generate_signer(conf);
        let tran_ref = object::generate_transfer_ref(conf);
        let del_ref = none<DeleteRef>();
        let exten_ref = object::generate_extend_ref(conf);
        move_to(&obj_signer,Cap_of_obj{
            tran_cap:tran_ref,
            exten_cap:exten_ref,
            del_cap:del_ref
        });

         let conf = &object::create_named_object(caller,Limit_Seed);
         let obj_signer = object::generate_signer(conf);
         let tran_ref = object::generate_transfer_ref(conf);
         let del_ref = none<DeleteRef>();
         let exten_ref = object::generate_extend_ref(conf);
         move_to(&obj_signer,Cap_of_obj{
             tran_cap:tran_ref,
             exten_cap:exten_ref,
             del_cap:del_ref
         });
    }

    #[test_only]
    public fun call_package_init(caller:&signer){
        init_module(caller);
    }
}