keys & values functions

Author: NouDaimon

contracts/data/EnumerableMap.sol

@@ -177,6 +177,70 @@ library EnumerableMap {
         }
     }
 
+    function keys(AddressToAddressMap storage map)
+        internal
+        view
+        returns (address[] memory keys)
+    {
+        uint256 len = map._inner._entries.length;
+        keys = new address[](len);
+
+        unchecked {
+            for (uint256 i; i < len; ++i) {
+                keys[i] = address(
+                    uint160(uint256(map._inner._entries[i]._key))
+                );
+            }
+        }
+    }
+
+    function keys(UintToAddressMap storage map)
+        internal
+        view
+        returns (uint256[] memory keys)
+    {
+        uint256 len = map._inner._entries.length;
+        keys = new uint256[](len);
+
+        unchecked {
+            for (uint256 i; i < len; ++i) {
+                keys[i] = uint256(map._inner._entries[i]._key);
+            }
+        }
+    }
+
+    function values(AddressToAddressMap storage map)
+        internal
+        view
+        returns (address[] memory values)
+    {
+        uint256 len = map._inner._entries.length;
+        values = new address[](len);
+        unchecked {
+            for (uint256 i; i < len; ++i) {
+                values[i] = address(
+                    uint160(uint256(map._inner._entries[i]._value))
+                );
+            }
+        }
+    }
+
+    function values(UintToAddressMap storage map)
+        internal
+        view
+        returns (address[] memory values)
+    {
+        uint256 len = map._inner._entries.length;
+        values = new address[](len);
+        unchecked {
+            for (uint256 i; i < len; ++i) {
+                values[i] = address(
+                    uint160(uint256(map._inner._entries[i]._value))
+                );
+            }
+        }
+    }
+
     function _at(Map storage map, uint256 index)
         private
         view