Complete guard service

Cargo.toml

@@ -43,7 +43,7 @@ features = ["full"]
 package = "darwinia-bridge-primitives"
 version = "0.0.10"
 git = "https://github.com/darwinia-network/bridge-primitives.git"
-# branch = "bump"
+# branch = "sudo"
 # path = "../bridge-primitives"
 features = ["runtime"]
 

src/api/darwinia.rs

@@ -1,27 +1,51 @@
 //! Darwinia API
 use crate::{pool::EthereumTransaction, result::Result, Config};
+use core::marker::PhantomData;
 use primitives::{
     chain::eth::{EthereumReceiptProofThing, HeaderStuff, RedeemFor},
-    frame::ethereum::{
-        backing::{RedeemCallExt, VerifiedProofStoreExt},
-        game::{EthereumRelayerGame, PendingHeadersStoreExt, ProposalsStoreExt},
-        relay::{ConfirmedBlockNumbersStoreExt, SubmitProposalCallExt},
+    frame::{
+        collective::{ExecuteCallExt, MembersStoreExt},
+        ethereum::{
+            backing::{RedeemCallExt, VerifiedProofStoreExt},
+            game::{EthereumRelayerGame, PendingHeadersStoreExt, ProposalsStoreExt},
+            relay::{
+                ApprovePendingHeader, ConfirmedBlockNumbersStoreExt, RejectPendingHeader,
+                SubmitProposalCallExt,
+            },
+        },
+        sudo::{KeyStoreExt, SudoCallExt},
     },
     runtime::DarwiniaRuntime,
 };
 use sp_keyring::sr25519::sr25519::Pair;
-use substrate_subxt::{sp_core::Pair as PairTrait, Client, ClientBuilder, PairSigner};
+use substrate_subxt::{
+    sp_core::{Encode, Pair as PairTrait},
+    Client, ClientBuilder, PairSigner,
+};
 use web3::types::H256;
 
 // Types
 type PendingHeader = <DarwiniaRuntime as EthereumRelayerGame>::PendingHeader;
 type RelayProposal = <DarwiniaRuntime as EthereumRelayerGame>::RelayProposal;
 
+/// Account Role
+#[derive(PartialEq, Eq)]
+pub enum Role {
+    /// Sudo Account
+    Sudo,
+    /// Council Member
+    Council,
+    /// Normal Account
+    Normal,
+}
+
 /// Dawrinia API
 pub struct Darwinia {
     client: Client<DarwiniaRuntime>,
     /// Keyring signer
     pub signer: PairSigner<DarwiniaRuntime, Pair>,
+    /// Account Role
+    pub role: Role,
 }
 
 impl Darwinia {
@@ -34,7 +58,55 @@ impl Darwinia {
             .build()
             .await?;
 
-        Ok(Darwinia { client, signer })
+        let pk = signer.signer().public().to_string();
+        let sudo = client.key(None).await?.to_string();
+        let council = client.members(None).await?;
+
+        Ok(Darwinia {
+            client,
+            signer,
+            role: if sudo == pk {
+                Role::Sudo
+            } else if council.iter().any(|cpk| cpk.to_string() == pk) {
+                Role::Council
+            } else {
+                Role::Normal
+            },
+        })
+    }
+
+    /// Approve pending header
+    pub async fn approve_pending_header(&self, pending: u64) -> Result<H256> {
+        let ex = self.client.encode(ApprovePendingHeader {
+            pending,
+            _runtime: PhantomData::default(),
+        })?;
+        Ok(match self.role {
+            Role::Sudo => self.client.sudo(&self.signer, &ex).await?,
+            Role::Council => {
+                self.client
+                    .execute(&self.signer, &ex, ex.size_hint() as u32)
+                    .await?
+            }
+            Role::Normal => H256::from([0; 32]),
+        })
+    }
+
+    /// Reject pending header
+    pub async fn reject_pending_header(&self, pending: u64) -> Result<H256> {
+        let ex = self.client.encode(RejectPendingHeader {
+            pending,
+            _runtime: PhantomData::default(),
+        })?;
+        Ok(match self.role {
+            Role::Sudo => self.client.sudo(&self.signer, &ex).await?,
+            Role::Council => {
+                self.client
+                    .execute(&self.signer, &ex, ex.size_hint() as u32)
+                    .await?
+            }
+            Role::Normal => H256::from([0; 32]),
+        })
     }
 
     /// Get relay proposals

src/api/mod.rs

@@ -2,4 +2,7 @@
 mod darwinia;
 mod shadow;
 
-pub use self::{darwinia::Darwinia, shadow::Shadow};
+pub use self::{
+    darwinia::{Darwinia, Role},
+    shadow::Shadow,
+};