{-# LANGUAGE
    BangPatterns
  , DataKinds
  , DeriveGeneric
  , GADTs
  , MagicHash
  , StandaloneDeriving
  , TypeApplications
#-}

{-# OPTIONS_GHC
    -fno-warn-orphans
#-}

import Test.Tasty.Bench (defaultMain, bench, nf)

import Data.Bytes.Types
import Control.DeepSeq
import GHC.Generics
import Control.Applicative (ZipList(..))
import qualified Data.Bytes.Text.Latin1 as BytesL
import qualified Url
import qualified Url.Unsafe
import qualified Data.ByteString.Char8 as BS
import qualified URI.ByteString as URI
import qualified Weigh
import qualified Dormouse.Uri as Dormouse

instance NFData (URI.URIRef a) where
  rnf (URI.URI a b c d e) = rnf a `seq` rnf b `seq` rnf c `seq` rnf d `seq` rnf e
  rnf (URI.RelativeRef b c d e) = rnf b `seq` rnf c `seq` rnf d `seq` rnf e

instance NFData URI.Authority
instance NFData URI.Host
instance NFData URI.UserInfo
instance NFData URI.SchemaError
instance NFData URI.URIParseError
instance NFData URI.Scheme
instance NFData URI.Port
instance NFData URI.Query

deriving instance Generic Url.ParseError
deriving instance Generic Url.Url
deriving instance Generic Bytes
instance NFData Url.ParseError
instance NFData Bytes
instance NFData Url.Unsafe.Url where
  rnf (Url.Unsafe.Url a _ _ _ _ _ _ _ _) = rnf a

deriving instance Generic (Dormouse.Path a)
deriving instance Generic Dormouse.Authority
deriving instance Generic Dormouse.Fragment
deriving instance Generic Dormouse.Host
deriving instance Generic Dormouse.PathSegment
deriving instance Generic Dormouse.Query
deriving instance Generic Dormouse.Scheme
deriving instance Generic Dormouse.Uri
deriving instance Generic Dormouse.UserInfo
instance NFData (Dormouse.Path a)
instance NFData Dormouse.Authority
instance NFData Dormouse.Fragment
instance NFData Dormouse.Host
instance NFData Dormouse.PathSegment
instance NFData Dormouse.Query
instance NFData Dormouse.Scheme
instance NFData Dormouse.Uri
instance NFData Dormouse.UserInfo

main :: IO ()
main = do
  putStr "\n"
  putStr "Memory usage:"
  Weigh.mainWith $ do
    Weigh.func "url-bytes 1,000 [Maybe Url]" (fmap Url.decodeUrl) bytesUrls
    Weigh.func "uri-bytestring 1,000 [Maybe URI]" (fmap $ URI.parseURI URI.strictURIParserOptions) bsUrls
    Weigh.func "dormouse-uri 1,000 [Maybe Uri]" (fmap (Dormouse.parseUri @Maybe)) bsUrls
  putStr "\n"

  defaultMain
    [ bench "url-bytes 1,000" $ nf (fmap Url.decodeUrl) bytesUrls
    , bench "uri-bytestring strict 1,000" $ nf (fmap $ URI.parseURI URI.strictURIParserOptions) bsUrls
    , bench "uri-bytestring lax 1,000" $ nf (fmap $ URI.parseURI URI.laxURIParserOptions) bsUrls
    , bench "dormouse-uri 1,000" $ nf (fmap (Dormouse.parseUri @Maybe)) bsUrls
    ]
  where
  !permUrls = take 1000 $ getZipList $ 
    (\a b c d -> a <> b <> c <> d)
    <$> ZipList (cycle schemes) 
    <*> ZipList (cycle urls)
    <*> ZipList (cycle paths)
    <*> ZipList (cycle queryParams)
  !bsUrls = BS.pack <$> permUrls
  !bytesUrls = BytesL.fromString <$> permUrls

paths :: [String]
paths =
  [ "/example"
  , "//foo/bar@:"
  , ""
  , "/"
  , "/en/wikipedia/Foo.txt"
  ]

queryParams :: [String]
queryParams =
  [ "?foo=bar&qux=quine"
  , ""
  , "?"
  ]

schemes :: [String]
schemes =
  [ "http://"
  , "https://"
  ]

urls :: [String]
urls =
  [ "ggoogle.com"
  , "guser:password@facebook.org:322"
  , "en.wikipedia.org"
  , "github.com"
  , "www.youtube.com"
  , "www.facebook.com"
  , "www.baidu.com"
  , "www.yahoo.com"
  , "www.amazon.com"
  , "www.wikipedia.org"
  , "www.qq.com"
  , "www.google.co.in"
  , "www.twitter.com"
  , "www.live.com"
  , "www.taobao.com"
  , "www.bing.com#foo"
  , "www.instagram.com"
  , "www.weibo.com"
  , "www.sina.com.cn"
  , "www.linkedin.com"
  , "www.yahoo.co.jp"
  , "www.msn.com"
  , "www.vk.com"
  , "www.google.de"
  , "www.yandex.ru"
  , "www.hao123.com"
  , "www.google.co.uk"
  , "www.reddit.com"
  , "www.ebay.com"
  , "www.google.fr"
  , "www.t.co"
  , "www.tmall.com"
  , "www.google.com.br"
  , "www.360.cn"
  , "www.sohu.com"
  , "www.amazon.co.jp"
  , "www.pinterest.com"
  , "www.netflix.com"
  , "www.google.it"
  , "www.google.ru"
  , "www.microsoft.com"
  , "www.google.es"
  , "www.wordpress.com"
  , "www.gmw.cn"
  , "www.tumblr.com"
  , "www.paypal.com"
  , "www.blogspot.com"
  , "www.imgur.com"
  , "www.stackoverflow.com"
  , "www.aliexpress.com"
  , "www.naver.com"
  , "www.ok.ru"
  , "www.apple.com"
  , "www.github.com"
  , "www.chinadaily.com.cn"
  , "www.imdb.com"
  , "www.google.co.kr"
  , "www.fc2.com"
  , "www.jd.com"
  , "www.blogger.com"
  , "www.163.com"
  , "www.google.ca"
  , "www.whatsapp.com"
  , "www.amazon.in"
  , "www.office.com"
  , "www.tianya.cn"
  , "www.google.co.id"
  , "www.youku.com"
  , "www.rakuten.co.jp"
  , "www.craigslist.org"
  , "www.amazon.de"
  , "www.nicovideo.jp"
  , "www.google.pl"
  , "www.soso.com"
  , "www.bilibili.com"
  , "www.dropbox.com"
  , "www.xinhuanet.com"
  , "www.outbrain.com"
  , "www.pixnet.net"
  , "www.alibaba.com"
  , "www.alipay.com"
  , "www.microsoftonline.com"
  , "www.booking.com"
  , "www.googleusercontent.com"
  , "www.google.com.au"
  , "www.popads.net"
  , "www.cntv.cn"
  , "www.zhihu.com"
  , "www.amazon.co.uk"
  , "www.diply.com"
  , "www.coccoc.com"
  , "www.cnn.com"
  , "www.bbc.co.uk"
  , "www.twitch.tv"
  , "www.wikia.com"
  , "www.google.co.th"
  , "www.go.com"
  , "www.google.com.ph"
  , "www.doubleclick.net"
  , "www.onet.pl"
  , "www.googleadservices.com"
  , "www.accuweather.com"
  , "www.googleweblight.com"
  , "www.answers.yahoo.com"
  ]