feat: track registration certificates and create account (#634)

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

Author: wolf31o2

database/certs.go

@@ -52,6 +52,15 @@ func (d *Database) SetPoolRetirement(
 	return d.metadata.SetPoolRetirement(cert, slot, txn.Metadata())
 }
 
+// SetRegistration saves a registration certificate
+func (d *Database) SetRegistration(
+	cert *lcommon.RegistrationCertificate,
+	slot, deposit uint64,
+	txn *Txn,
+) error {
+	return d.metadata.SetRegistration(cert, slot, deposit, txn.Metadata())
+}
+
 // SetStakeDelegation saves a stake delegation certificate
 func (d *Database) SetStakeDelegation(
 	cert *lcommon.StakeDelegationCertificate,

database/plugin/metadata/sqlite/certs.go

@@ -295,3 +295,37 @@ func (d *MetadataStoreSqlite) SetStakeRegistration(
 	}
 	return nil
 }
+
+// SetRegistration saves a registration certificate
+func (d *MetadataStoreSqlite) SetRegistration(
+	cert *lcommon.RegistrationCertificate,
+	slot, deposit uint64,
+	txn *gorm.DB,
+) error {
+	tmpAccount := models.Account{
+		StakingKey:    cert.StakeCredential.Credential.Bytes(),
+		AddedSlot:     slot,
+		DepositAmount: deposit,
+	}
+	tmpItem := models.Registration{
+		StakingKey:    cert.StakeCredential.Credential.Bytes(),
+		AddedSlot:     slot,
+		DepositAmount: deposit,
+	}
+	if txn != nil {
+		if accountResult := txn.Create(&tmpAccount); accountResult.Error != nil {
+			return accountResult.Error
+		}
+		if result := txn.Create(&tmpItem); result.Error != nil {
+			return result.Error
+		}
+	} else {
+		if accountResult := d.DB().Create(&tmpAccount); accountResult.Error != nil {
+			return accountResult.Error
+		}
+		if result := d.DB().Create(&tmpItem); result.Error != nil {
+			return result.Error
+		}
+	}
+	return nil
+}

database/plugin/metadata/store.go

@@ -92,6 +92,12 @@ type MetadataStore interface {
 		uint64, // epoch
 		*gorm.DB,
 	) error
+	SetRegistration(
+		*lcommon.RegistrationCertificate,
+		uint64, // slot
+		uint64, // deposit
+		*gorm.DB,
+	) error
 	SetStakeDelegation(
 		*lcommon.StakeDelegationCertificate,
 		uint64, // slot

ledger/certs.go

@@ -1,4 +1,4 @@
-// Copyright 2024 Blink Labs Software
+// Copyright 2025 Blink Labs Software
 //
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.
@@ -56,6 +56,16 @@ func (ls *LedgerState) processTransactionCertificates(
 			if err != nil {
 				return err
 			}
+		case *lcommon.RegistrationCertificate:
+			err := ls.db.SetRegistration(
+				cert,
+				blockPoint.Slot,
+				certDeposit,
+				txn,
+			)
+			if err != nil {
+				return err
+			}
 		case *lcommon.StakeDelegationCertificate:
 			err := ls.db.SetStakeDelegation(
 				cert,

ledger/eras/conway.go

@@ -132,6 +132,8 @@ func CertDepositConway(
 	switch cert.(type) {
 	case *lcommon.PoolRegistrationCertificate:
 		return uint64(tmpPparams.PoolDeposit), nil
+	case *lcommon.RegistrationCertificate:
+		return uint64(tmpPparams.KeyDeposit), nil
 	case *lcommon.StakeRegistrationCertificate:
 		return uint64(tmpPparams.KeyDeposit), nil
 	default: