feat: expose module account setter

terra-money · Oct 13, 2023 · 97f7d08 · 97f7d08
1 parent f1b76a2
commit 97f7d08
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/x/auth/keeper/keeper.go b/x/auth/keeper/keeper.go
@@ -216,6 +216,21 @@ func (ak AccountKeeper) GetModuleAccountAndPermissions(ctx sdk.Context, moduleNa
 	return maccI, perms
 }
 
+// SetModuleAccountAndPermissions the module account from the auth account store and its
+// registered permissions
+func (ak AccountKeeper) SetModuleAccountAndPermissions(ctx sdk.Context, moduleName string, perms ...string) types.ModuleAccountI {
+	// create a new empty module account
+	macc := types.NewEmptyModuleAccount(moduleName, perms...)
+	// set the account number
+	maccI := (ak.NewAccount(ctx, macc)).(types.ModuleAccountI)
+	// Add the account to the store
+	ak.SetModuleAccount(ctx, maccI)
+	// Add the permissions for the module account
+	ak.permAddrs[moduleName] = types.NewPermissionsForAddress(maccI.GetName(), perms)
+
+	return maccI
+}
+
 // GetModuleAccount gets the module account from the auth account store, if the account does not
 // exist in the AccountKeeper, then it is created.
 func (ak AccountKeeper) GetModuleAccount(ctx sdk.Context, moduleName string) types.ModuleAccountI {