module Main where
import Text.Printf
import Test.Hspec
import Network.HostAndPort


isLeft :: Either a b -> Bool
isLeft (Left _) = True
isLeft (Right _) = False


main :: IO ()
main = hspec spec


spec :: Spec
spec = do
    describe "isIPv4Address" $ do
        valid isIPv4Address "127.0.0.1"
        valid isIPv4Address "255.255.255.255"

        invalid isIPv4Address " 127.0.0.1"
        invalid isIPv4Address "127.0.0.1 "

        invalid isIPv4Address "9.9.256.1"

    describe "isIPv6Address" $ do
        valid isIPv6Address "::1"
        valid isIPv6Address "::1:2"
        valid isIPv6Address "::1:2:3"
        valid isIPv6Address "::1:2:3:4"
        valid isIPv6Address "::1:2:3:4:5"
        valid isIPv6Address "::1:2:3:4:5:6"
        valid isIPv6Address "::1:2:3:4:5:6:7"
        valid isIPv6Address "1:2:3:4:5:6:7:8"
        valid isIPv6Address "::192.168.1.10"
        valid isIPv6Address "1::192.168.1.10"
        valid isIPv6Address "1:2::192.168.1.10"
        valid isIPv6Address "1:2:3::192.168.1.10"
        valid isIPv6Address "1:2:3:4::192.168.1.10"
        valid isIPv6Address "1:2:3:4:5::192.168.1.10"
        valid isIPv6Address "::1:192.168.1.10"
        valid isIPv6Address "::1:2:192.168.1.10"
        valid isIPv6Address "::1:2:3:192.168.1.10"
        valid isIPv6Address "::1:2:3:4:192.168.1.10"
        valid isIPv6Address "::1:2:3:4:5:192.168.1.10"
        valid isIPv6Address "::1:2:3:4:5:192.168.1.10"
        valid isIPv6Address "ff::192.168.1.10"
        valid isIPv6Address "1:2:3:4:5:6:127.0.0.1"
        valid isIPv6Address "1::"
        valid isIPv6Address "1:2::"
        valid isIPv6Address "1:2:3::"
        valid isIPv6Address "1:2:3:4::"
        valid isIPv6Address "1:2:3:4:5::"
        valid isIPv6Address "1:2:3:4:5:6::"
        valid isIPv6Address "1:2:3:4:5:6:7::"
        valid isIPv6Address "1::f"
        valid isIPv6Address "1:2::f"
        valid isIPv6Address "1:2:3::f"
        valid isIPv6Address "1:2:3:4::f"
        valid isIPv6Address "1:2:3:4:5::f"
        valid isIPv6Address "1:2:3:4:5:6::f"
        valid isIPv6Address "1:2:3:4:5::6:f"
        valid isIPv6Address "1:2:3:4::5:6:f"
        valid isIPv6Address "1:2:3::4:5:6:f"
        valid isIPv6Address "1:2::3:4:5:6:f"
        valid isIPv6Address "1::2:3:4:5:6:f"
        valid isIPv6Address "::1:2:3:4:5:6:f"
        valid isIPv6Address "::ffff:0:255.255.255.255"
        valid isIPv6Address "fe80::7:8%eth0"
        valid isIPv6Address "fe80::7:8%11"

        invalid isIPv6Address "::"
        invalid isIPv6Address "1:2:3:4:5:6:7:8:9"
        invalid isIPv6Address "1:2:3:4:5:6:7:"
        invalid isIPv6Address "1:2:3:4:5:6:7"
        invalid isIPv6Address ":1:2:3:4:5:6:7"
        invalid isIPv6Address "1:2:3:4:5:6:7:10.0.0.1"
        invalid isIPv6Address "::1:2:3:4:5:6:7:8"
        invalid isIPv6Address "1:2:3:4:5:6::192.168.1.10"
        invalid isIPv6Address "1:2:3:4:5:6:7:8::"

    describe "defaultHostAndPort" $ do
        it "parsing hosts" $ do
            defaultHostAndPort "40" "localhost:25" `shouldBe` Just ("localhost", "25")

        it "parsing ipv4" $ do
            defaultHostAndPort "80" "127.0.0.1:8080" `shouldBe` Just ("127.0.0.1", "8080")
            defaultHostAndPort "80" "127.0.0.1" `shouldBe` Just ("127.0.0.1", "80")

        it "parsing ipv6" $ do
            defaultHostAndPort "26000" "[f08e::7:8:1]" `shouldBe` Just ("f08e::7:8:1", "26000")
            defaultHostAndPort "26000" "[f08e::7:8:1]:27000" `shouldBe` Just ("f08e::7:8:1", "27000")

        it "testing invalid" $ do
            defaultHostAndPort "60" "localhost:99999" `shouldBe` Nothing
            defaultHostAndPort "60" "[bad]:25" `shouldBe` Nothing

    describe "detailedHostAndPort" $ do
        it "parsing hosts" $ do
            detailedHostAndPort "localhost" `shouldBe` Right (HostName "localhost", Nothing)
            detailedHostAndPort "some-domain.com" `shouldBe` Right (HostName "some-domain.com", Nothing)
            detailedHostAndPort "42.another-domain.com:3030" `shouldBe` Right (HostName "42.another-domain.com", Just "3030")

        it "parsing ipv6address" $ do
            detailedHostAndPort "[::1]:8080" `shouldBe` Right (IPv6Address "::1", Just "8080")
            detailedHostAndPort "[10::1:f]:8080" `shouldBe` Right (IPv6Address "10::1:f", Just "8080")
            detailedHostAndPort "[::1]:65536" `shouldSatisfy` isLeft

        it "parsing ipv4address" $ do
            detailedHostAndPort "127.0.0.1" `shouldBe` Right (IPv4Address "127.0.0.1", Nothing)
            detailedHostAndPort "127.0.0.1:6060" `shouldBe` Right (IPv4Address "127.0.0.1", Just "6060")
  where
    valid f s = it (printf "valid \"%s\"" s) (f s `shouldBe` True)
    invalid f s = it (printf "invalid \"%s\"" s) (f s `shouldBe` False)