Hide away support modules.

substrate/wasm-runtime/polkadot/src/support/function.rs

@@ -0,0 +1,50 @@
+use staking;
+use consensus;
+use primitives::AccountID;
+use streamreader::StreamReader;
+
+/// The functions that a transaction can call (and be dispatched to).
+#[cfg_attr(test, derive(PartialEq, Debug))]
+#[derive(Clone, Copy)]
+pub enum Function {
+	StakingStake,
+	StakingUnstake,
+	StakingTransferStake,
+	ConsensusSetSessionKey,
+}
+
+impl Function {
+	pub fn from_u8(value: u8) -> Option<Function> {
+		match value {
+			x if x == Function::StakingStake as u8 => Some(Function::StakingStake),
+			x if x == Function::StakingUnstake as u8 => Some(Function::StakingUnstake),
+			x if x == Function::StakingTransferStake as u8 => Some(Function::StakingTransferStake),
+			x if x == Function::ConsensusSetSessionKey as u8 => Some(Function::ConsensusSetSessionKey),
+			_ => None,
+		}
+	}
+}
+
+impl Function {
+	/// Dispatch the function.
+	pub fn dispatch(&self, transactor: &AccountID, data: &[u8]) {
+		let mut params = StreamReader::new(data);
+		match *self {
+			Function::StakingStake => {
+				staking::stake(transactor);
+			}
+			Function::StakingUnstake => {
+				staking::unstake(transactor);
+			}
+			Function::StakingTransferStake => {
+				let dest = params.read().unwrap();
+				let value = params.read().unwrap();
+				staking::transfer_stake(transactor, &dest, value);
+			}
+			Function::ConsensusSetSessionKey => {
+				let session = params.read().unwrap();
+				consensus::set_session_key(transactor, &session);
+			}
+		}
+	}
+}