lib.rs 3.73 KB
Newer Older
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
// Copyright 2018-2019 Parity Technologies (UK) Ltd.
// This file is part of ink!.
//
// ink! is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// ink! is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with ink!.  If not, see <http://www.gnu.org/licenses/>.

#![cfg_attr(not(feature = "std"), no_std)]

use ink_core::{
    memory::format,
    storage,
};
use ink_lang::contract;

#[derive(Debug, Copy, Clone, PartialEq, Eq, scale::Encode, scale::Decode)]
#[cfg_attr(feature = "ink-generate-abi", derive(type_metadata::Metadata))]
pub enum Which {
    Adder,
    Subber,
}

impl ink_core::storage::Flush for Which {
    fn flush(&mut self) {}
}

contract! {
    #![env = ink_core::env::DefaultSrmlTypes]

    /// Delegates calls to an adder or subber contract to mutate
    /// a value in an accumulator contract.
    ///
    /// In order to deploy the delegator smart contract we first
    /// have to manually put the code of the accumulator, adder
    /// and subber smart contracts, receive their code hashes from
    /// the signalled events and put their code hash into our
    /// delegator smart contract.
    struct Delegator {
        /// Says which of adder or subber is currently in use.
        which: storage::Value<Which>,
        /// The accumulator smart contract.
        accumulator: storage::Value<accumulator::Accumulator>,
        /// The adder smart contract.
        adder: storage::Value<adder::Adder>,
        /// The subber smart contract.
        subber: storage::Value<subber::Subber>,
    }

    impl Deploy for Delegator {
        /// Initializes the value to the initial value.
        fn deploy(
            &mut self,
            init_value: i32,
            accumulator_code_hash: Hash,
            adder_code_hash: Hash,
            subber_code_hash: Hash,
        ) {
            self.which.set(Which::Adder);
            let total_balance = env.balance();
            let accumulator = accumulator::Accumulator::new(accumulator_code_hash, init_value)
                .value(total_balance / 4)
                .create()
                .expect("failed at instantiating the accumulator contract");
            self.accumulator.set(accumulator.clone());
            self.adder.set(
                adder::Adder::new(adder_code_hash, accumulator.account_id())
                    .value(total_balance / 4)
                    .create()
                    .expect("failed at instantiating the adder contract")
            );
            self.subber.set(
                subber::Subber::new(subber_code_hash, accumulator.account_id())
                    .value(total_balance / 4)
                    .create()
                    .expect("failed at instantiating the subber contract")
            );
        }
    }

    impl Delegator {
        /// Delegates the call.
        pub(external) fn delegate(&mut self, by: i32) {
            match &*self.which {
                Which::Adder => self.adder.inc(by),
                Which::Subber => self.subber.dec(by),
            }
        }

        /// Switches the delegator.
        pub(external) fn switch(&mut self) {
            match *self.which {
                Which::Adder => {
                    *self.which = Which::Subber;
                }
                Which::Subber => {
                    *self.which = Which::Adder;
                }
            }
        }
    }
}