update rewards and governed pool MIP-124

aptos-move/framework/aptos-framework/sources/governed_gas_pool.move

@@ -20,6 +20,8 @@ module aptos_framework::governed_gas_pool {
     #[test_only]
     use aptos_framework::aptos_coin::Self;
 
+    friend aptos_framework::stake;
+
     const MODULE_SALT: vector<u8> = b"aptos_framework::governed_gas_pool";
 
     /// The Governed Gas Pool
@@ -170,6 +172,27 @@ module aptos_framework::governed_gas_pool {
         coin::balance<CoinType>(pool_address)
     }
 
+    /// Withdraws coins from the governed gas pool.
+    ///
+    /// This function allows friend modules to withdraw a specified amount of a given
+    /// `CoinType` from the governed gas pool. It uses the internal signer of the
+    /// governed gas pool to authorize the withdrawal.
+    ///
+    /// @param amount The amount of coins to withdraw from the pool.
+    /// @return A `Coin<CoinType>` resource containing the withdrawn amount.
+    public(friend) fun withdraw_from_pool<CoinType>(
+        amount: u64
+    ): Coin<CoinType> acquires GovernedGasPool {
+        let s = governed_gas_signer(); // uses the private signer function
+        coin::withdraw<CoinType>(&s, amount)
+    }
+
+    /// Register Aptos coin with Gouverned gas signer.
+    public(friend) fun register_coin<CoinType>() acquires GovernedGasPool {
+        let s = governed_gas_signer();
+        coin::register<CoinType>(&s);
+    }
+
     #[test_only]
     /// The AptosCoin mint capability
     struct AptosCoinMintCapability has key {
@@ -361,4 +384,4 @@ module aptos_framework::governed_gas_pool {
         // initialize the governed gas pool again, no abort
         initialize(aptos_framework, vector::empty<u8>());
     }
-}
+}