X-Git-Url: http://gitweb.michael.orlitzky.com/?p=dead%2Fharbl.git;a=blobdiff_plain;f=src%2FIPv4Pattern.hs;h=8c605ff97cbfb8881301298897da51b88bb8edae;hp=06184534c658c724f7421a9345c6ddd5f72d6e4c;hb=e093d003defb7948f17927091c3e73a250d53e6c;hpb=a50f9a8628239901278c2c971919132a5747cbd4

diff --git a/src/IPv4Pattern.hs b/src/IPv4Pattern.hs
index 0618453..8c605ff 100644
--- a/src/IPv4Pattern.hs
+++ b/src/IPv4Pattern.hs
@@ -1,5 +1,3 @@
-{-# LANGUAGE FlexibleInstances #-}
-
 -- | An IPv4 address pattern has four fields separated by ".".  Each
 --   field is either a decimal number, or a sequence inside "[]" that
 --   contains one or more ";"-separated decimal numbers or
@@ -23,14 +21,17 @@
 --   v4sequence = v4seq_member | v4sequence ";" v4seq_member
 --   v4seq_member = v4octet | v4octet ".." v4octet
 --
-module IPv4Pattern
+module IPv4Pattern (
+  IPv4Pattern,
+  addresses,
+  ipv4pattern_tests,
+  v4pattern)
 where
 
 
 import Test.Tasty ( TestTree, testGroup )
 import Test.Tasty.HUnit ( (@?=), testCase )
 import Text.Parsec (
-  ParseError,
   (<|>),
   char,
   digit,
@@ -42,25 +43,7 @@ import Text.Parsec (
 import Text.Parsec.String ( Parser )
 import Text.Read ( readMaybe )
 
-
-class Pretty a where
-  -- | Obtain a pretty 'String' representation of the given thingy.
-  prettyshow :: a -> String
-
-  -- | Pretty-print the given thingy.
-  pp :: a -> IO ()
-  pp = putStrLn . prettyshow
-
-
--- | Define a 'Pretty' instance for the result of 'parse'. This lets
---   us pretty-print the result of a parse attempt without worrying
---   about whether or not it failed. If the parse failed, you get the
---   same output that you usually would. Otherwise we pretty-print the
---   parsed value.
---
-instance Pretty a => Pretty (Either ParseError a) where
-  prettyshow (Left err) = show err
-  prettyshow (Right v)  = prettyshow v
+import Pretty ( Pretty(..) )
 
 
 -- * Octets
@@ -74,7 +57,7 @@ newtype IPv4Octet = IPv4Octet Int
 
 
 instance Pretty IPv4Octet where
-  prettyshow (IPv4Octet x) = show x
+  pretty_show (IPv4Octet x) = show x
 
 
 -- | Parse an IPv4 octet, which should contain a string of digits.
@@ -147,9 +130,9 @@ data IPv4SequenceMember =
 
 
 instance Pretty IPv4SequenceMember where
-  prettyshow (IPv4SequenceMemberOctet octet) = prettyshow octet
-  prettyshow (IPv4SequenceMemberOctetRange octet1 octet2) =
-    (prettyshow octet1) ++ ".." ++ (prettyshow octet2)
+  pretty_show (IPv4SequenceMemberOctet octet) = pretty_show octet
+  pretty_show (IPv4SequenceMemberOctetRange octet1 octet2) =
+    (pretty_show octet1) ++ ".." ++ (pretty_show octet2)
 
 
 -- | Parse an IPv4 \"sequence member\". A sequence member is either an
@@ -193,9 +176,9 @@ data IPv4Sequence =
 
 
 instance Pretty IPv4Sequence where
-  prettyshow (IPv4SequenceSingleMember member) = prettyshow member
-  prettyshow (IPv4SequenceOptions member subsequence) =
-    (prettyshow member) ++ ";" ++ (prettyshow subsequence)
+  pretty_show (IPv4SequenceSingleMember member) = pretty_show member
+  pretty_show (IPv4SequenceOptions member subsequence) =
+    (pretty_show member) ++ ";" ++ (pretty_show subsequence)
 
 
 -- | Parse an IPv4 \"sequence\". A sequence is whatever is allowed
@@ -212,14 +195,14 @@ instance Pretty IPv4Sequence where
 --   >>> parseTest v4sequence "1"
 --   IPv4SequenceSingleMember (IPv4SequenceMemberOctet (IPv4Octet 1))
 --
---   >>> pp $ parse v4sequence "" "1..2"
+--   >>> pretty_print $ parse v4sequence "" "1..2"
 --   1..2
 --
---   >>> pp $ parse v4sequence "" "1..2;8"
+--   >>> pretty_print $ parse v4sequence "" "1..2;8"
 --   1..2;8
 --
 v4sequence :: Parser IPv4Sequence
-v4sequence = try both <|> just_one -- Maybe sepBy is appropriate here?
+v4sequence = try both <|> just_one
   where
     both = do
       sm <- v4seq_member
@@ -238,8 +221,8 @@ data IPv4Field = IPv4FieldOctet IPv4Octet | IPv4FieldSequence IPv4Sequence
 
 
 instance Pretty IPv4Field where
-  prettyshow (IPv4FieldOctet octet) = prettyshow octet
-  prettyshow (IPv4FieldSequence s) = "[" ++ (prettyshow s) ++ "]"
+  pretty_show (IPv4FieldOctet octet) = pretty_show octet
+  pretty_show (IPv4FieldSequence s) = "[" ++ (pretty_show s) ++ "]"
 
 
 -- | Parse an IPv4 \"field\", which is either a boring old octet, or a
@@ -251,7 +234,7 @@ instance Pretty IPv4Field where
 --   >>> parseTest v4field "127"
 --   IPv4FieldOctet (IPv4Octet 127)
 --
---   >>> pp $ parse v4field "" "[127]"
+--   >>> pretty_print $ parse v4field "" "[127]"
 --   [127]
 --
 v4field :: Parser IPv4Field
@@ -275,13 +258,13 @@ data IPv4Pattern =
 
 
 instance Pretty IPv4Pattern where
-  prettyshow (IPv4Pattern f1 f2 f3 f4) =
-    (prettyshow f1) ++ "."
-                    ++ (prettyshow f2)
+  pretty_show (IPv4Pattern f1 f2 f3 f4) =
+    (pretty_show f1) ++ "."
+                    ++ (pretty_show f2)
                     ++ "."
-                    ++ (prettyshow f3)
+                    ++ (pretty_show f3)
                     ++ "."
-                    ++ (prettyshow f4)
+                    ++ (pretty_show f4)
 
 
 -- | Parse an ipv4 address pattern. This consists of four fields,
@@ -292,13 +275,13 @@ instance Pretty IPv4Pattern where
 --
 --   ==== _Examples_
 --
---   >>> pp $ parse v4pattern "" "127.0.0.1"
+--   >>> pretty_print $ parse v4pattern "" "127.0.0.1"
 --   127.0.0.1
 --
---   >>> pp $ parse v4pattern "" "127.0.[1..3].1"
+--   >>> pretty_print $ parse v4pattern "" "127.0.[1..3].1"
 --   127.0.[1..3].1
 --
---   >>> pp $ parse v4pattern "" "127.0.[1..3;8].1"
+--   >>> pretty_print $ parse v4pattern "" "127.0.[1..3;8].1"
 --   127.0.[1..3;8].1
 --
 --   In the module intro, it is mentioned that this is invalid:
@@ -312,7 +295,7 @@ instance Pretty IPv4Pattern where
 --   This one is /also/ invalid; however, we'll parse the valid part off
 --   the front of it:
 --
---   >>> pp $ parse v4pattern "" "1.2.3.3[6..9]"
+--   >>> pretty_print $ parse v4pattern "" "1.2.3.3[6..9]"
 --   1.2.3.3
 --
 v4pattern :: Parser IPv4Pattern
@@ -437,6 +420,11 @@ addresses (IPv4Pattern field1 field2 field3 field4) = do
 
 -- * Tests
 
+ipv4pattern_tests :: TestTree
+ipv4pattern_tests =
+  testGroup "IPv4Pattern Tests" [ v4octet_tests ]
+
+
 v4octet_tests :: TestTree
 v4octet_tests =
   testGroup