SmartContract/intercontract.py at master · XTZOptions/SmartContract · 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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
import smartpy as sp

class ALACoin(sp.Contract):
    def __init__(self, admin):
        self.init(balances = sp.big_map(), administrator = admin, totalSupply = 0,contract = sp.big_map())

    @sp.entry_point
    def transfer(self, params):
        sp.verify((sp.sender == self.data.administrator) |
            (((params.fromAddr == sp.sender) |
                 (self.data.balances[params.fromAddr].approvals[sp.sender] >= params.amount))))

        self.addAddressIfNecessary(params.toAddr)
        sp.verify(self.data.balances[params.fromAddr].balance >= params.amount)
        self.data.balances[params.fromAddr].balance -= params.amount
        self.data.balances[params.toAddr].balance += params.amount
        sp.if (params.fromAddr != sp.sender) & (self.data.administrator != sp.sender):
            self.data.balances[params.fromAddr].approvals[params.toAddr] -= params.amount


    @sp.entry_point
    def approve(self, params):
        sp.verify((sp.sender == self.data.administrator) |
                  (params.fromAddr == sp.sender))
        sp.verify(self.data.balances[params.fromAddr].approvals.get(params.toAddr, 0) == 0)
        self.data.balances[params.fromAddr].approvals[params.toAddr] = params.amount

    @sp.entry_point
    def setAdministrator(self, params):
        sp.verify(sp.sender == self.data.administrator)
        self.data.administrator = params

    @sp.entry_point
    def mint(self, params):

        tezValue=sp.tez(sp.as_nat(params.amount))
        sp.verify(sp.amount == tezValue)
        self.addAddressIfNecessary(params.address)
        self.data.balances[params.address].balance += params.amount*10000
        self.data.totalSupply += params.amount*10000

    @sp.entry_point
    def withdraw(self):
        sp.verify(self.data.balances.contains(sp.sender))
        sp.send(sp.sender,sp.tez(1))

        self.data.balances[sp.sender].balance = sp.to_int(self.data.balances[sp.sender].balance % 10000)

    @sp.entry_point
    def burn(self, params):
        sp.verify(sp.sender == self.data.administrator)
        sp.verify(self.data.balances[params.address].balance >= params.amount)
        self.data.balances[params.address].balance -= params.amount
        self.data.totalSupply -= params.amount


    @sp.entry_point
    def AddContract(self,contract):
        sp.verify(sp.sender == self.data.administrator)
        self.data.contract[contract] = 1

    @sp.entry_point
    def LockToken(self,params):
        sp.verify(sp.sender == self.data.administrator)
        sp.verify(self.data.balances[params.address].balance >= params.amount)
        self.data.balances[params.address].balance -= params.amount

    @sp.entry_point
    def UnlockToken(self,params):

        self.data.balances[params.address].balance += params.amount
        self.data.balances[sp.sender].balance += params.amount

    def addAddressIfNecessary(self, address):
        sp.if ~ self.data.balances.contains(address):
            self.data.balances[address] = sp.record(balance = 0, approvals = {})

class Options(sp.Contract):

    def __init__(self, token,admin):
        self.init(tokenContract=token,admin=admin,date=sp.int(0))

    @sp.entry_point
    def Payment(self,params):
        self.data.date = 100
        c = sp.contract(sp.TRecord(address = sp.TAddress, amount = sp.TInt), self.data.tokenContract, entry_point = "UnlockToken").open_some()
        mydata = sp.record(address = sp.sender,amount=10)
        sp.transfer(mydata, sp.mutez(0), c)

@sp.add_test(name = "ALA Coin")
def test():

    scenario = sp.test_scenario()
    scenario.h1("ALA Contract")
    value = 1
    admin = sp.address("tz123")
    alice = sp.address("tz1456")
    bob   = sp.address("tz1678")


    c1 = ALACoin(admin)
    c2 = Options(c1.address,admin)

    scenario += c1
    scenario += c2

    scenario += c1.mint(address = alice, amount = 12).run(sender = alice,amount = sp.tez(12))
    scenario += c1.mint(address = bob, amount = 10).run(sender = alice,amount = sp.tez(10))
    scenario += c1.AddContract(c2.address).run(sender=admin)
    scenario += c2.Payment().run(sender=alice)
    # scenario += c1.LockToken(address = bob, amount = 10).run(sender = admin)
    # scenario += c1.UnlockToken(address = bob, amount = 10).run(sender = admin)
    # scenario += c1.withdraw().run(sender = bob)