add tps to address book

Author: RodrigoAD

deployment/changesets/cd_deploy_tp_and_configure.go

@@ -120,8 +120,51 @@ func (d DeployTPAndConfigure) Apply(e cldf.Environment, config DeployTPAndConfig
 			if err != nil {
 				return cldf.ChangesetOutput{}, fmt.Errorf("failed to save BnMTokenPoolOwnerCapId address %s for Sui chain %d: %w", tokenPoolReport.Output.DeployBurnMintTokenPoolOutput.Objects.OwnerCapObjectId, config.SuiChainSelector, err)
 			}
+
+		case "lnr":
+			// save LnR Pool to the addressbook
+			typeAndVersionLnRTokenPool := cldf.NewTypeAndVersion(deployment.SuiLnRTokenPoolType, deployment.Version1_0_0)
+			err = ab.Save(config.SuiChainSelector, tokenPoolReport.Output.LockReleaseTPPackageID, typeAndVersionLnRTokenPool)
+			if err != nil {
+				return cldf.ChangesetOutput{}, fmt.Errorf("failed to save LnRTokenPool address %s for Sui chain %d: %w", tokenPoolReport.Output.LockReleaseTPPackageID, config.SuiChainSelector, err)
+			}
+
+			// save LnR Pool State to the addressBook
+			typeAndVersionLnRTokenPoolState := cldf.NewTypeAndVersion(deployment.SuiLnRTokenPoolStateType, deployment.Version1_0_0)
+			err = ab.Save(config.SuiChainSelector, tokenPoolReport.Output.DeployLockReleaseTokenPoolOutput.Objects.StateObjectId, typeAndVersionLnRTokenPoolState)
+			if err != nil {
+				return cldf.ChangesetOutput{}, fmt.Errorf("failed to save LnRTokenPoolState address %s for Sui chain %d: %w", tokenPoolReport.Output.DeployLockReleaseTokenPoolOutput.Objects.StateObjectId, config.SuiChainSelector, err)
+			}
+
+			// save LnR Pool OwnerId to the addressBook
+			typeAndVersionLnRTokenPoolOwnerId := cldf.NewTypeAndVersion(deployment.SuiLnRTokenPoolOwnerIDType, deployment.Version1_0_0)
+			err = ab.Save(config.SuiChainSelector, tokenPoolReport.Output.DeployLockReleaseTokenPoolOutput.Objects.OwnerCapObjectId, typeAndVersionLnRTokenPoolOwnerId)
+			if err != nil {
+				return cldf.ChangesetOutput{}, fmt.Errorf("failed to save LnRTokenPoolOwnerCapId address %s for Sui chain %d: %w", tokenPoolReport.Output.DeployLockReleaseTokenPoolOutput.Objects.OwnerCapObjectId, config.SuiChainSelector, err)
+			}
+
+		case "managed":
+			// save Managed Pool to the addressbook
+			typeAndVersionManagedTokenPool := cldf.NewTypeAndVersion(deployment.SuiManagedTokenPoolType, deployment.Version1_0_0)
+			err = ab.Save(config.SuiChainSelector, tokenPoolReport.Output.ManagedTPPackageId, typeAndVersionManagedTokenPool)
+			if err != nil {
+				return cldf.ChangesetOutput{}, fmt.Errorf("failed to save ManagedTokenPool address %s for Sui chain %d: %w", tokenPoolReport.Output.ManagedTPPackageId, config.SuiChainSelector, err)
+			}
+
+			// save Managed Pool State to the addressBook
+			typeAndVersionManagedTokenPoolState := cldf.NewTypeAndVersion(deployment.SuiManagedTokenPoolStateType, deployment.Version1_0_0)
+			err = ab.Save(config.SuiChainSelector, tokenPoolReport.Output.DeployManagedTokenPoolOutput.Objects.StateObjectId, typeAndVersionManagedTokenPoolState)
+			if err != nil {
+				return cldf.ChangesetOutput{}, fmt.Errorf("failed to save ManagedTokenPoolState address %s for Sui chain %d: %w", tokenPoolReport.Output.DeployManagedTokenPoolOutput.Objects.StateObjectId, config.SuiChainSelector, err)
+			}
+
+			// save Managed Pool OwnerId to the addressBook
+			typeAndVersionManagedTokenPoolOwnerId := cldf.NewTypeAndVersion(deployment.SuiManagedTokenPoolOwnerIDType, deployment.Version1_0_0)
+			err = ab.Save(config.SuiChainSelector, tokenPoolReport.Output.DeployManagedTokenPoolOutput.Objects.OwnerCapObjectId, typeAndVersionManagedTokenPoolOwnerId)
+			if err != nil {
+				return cldf.ChangesetOutput{}, fmt.Errorf("failed to save ManagedTokenPoolOwnerCapId address %s for Sui chain %d: %w", tokenPoolReport.Output.DeployManagedTokenPoolOutput.Objects.OwnerCapObjectId, config.SuiChainSelector, err)
+			}
 		}
-		// Note: Address book saving for "lnr" and "managed" token pools can be added here if needed
 	}
 
 	return cldf.ChangesetOutput{

deployment/types.go

@@ -42,9 +42,21 @@ var (
 	SuiLinkTokenTreasuryCapID    deployment.ContractType = "SuiLinkTokenTreasuryCapID"
 	SuiMCMSType                  deployment.ContractType = "SuiManyChainMultisig"
 	SuiLinkTokenType             deployment.ContractType = "SuiLinkToken"
-	SuiBnMTokenPoolType          deployment.ContractType = "SuiBnMTokenPool"
-	SuiBnMTokenPoolStateType     deployment.ContractType = "SuiBnMTokenPoolState"
-	SuiBnMTokenPoolOwnerIDType   deployment.ContractType = "SuiBnMTokenPoolOwnerID"
+
+	// BnM Token Pool related
+	SuiBnMTokenPoolType        deployment.ContractType = "SuiBnMTokenPool"
+	SuiBnMTokenPoolStateType   deployment.ContractType = "SuiBnMTokenPoolState"
+	SuiBnMTokenPoolOwnerIDType deployment.ContractType = "SuiBnMTokenPoolOwnerID"
+
+	// LnR Token Pool related
+	SuiLnRTokenPoolType        deployment.ContractType = "SuiLnRTokenPool"
+	SuiLnRTokenPoolStateType   deployment.ContractType = "SuiLnRTokenPoolState"
+	SuiLnRTokenPoolOwnerIDType deployment.ContractType = "SuiLnRTokenPoolOwnerID"
+
+	// Managed Token Pool related
+	SuiManagedTokenPoolType        deployment.ContractType = "SuiManagedTokenPool"
+	SuiManagedTokenPoolStateType   deployment.ContractType = "SuiManagedTokenPoolState"
+	SuiManagedTokenPoolOwnerIDType deployment.ContractType = "SuiManagedTokenPoolOwnerID"
 )
 
 var (