{-# LANGUAGE OverloadedLists #-}
{-# OPTIONS_GHC -Wno-orphans #-}

module Props.PointedHask where

import Data.Void (Void)
import Test.Falsify.Generator (Function (..), oneof)
import Test.Tasty (TestTree, testGroup)
import Prelude

import Proarrow.Category.Instance.PointedHask (FromPointed (..), POINTED (..), Pointed (..), These (..))
import Proarrow.Core (CategoryOf (..), UN)

import Proarrow.Testing
  ( GenTotal (..)
  , Testable (..)
  , TestableProfunctor
  , TestableType (..)
  , TestingEqShow (..)
  , genSomeDef
  , invmap
  , pattern GenNonEmpty
  )
import Proarrow.Testing.Laws
import Props.Hask ()

test :: TestTree
test =
  testGroup
    "Pointed Hask"
    [ testCategory @POINTED
    , testTerminalObject @POINTED
    , testInitialObject @POINTED
    , testBinaryProducts @POINTED (\r -> r)
    , testBinaryCoproducts @POINTED (\r -> r)
    , testMonoidal @POINTED (\r -> r)
    , testSymMonoidal @POINTED (\r -> r)
    , testCopyDiscard @POINTED (\r -> r)
    , testMonoid @(P Void) (\r -> r)
    , testMonoid @(P ()) (\r -> r)
    , testMonoid @(P [()]) (\r -> r)
    , testFunctor @(FromPointed []) (\r -> r)
    ]

instance (TestOb a, TestOb b) => TestableType (Pointed a b) where
  gen = invmap Pt unPt gen
instance (TestOb a, TestOb b) => TestingEqShow (Pointed a b) where
  eqP (Pt l) (Pt r) = eqP l r
  showP _ = "<pointed function>"
instance TestableProfunctor Pointed

instance Testable POINTED where
  type TestOb a = (Ob a, TestOb (UN P a))
  showOb @(P a) = showOb @_ @a
  genSome = genSomeDef @'[P Bool, P (Bool, Bool), P (Maybe Bool)]

instance (TestableType a, TestableType b) => TestableType (These a b) where
  gen = case (gen @a, gen @b) of
    (GenEmpty l, GenEmpty r) -> GenEmpty \case
      This x -> l x
      That y -> r y
      These x _ -> l x
    (GenNonEmpty ga, GenEmpty _) -> GenNonEmpty (This <$> ga)
    (GenEmpty _, GenNonEmpty gb) -> GenNonEmpty (That <$> gb)
    (GenNonEmpty ga, GenNonEmpty gb) -> GenNonEmpty (oneof [This <$> ga, That <$> gb, These <$> ga <*> gb])
instance (TestingEqShow a, TestingEqShow b) => TestingEqShow (These a b) where
  eqP (This l) (This r) = eqP l r
  eqP (That l) (That r) = eqP l r
  eqP (These l1 l2) (These r1 r2) = liftA2 (&&) (eqP l1 r1) (eqP l2 r2)
  eqP _ _ = pure False
  showP (This a) = "This " ++ showP a
  showP (That b) = "That " ++ showP b
  showP (These a b) = "These " ++ showP a ++ " " ++ showP b
instance (Function a, Function b) => Function (These a b)

instance (TestOb (a :: POINTED)) => TestableType (FromPointed [] a) where
  gen = invmap FromPointed unFromPointed gen
instance (TestOb (a :: POINTED)) => TestingEqShow (FromPointed [] a) where
  eqP (FromPointed l) (FromPointed r) = eqP l r
  showP (FromPointed xs) = "FromPointed " ++ showP xs
instance Function (FromPointed [] a) where
  function = error "Function (FromPointed [] a): unused"