feat(database): deregistration certificate (#663)

Signed-off-by: Chris Gianelloni <[email protected]>

Author: wolf31o2

database/account.go

@@ -70,6 +70,19 @@ func (d *Database) SetAccount(
 	)
 }
 
+// SetDeregistration saves a deregistration certificate
+func (d *Database) SetDeregistration(
+	cert *lcommon.DeregistrationCertificate,
+	slot uint64,
+	txn *Txn,
+) error {
+	return d.metadata.SetDeregistration(
+		cert,
+		slot,
+		txn.Metadata(),
+	)
+}
+
 // SetRegistration saves a registration certificate
 func (d *Database) SetRegistration(
 	cert *lcommon.RegistrationCertificate,

database/plugin/metadata/sqlite/account.go

@@ -66,6 +66,40 @@ func (d *MetadataStoreSqlite) SetAccount(
 	return nil
 }
 
+// SetDeregistration saves a deregistration certificate
+func (d *MetadataStoreSqlite) SetDeregistration(
+	cert *lcommon.DeregistrationCertificate,
+	slot uint64,
+	txn *gorm.DB,
+) error {
+	stakeKey := cert.StakeCredential.Credential.Bytes()
+	tmpAccount, err := d.GetAccount(stakeKey, txn)
+	if err != nil {
+		return err
+	}
+	tmpItem := models.Deregistration{
+		StakingKey: stakeKey,
+		AddedSlot:  slot,
+	}
+	tmpAccount.Active = false
+	if txn != nil {
+		if accountErr := txn.Save(&tmpAccount); accountErr.Error != nil {
+			return accountErr.Error
+		}
+		if result := txn.Create(&tmpItem); result.Error != nil {
+			return result.Error
+		}
+	} else {
+		if accountErr := d.DB().Save(&tmpAccount); accountErr.Error != nil {
+			return accountErr.Error
+		}
+		if result := d.DB().Create(&tmpItem); result.Error != nil {
+			return result.Error
+		}
+	}
+	return nil
+}
+
 // SetStakeRegistration saves a stake registration certificate and account
 func (d *MetadataStoreSqlite) SetStakeRegistration(
 	cert *lcommon.StakeRegistrationCertificate,

database/plugin/metadata/store.go

@@ -70,6 +70,11 @@ type MetadataStore interface {
 		bool, // active
 		*gorm.DB,
 	) error
+	SetDeregistration(
+		*lcommon.DeregistrationCertificate,
+		uint64, // slot
+		*gorm.DB,
+	) error
 	SetEpoch(
 		uint64, // slot
 		uint64, // epoch

ledger/certs.go

@@ -37,6 +37,15 @@ func (ls *LedgerState) processTransactionCertificates(
 			return err
 		}
 		switch cert := tmpCert.(type) {
+		case *lcommon.DeregistrationCertificate:
+			err := ls.db.SetDeregistration(
+				cert,
+				blockPoint.Slot,
+				txn,
+			)
+			if err != nil {
+				return err
+			}
 		case *lcommon.PoolRegistrationCertificate:
 			err := ls.db.SetPoolRegistration(
 				cert,