Merge pull request #1169 from input-output-hk/add-pool-retirement-registration

Add pool retirement and pool update certificates graphql types

Author: NicolasDP

jormungandr/src/explorer/graphql/mod.rs

@@ -505,7 +505,7 @@ impl StakeDelegation {
             .map(|addr| Address::from(&ExplorerAddress::New(addr)))
     }
 
-    pub fn pool(&self, context: &Context) -> Vec<Pool> {
+    pub fn pools(&self, context: &Context) -> Vec<Pool> {
         use chain_impl_mockchain::account::DelegationType;
         use std::iter::FromIterator as _;
 
@@ -666,7 +666,7 @@ impl From<certificate::OwnerStakeDelegation> for OwnerStakeDelegation {
     Context = Context,
 )]
 impl OwnerStakeDelegation {
-    fn pool(&self) -> Vec<Pool> {
+    fn pools(&self) -> Vec<Pool> {
         use chain_impl_mockchain::account::DelegationType;
         use std::iter::FromIterator as _;
 
@@ -684,12 +684,63 @@ impl OwnerStakeDelegation {
     }
 }
 
+/// Retirement info for a pool
+struct PoolRetirement {
+    pool_retirement: certificate::PoolRetirement,
+}
+
+impl From<certificate::PoolRetirement> for PoolRetirement {
+    fn from(pool_retirement: certificate::PoolRetirement) -> PoolRetirement {
+        PoolRetirement { pool_retirement }
+    }
+}
+
+#[juniper::object(
+    Context = Context,
+)]
+impl PoolRetirement {
+    pub fn pool_id(&self) -> PoolId {
+        PoolId(format!("{}", self.pool_retirement.pool_id))
+    }
+
+    pub fn retirement_time(&self) -> TimeOffsetSeconds {
+        self.pool_retirement.retirement_time.into()
+    }
+}
+
+struct PoolUpdate {
+    pool_update: certificate::PoolUpdate,
+}
+
+impl From<certificate::PoolUpdate> for PoolUpdate {
+    fn from(pool_update: certificate::PoolUpdate) -> PoolUpdate {
+        PoolUpdate { pool_update }
+    }
+}
+
+#[juniper::object(
+    Context = Context,
+)]
+impl PoolUpdate {
+    pub fn pool_id(&self) -> PoolId {
+        PoolId(format!("{}", self.pool_update.pool_id))
+    }
+
+    pub fn start_validity(&self) -> TimeOffsetSeconds {
+        self.pool_update.start_validity.into()
+    }
+
+    // TODO: Previous keys?
+    // TODO: Updated keys?
+}
+
 // TODO can we use jormungandr-lib Certificate ?
 enum Certificate {
     StakeDelegation(StakeDelegation),
     OwnerStakeDelegation(OwnerStakeDelegation),
     PoolRegistration(PoolRegistration),
-    // TODO: PoolManagement
+    PoolRetirement(PoolRetirement),
+    PoolUpdate(PoolUpdate),
 }
 
 impl TryFrom<chain_impl_mockchain::certificate::Certificate> for Certificate {
@@ -707,8 +758,12 @@ impl TryFrom<chain_impl_mockchain::certificate::Certificate> for Certificate {
             certificate::Certificate::PoolRegistration(c) => {
                 Ok(Certificate::PoolRegistration(PoolRegistration::from(c)))
             }
-            certificate::Certificate::PoolRetirement(_) => Err(ErrorKind::Unimplemented.into()),
-            certificate::Certificate::PoolUpdate(_) => Err(ErrorKind::Unimplemented.into()),
+            certificate::Certificate::PoolRetirement(c) => {
+                Ok(Certificate::PoolRetirement(PoolRetirement::from(c)))
+            }
+            certificate::Certificate::PoolUpdate(c) => {
+                Ok(Certificate::PoolUpdate(PoolUpdate::from(c)))
+            }
         }
     }
 }
@@ -720,6 +775,8 @@ graphql_union!(Certificate: Context |&self| {
         &StakeDelegation => match *self { Certificate::StakeDelegation(ref c) => Some(c), _ => None },
         &OwnerStakeDelegation => match *self { Certificate::OwnerStakeDelegation(ref c) => Some(c), _ => None },
         &PoolRegistration => match *self { Certificate::PoolRegistration(ref c) => Some(c), _ => None },
+        &PoolUpdate => match *self { Certificate::PoolUpdate(ref c) => Some(c), _ => None},
+        &PoolRetirement => match *self { Certificate::PoolRetirement(ref c) => Some(c), _ => None}
     }
 });