Update Main and Cidr to use the new modules instead of converting to/from bit strings.

[hath.git] / src / Main.hs

diff --git a/src/Main.hs b/src/Main.hs
index 7b3bd7b0884847a456545903594130e9a8504539..a86eebf90081f0807a8628d35cff6c7a677d20c2 100644 (file)
--- a/src/Main.hs
+++ b/src/Main.hs
@@ -2,17 +2,9 @@ import Data.List (intercalate, intersperse)
 import System.Exit (ExitCode(..), exitWith)
 import System.IO (stderr, hPutStrLn)
 
-import Cidr (Cidr,
-             from_string,
-             is_valid_cidr,
-             min_first_octet,
-             min_second_octet,
-             min_third_octet,
-             min_fourth_octet,
-             max_first_octet,
-             max_second_octet,
-             max_third_octet,
-             max_fourth_octet)
+import Cidr (Cidr(..),
+             cidr_from_string,
+             is_valid_cidr)
 
 import CommandLine (help_set,
                     help_text,
@@ -20,7 +12,10 @@ import CommandLine (help_set,
                     Mode(..),
                     parse_errors,
                     parse_mode)
-
+    
+import IPv4Address
+import Octet
+    
 -- Some exit codes, used in the ExitFailure constructor.
 exit_invalid_cidr :: Int
 exit_invalid_cidr = 1
@@ -59,14 +54,14 @@ cidr_to_regex cidr =
       range2 = numeric_range min2 max2
       range3 = numeric_range min3 max3
       range4 = numeric_range min4 max4
-      min1   = min_first_octet cidr
-      min2   = min_second_octet cidr
-      min3   = min_third_octet cidr
-      min4   = min_fourth_octet cidr
-      max1   = max_first_octet cidr
-      max2   = max_second_octet cidr
-      max3   = max_third_octet cidr
-      max4   = max_fourth_octet cidr
+      min1   = octet_to_int (min_octet1 (ipv4address cidr) (maskbits cidr))
+      min2   = octet_to_int (min_octet2 (ipv4address cidr) (maskbits cidr))
+      min3   = octet_to_int (min_octet3 (ipv4address cidr) (maskbits cidr))
+      min4   = octet_to_int (min_octet4 (ipv4address cidr) (maskbits cidr))
+      max1   = octet_to_int (max_octet1 (ipv4address cidr) (maskbits cidr))
+      max2   = octet_to_int (max_octet2 (ipv4address cidr) (maskbits cidr))
+      max3   = octet_to_int (max_octet3 (ipv4address cidr) (maskbits cidr))
+      max4   = octet_to_int (max_octet4 (ipv4address cidr) (maskbits cidr))
 
 
 
@@ -125,7 +120,7 @@ main = do
 
   let cidr_strings = lines input
   mapM validate_or_die cidr_strings
-  let cidrs = map Cidr.from_string cidr_strings
+  let cidrs = map Cidr.cidr_from_string cidr_strings
 
   -- Get the mode of operation.
   mode <- CommandLine.parse_mode