become delegate on every add_stake op

Author: camfairchild

pallets/subtensor/src/staking/add_stake.rs

@@ -142,10 +142,15 @@ impl<T: Config> Pallet<T> {
             allow_partial,
         )?;
 
-        // 4. Ensure the remove operation from the coldkey is a success.
+        // 4. If the coldkey is not the owner, make the hotkey a delegate.
+        if Self::get_owning_coldkey_for_hotkey(&hotkey) != coldkey {
+            Self::maybe_become_delegate(&hotkey);
+        }
+
+        // 5. Ensure the remove operation from the coldkey is a success.
         let tao_staked: u64 = Self::remove_balance_from_coldkey_account(&coldkey, possible_stake)?;
 
-        // 5. Swap the stake into alpha on the subnet and increase counters.
+        // 6. Swap the stake into alpha on the subnet and increase counters.
         // Emit the staking event.
         let fee = DefaultStakingFee::<T>::get();
         Self::stake_into_subnet(&hotkey, &coldkey, netuid, tao_staked, fee);

pallets/subtensor/src/staking/helpers.rs

@@ -99,6 +99,13 @@ impl<T: Config> Pallet<T> {
         }
     }
 
+    //// If the hotkey is not a delegate, make it a delegate.
+    pub fn maybe_become_delegate(hotkey: &T::AccountId) {
+        if !Self::hotkey_is_delegate(hotkey) {
+            Self::delegate_hotkey(hotkey, Self::get_hotkey_take(hotkey));
+        }
+    }
+
     /// Returns the coldkey owning this hotkey. This function should only be called for active accounts.
     ///
     /// # Arguments

pallets/subtensor/src/staking/move_stake.rs

@@ -344,6 +344,11 @@ impl<T: Config> Pallet<T> {
         // Because of the fee, the tao_unstaked may be too low if initial stake is low. In that case,
         // do not restake.
         if tao_unstaked >= DefaultMinStake::<T>::get().saturating_add(fee) {
+            // If the coldkey is not the owner, make the hotkey a delegate.
+            if Self::get_owning_coldkey_for_hotkey(destination_hotkey) != *destination_coldkey {
+                Self::maybe_become_delegate(destination_hotkey);
+            }
+
             Self::stake_into_subnet(
                 destination_hotkey,
                 destination_coldkey,