From 9f7577da5455d1049e9ccfac16277135568d021f Mon Sep 17 00:00:00 2001
From: 0xd34df00d <0xd34df00d@gmail.com>
Date: Thu, 22 Aug 2024 17:24:11 -0500
Subject: [PATCH] Derive `IsList` instances to `EnumSet` and `EnumMap`

---
 Data/EnumMap/Base.hs | 16 ++++++++++++++--
 Data/EnumSet.hs      | 15 +++++++++++++--
 2 files changed, 27 insertions(+), 4 deletions(-)

diff --git a/Data/EnumMap/Base.hs b/Data/EnumMap/Base.hs
index 8a1a141..3d13634 100644
--- a/Data/EnumMap/Base.hs
+++ b/Data/EnumMap/Base.hs
@@ -2,6 +2,7 @@
 {-# LANGUAGE DeriveDataTypeable         #-}
 {-# LANGUAGE GeneralizedNewtypeDeriving #-}
 {-# LANGUAGE DeriveTraversable          #-}
+{-# LANGUAGE TypeFamilies               #-}
 
 -- |
 -- Module      :  $Header$
@@ -184,6 +185,7 @@ import Data.Semigroup ( Semigroup )
 import Data.Traversable ( Traversable )
 import Data.Typeable ( Typeable )
 import Data.Aeson ( FromJSON(..), ToJSON(..) )
+import qualified GHC.Exts as IL
 import Text.Read
 
 -- | Wrapper for 'IntMap' with 'Enum' keys.
@@ -207,6 +209,16 @@ instance (ToJSON a) => ToJSON (EnumMap k a) where
 instance (FromJSON a) => FromJSON (EnumMap k a) where
     parseJSON = fmap (EnumMap . I.fromList) . parseJSON
 
+instance (Enum k) => IL.IsList (EnumMap k a) where
+  type Item (EnumMap k a) = (k, a)
+
+  toList = P.map (first toEnum) . I.toList . unWrap
+  {-# INLINE toList #-}
+
+  fromList = EnumMap . I.fromList . P.map (first fromEnum)
+  {-# INLINE fromList #-}
+
+
 --
 -- Conversion to/from 'IntMap'.
 --
@@ -637,7 +649,7 @@ assocs = P.map (first toEnum) . I.assocs . unWrap
 {-# INLINE assocs #-}
 
 toList :: (Enum k) => EnumMap k a -> [(k, a)]
-toList = P.map (first toEnum) . I.toList . unWrap
+toList = IL.toList
 {-# INLINE toList #-}
 
 toAscList :: (Enum k) => EnumMap k a -> [(k, a)]
@@ -649,7 +661,7 @@ toDescList = P.map (first toEnum) . I.toDescList . unWrap
 {-# INLINE toDescList #-}
 
 fromList :: (Enum k) => [(k, a)] -> EnumMap k a
-fromList = EnumMap . I.fromList . P.map (first fromEnum)
+fromList = IL.fromList
 {-# INLINE fromList #-}
 
 fromListWith :: (Enum k) => (a -> a -> a) -> [(k, a)] -> EnumMap k a
diff --git a/Data/EnumSet.hs b/Data/EnumSet.hs
index 47c15b4..8351cd7 100644
--- a/Data/EnumSet.hs
+++ b/Data/EnumSet.hs
@@ -4,6 +4,7 @@
 {-# LANGUAGE DeriveAnyClass #-}
 {-# LANGUAGE StandaloneDeriving #-}
 {-# LANGUAGE DerivingStrategies #-}
+{-# LANGUAGE TypeFamilies #-}
 
 -- |
 -- Module      :  $Header$
@@ -113,6 +114,7 @@ import Data.Aeson
     ( FromJSON(parseJSON), ToJSON(toEncoding, toJSON) )
 import Text.Read
 import GHC.Generics (Generic)
+import qualified GHC.Exts as IL
 
 -- | Wrapper for 'IntSet' with 'Enum' elements.
 newtype EnumSet k = EnumSet { unWrap :: IntSet }
@@ -143,6 +145,15 @@ instance ToJSON (EnumSet a) where
 instance (FromJSON a) => FromJSON (EnumSet a) where
     parseJSON = fmap (EnumSet . I.fromList) . parseJSON
 
+instance (Enum a) => IL.IsList (EnumSet a) where
+  type Item (EnumSet a) = a
+
+  toList = P.map toEnum . I.toList . unWrap
+  {-# INLINE toList #-}
+
+  fromList = EnumSet . I.fromList . P.map fromEnum
+  {-# INLINE fromList #-}
+
 --
 -- Conversion to/from 'IntSet'.
 --
@@ -316,7 +327,7 @@ elems = P.map toEnum . I.elems . unWrap
 {-# INLINE elems #-}
 
 toList :: (Enum k) => EnumSet k -> [k]
-toList = P.map toEnum . I.toList . unWrap
+toList = IL.toList
 {-# INLINE toList #-}
 
 toAscList :: (Enum k) => EnumSet k -> [k]
@@ -328,7 +339,7 @@ toDescList = P.map toEnum . I.toDescList . unWrap
 {-# INLINE toDescList #-}
 
 fromList :: (Enum k) => [k] -> EnumSet k
-fromList = EnumSet . I.fromList . P.map fromEnum
+fromList = IL.fromList
 {-# INLINE fromList #-}
 
 fromAscList :: (Enum k) => [k] -> EnumSet k