predicate-typed (empty) → 0.1.0.0
raw patch · 19 files changed
+11286/−0 lines, 19 filesdep +QuickCheckdep +aesondep +basesetup-changed
Dependencies added: QuickCheck, aeson, base, binary, bytestring, comonad, containers, deepseq, directory, discrimination, doctest, ghc-prim, lens, mtl, pcre-heavy, pcre-light, predicate-typed, pretty, pretty-terminal, pretty-tree, safe, semialign, stm, tasty, tasty-hunit, tasty-quickcheck, template-haskell, text, th-lift, th-orphans, these, these-lens, time, tree-view
Files
- ChangeLog.md +3/−0
- LICENSE +29/−0
- README.md +220/−0
- Setup.hs +2/−0
- doctests.hs +11/−0
- predicate-typed.cabal +162/−0
- src/Predicate.hs +6618/−0
- src/Refined.hs +427/−0
- src/Refined3.hs +742/−0
- src/Refined3Helper.hs +334/−0
- src/TH_Orphans.hs +28/−0
- src/UtilP.hs +994/−0
- src/UtilP_TH.hs +83/−0
- test/TastyExtras.hs +78/−0
- test/TestJson.hs +94/−0
- test/TestPredicate.hs +919/−0
- test/TestRefined.hs +142/−0
- test/TestRefined3.hs +387/−0
- test/TestSpec.hs +13/−0
+ ChangeLog.md view
@@ -0,0 +1,3 @@+# Changelog for predicate-typed + +## Unreleased changes
+ LICENSE view
@@ -0,0 +1,29 @@+BSD 3-Clause License + +Copyright (c) 2018, Grant Weyburne +All rights reserved. + +Redistribution and use in source and binary forms, with or without +modification, are permitted provided that the following conditions are met: + +1. Redistributions of source code must retain the above copyright notice, this + list of conditions and the following disclaimer. + +2. Redistributions in binary form must reproduce the above copyright notice, + this list of conditions and the following disclaimer in the documentation + and/or other materials provided with the distribution. + +3. Neither the name of the copyright holder nor the names of its + contributors may be used to endorse or promote products derived from + this software without specific prior written permission. + +THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" +AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE +IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE +DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE +FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL +DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR +SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER +CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, +OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE +OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ README.md view
@@ -0,0 +1,220 @@+# predicate-typed + +what this library provides: +1. a dsl for building refinement types +2. Refined is simple refinement type that just validates the input against a predicate +3. Refined3 is a more complex refinement type that allows you to change the input +4. validation against input values +5. visualisation of each step in the process +6 template haskell methods for creating the refinement types at compile time +7. ToJSON and FromJSON instances for Refined and Refined3 +8. Read and Show instance for Refined and Refined3 +9. Binary instances for Refined and Refined3 +10. database encoders and decoders using odbc(sqlhandler-odbcalt) or hdbc((sqlhandler-odbc) +11. quickcheck arbitrary methods + +```haskell +data Refined p a = Refined a +``` +* **_a_** is the input type +* **_p_** predicate on **_a_** + +### Examples of Refined (for more information see [doctests](src/Refined.hs)) +1. reads in a number and checks to see that it is greater than 99 +```haskell +>prtRefinedIO @(ReadP Int >> Id > 99) ol "123" +Right (Refined {unRefined = "123"}) +``` + +2. reads in a number but fails at compile-time +```haskell +>prtRefinedIO @(ReadP Int >> Id > 99) ol "1x2y3" +Left (FailP "ReadP Int (1x2y3) failed") +``` + +3. reads in a hexadecimal string and checks to see that it is between 99 and 256 +```haskell +>prtRefinedIO @(ReadBase Int 16 >> Between 99 256) ol "000fe" +Right (Refined {unRefined = "000fe"}) +``` + +4. reads in a hexadecimal string but fails the predicate check so doesnt compile +```haskell +>prtRefinedIO @(ReadBase Int 16 >> Between 99 253) ol "000fe" +Left FalseP +``` + +5. same as 4. above but now we get details of where it went wrong +```haskell +>prtRefinedIO @(ReadBase Int 16 >> Between 99 253) o2 "000fe" +``` + +6. reads in a string as time and does simple validation +```haskell +>prtRefinedIO @(Resplit ":" Id >> Map (ReadP Int) Id >> Len == 3) ol "12:01:05" +Right (Refined {unRefined = "12:01:05"}) +``` + * `Resplit ":" Id` + split using regex using a colon as a delimiter ["12","01","05"] + * `Map (ReadP Int) Id` + Read in the values as Ints [12,1,5] + * `Len == 3` + Check to see that the length of the list of Ints is 3 + + +### Testing out predicates +When using _Refined_ the expression in _p_ must result in a True/False\ +_pe2_ does not have that restriction so you can run the whole thing or the individual pieces\ +(for less detail use _pl_) + +```haskell +>pe2 @(Resplit ":" Id >> Map (ReadP Int) Id >> Len == 3) "12:01:05" + +>pe2 @(Resplit ":" Id) "12:01:05" + +>pe2 @(Map (ReadP Int) Id) ["12","01","05"] + +>pe2 @(Len == 3) [12,1,5] +``` + +### An example using Refined3 (for more information see [doctests](src/Refined3.hs) and [doctests](src/Refined3Helper.hs)) + +```haskell +>type Hex = '(ReadBase Int 16, Between 0 255, ShowBase 16, String) + +>prtEval3PIO (Proxy @Hex) ol "0000fe" +Refined3 {in3 = 254, out3 = "fe"} +``` +1. `ReadBase Int 16` + reads a hexadecimal string and returns 254 +2. `Between 0 255` + checks to make sure the predicate holds ie the number is between 0 and 255 +3. `ShowBase 16` + formats the output as "fe" which is compatible with the input + +run this to get details in color of each evaluation step: +```haskell +>prtEval3PIO (Proxy @Hex) o2 "0000fe" + +***Step 1. Success Initial Conversion(ip) [254] *** + +P ReadBase(Int) 16 254 | "0000fe" +| +`- P Id "0000fe" + +***Step 2. Success Boolean Check(op) *** + +True True && True +| ++- True 254 >= 0 +| | +| +- P I +| | +| `- P '0 +| +`- True 254 <= 255 + | + +- P I + | + `- P '255 + +***Step 3. Success Output Conversion(fmt) = "fe" *** + +P ShowBase 16 fe | 254 +``` + +Read in the string "0000fe" as input to `ReadBase Int 16` and produce 254 as output +```haskell +>pe2 @(ReadBase Int 16) "0000fe" +PresentT 254 + +>pe2 @(Between 0 255) 254 +TrueT + +>pe2 @(ShowBase 16) 254 = "fe" +PresentT "fe" +``` + +### Template Haskell versions + +```haskell +ex1 :: Refined (ReadP Int >> Id > 99) String +ex1 = $$(refinedTH "123") +``` + +```haskell +type Hex = '(ReadBase Int 16, Between 0 255, ShowBase 16, String) + +ex2 :: MakeR3 Hex +ex2 = $$(refined3TH "0000fe") +``` + +### Any valid Read/Show instance can be used with Refined3 +```haskell +>$$(refined3TH "13 % 3") :: ReadShowR Rational +Refined3 {r3In = 13 % 3, r3Out = "13 % 3"} + +>$$(refined3TH "2016-11-09") :: ReadShowR Day +Refined3 {r3In = 2016-11-09, r3Out = "2016-11-09"} +``` + +An example of an invalid refined3TH call +```haskell +>$$(refined3TH "2016-xy-09") :: ReadShowR Day + +<interactive>:171:4: error: + * refined3TH: predicate failed with Step 1. Initial Conversion(ip) Failed | ReadP Day (2016-xy-09) failed + * In the Template Haskell splice $$(refined3TH "2016-xy-09") + In the expression: $$(refined3TH "2016-xy-09") :: ReadShowR Day + In an equation for `it': + it = $$(refined3TH "2016-xy-09") :: ReadShowR Day +``` + +### Json decoding + +#### This example is successful as it is a valid hexadecimal and is in the range 10 though 256 +```haskell +>eitherDecode' @(Refined3 (ReadBase Int 16) (Id > 10 && Id < 256) ShowP String) "\"00fe\"" +Right (Refined3 {in3 = 254, out3 = "254"}) +``` + +#### This example fails as the value is not a valid hexadecimal string +```haskell +>either putStrLn print $ eitherDecode' @(Refined3 (ReadBase Int 16) 'True ShowP String) "\"00feg\"" +Error in $: Refined3:Step 1. Initial Conversion(ip) Failed | invalid base 16 + +***Step 1. Initial Conversion(ip) Failed *** + +[Error invalid base 16] ReadBase(Int) 16 as=00feg err=[(254,"g")] +| +`- P Id "00feg" +``` + +#### This example fails as the hexadecimal value is valid but is not between 10 and 256 + +```haskell +>either putStrLn print $ eitherDecode' @(Refined3 (ReadBase Int 16) (Id > 10 && Id < 256) ShowP String) "\"00fe443a\"" +Error in $: Refined3:Step 2. False Boolean Check(op) | FalseP + +***Step 1. Success Initial Conversion(ip) [16663610] *** + +P ReadBase(Int,16) 16663610 | "00fe443a" +| +`- P Id "00fe443a" + +***Step 2. False Boolean Check(op) = FalseP *** + +False True && False +| ++- True 16663610 > 10 +| | +| +- P Id 16663610 +| | +| `- P '10 +| +`- False 16663610 < 256 + | + +- P Id 16663610 + | + `- P '256 +```
+ Setup.hs view
@@ -0,0 +1,2 @@+import Distribution.Simple +main = defaultMain
+ doctests.hs view
@@ -0,0 +1,11 @@+import Test.DocTest +-- main = doctest ["src", "--verbose"] +main :: IO () +main = doctest ["-isrc", "src"] + +{- +C:\haskell\predicate-typed>stack exec doctest -- "src/Predicate.hs" + +-- just specify the directory +C:\haskell\predicate-typed>stack exec doctest -- src +-}
+ predicate-typed.cabal view
@@ -0,0 +1,162 @@+cabal-version: 1.12 ++-- This file has been generated from package.yaml by hpack version 0.31.1.+--+-- see: https://github.com/sol/hpack+--+-- hash: 9a478dac9f97bf4a987d2f412413056ea9656a469275286e14b91c4c1744b6e7++name: predicate-typed+version: 0.1.0.0+synopsis: Predicates, Refinement types and Dsl+description: Please see the README on GitHub at <https://github.com/gbwey/predicate-typed#readme>+homepage: https://github.com/github.com/gbwey#readme+bug-reports: https://github.com/github.com/gbwey/issues+author: Grant Weyburne+maintainer: gbwey9@gmail.com+copyright: 2019 gbwey+license: BSD3+license-file: LICENSE+build-type: Simple+extra-source-files:+ README.md+ ChangeLog.md++source-repository head+ type: git+ location: https://github.com/github.com/gbwey+ subdir: predicate-typed++library+ exposed-modules:+ Predicate+ Refined+ Refined3+ Refined3Helper+ TH_Orphans+ UtilP+ UtilP_TH+ other-modules:+ Paths_predicate_typed+ hs-source-dirs:+ src+ ghc-options: -fomit-interface-pragmas+ build-depends:+ QuickCheck+ , aeson+ , base >=4.7 && <5+ , binary+ , bytestring+ , comonad+ , containers+ , deepseq+ , directory+ , discrimination+ , ghc-prim+ , lens+ , mtl+ , pcre-heavy+ , pcre-light+ , pretty+ , pretty-terminal+ , pretty-tree+ , safe+ , semialign+ , template-haskell+ , text+ , th-lift+ , th-orphans+ , these+ , these-lens+ , time+ , tree-view+ default-language: Haskell2010++test-suite doctests+ type: exitcode-stdio-1.0+ main-is: doctests.hs+ hs-source-dirs:+ ./.+ src+ ghc-options: -threaded -fomit-interface-pragmas+ build-depends:+ QuickCheck+ , aeson+ , base >=4.7 && <5+ , binary+ , bytestring+ , comonad+ , containers+ , deepseq+ , directory+ , discrimination+ , doctest+ , ghc-prim+ , lens+ , mtl+ , pcre-heavy+ , pcre-light+ , predicate-typed+ , pretty+ , pretty-terminal+ , pretty-tree+ , safe+ , semialign+ , template-haskell+ , text+ , th-lift+ , th-orphans+ , these+ , these-lens+ , time+ , tree-view+ default-language: Haskell2010++test-suite predicate-typed-test+ type: exitcode-stdio-1.0+ main-is: TestSpec.hs+ other-modules:+ TastyExtras+ TestJson+ TestPredicate+ TestRefined+ TestRefined3+ Paths_predicate_typed+ hs-source-dirs:+ test+ ghc-options: -threaded -rtsopts -with-rtsopts=-N -fomit-interface-pragmas+ build-depends:+ QuickCheck+ , aeson+ , base >=4.7 && <5+ , binary+ , bytestring+ , comonad+ , containers+ , deepseq+ , directory+ , discrimination+ , ghc-prim+ , lens+ , mtl+ , pcre-heavy+ , pcre-light+ , predicate-typed+ , pretty+ , pretty-terminal+ , pretty-tree+ , safe+ , semialign+ , stm+ , tasty+ , tasty-hunit+ , tasty-quickcheck+ , template-haskell+ , text+ , th-lift+ , th-orphans+ , these+ , these-lens+ , time+ , tree-view+ default-language: Haskell2010
+ src/Predicate.hs view
@@ -0,0 +1,6618 @@+{-# OPTIONS -Wall #-} +{-# OPTIONS -Wcompat #-} +{-# OPTIONS -Wincomplete-record-updates #-} +{-# OPTIONS -Wincomplete-uni-patterns #-} +{-# OPTIONS -Wredundant-constraints #-} +{-# LANGUAGE TypeOperators #-} +{-# LANGUAGE UndecidableInstances #-} +{-# LANGUAGE FlexibleContexts #-} +{-# LANGUAGE AllowAmbiguousTypes #-} +{-# LANGUAGE FlexibleInstances #-} +{-# LANGUAGE MultiParamTypeClasses #-} +{-# LANGUAGE TypeApplications #-} +{-# LANGUAGE DataKinds #-} +{-# LANGUAGE GADTs #-} +{-# LANGUAGE TypeFamilies #-} +{-# LANGUAGE PolyKinds #-} +{-# LANGUAGE ScopedTypeVariables #-} +{-# LANGUAGE LambdaCase #-} +{-# LANGUAGE RankNTypes #-} +{-# LANGUAGE OverloadedStrings #-} +{-# LANGUAGE ConstraintKinds #-} +{-# LANGUAGE MultiWayIf #-} +{-# LANGUAGE TypeFamilyDependencies #-} +{-# LANGUAGE TupleSections #-} +{-# LANGUAGE ViewPatterns #-} +{-# LANGUAGE NoStarIsType #-} +{-# LANGUAGE OverloadedLists #-} +{- | +Module : Predicate +Description : Dsl for evaluating and displaying type level expressions +Copyright : (c) Grant Weyburne, 2019 +License : BSD-3 +Maintainer : gbwey9@gmail.com + +class P is the main class. Most of this code contains instances of this class +that evaluation of expressions at the type level. +-} +module Predicate where +import UtilP +import Safe +import GHC.TypeLits (Symbol,Nat,KnownSymbol,KnownNat,ErrorMessage((:$$:),(:<>:))) +import qualified GHC.TypeLits as GL +import qualified GHC.TypeNats as GN +import Control.Lens hiding (strict,iall) +import Data.List +import Data.Text.Lens +import Data.Proxy +import Control.Applicative +import Data.Typeable +import Control.Monad.Except +import qualified Control.Exception as E +import Data.Kind (Type) +import qualified Text.Regex.PCRE.Heavy as RH +import Data.String +import Data.Foldable +import Data.Maybe +import Control.Arrow +import qualified Data.Semigroup as SG +import Numeric +import Data.Char +import Data.Function +import Data.These +import Data.These.Lens () +import qualified Data.Align as TA +import Data.Ratio +import Data.Time +import Data.Coerce +import Data.Void +import qualified Data.Sequence as Seq +import Text.Printf +import System.Directory +import Control.Comonad +import System.IO +import System.Environment +import qualified GHC.Exts as Ge +import Data.Bool +import Data.Either +import qualified Data.Type.Equality as DE +import Data.Time.Calendar.WeekDate + +-- | This is the core class. Each instance of this class can be combined into a dsl using 'Main.>>' +class P p a where + type PP (p :: k) a :: Type -- PP is the output type + eval :: MonadEval m => Proxy p -> POpts -> a -> m (TT (PP p a)) -- ^ returns a tree of results + +-- | A specialised form of 'eval' that works only on predicates +evalBool :: (MonadEval m, P p a, PP p a ~ Bool) => Proxy p -> POpts -> a -> m (TT (PP p a)) +evalBool p opts a = fixBoolT <$> eval p opts a + +-- | a type level predicate for a monotonic increasing list +type Asc = Map (Fst <= Snd) Pairs >> Ands +-- | a type level predicate for a strictly increasing list +type Asc' = Map (Fst < Snd) Pairs >> Ands +-- | a type level predicate for a monotonic decreasing list +type Desc = Map (Fst >= Snd) Pairs >> Ands +-- | a type level predicate for a strictly decreasing list +type Desc' = Map (Fst > Snd) Pairs >> Ands + +-- | A predicate that determines if the value is between \'p\' and \'q\' +-- The values can be rational numbers using 'Rat' or plain Natural numbers +type Between p q = Ge p && Le q +-- | This is the same as 'Between' but where \'r\' is 'Id' +type Between' p q r = r >= p && r <= q + +-- | a type level predicate for all positive elements in a list +type AllPositive = Map Positive >> Ands +-- | a type level predicate for all negative elements in a list +type AllNegative = Map Negative >> Ands +type Positive = Ge 0 +type Negative = Le 0 + +type AllPositive' = FoldMap SG.All (Map Positive Id) +type AllNegative' = FoldMap SG.All (Map Negative Id) + +type All x = Map x Id >> Ands +type Any x = Map x Id >> Ors + +-- | 'unzip' equivalent +type Unzip = (Map Fst Id, Map Snd Id) + +-- | represents a predicate using a 'Symbol' as a regular expression +-- evaluates 'Re' and returns True if there is a match +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Re "^\\d{2}:\\d{2}:\\d{2}$" Id) "13:05:25" +-- True +-- TrueT +-- +data Re' (rs :: [ROpt]) p q +type Re p q = Re' '[] p q + +instance (GetROpts rs + , PP p x ~ String + , PP q x ~ String + , P p x + , P q x + ) => P (Re' rs p q) x where + type PP (Re' rs p q) x = Bool + eval _ opts x = do + let msg0 = "Re" <> (if null rs then "' " <> show rs else "") + rs = getROpts @rs + lr <- runPQ msg0 (Proxy @p) (Proxy @q) opts x + pure $ case lr of + Left e -> e + Right (p,q,pp,qq) -> + let msg1 = msg0 <> " (" <> p <> ")" + hhs = [hh pp, hh qq] + in case compileRegex @rs opts msg1 p hhs of + Left tta -> tta + Right regex -> + let b = q RH.=~ regex + in mkNodeB opts b [msg1 <> showLit opts " | " q] hhs + +-- only way with rescan is to be explicit: no repeats! and useanchors but not (?m) +-- or just use Re' but then we only get a bool ie doesnt capture groups +-- rescan returns Right [] as an failure! +-- [] is failure! + + +-- | runs a regex matcher returning the original values and optionally any groups +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Rescan "^(\\d{2}):(\\d{2}):(\\d{2})$" Id) "13:05:25" +-- Present [("13:05:25",["13","05","25"])] +-- PresentT [("13:05:25",["13","05","25"])] +-- +data Rescan' (rs :: [ROpt]) p q +type Rescan p q = Rescan' '[] p q + +instance (GetROpts rs + , PP p x ~ String + , PP q x ~ String + , P p x + , P q x + ) => P (Rescan' rs p q) x where + type PP (Rescan' rs p q) x = [(String, [String])] + eval _ opts x = do + let msg0 = "Rescan" <> (if null rs then "' " <> show rs else "") + rs = getROpts @rs + lr <- runPQ msg0 (Proxy @p) (Proxy @q) opts x + pure $ case lr of + Left e -> e + Right (p,q,pp,qq) -> + let msg1 = msg0 <> " (" <> p <> ")" + hhs = [hh pp, hh qq] + in case compileRegex @rs opts msg1 p hhs of + Left tta -> tta + Right regex -> + case splitAt _MX $ RH.scan regex q of + (b, _:_) -> mkNode opts (FailT "Regex looping") [msg1 <> " Looping? " <> show (take 10 b) <> "..." <> showA opts " | " q] hhs + ([], _) -> -- this is a failure cos empty string returned: so reuse p? + mkNode opts (FailT "Regex no results") [msg1 <> " no match" <> showA opts " | " q] [hh pp, hh qq] + (b, _) -> mkNode opts (PresentT b) [msg1 <> show0 opts " " b <> showLit opts " | " q] [hh pp, hh qq] + + +-- | similar to 'Rescan' but gives the column start and ending positions instead of values +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(RescanRanges "^(\\d{2}):(\\d{2}):(\\d{2})$" Id) "13:05:25" +-- Present [((0,8),[(0,2),(3,5),(6,8)])] +-- PresentT [((0,8),[(0,2),(3,5),(6,8)])] +-- +data RescanRanges' (rs :: [ROpt]) p q +type RescanRanges p q = RescanRanges' '[] p q + +instance (GetROpts rs + , PP p x ~ String + , PP q x ~ String + , P p x + , P q x + ) => P (RescanRanges' rs p q) x where + type PP (RescanRanges' rs p q) x = [((Int,Int), [(Int,Int)])] + eval _ opts x = do + let msg0 = "RescanRanges" <> (if null rs then "' " <> show rs else "") + rs = getROpts @rs + lr <- runPQ msg0 (Proxy @p) (Proxy @q) opts x + pure $ case lr of + Left e -> e + Right (p,q,pp,qq) -> + let msg1 = msg0 <> " (" <> p <> ")" + hhs = [hh pp, hh qq] + in case compileRegex @rs opts msg1 p hhs of + Left tta -> tta + Right regex -> + case splitAt _MX $ RH.scanRanges regex q of + (b, _:_) -> mkNode opts (FailT "Regex looping") [msg1 <> " Looping? " <> show (take 10 b) <> "..." <> showA opts " | " q] hhs + ([], _) -> -- this is a failure cos empty string returned: so reuse p? + mkNode opts (FailT "Regex no results") [msg1 <> " no match" <> showA opts " | " q] hhs + (b, _) -> mkNode opts (PresentT b) [msg1 <> show0 opts " " b <> showLit opts " | " q] hhs + +-- | splits a string on a regex delimiter +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Resplit "\\." Id) "141.201.1.22" +-- Present ["141","201","1","22"] +-- PresentT ["141","201","1","22"] +-- +data Resplit' (rs :: [ROpt]) p q +type Resplit p q = Resplit' '[] p q + +instance (GetROpts rs + , PP p x ~ String + , PP q x ~ String + , P p x + , P q x + ) => P (Resplit' rs p q) x where + type PP (Resplit' rs p q) x = [String] + eval _ opts x = do + let msg0 = "Resplit" <> (if null rs then "' " <> show rs else "") + rs = getROpts @rs + lr <- runPQ msg0 (Proxy @p) (Proxy @q) opts x + pure $ case lr of + Left e -> e + Right (p,q,pp,qq) -> + let msg1 = msg0 <> " (" <> p <> ")" + hhs = [hh pp, hh qq] + in case compileRegex @rs opts msg1 p hhs of + Left tta -> tta + Right regex -> + case splitAt _MX $ RH.split regex q of + (b, _:_) -> mkNode opts (FailT "Regex looping") [msg1 <> " Looping? " <> show (take 10 b) <> "..." <> showA opts " | " q] hhs + ([], _) -> -- this is a failure cos empty string returned: so reuse p? + mkNode opts (FailT "Regex no results") [msg1 <> " no match" <> showA opts " | " q] hhs + (b, _) -> mkNode opts (PresentT b) [msg1 <> show0 opts " " b <> showLit opts " | " q] hhs + +_MX :: Int +_MX = 100 + +-- | replaces regex \'s\' with a string \'s1\' inside the value +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(ReplaceAllString "\\." ":" Id) "141.201.1.22" +-- Present "141:201:1:22" +-- PresentT "141:201:1:22" +-- +data ReplaceImpl (alle :: Bool) (rs :: [ROpt]) p q r +type ReplaceAll' (rs :: [ROpt]) p q r = ReplaceImpl 'True rs p q r +type ReplaceAll p q r = ReplaceAll' '[] p q r +type ReplaceOne' (rs :: [ROpt]) p q r = ReplaceImpl 'False rs p q r +type ReplaceOne p q r = ReplaceOne' '[] p q r + +type ReplaceAllString' (rs :: [ROpt]) p q r = ReplaceAll' rs p (MakeRR q) r +type ReplaceAllString p q r = ReplaceAllString' '[] p q r + +type ReplaceOneString' (rs :: [ROpt]) p q r = ReplaceOne' rs p (MakeRR q) r +type ReplaceOneString p q r = ReplaceOneString' '[] p q r + +-- | Simple replacement string: see 'ReplaceAllString' and 'ReplaceOneString' +-- +data MakeRR p + +instance (PP p x ~ String + , P p x) => P (MakeRR p) x where + type PP (MakeRR p) x = RR + eval _ opts x = do + let msg0 = "MakeRR" + pp <- eval (Proxy @p) opts x + pure $ case getValueLR opts msg0 pp [] of + Left e -> e + Right as -> + let b = RR as + in mkNode opts (PresentT b) [msg0 <> showA opts " | " as] [hh pp] + +-- | A replacement function (String -> [String] -> String) which returns the whole match and the groups +-- Used by 'RH.sub' and 'RH.sub' +-- Requires "Text.Show.Functions" +-- +data MakeRR1 p + +instance (PP p x ~ (String -> [String] -> String) + , P p x) => P (MakeRR1 p) x where + type PP (MakeRR1 p) x = RR + eval _ opts x = do + let msg0 = "MakeRR1 (String -> [String] -> String)" + pp <- eval (Proxy @p) opts x + pure $ case getValueLR opts msg0 pp [] of + Left e -> e + Right f -> mkNode opts (PresentT (RR1 f)) [msg0] [hh pp] + +-- | A replacement function (String -> String) that yields the whole match +-- Used by 'RH.sub' and 'RH.sub' +-- Requires "Text.Show.Functions" +-- +-- >>> :m + Text.Show.Functions +-- >>> pl @(ReplaceAll "\\." (MakeRR2 Fst) Snd) (\x -> x <> ":" <> x, "141.201.1.22") +-- Present "141.:.201.:.1.:.22" +-- PresentT "141.:.201.:.1.:.22" +-- +data MakeRR2 p + +instance (PP p x ~ (String -> String) + , P p x) => P (MakeRR2 p) x where + type PP (MakeRR2 p) x = RR + eval _ opts x = do + let msg0 = "MakeRR2 (String -> String)" + pp <- eval (Proxy @p) opts x + pure $ case getValueLR opts msg0 pp [] of + Left e -> e + Right f -> mkNode opts (PresentT (RR2 f)) [msg0] [hh pp] + +-- | A replacement function ([String] -> String) which yields the groups +-- Used by 'RH.sub' and 'RH.sub' +-- Requires "Text.Show.Functions" +-- +-- >>> :m + Text.Show.Functions +-- >>> pl @(ReplaceAll "^(\\d+)\\.(\\d+)\\.(\\d+)\\.(\\d+)$" (MakeRR3 Fst) Snd) (\ys -> intercalate " | " $ map (show . succ . read @Int) ys, "141.201.1.22") +-- Present "142 | 202 | 2 | 23" +-- PresentT "142 | 202 | 2 | 23" +-- +data MakeRR3 p + +instance (PP p x ~ ([String] -> String) + , P p x) => P (MakeRR3 p) x where + type PP (MakeRR3 p) x = RR + eval _ opts x = do + let msg0 = "MakeRR3 ([String] -> String)" + pp <- eval (Proxy @p) opts x + pure $ case getValueLR opts msg0 pp [] of + Left e -> e + Right f -> mkNode opts (PresentT (RR3 f)) [msg0] [hh pp] + +instance (GetBool b + , GetROpts rs + , PP p x ~ String + , PP q x ~ RR + , PP r x ~ String + , P p x + , P q x + , P r x + ) => P (ReplaceImpl b rs p q r) x where + type PP (ReplaceImpl b rs p q r) x = String + eval _ opts x = do + let msg0 = "Replace" <> (if alle then "All" else "One") <> (if null rs then "' " <> show rs else "") + rs = getROpts @rs + alle = getBool @b + lr <- runPQ msg0 (Proxy @p) (Proxy @q) opts x + case lr of + Left e -> pure e + Right (p,q,pp,qq) -> + let msg1 = msg0 <> " (" <> p <> ")" + hhs = [hh pp, hh qq] + in case compileRegex @rs opts msg1 p hhs of + Left tta -> pure tta + Right regex -> do + rr <- eval (Proxy @r) opts x + pure $ case getValueLR opts msg0 rr hhs of + Left e -> e + Right r -> + let ret :: String + ret = case q of + RR s -> (if alle then RH.gsub else RH.sub) regex s r + RR1 s -> (if alle then RH.gsub else RH.sub) regex s r + RR2 s -> (if alle then RH.gsub else RH.sub) regex s r + RR3 s -> (if alle then RH.gsub else RH.sub) regex s r + in mkNode opts (PresentT ret) [msg1 <> showLit opts " " r <> showLit opts " | " ret] (hhs <> [hh rr]) + +-- | a predicate for determining if a string 'Data.Text.IsText' belongs to the given character set +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> import qualified Data.Text as T +-- >>> pl @IsLower "abc" +-- True +-- TrueT +-- +-- >>> pl @IsLower "abcX" +-- False +-- FalseT +-- +-- >>> pl @IsLower (T.pack "abcX") +-- False +-- FalseT +-- +-- >>> pl @IsHexDigit "01efA" +-- True +-- TrueT +-- +-- >>> pl @IsHexDigit "01egfA" +-- False +-- FalseT +-- +data IsCharSet (cs :: CharSet) + +data CharSet = CLower + | CUpper + | CNumber + | CSpace + | CPunctuation + | CControl + | CHexDigit + | COctDigit + | CSeparator + | CLatin1 + deriving Show + +class GetCharSet (cs :: CharSet) where + getCharSet :: (CharSet, Char -> Bool) +instance GetCharSet 'CLower where + getCharSet = (CLower, isLower) +instance GetCharSet 'CUpper where + getCharSet = (CUpper, isUpper) +instance GetCharSet 'CNumber where + getCharSet = (CNumber, isNumber) +instance GetCharSet 'CPunctuation where + getCharSet = (CPunctuation, isPunctuation) +instance GetCharSet 'CControl where + getCharSet = (CControl, isControl) +instance GetCharSet 'CHexDigit where + getCharSet = (CHexDigit, isHexDigit) +instance GetCharSet 'COctDigit where + getCharSet = (COctDigit, isOctDigit) +instance GetCharSet 'CSeparator where + getCharSet = (CSeparator, isSeparator) +instance GetCharSet 'CLatin1 where + getCharSet = (CLatin1, isLatin1) + +type IsLower = IsCharSet 'CLower +type IsUpper = IsCharSet 'CUpper +type IsNumber = IsCharSet 'CNumber +type IsSpace = IsCharSet 'CSpace +type IsPunctuation = IsCharSet 'CPunctuation +type IsControl = IsCharSet 'CControl +type IsHexDigit = IsCharSet 'CHexDigit +type IsOctDigit = IsCharSet 'COctDigit +type IsSeparator = IsCharSet 'CSeparator +type IsLatin1 = IsCharSet 'CLatin1 + +instance (GetCharSet cs + , Show a + , IsText a + ) => P (IsCharSet cs) a where + type PP (IsCharSet cs) a = Bool + eval _ opts as = + let b = allOf text f as + msg0 = "IsCharSet " ++ show cs + (cs,f) = getCharSet @cs + in pure $ mkNodeB opts b [msg0 <> showA opts " | " as] [] + + +-- | converts a string 'Data.Text.Lens.IsText' value to lower case +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @ToLower "HeLlO wOrld!" +-- Present "hello world!" +-- PresentT "hello world!" +-- +data ToLower + +instance (Show a, IsText a) => P ToLower a where + type PP ToLower a = a + eval _ opts as = + let xs = as & text %~ toLower + in pure $ mkNode opts (PresentT xs) ["ToLower" <> show0 opts " " xs <> showA opts " | " as] [] + +-- | converts a string 'Data.Text.Lens.IsText' value to upper case +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @ToUpper "HeLlO wOrld!" +-- Present "HELLO WORLD!" +-- PresentT "HELLO WORLD!" +-- +data ToUpper + +instance (Show a, IsText a) => P ToUpper a where + type PP ToUpper a = a + eval _ opts as = + let xs = as & text %~ toUpper + in pure $ mkNode opts (PresentT xs) ["ToUpper" <> show0 opts " " xs <> showA opts " | " as] [] + + +-- | similar to 'Data.List.inits' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @Inits [4,8,3,9] +-- Present [[],[4],[4,8],[4,8,3],[4,8,3,9]] +-- PresentT [[],[4],[4,8],[4,8,3],[4,8,3,9]] +-- +-- >>> pl @Inits [] +-- Present [[]] +-- PresentT [[]] +-- +data Inits + +instance Show a => P Inits [a] where + type PP Inits [a] = [[a]] + eval _ opts as = + let xs = inits as + in pure $ mkNode opts (PresentT xs) ["Inits" <> show0 opts " " xs <> showA opts " | " as] [] + +-- | similar to 'Data.List.tails' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @Tails [4,8,3,9] +-- Present [[4,8,3,9],[8,3,9],[3,9],[9],[]] +-- PresentT [[4,8,3,9],[8,3,9],[3,9],[9],[]] +-- +-- >>> pl @Tails [] +-- Present [[]] +-- PresentT [[]] +-- +data Tails + +instance Show a => P Tails [a] where + type PP Tails [a] = [[a]] + eval _ opts as = + let xs = tails as + in pure $ mkNode opts (PresentT xs) ["Tails" <> show0 opts " " xs <> showA opts " | " as] [] + +-- | split a list into single values +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @Ones [4,8,3,9] +-- Present [[4],[8],[3],[9]] +-- PresentT [[4],[8],[3],[9]] +-- +-- >>> pl @Ones [] +-- Present [] +-- PresentT [] +-- +data Ones + +instance (as ~ [a], Show a) => P Ones as where + type PP Ones as = [as] + eval _ opts as = + let xs = map (:[]) as + in pure $ mkNode opts (PresentT xs) ["Ones" <> show0 opts " " xs <> showA opts " | " as] [] + +-- | similar to 'show' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @ShowP [4,8,3,9] +-- Present "[4,8,3,9]" +-- PresentT "[4,8,3,9]" +-- +-- >>> pl @ShowP 'x' +-- Present "'x'" +-- PresentT "'x'" +-- +data ShowP + +instance Show as => P ShowP as where + type PP ShowP as = String + eval _ opts as = + let x = show as + in pure $ mkNode opts (PresentT x) ["ShowP" <> showLit0 opts " " x <> showA opts " | " as] [] + +-- | type level expression representing a formatted time +-- similar to 'Data.Time.formatTime' using a type level 'Symbol' to get the formatting string +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(FormatTimeP "%F %T" Id) (read "2019-05-24 05:19:59" :: LocalTime) +-- Present "2019-05-24 05:19:59" +-- PresentT "2019-05-24 05:19:59" +-- +-- >>> pl @(FormatTimeP Fst Snd) ("the date is %d/%m/%Y", read "2019-05-24" :: Day) +-- Present "the date is 24/05/2019" +-- PresentT "the date is 24/05/2019" +-- +data FormatTimeP p q + +instance (PP p x ~ String + , FormatTime (PP q x) + , P p x + , Show (PP q x) + , P q x + ) => P (FormatTimeP p q) x where + type PP (FormatTimeP p q) x = String + eval _ opts x = do + let msg0 = "FormatTimeP" + lr <- runPQ msg0 (Proxy @p) (Proxy @q) opts x + pure $ case lr of + Left e -> e + Right (p,q,pp,qq) -> + let msg1 = msg0 <> " (" <> p <> ")" + b = formatTime defaultTimeLocale p q + in mkNode opts (PresentT b) [msg1 <> showLit0 opts " " b <> showA opts " | " q] [hh pp, hh qq] + +-- | similar to 'Data.Time.parseTimeM' where \'t\' is the 'Data.Time.ParseTime' type, \'p\' is the datetime format and \'q\' points to the content to parse +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(ParseTimeP LocalTime "%F %T" Id) "2019-05-24 05:19:59" +-- Present 2019-05-24 05:19:59 +-- PresentT 2019-05-24 05:19:59 +-- +-- >>> pl @(ParseTimeP LocalTime "%F %T" "2019-05-24 05:19:59") (Right "we ignore this using Symbol and not Id") +-- Present 2019-05-24 05:19:59 +-- PresentT 2019-05-24 05:19:59 +-- +-- keeping \'q\' as we might want to extract from a tuple +data ParseTimeP' t p q +type ParseTimeP (t :: Type) p q = ParseTimeP' (Hole t) p q + +instance (ParseTime (PP t a) + , Typeable (PP t a) + , Show (PP t a) + , P p a + , P q a + , PP p a ~ String + , PP q a ~ String + ) => P (ParseTimeP' t p q) a where + type PP (ParseTimeP' t p q) a = PP t a + eval _ opts a = do + let msg0 = "ParseTimeP " <> t + t = showT @(PP t a) + lr <- runPQ msg0 (Proxy @p) (Proxy @q) opts a + pure $ case lr of + Left e -> e + Right (p,q,pp,qq) -> + let msg1 = msg0 <> " (" <> p <> ")" + hhs = [hh pp, hh qq] + in case parseTimeM @Maybe @(PP t a) True defaultTimeLocale p q of + Just b -> mkNode opts (PresentT b) [msg1 <> show0 opts " " b <> showLit0 opts " | fmt=" p <> showA opts " | " q] hhs + Nothing -> mkNode opts (FailT (msg1 <> " failed to parse")) [msg1 <> " failed"] hhs + +-- | A convenience method to match against many different datetime formats to find a match +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(ParseTimes LocalTime '["%Y-%m-%d %H:%M:%S", "%m/%d/%y %H:%M:%S", "%B %d %Y %H:%M:%S", "%Y-%m-%dT%H:%M:%S"] "03/11/19 01:22:33") () +-- Present 2019-03-11 01:22:33 +-- PresentT 2019-03-11 01:22:33 +-- +-- >>> pl @(ParseTimes LocalTime Fst Snd) (["%Y-%m-%d %H:%M:%S", "%m/%d/%y %H:%M:%S", "%B %d %Y %H:%M:%S", "%Y-%m-%dT%H:%M:%S"], "03/11/19 01:22:33") +-- Present 2019-03-11 01:22:33 +-- PresentT 2019-03-11 01:22:33 +-- +data ParseTimes' t p q +type ParseTimes (t :: Type) p q = ParseTimes' (Hole t) p q + +instance (ParseTime (PP t a) + , Typeable (PP t a) + , Show (PP t a) + , P p a + , P q a + , PP p a ~ [String] + , PP q a ~ String + ) => P (ParseTimes' t p q) a where + type PP (ParseTimes' t p q) a = PP t a + eval _ opts a = do + let msg0 = "ParseTimes " <> t + t = showT @(PP t a) + lr <- runPQ msg0 (Proxy @p) (Proxy @q) opts a + pure $ case lr of + Left e -> e + Right (p,q,pp,qq) -> + let msg1 = msg0 + hhs = [hh pp, hh qq] + zs = map (\d -> (d,) <$> parseTimeM @Maybe @(PP t a) True defaultTimeLocale d q) p + in case catMaybes zs of + [] -> mkNode opts (FailT ("no match on [" ++ q ++ "]")) [msg1 <> " no match"] hhs + (d,b):_ -> mkNode opts (PresentT b) [msg1 <> show0 opts " " b <> showLit0 opts " | fmt=" d <> showA opts " | " q] hhs + +-- | create a 'Day' from three int values passed in as year month and day +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> :set -XTypeOperators +-- >>> pl @(MkDay Fst (Snd >> Fst) (Snd >> Snd)) (2019,(12,30)) +-- Present Just (2019-12-30,(1,1)) +-- PresentT (Just (2019-12-30,(1,1))) +-- +-- >>> pl @(MkDay Fst (Snd >> Fst) (Snd >> Snd)) (2019,(99,99999)) +-- Present Nothing +-- PresentT Nothing +-- +-- >>> pl @(MkDay Fst (Snd >> Fst) (Snd >> Snd)) (1999,(3,13)) +-- Present Just (1999-03-13,(10,6)) +-- PresentT (Just (1999-03-13,(10,6))) +-- +data MkDay p q r +instance (P p x + , P q x + , P r x + , PP p x ~ Int + , PP q x ~ Int + , PP r x ~ Int + ) => P (MkDay p q r) x where + type PP (MkDay p q r) x = Maybe (Day, (Int, Int)) + eval _ opts x = do + let msg0 = "MkDay" + lr <- runPQ msg0 (Proxy @p) (Proxy @q) opts x + case lr of + Left e -> pure e + Right (p,q,pp,qq) -> do + let hhs = [hh pp, hh qq] + rr <- eval (Proxy @r) opts x + pure $ case getValueLR opts msg0 rr hhs of + Left e -> e + Right r -> + let mday = fromGregorianValid (fromIntegral p) q r + b = mday <&> \day -> + let (_, week, dow) = toWeekDate day + in (day, (week, dow)) + in mkNode opts (PresentT b) [msg0 <> show0 opts " " b <> showA opts " | (y,m,d)=" (p,q,r)] (hhs <> [hh rr]) + +-- | uncreate a 'Day' returning year month and day +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @UnMkDay (read "2019-12-30") +-- Present (2019,(12,30)) +-- PresentT (2019,(12,30)) +-- +data UnMkDay +instance a ~ Day => P UnMkDay a where + type PP UnMkDay a = (Int, (Int, Int)) + eval _ opts day = + let msg0 = "UnMkDay" + (y,m,d) = toGregorian day + b = (fromIntegral y,(m,d)) + in pure $ mkNode opts (PresentT b) [msg0 <> show0 opts " " b <> showA opts " | " day] [] + +-- | uses the 'Read' of the given type \'t\' and \'p\' which points to the content to read +-- +-- >>> :set -XTypeApplications +-- >>> :set -XTypeOperators +-- >>> :set -XDataKinds +-- >>> pl @(ReadP Rational) "4 % 5" +-- Present 4 % 5 +-- PresentT (4 % 5) +-- +-- >>> pl @(ReadP' Day Id >> Between (ReadP' Day "2017-04-11") (ReadP' Day "2018-12-30")) "2018-10-12" +-- True +-- TrueT +-- +-- >>> pl @(ReadP' Day Id >> Between (ReadP' Day "2017-04-11") (ReadP' Day "2018-12-30")) "2016-10-12" +-- False +-- FalseT +-- +data ReadP'' t p +type ReadP (t :: Type) = ReadP'' (Hole t) Id +type ReadP' (t :: Type) p = ReadP'' (Hole t) p + +instance (P p x + , PP p x ~ String + , Typeable (PP t x) + , Show (PP t x) + , Read (PP t x) + ) => P (ReadP'' t p) x where + type PP (ReadP'' t p) x = PP t x + eval _ opts x = do + let msg0 = "ReadP " <> t + t = showT @(PP t x) + pp <- eval (Proxy @p) opts x + pure $ case getValueLR opts msg0 pp [] of + Left e -> e + Right s -> + let msg1 = msg0 <> " (" <> s <> ")" + in case reads @(PP t x) s of + [(b,"")] -> mkNode opts (PresentT b) [msg1 <> show0 opts " " b <> showLit opts " | " s] [hh pp] + _ -> mkNode opts (FailT (msg1 <> " failed")) [msg1 <> " failed"] [hh pp] + +-- | similar to 'minimum' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @Min [10,4,5,12,3,4] +-- Present 3 +-- PresentT 3 +-- +-- >>> pl @Min [] +-- Error empty list +-- FailT "empty list" +-- +data Min + +instance (Ord a, Show a) => P Min [a] where + type PP Min [a] = a + eval _ opts as' = + pure $ case as' of + [] -> mkNode opts (FailT "empty list") ["Min(empty list)"] [] + as@(_:_) -> + let v = minimum as + in mkNode opts (PresentT v) ["Min" <> show0 opts " " v <> showA opts " | " as] [] + +-- | similar to 'maximum' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @Max [10,4,5,12,3,4] +-- Present 12 +-- PresentT 12 +-- +-- >>> pl @Max [] +-- Error empty list +-- FailT "empty list" +-- + +data Max +type Max' t = FoldMap (SG.Max t) Id + +instance (Ord a, Show a) => P Max [a] where + type PP Max [a] = a + eval _ opts as' = + pure $ case as' of + [] -> mkNode opts (FailT "empty list") ["Max(empty list)"] [] + as@(_:_) -> + let v = maximum as + in mkNode opts (PresentT v) ["Max" <> show0 opts " " v <> showA opts " | " as] [] + +-- | sort a list +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(SortOn Fst Id) [(10,"abc"), (3,"def"), (4,"gg"), (10,"xyz"), (1,"z")] +-- Present [(1,"z"),(3,"def"),(4,"gg"),(10,"abc"),(10,"xyz")] +-- PresentT [(1,"z"),(3,"def"),(4,"gg"),(10,"abc"),(10,"xyz")] +-- +data SortBy p q +type SortOn p q = SortBy (OrdA p) q +type SortOnDesc p q = SortBy (Swap >> OrdA p) q + +type SortByHelper p = Partition (p >> Id == 'GT) Id + +instance (P p (a,a) + , P q x + , Show a + , PP q x ~ [a] + , PP p (a,a) ~ Ordering + ) => P (SortBy p q) x where + type PP (SortBy p q) x = PP q x + eval _ opts x = do + let msg0 = "SortBy" + qq <- eval (Proxy @q) opts x + case getValueLR opts (msg0 <> " q failed") qq [] of + Left e -> pure e + Right as -> do + let ff :: MonadEval m => [a] -> m (TT [a]) + ff = \case + [] -> pure $ mkNode opts mempty [msg0 <> " empty"] [] + [w] -> pure $ mkNode opts (PresentT [w]) [msg0 <> " one element " <> show w] [] + w:ys@(_:_) -> do + pp <- (if oDebug opts >= 3 then + eval (Proxy @(SortByHelper p)) + else eval (Proxy @(Hide (SortByHelper p)))) opts (map (w,) ys) +-- pp <- eval (Proxy @(Hide (Partition (p >> Id == 'GT) Id))) opts (map (w,) ys) +-- too much output: dont need (Map Snd *** Map Snd) -- just do map snd in code +-- pp <- eval (Proxy @(Partition (p >> (Id == 'GT)) Id >> (Map Snd *** Map Snd))) opts (map (w,) ys) + case getValueLR opts msg0 pp [] of + Left e -> pure e + Right (ll', rr') -> do + lhs <- ff (map snd ll') + case getValueLR opts msg0 lhs [] of + Left _ -> pure lhs -- dont rewrap + Right ll -> do + rhs <- ff (map snd rr') + case getValueLR opts msg0 rhs [] of + Left _ -> pure rhs + Right rr -> do + pure $ mkNode opts (PresentT (ll ++ w : rr)) + [msg0 <> show0 opts " lhs=" ll <> " pivot " <> show w <> show0 opts " rhs=" rr] + (hh pp : [hh lhs | length ll > 1] ++ [hh rhs | length rr > 1]) + ret <- ff as + pure $ case getValueLR opts msg0 ret [hh qq] of + Left _e -> ret -- dont rewrap the error + Right xs -> mkNode opts (_tBool ret) [msg0 <> show0 opts " " xs] [hh qq, hh ret] + +-- | similar to 'length' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @Len [10,4,5,12,3,4] +-- Present 6 +-- PresentT 6 +-- +-- >>> pl @Len [] +-- Present 0 +-- PresentT 0 +-- +data Len +instance (Show a, as ~ [a]) => P Len as where + type PP Len as = Int + eval _ opts as = + let n = length as + in pure $ mkNode opts (PresentT n) ["Len" <> show0 opts " " n <> showA opts " | " as] [] + +-- | similar to 'length' for 'Foldable' instances +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Length Id) (Left "aa") +-- Present 0 +-- PresentT 0 +-- +-- >>> pl @(Length Id) (Right "aa") +-- Present 1 +-- PresentT 1 +-- +-- >>> pl @(Length (Right' Id)) (Right "abcd") +-- Present 4 +-- PresentT 4 +-- +data Length p + +instance (PP p x ~ t a + , P p x + , Show (t a) + , Foldable t) => P (Length p) x where + type PP (Length p) x = Int + eval _ opts x = do + let msg0 = "Length" + pp <- eval (Proxy @p) opts x + pure $ case getValueLR opts msg0 pp [] of + Left e -> e + Right as -> + let n = length as + in mkNode opts (PresentT n) [msg0 <> show0 opts " " n <> showA opts " | " as] [] + +-- | similar to 'Control.Lens._1' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(FstL _ Id) (10,"Abc") +-- Present 10 +-- PresentT 10 +-- +data FstL' t p +type FstL (t :: Type) p = FstL' (Hole t) p + +instance (PP p x ~ s + , P p x + , Show s + , Field1 s s (PP t x) (PP t x) + , Show (PP t x) + ) => P (FstL' t p) x where + type PP (FstL' t p) x = PP t x + eval _ opts x = do + let msg0 = "FstL" + pp <- eval (Proxy @p) opts x + pure $ case getValueLR opts msg0 pp [] of + Left e -> e + Right p -> + let a = p ^. _1 + in mkNode opts (PresentT a) [msg0 <> show0 opts " " a <> showA opts " | " p] [hh pp] + +-- | similar to 'Control.Lens._2' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(SndL _ Id) (10,"Abc") +-- Present "Abc" +-- PresentT "Abc" +-- +data SndL' t p +type SndL (t :: Type) p = SndL' (Hole t) p + +instance (PP p x ~ s + , P p x + , Show s + , Field2 s s (PP t x) (PP t x) + , Show (PP t x) + ) => P (SndL' t p) x where + type PP (SndL' t p) x = PP t x + eval _ opts x = do + let msg0 = "SndL" + pp <- eval (Proxy @p) opts x + pure $ case getValueLR opts msg0 pp [] of + Left e -> e + Right p -> + let a = p ^. _2 + in mkNode opts (PresentT a) [msg0 <> show0 opts " " a <> showA opts " s=" p] [hh pp] + +-- | similar to 'Control.Lens._3' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(ThdL _ Id) (10,"Abc",'x') +-- Present 'x' +-- PresentT 'x' +-- +data ThdL' t p +type ThdL (t :: Type) p = ThdL' (Hole t) p + +instance (PP p x ~ s + , P p x + , Show s + , Field3 s s (PP t x) (PP t x) + , Show (PP t x) + ) => P (ThdL' t p) x where + type PP (ThdL' t p) x = PP t x + eval _ opts x = do + let msg0 = "ThdL" + pp <- eval (Proxy @p) opts x + pure $ case getValueLR opts msg0 pp [] of + Left e -> e + Right p -> + let a = p ^. _3 + in mkNode opts (PresentT a) [msg0 <> show0 opts " " a <> showA opts " | " p] [hh pp] + +-- since we support '(,,,) so have to support Field4 to be able to retrieve those values + +-- | similar to 'Control.Lens._4' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(FthL _ Id) (10,"Abc",'x',True) +-- Present True +-- PresentT True +-- +data FthL' t p +type FthL (t :: Type) p = FthL' (Hole t) p + +instance (PP p x ~ s + , P p x + , Show s + , Field4 s s (PP t x) (PP t x) + , Show (PP t x) + ) => P (FthL' t p) x where + type PP (FthL' t p) x = PP t x + eval _ opts x = do + let msg0 = "FthL" + pp <- eval (Proxy @p) opts x + pure $ case getValueLR opts msg0 pp [] of + Left e -> e + Right p -> + let a = p ^. _4 + in mkNode opts (PresentT a) [msg0 <> show0 opts " " a <> showA opts " | " p] [hh pp] + +-- | similar to 'fst' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @Fst (10,"Abc") +-- Present 10 +-- PresentT 10 +-- +data Fst + +instance (Show x, Show a) => P Fst (a,x) where + type PP Fst (a,x) = a + eval _ opts (a,x) = + pure $ mkNode opts (PresentT a) ["Fst" <> show0 opts " " a <> showA opts " | " (a,x)] [] + +-- | similar to 'snd' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @Snd (10,"Abc") +-- Present "Abc" +-- PresentT "Abc" +-- +data Snd + +instance (Show x, Show b) => P Snd (x,b) where + type PP Snd (x,b) = b + eval _ opts (x,b) = + pure $ mkNode opts (PresentT b) ["Snd" <> show0 opts " " b <> showA opts " | " (x,b)] [] + + +-- | 'fst' for a 3-tuple +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @Fst3 (10,"Abc",True) +-- Present 10 +-- PresentT 10 +-- +data Fst3 + +instance (Show x, Show y, Show a) => P Fst3 (a,x,y) where + type PP Fst3 (a,x,y) = a + eval _ opts (a,x,y) = + pure $ mkNode opts (PresentT a) ["Fst3" <> show0 opts " " a <> showA opts " | " (a,x,y)] [] + +-- | 'snd' for a 3-tuple +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @Snd3 (10,"Abc",True) +-- Present "Abc" +-- PresentT "Abc" +-- +data Snd3 + +instance (Show x, Show y, Show b) => P Snd3 (x,b,y) where + type PP Snd3 (x,b,y) = b + eval _ opts (x,b,y) = + pure $ mkNode opts (PresentT b) ["Snd3" <> show0 opts " " b <> showA opts " | " (x,b,y)] [] + +-- | access to third element in a 3-tuple +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @Thd3 (10,True,"Abc") +-- Present "Abc" +-- PresentT "Abc" +-- +data Thd3 + +instance (Show x, Show y, Show b) => P Thd3 (x,y,b) where + type PP Thd3 (x,y,b) = b + eval _ opts (x,y,b) = + pure $ mkNode opts (PresentT b) ["Thd3" <> show0 opts " " b <> showA opts " | " (x,y,b)] [] + +-- | identity function +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @I 23 +-- Present 23 +-- PresentT 23 +data I +instance P I a where + type PP I a = a + eval _ opts a = + pure $ mkNode opts (PresentT a) ["I"] [] + + +-- | identity function that displays the input +-- +-- even more constraints than 'I' so we might need to add explicit type signatures +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @Id 23 +-- Present 23 +-- PresentT 23 +data Id -- showable version of I +instance Show a => P Id a where + type PP Id a = a + eval _ opts a = pure $ mkNode opts (PresentT a) ["Id" <> show0 opts " " a] [] + + +-- | identity function that also displays the type information for debugging +-- +-- even more constraints than 'Id' so we might need to explicitly add types (Typeable) +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @IdT 23 +-- Present 23 +-- PresentT 23 +data IdT +instance (Typeable a, Show a) => P IdT a where + type PP IdT a = a + eval _ opts a = + let t = showT @a + in pure $ mkNode opts (PresentT a) ["IdT(" <> t <> ")" <> show0 opts " " a] [] + +-- | 'fromString' function +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> :set -XOverloadedStrings +-- >>> pl @(FromStringP (Identity _) Id) "abc" +-- Present Identity "abc" +-- PresentT (Identity "abc") +-- +-- >>> pl @(FromStringP (Seq.Seq _) Id) "abc" +-- Present fromList "abc" +-- PresentT (fromList "abc") +data FromStringP' t s +type FromStringP (t :: Type) p = FromStringP' (Hole t) p + +instance (P s a + , PP s a ~ String + , Show (PP t a) + , IsString (PP t a) + ) => P (FromStringP' t s) a where + type PP (FromStringP' t s) a = PP t a + eval _ opts a = do + let msg0 = "FromStringP" + ss <- eval (Proxy @s) opts a + pure $ case getValueLR opts msg0 ss [] of + Left e -> e + Right s -> + let b = fromString @(PP t a) s + in mkNode opts (PresentT b) [msg0 <> show0 opts " " b] [hh ss] + + +-- | 'fromInteger' function +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(FromInteger (SG.Sum _) Id) 23 +-- Present Sum {getSum = 23} +-- PresentT (Sum {getSum = 23}) +data FromInteger' t n +type FromInteger (t :: Type) p = FromInteger' (Hole t) p +type FromIntegerP n = FromInteger' Unproxy n + +instance (Num (PP t a) + , Integral (PP n a) + , P n a + , Show (PP t a) + ) => P (FromInteger' t n) a where + type PP (FromInteger' t n) a = PP t a + eval _ opts a = do + let msg0 = "FromInteger" + nn <- eval (Proxy @n) opts a + pure $ case getValueLR opts msg0 nn [] of + Left e -> e + Right n -> + let b = fromInteger (fromIntegral n) + in mkNode opts (PresentT b) [msg0 <> show0 opts " " b] [hh nn] + +-- | 'fromIntegral' function +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(FromIntegral (SG.Sum _) Id) 23 +-- Present Sum {getSum = 23} +-- PresentT (Sum {getSum = 23}) +data FromIntegral' t n +type FromIntegral (t :: Type) p = FromIntegral' (Hole t) p + +instance (Num (PP t a) + , Integral (PP n a) + , P n a + , Show (PP t a) + , Show (PP n a) + ) => P (FromIntegral' t n) a where + type PP (FromIntegral' t n) a = PP t a + eval _ opts a = do + let msg0 = "FromIntegral" + nn <- eval (Proxy @n) opts a + pure $ case getValueLR opts msg0 nn [] of + Left e -> e + Right n -> + let b = fromIntegral n + in mkNode opts (PresentT b) [msg0 <> show0 opts " " b <> showA opts " | " n] [hh nn] + +-- | 'toRational' function +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(ToRational Id) 23.5 +-- Present 47 % 2 +-- PresentT (47 % 2) + +data ToRational p + +instance (a ~ PP p x + , Show a + , Real a + , P p x) + => P (ToRational p) x where + type PP (ToRational p) x = Rational + eval _ opts x = do + let msg0 = "ToRational" + pp <- eval (Proxy @p) opts x + pure $ case getValueLR opts msg0 pp [] of + Left e -> e + Right a -> + let r = (toRational a) + in mkNode opts (PresentT r) [msg0 <> show0 opts " " r <> showA opts " | " a] [hh pp] + +-- | 'fromRational' function +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(FromRational Rational Id) 23.5 +-- Present 47 % 2 +-- PresentT (47 % 2) +data FromRational' t r +type FromRational (t :: Type) p = FromRational' (Hole t) p + +instance (P r a + , PP r a ~ Rational + , Show (PP t a) + , Fractional (PP t a) + ) => P (FromRational' t r) a where + type PP (FromRational' t r) a = PP t a + eval _ opts a = do + let msg0 = "FromRational" + rr <- eval (Proxy @r) opts a + pure $ case getValueLR opts msg0 rr [] of + Left e -> e + Right r -> + let b = fromRational @(PP t a) r + in mkNode opts (PresentT b) [msg0 <> show0 opts " " b <> showA opts " | " r] [hh rr] + +-- | 'truncate' function +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Truncate Int Id) (23 % 5) +-- Present 4 +-- PresentT 4 +data Truncate' t p +type Truncate (t :: Type) p = Truncate' (Hole t) p + +instance (Show (PP p x) + , P p x + , Show (PP t x) + , RealFrac (PP p x) + , Integral (PP t x) + ) => P (Truncate' t p) x where + type PP (Truncate' t p) x = PP t x + eval _ opts x = do + let msg0 = "Truncate" + pp <- eval (Proxy @p) opts x + pure $ case getValueLR opts msg0 pp [] of + Left e -> e + Right p -> + let b = truncate p + in mkNode opts (PresentT b) [msg0 <> show0 opts " " b <> showA opts " | " p] [hh pp] + +-- | 'ceiling' function +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Ceiling Int Id) (23 % 5) +-- Present 5 +-- PresentT 5 +data Ceiling' t p +type Ceiling (t :: Type) p = Ceiling' (Hole t) p + +instance (Show (PP p x) + , P p x + , Show (PP t x) + , RealFrac (PP p x) + , Integral (PP t x) + ) => P (Ceiling' t p) x where + type PP (Ceiling' t p) x = PP t x + eval _ opts x = do + let msg0 = "Ceiling" + pp <- eval (Proxy @p) opts x + pure $ case getValueLR opts msg0 pp [] of + Left e -> e + Right p -> + let b = ceiling p + in mkNode opts (PresentT b) [msg0 <> show0 opts " " b <> showA opts " | " p] [hh pp] + +-- | 'floor' function +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Floor Int Id) (23 % 5) +-- Present 4 +-- PresentT 4 +data Floor' t p +type Floor (t :: Type) p = Floor' (Hole t) p + +instance (Show (PP p x) + , P p x + , Show (PP t x) + , RealFrac (PP p x) + , Integral (PP t x) + ) => P (Floor' t p) x where + type PP (Floor' t p) x = PP t x + eval _ opts x = do + let msg0 = "Floor" + pp <- eval (Proxy @p) opts x + pure $ case getValueLR opts msg0 pp [] of + Left e -> e + Right p -> + let b = floor p + in mkNode opts (PresentT b) [msg0 <> show0 opts " " b <> showA opts " | " p] [hh pp] + +-- Start non-Type kinds +----------------------- +----------------------- +----------------------- + +-- | pulls the type level 'Bool' to the value level +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @'True "ignore this" +-- True +-- TrueT +-- +-- >>> pl @'False () +-- False +-- FalseT +instance GetBool b => P (b :: Bool) a where + type PP b a = Bool + eval _ opts _ = + let b = getBool @b + in pure $ mkNodeB opts b ["'" <> show b] [] + +-- | pulls the type level 'Symbol' to the value level +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @"hello world" () +-- Present "hello world" +-- PresentT "hello world" +instance KnownSymbol s => P (s :: Symbol) a where + type PP s a = String + eval _ opts _ = + let s = symb @s + in pure $ mkNode opts (PresentT s) ["'" <> showLit0 opts "" s] [] + +-- | run the predicates in a promoted 2-tuple; similar to 'Control.Arrow.&&&' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @'(Snd, Fst) ("helo",123) +-- Present (123,"helo") +-- PresentT (123,"helo") +-- +-- >>> :set -XTypeOperators +-- >>> pl @'(Len, Id <> "|" <> Reverse) "helo" +-- Present (4,"helo|oleh") +-- PresentT (4,"helo|oleh") +instance (P p a, P q a) => P '(p,q) a where + type PP '(p,q) a = (PP p a, PP q a) + eval _ opts a = do + let msg = "'(,)" + lr <- runPQ msg (Proxy @p) (Proxy @q) opts a + pure $ case lr of + Left e -> e + Right (p,q,pp,qq) -> + mkNode opts (PresentT (p,q)) [msg] [hh pp, hh qq] + +-- | run the predicates in a promoted 3-tuple +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @'(Len, Id, Reverse) "helo" +-- Present (4,"helo","oleh") +-- PresentT (4,"helo","oleh") +instance (P p a + , P q a + , P r a + ) => P '(p,q,r) a where + type PP '(p,q,r) a = (PP p a, PP q a, PP r a) + eval _ opts a = do + let msg = "'(,,)" + lr <- runPQ msg (Proxy @p) (Proxy @q) opts a + case lr of + Left e -> pure e + Right (p,q,pp,qq) -> do + let hhs = [hh pp, hh qq] + rr <- eval (Proxy @r) opts a + pure $ case getValueLR opts msg rr hhs of + Left e -> e + Right r -> mkNode opts (PresentT (p,q,r)) [msg] (hhs <> [hh rr]) + +-- | run the predicates in a promoted 4-tuple +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @'(Len, Id, "inj", 999) "helo" +-- Present (4,"helo","inj",999) +-- PresentT (4,"helo","inj",999) +instance (P p a + , P q a + , P r a + , P s a + ) => P '(p,q,r,s) a where + type PP '(p,q,r,s) a = (PP p a, PP q a, PP r a, PP s a) + eval _ opts a = do + let msg = "'(,,)" + lr <- runPQ msg (Proxy @p) (Proxy @q) opts a + case lr of + Left e -> pure e + Right (p,q,pp,qq) -> do + lr1 <- runPQ msg (Proxy @r) (Proxy @s) opts a + pure $ case lr1 of + Left e -> e + Right (r,s,rr,ss) -> + mkNode opts (PresentT (p,q,r,s)) [msg] [hh pp, hh qq, hh rr, hh ss] + +-- | extracts the value level representation of the promoted 'Ordering' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @'LT "not used" +-- Present LT +-- PresentT LT +-- +-- >>> pl @'EQ () +-- Present EQ +-- PresentT EQ +instance GetOrdering cmp => P (cmp :: Ordering) a where + type PP cmp a = Ordering + eval _ opts _a = + let cmp = getOrdering @cmp + msg = "'" <> show cmp + in pure $ mkNode opts (PresentT cmp) [msg] [] + +-- | extracts the value level representation of the type level 'Nat' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @123 () +-- Present 123 +-- PresentT 123 +instance KnownNat n => P (n :: Nat) a where + type PP n a = Int + eval _ opts _ = + let n = nat @n + in pure $ mkNode opts (PresentT n) ["'" <> show n] [] + +-- | extracts the value level representation of the type level \'() +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @'() () +-- Present () +-- PresentT () +instance P '() a where + type PP '() a = () + eval _ opts _ = pure $ mkNode opts (PresentT ()) ["'()"] [] + +-- todo: the type has to be [a] so we still need type PP '[p] a = [PP p a] to keep the types in line + +-- | extracts the value level representation of the type level \'[] +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @'[] False +-- Present [] +-- PresentT [] +instance P ('[] :: [k]) a where + type PP ('[] :: [k]) a = [a] + eval _ opts _ = pure $ mkNode opts mempty ["'[]"] [] + +-- | runs each predicate in turn from the promoted list +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> :set -XNoStarIsType +-- >>> pl @'[1, 2, 3] 999 +-- Present [1,2,3] +-- PresentT [1,2,3] +-- +-- >>> pl @'[W 1, W 2, W 3, Id] 999 +-- Present [1,2,3,999] +-- PresentT [1,2,3,999] +-- +-- >>> pl @'[W 1, W 2, W 3, Id * 4, Pred] 999 +-- Present [1,2,3,3996,998] +-- PresentT [1,2,3,3996,998] +-- +-- >>> :set -XTypeOperators +-- >>> pl @'[Id * 4, Pred] 999 +-- Present [3996,998] +-- PresentT [3996,998] +instance (Show (PP p a), Show a, P p a) => P '[p] a where + type PP '[p] a = [PP p a] + eval _ opts a = do + pp <- eval (Proxy @p) opts a + let msg = "" -- "'[](end)" + pure $ case getValueLR opts msg pp [] of + Left e -> e + Right b -> mkNode opts (PresentT [b]) [msg <> show0 opts " " b <> showA opts " | " a] [hh pp] -- <> show0 opts " " a <> showA opts " b=" b]) [hh pp] + +instance (Show (PP p a) + , Show a + , P (p1 ': ps) a + , PP (p1 ': ps) a ~ [PP p1 a] + , P p a + , PP p a ~ PP p1 a + ) => P (p ': p1 ': ps) a where + type PP (p ': p1 ': ps) a = [PP p a] + eval _ opts a = do + let msg = "'" + -- len = 2 + getLen @ps + lr <- runPQ msg (Proxy @p) (Proxy @(p1 ': ps)) opts a + pure $ case lr of + Left e -> e + Right (p,q,pp,qq) -> + mkNode opts (PresentT (p:q)) [msg <> show0 opts "" (p:q) <> showA opts " | " a] [hh pp, hh qq] + +-- | extracts the \'a\' from type level \'Maybe a\' if the value exists +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @('Just Id) (Just 123) +-- Present 123 +-- PresentT 123 +-- +-- >>> pl @('Just Not) (Just True) +-- Present False +-- PresentT False +-- +-- >>> pl @('Just Id) Nothing +-- Error 'Just found Nothing +-- FailT "'Just found Nothing" +-- +instance (Show (PP p a) + , P p a + , Show a + ) => P ('Just p) (Maybe a) where + type PP ('Just p) (Maybe a) = PP p a + eval _ opts ma = do + let msg = "'Just" + case ma of + Just a -> do + pp <- eval (Proxy @p) opts a + pure $ case getValueLR opts msg pp [] of + Left e -> e + Right b -> mkNode opts (PresentT b) [msg <> show0 opts " " b <> showA opts " | " ma] [hh pp] + Nothing -> pure $ mkNode opts (FailT (msg <> " found Nothing")) [msg <> " found Nothing"] [] + +-- | expects Nothing otherwise it fails +-- if the value is Nothing then it returns \'Proxy a\' as this provides more information than '()' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @'Nothing Nothing +-- Present Proxy +-- PresentT Proxy +-- +-- >>> pl @'Nothing (Just True) +-- Error 'Nothing found Just +-- FailT "'Nothing found Just" +-- +instance P 'Nothing (Maybe a) where + type PP 'Nothing (Maybe a) = Proxy a -- () gives us less information + eval _ opts ma = + let msg = "'Nothing" + in pure $ case ma of + Nothing -> mkNode opts (PresentT Proxy) [msg] [] + Just _ -> mkNode opts (FailT (msg <> " found Just")) [msg <> " found Just"] [] + +-- omitted Show x so we can have less ambiguity +-- | extracts the \'a\' from type level \'Either a b\' if the value exists +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @('Left Id) (Left 123) +-- Present 123 +-- PresentT 123 +-- +-- >>> pl @('Left Id) (Right "aaa") +-- Error 'Left found Right +-- FailT "'Left found Right" +-- +instance (Show a + , Show (PP p a) + , P p a + ) => P ('Left p) (Either a x) where + type PP ('Left p) (Either a x) = PP p a + eval _ opts lr = + let msg = "'Left" + in case lr of + Right _ -> pure $ mkNode opts (FailT (msg <> " found Right")) [msg <> " found Right"] [] + Left a -> do + pp <- eval (Proxy @p) opts a + pure $ case getValueLR opts msg pp [] of + Left e -> e + Right b -> mkNode opts (_tBool pp) [msg <> show0 opts " " b <> showA opts " | Left " a] [hh pp] + +-- | extracts the \'b\' from type level \'Either a b\' if the value exists +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @('Right Id) (Right 123) +-- Present 123 +-- PresentT 123 +-- +-- >>> pl @('Right Id) (Left "aaa") +-- Error 'Right found Left +-- FailT "'Right found Left" +-- +instance (Show a + , Show (PP p a) + , P p a + ) => P ('Right p) (Either x a) where + type PP ('Right p) (Either x a) = PP p a + eval _ opts lr = do + let msg = "'Right" + case lr of + Left _ -> pure $ mkNode opts (FailT (msg <> " found Left")) [msg <> " found Left"] [] + Right a -> do + pp <- eval (Proxy @p) opts a + pure $ case getValueLR opts msg pp [] of + Left e -> e + Right b -> mkNode opts (_tBool pp) [msg <> show0 opts " " b <> showA opts " | Right " a] [hh pp] + +-- removed Show x: else ambiguity errors in TestPredicate + +-- | extracts the \'a\' from type level \'These a b\' if the value exists +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @('This Id) (This 123) +-- Present 123 +-- PresentT 123 +-- +-- >>> pl @('This Id) (That "aaa") +-- Error 'This found That +-- FailT "'This found That" +-- +-- >>> pl @('This Id) (These 999 "aaa") +-- Error 'This found These +-- FailT "'This found These" +-- +instance (Show a + , Show (PP p a) + , P p a + ) => P ('This p) (These a x) where + type PP ('This p) (These a x) = PP p a + eval _ opts th = do + let msg = "'This" + case th of + This a -> do + pp <- eval (Proxy @p) opts a + pure $ case getValueLR opts msg pp [] of + Left e -> e + Right b -> mkNode opts (_tBool pp) [msg <> show0 opts " " b <> showA opts " | This " a] [hh pp] + _ -> pure $ mkNode opts (FailT (msg <> " found " <> showThese th)) [msg <> " found " <> showThese th] [] + +-- | extracts the \'b\' from type level \'These a b\' if the value exists +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @('That Id) (That 123) +-- Present 123 +-- PresentT 123 +-- +-- >>> pl @('That Id) (This "aaa") +-- Error 'That found This +-- FailT "'That found This" +-- +-- >>> pl @('That Id) (These 44 "aaa") +-- Error 'That found These +-- FailT "'That found These" +-- +instance (Show a + , Show (PP p a) + , P p a + ) => P ('That p) (These x a) where + type PP ('That p) (These x a) = PP p a + eval _ opts th = do + let msg = "'That" + case th of + That a -> do + pp <- eval (Proxy @p) opts a + pure $ case getValueLR opts msg pp [] of + Left e -> e + Right b -> mkNode opts (_tBool pp) [msg <> show0 opts " " b <> showA opts " | That " a] [hh pp] + _ -> pure $ mkNode opts (FailT (msg <> " found " <> showThese th)) [msg <> " found " <> showThese th] [] + + +-- | extracts the (a,b) from type level 'These a b' if the value exists +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @('These Id Id) (These 123 "abc") +-- Present (123,"abc") +-- PresentT (123,"abc") +-- +-- >>> pl @('These Pred Len) (These 123 "abcde") +-- Present (122,5) +-- PresentT (122,5) +-- +-- >>> pl @('These Id Id) (This "aaa") +-- Error 'These found This +-- FailT "'These found This" +-- +-- >>> pl @('These Id Id) (That "aaa") +-- Error 'These found That +-- FailT "'These found That" +-- +instance (Show a + , Show b + , P p a + , P q b + , Show (PP p a) + , Show (PP q b) + ) => P ('These p q) (These a b) where + type PP ('These p q) (These a b) = (PP p a, PP q b) + eval _ opts th = do + let msg = "'These" + case th of + These a b -> do + pp <- eval (Proxy @p) opts a + case getValueLR opts msg pp [] of + Left e -> pure e + Right p -> do + qq <- eval (Proxy @q) opts b + pure $ case getValueLR opts (msg <> " q failed p=" <> show p) qq [hh pp] of + Left e -> e + Right q -> mkNode opts (PresentT (p,q)) [msg <> show0 opts " " (p,q) <> showA opts " | " (These a b)] [hh pp, hh qq] + _ -> pure $ mkNode opts (FailT (msg <> " found " <> showThese th)) [msg <> " found " <> showThese th] [] + +-- | converts the value to the corresponding 'Proxy' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @'Proxy 'x' +-- Present Proxy +-- PresentT Proxy +-- +instance Show a => P 'Proxy a where + type PP 'Proxy a = Proxy a + eval _ opts a = + let b = Proxy @a + in pure $ mkNode opts (PresentT b) ["'Proxy" <> showA opts " | " a] [] + +-- End non-Type kinds +----------------------- +----------------------- +----------------------- + +-- | converts a value to a 'Proxy': the same as '\'Proxy' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @MkProxy 'x' +-- Present Proxy +-- PresentT Proxy +-- +data MkProxy + +instance Show a => P MkProxy a where + type PP MkProxy a = Proxy a + eval _ opts a = + let b = Proxy @a + in pure $ mkNode opts (PresentT b) ["MkProxy" <> showA opts " | " a] [] + +type family DoExpandT (ps :: [k]) :: Type where + DoExpandT '[] = GL.TypeError ('GL.Text "'[] invalid: requires at least one predicate in the list") + DoExpandT '[p] = Id >> p -- need this else fails cos 1 is nat and would mean that the result is nat not Type! + -- if p >> Id then turns TrueT to PresentT True + DoExpandT (p ': p1 ': ps) = p >> DoExpandT (p1 ': ps) + +-- | processes a type level list predicates running each in sequence: see 'Predicate.>>' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Do [Pred, ShowP, Id &&& Len]) 9876543 +-- Present ("9876542",7) +-- PresentT ("9876542",7) +-- +data Do (ps :: [k]) +instance (P (DoExpandT ps) a) => P (Do ps) a where + type PP (Do ps) a = PP (DoExpandT ps) a + eval _ = eval (Proxy @(DoExpandT ps)) + +-- | Convenient method to convert a value \'p\' to a 'Maybe' based on a predicate '\b\' +-- if '\b\' then Just \'p'\ else Nothing +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(MaybeB (Id > 4) Id) 24 +-- Present Just 24 +-- PresentT (Just 24) +-- +-- >>> pl @(MaybeB (Id > 4) Id) (-5) +-- Present Nothing +-- PresentT Nothing +-- +data MaybeB b p + +instance (Show (PP p a) + , P b a + , P p a + , PP b a ~ Bool + ) => P (MaybeB b p) a where + type PP (MaybeB b p) a = Maybe (PP p a) + eval _ opts z = do + let msg0 = "MaybeB" + bb <- evalBool (Proxy @b) opts z + case getValueLR opts (msg0 <> " b failed") bb [] of + Left e -> pure e + Right True -> do + pp <- eval (Proxy @p) opts z + pure $ case getValueLR opts (msg0 <> " p failed") pp [hh bb] of + Left e -> e + Right p -> mkNode opts (PresentT (Just p)) [msg0 <> "(False)" <> show0 opts " Just " p] [hh bb, hh pp] + Right False -> pure $ mkNode opts (PresentT Nothing) [msg0 <> "(True)"] [hh bb] + +-- | Convenient method to convert a \'p\' or '\q'\ to a 'Either' based on a predicate '\b\' +-- if \'b\' then Right \'p\' else Left '\q\' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(EitherB (Fst > 4) (Snd >> Fst) (Snd >> Snd)) (24,(-1,999)) +-- Present Right 999 +-- PresentT (Right 999) +-- +-- >>> pl @(EitherB (Fst > 4) (Snd >> Fst) (Snd >> Snd)) (1,(-1,999)) +-- Present Left (-1) +-- PresentT (Left (-1)) +-- +data EitherB b p q + +instance (Show (PP p a) + , P p a + , Show (PP q a) + , P q a + , P b a + , PP b a ~ Bool + ) => P (EitherB b p q) a where + type PP (EitherB b p q) a = Either (PP p a) (PP q a) + eval _ opts z = do + let msg0 = "EitherB" + bb <- evalBool (Proxy @b) opts z + case getValueLR opts (msg0 <> " b failed") bb [] of + Left e -> pure e + Right False -> do + pp <- eval (Proxy @p) opts z + pure $ case getValueLR opts (msg0 <> " p failed") pp [hh bb] of + Left e -> e + Right p -> mkNode opts (PresentT (Left p)) [msg0 <> "(False)" <> show0 opts " Left " p] [hh bb, hh pp] + Right True -> do + qq <- eval (Proxy @q) opts z + pure $ case getValueLR opts (msg0 <> " p failed") qq [hh bb] of + Left e -> e + Right q -> mkNode opts (PresentT (Right q)) [msg0 <> "(True)" <> show0 opts " Right " q] [hh bb, hh qq] + +-- | create inductive tuples from a type level list of predicates +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(TupleI '[Id,ShowP,Pred,W "str", W 999]) 666 +-- Present (666,("666",(665,("str",(999,()))))) +-- PresentT (666,("666",(665,("str",(999,()))))) +-- +data TupleI (ps :: [k]) -- make it an inductive tuple + +instance P (TupleI ('[] :: [k])) a where + type PP (TupleI ('[] :: [k])) a = () + eval _ opts _ = pure $ mkNode opts (PresentT ()) ["TupleI(done)"] [] + +instance (P p a + , P (TupleI ps) a + , Show a + ) => P (TupleI (p ': ps)) a where + type PP (TupleI (p ': ps)) a = (PP p a, PP (TupleI ps) a) + eval _ opts a = do + pp <- eval (Proxy @p) opts a + let msg = "TupleI" -- "'[](" <> show len <> ")" + case getValueLR opts msg pp [] of + Left e -> pure e + Right w -> do + qq <- eval (Proxy @(TupleI ps)) opts a + pure $ case getValueLR opts msg qq [hh pp] of + Left e -> e + -- only PresentP makes sense here (ie not TrueP/FalseP: ok in base case tho + Right ws -> mkNode opts (PresentT (w,ws)) [msg <> show0 opts " " a] [hh pp, hh qq] + +-- | type level representation of signed rational numbers/integers +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> :set -XNoStarIsType +-- >>> pl @(NegR 14 3) () +-- Present (-14) % 3 +-- PresentT ((-14) % 3) +-- +-- >>> pl @(PosR 14 3) () +-- Present 14 % 3 +-- PresentT (14 % 3) +-- +-- >>> pl @(CmpRat (NegR 14 3) (Neg 5)) () +-- Present GT +-- PresentT GT +-- +-- >>> pl @(NegR 14 3 * Neg 5) () +-- Present 70 % 3 +-- PresentT (70 % 3) +-- +-- >>> pl @(NegR 14 3 - Pos 5) () +-- Present (-29) % 3 +-- PresentT ((-29) % 3) +-- +-- >>> pl @(CmpRat (PosR 14 3) 5) () +-- Present LT +-- PresentT LT + +data Rat (pos :: Bool) (num :: Nat) (den :: Nat) +-- | constructs a positive integer as a rational number 'Rat' +type Pos (n :: Nat) = Rat 'True n 1 +-- | constructs a negative integer as a rational number 'Rat' +type Neg (n :: Nat) = Rat 'False n 1 + +-- | constructs a valid positive rational number 'Rat' +type family PosR (n :: Nat) (d :: Nat) where + PosR n 0 = GL.TypeError ('GL.Text "PosR has a 0 denominator where numerator=" ':<>: 'GL.ShowType n) + PosR n d = Rat 'True n d + +-- | constructs a valid negative rational number 'Rat' +type family NegR (n :: Nat) (d :: Nat) where + NegR n 0 = GL.TypeError ('GL.Text "NegR has a 0 denominator where numerator=" ':<>: 'GL.ShowType n) + NegR n d = Rat 'False n d + +instance (GetBool pos + , KnownNat num + , KnownNat den + , NotZeroT den + ) => P (Rat pos num den) a where + type PP (Rat pos num den) a = Rational + eval _ opts _ = + let pos = getBool @pos + num = nat @num + den = nat @den + msg = "Rat " <> show r + r = (if pos then id else negate) (num % den) + in pure $ mkNode opts (PresentT r) [msg] [] + +-- | compares 2 numbers where the numbers are type level signed rationals or Nats +type family CmpRat (m :: k) (n :: k1) :: Ordering where + CmpRat (Rat x n 0) z = GL.TypeError ('GL.Text "CmpRat: lhs has 0 denominator" ':$$: 'GL.ShowType (Rat x n 0) ':<>: 'GL.Text " `CmpRat` " ':<>: 'GL.ShowType z) + CmpRat z (Rat x n 0) = GL.TypeError ('GL.Text "CmpRat: rhs has 0 denominator" ':$$: 'GL.ShowType z ':<>: 'GL.Text " `CmpRat` " ':<>: 'GL.ShowType (Rat x n 0)) + CmpRat (m :: Nat) (n :: Nat) = GN.CmpNat m n + CmpRat (Rat x n d) (w :: Nat) = CmpRat (Rat x n d) (Pos w) + CmpRat (w :: Nat) (Rat x n d) = CmpRat (Pos w) (Rat x n d) + CmpRat (Rat x 0 d) (Rat x1 0 d1) = 'EQ + CmpRat (Rat 'True n d) (Rat 'False n1 d1) = 'GT + CmpRat (Rat 'False n d) (Rat 'True n1 d1) = 'LT + CmpRat (Rat 'False n d) (Rat 'False n1 d1) = + CmpRat (Rat 'True n1 d1) (Rat 'True n d) + CmpRat (Rat 'True n d) (Rat 'True n1 d1) = + IfT (GN.CmpNat (GN.Div n d) (GN.Div n1 d1) DE.== 'EQ) + (GN.CmpNat (n GN.* d1) (n1 GN.* d)) + (GN.CmpNat (GN.Div n d) (GN.Div n1 d1)) + +-- | get a list of 'Rational's from the type level +class GetRats as where + getRats :: [Rational] +instance GetRats '[] where + getRats = [] +instance (GetRat n, GetRats ns) => GetRats (n ': ns) where + getRats = getRat @n : getRats @ns + +-- | get a 'Rational' from the type level +class GetRat a where + getRat :: Rational +instance KnownNat n => GetRat (n :: Nat) where + getRat = nat @n +instance (GetBool pos, KnownNat num, KnownNat den, NotZeroT den) => GetRat (Rat (pos :: Bool) (num :: Nat) (den :: Nat)) where + getRat = let s = getBool @pos + n = nat @num + d = nat @den + in (if s then 1 else (-1)) * n % d + +-- | add a message to give more context to the evaluation tree +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pe @(Msg' "somemessage" Id) 999 +-- P [somemessage] Id 999 +-- PresentT 999 +-- +data Msg prt p +type Msg' prt p = Msg (Printf "[%s] " prt) p -- put msg in square brackets + +instance (P prt a + , PP prt a ~ String + , P p a + ) => P (Msg prt p) a where + type PP (Msg prt p) a = PP p a + eval _ opts a = do + pp <- eval (Proxy @prt) opts a + case getValueLR opts "Msg" pp [] of + Left e -> pure e + Right msg -> prefixMsg msg <$> eval (Proxy @p) opts a + +-- | pad \'q\' with '\n'\ values from '\p'\ +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(PadL 5 999 Id) [12,13] +-- Present [999,999,999,12,13] +-- PresentT [999,999,999,12,13] +-- +-- >>> pl @(PadR 5 Fst '[12,13]) (999,'x') +-- Present [12,13,999,999,999] +-- PresentT [12,13,999,999,999] +-- +-- >>> pl @(PadR 2 Fst '[12,13,14]) (999,'x') +-- Present [12,13,14] +-- PresentT [12,13,14] +-- +data Pad (left :: Bool) n p q +type PadL n p q = Pad 'True n p q +type PadR n p q = Pad 'False n p q + +instance (P n a + , GetBool left + , Integral (PP n a) + , [PP p a] ~ PP q a + , P p a + , P q a + , Show (PP p a) + ) => P (Pad left n p q) a where + type PP (Pad left n p q) a = PP q a + eval _ opts a = do + let msg0 = "Pad" <> (if lft then "L" else "R") + lft = getBool @left + lr <- runPQ msg0 (Proxy @n) (Proxy @p) opts a + case lr of + Left e -> pure e + Right (fromIntegral -> n,p,nn,pp) -> do + let msg1 = msg0 <> show0 opts " " n <> " pad=" <> show p + hhs = [hh nn, hh pp] + qq <- eval (Proxy @q) opts a + pure $ case getValueLR opts (msg1 <> " q failed") qq hhs of + Left e -> e + Right q -> + let l = length q + diff = if n<=l then 0 else n-l + bs = if lft + then (replicate diff p) <> q + else q <> (replicate diff p) + in mkNode opts (PresentT bs) [msg1 <> show0 opts " " bs <> showA opts " | " q] (hhs <> [hh qq]) + +-- | split a list \'p\' into parts using the lengths in the type level list \'ns\' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(SplitAts '[2,3,1,1] Id) "hello world" +-- Present ["he","llo"," ","w","orld"] +-- PresentT ["he","llo"," ","w","orld"] +-- +-- >>> pl @(SplitAts '[2] Id) "hello world" +-- Present ["he","llo world"] +-- PresentT ["he","llo world"] +-- +-- >>> pl @(SplitAts '[10,1,1,5] Id) "hello world" +-- Present ["hello worl","d","",""] +-- PresentT ["hello worl","d","",""] +-- +data SplitAts ns p +instance (P ns x + , P p x + , PP p x ~ [a] + , Show n + , Show a + , PP ns x ~ [n] + , Integral n + ) => P (SplitAts ns p) x where + type PP (SplitAts ns p) x = [PP p x] + eval _ opts x = do + let msg = "SplitAts" + lr <- runPQ msg (Proxy @ns) (Proxy @p) opts x + pure $ case lr of + Left e -> e + Right (ns,p,nn,pp) -> + let zs = foldr (\n k s -> let (a,b) = splitAt (fromIntegral n) s + in a:k b + ) (\as -> if null as then [] else [as]) ns p + in mkNode opts (PresentT zs) [msg <> show0 opts " " zs <> showA opts " | ns=" ns <> showA opts " | " p] [hh nn, hh pp] + +-- | similar to 'splitAt' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(SplitAt 4 Id) "hello world" +-- Present ("hell","o world") +-- PresentT ("hell","o world") +-- +-- >>> pl @(SplitAt 20 Id) "hello world" +-- Present ("hello world","") +-- PresentT ("hello world","") +-- +-- >>> pl @(SplitAt 0 Id) "hello world" +-- Present ("","hello world") +-- PresentT ("","hello world") +-- +-- >>> pl @(SplitAt Snd Fst) ("hello world",4) +-- Present ("hell","o world") +-- PresentT ("hell","o world") +-- +data SplitAt n p +type Take n p = SplitAt n p >> Fst +type Drop n p = SplitAt n p >> Snd + +instance (PP p a ~ [b] + , P n a + , P p a + , Show b + , Integral (PP n a) + ) => P (SplitAt n p) a where + type PP (SplitAt n p) a = (PP p a, PP p a) + eval _ opts a = do + let msg0 = "SplitAt" + lr <- runPQ msg0 (Proxy @n) (Proxy @p) opts a + pure $ case lr of + Left e -> e -- (Left e, tt') + Right (fromIntegral -> n,p,pp,qq) -> + let msg1 = msg0 <> show0 opts " " n <> show0 opts " " p + (x,y) = splitAt n p + in mkNode opts (PresentT (x,y)) [msg1 <> show0 opts " " (x,y) <> showA opts " | n=" n <> showA opts " | " p] [hh pp, hh qq] + +type Tail = Uncons >> 'Just Snd +type Head = Uncons >> 'Just Fst +type Init = Unsnoc >> 'Just Fst +type Last = Unsnoc >> 'Just Snd + +-- | similar to 'Control.Arrow.&&&' +type p &&& q = W '(p, q) +infixr 3 &&& + +-- | similar to 'Control.Arrow.***' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Pred *** ShowP) (13, True) +-- Present (12,"True") +-- PresentT (12,"True") +-- +data (p :: k) *** (q :: k1) +type Star p q = p *** q +infixr 3 *** +type First p = Star p I +type Second q = Star I q + +instance (Show (PP p a) + , Show (PP q b) + , P p a + , P q b + , Show a + , Show b + ) => P (p *** q) (a,b) where + type PP (p *** q) (a,b) = (PP p a, PP q b) + eval _ opts (a,b) = do + let msg = "(***)" + pp <- eval (Proxy @p) opts a + case getValueLR opts msg pp [] of + Left e -> pure e + Right a1 -> do + qq <- eval (Proxy @q) opts b + pure $ case getValueLR opts msg qq [hh pp] of + Left e -> e + Right b1 -> mkNode opts (PresentT (a1,b1)) [msg <> show0 opts " " (a1,b1) <> showA opts " | " (a,b)] [hh pp, hh qq] + +-- | similar 'Control.Arrow.|||' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Pred ||| Id) (Left 13) +-- Present 12 +-- PresentT 12 +-- +-- >>> pl @(ShowP ||| Id) (Right "hello") +-- Present "hello" +-- PresentT "hello" +-- +data (|||) (p :: k) (q :: k1) +infixr 2 ||| +type EitherIn p q = p ||| q +type IsLeft = 'True ||| 'False +type IsRight = 'False ||| 'True + +instance (Show (PP p a) + , P p a + , P q b + , PP p a ~ PP q b + , Show a + , Show b + ) => P (p ||| q) (Either a b) where + type PP (p ||| q) (Either a b) = PP p a + eval _ opts (Left a) = do + let msg = "|||" + pp <- eval (Proxy @p) opts a + pure $ case getValueLR opts msg pp [] of + Left e -> e + Right a1 -> mkNode opts (_tBool pp) ["Left" <> show0 opts " " a1 <> showA opts " | " a] [hh pp] + eval _ opts (Right a) = do + let msg = "|||" + qq <- eval (Proxy @q) opts a + pure $ case getValueLR opts msg qq [] of + Left e -> e + Right a1 -> mkNode opts (_tBool qq) ["Right" <> show0 opts " " a1 <> showA opts " | " a] [hh qq] + +-- | similar 'Control.Arrow.+++' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Pred +++ Id) (Left 13) +-- Present Left 12 +-- PresentT (Left 12) +-- +-- >>> pl @(ShowP +++ Reverse) (Right "hello") +-- Present Right "olleh" +-- PresentT (Right "olleh") +-- +data (+++) (p :: k) (q :: k1) +infixr 2 +++ + +instance (Show (PP p a) + , Show (PP q b) + , P p a + , P q b + , Show a + , Show b + ) => P (p +++ q) (Either a b) where + type PP (p +++ q) (Either a b) = Either (PP p a) (PP q b) + eval _ opts (Left a) = do + let msg = "+++" + pp <- eval (Proxy @p) opts a + pure $ case getValueLR opts msg pp [] of + Left e -> e + Right a1 -> mkNode opts (PresentT (Left a1)) ["(+++) Left" <> show0 opts " Left " a1 <> showA opts " | " a] [hh pp] + eval _ opts (Right a) = do + let msg = "+++" + qq <- eval (Proxy @q) opts a + pure $ case getValueLR opts msg qq [] of + Left e -> e + Right a1 -> mkNode opts (PresentT (Right a1)) ["(+++) Right" <> show0 opts " Right" a1 <> showA opts " | " a] [hh qq] + +type Dup = '(Id, Id) + +data BinOp = BMult | BSub | BAdd deriving (Show,Eq) + +type Mult p q = Bin 'BMult p q +type Add p q = Bin 'BAdd p q +type Sub p q = Bin 'BSub p q + +type p + q = Add p q +infixl 6 + +type p - q = Sub p q +infixl 6 - +type p * q = Mult p q +infixl 7 * + +type p > q = Cmp 'Cgt p q +infix 4 > +type p >= q = Cmp 'Cge p q +infix 4 >= +type p == q = Cmp 'Ceq p q +infix 4 == +type p /= q = Cmp 'Cne p q +infix 4 /= +type p <= q = Cmp 'Cle p q +infix 4 <= +type p < q = Cmp 'Clt p q +infix 4 < + +type p >? q = CmpI 'Cgt p q +infix 4 >? +type p >=? q = CmpI 'Cge p q +infix 4 >=? +type p ==? q = CmpI 'Ceq p q +infix 4 ==? +type p /=? q = CmpI 'Cne p q +infix 4 /=? +type p <=? q = CmpI 'Cle p q +infix 4 <=? +type p <? q = CmpI 'Clt p q +infix 4 <? + +class GetBinOp (k :: BinOp) where + getBinOp :: (Num a, a ~ b) => (String, a -> b -> a) + +instance GetBinOp 'BMult where + getBinOp = ("*",(*)) +instance GetBinOp 'BSub where + getBinOp = ("-",(-)) +instance GetBinOp 'BAdd where + getBinOp = ("+",(+)) + +-- | addition, multiplication and subtraction +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> :set -XNoStarIsType +-- >>> pl @(Fst * Snd) (13,5) +-- Present 65 +-- PresentT 65 +-- +-- >>> pl @(Fst + 4 * (Snd >> Len) - 4) (3,"hello") +-- Present 19 +-- PresentT 19 +-- +data Bin (op :: BinOp) p q + +instance (GetBinOp op + , PP p a ~ PP q a + , P p a + , P q a + , Show (PP p a) + , Num (PP p a) + ) => P (Bin op p q) a where + type PP (Bin op p q) a = PP p a + eval _ opts a = do + let (s,f) = getBinOp @op + lr <- runPQ s (Proxy @p) (Proxy @q) opts a + pure $ case lr of + Left e -> e + Right (p,q,pp,qq) -> + let d = p `f` q + in mkNode opts (PresentT d) [show p <> " " <> s <> " " <> show q <> " = " <> show d] [hh pp, hh qq] + +-- | fractional division +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Fst / Snd) (13,2) +-- Present 6.5 +-- PresentT 6.5 +-- +-- >>> pl @(Pos 13 / Id) 0 +-- Error DivF zero denominator +-- FailT "DivF zero denominator" +-- +data DivF p q +type p / q = DivF p q +infixl 7 / + +instance (PP p a ~ PP q a + , Eq (PP q a) + , P p a + , P q a + , Show (PP p a) + , Fractional (PP p a) + ) => P (DivF p q) a where + type PP (DivF p q) a = PP p a + eval _ opts a = do + let msg = "DivF" + lr <- runPQ msg (Proxy @p) (Proxy @q) opts a + pure $ case lr of + Left e -> e + Right (p,q,pp,qq) + | q == 0 -> let msg1 = msg <> " zero denominator" + in mkNode opts (FailT msg1) [msg1] [hh pp, hh qq] + | otherwise -> + let d = p / q + in mkNode opts (PresentT d) [show p <> " / " <> show q <> " = " <> show d] [hh pp, hh qq] + +-- | similar to 'negate' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @Negate 14 +-- Present -14 +-- PresentT (-14) +-- +data Negate + +instance (Show a, Num a) => P Negate a where + type PP Negate a = a + eval _ opts a = + let d = negate a + in pure $ mkNode opts (PresentT d) ["Negate" <> show0 opts " " d <> showA opts " | " a] [] + +-- | similar to 'abs' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @Abs (-14) +-- Present 14 +-- PresentT 14 +-- +-- >>> pl @Abs 14 +-- Present 14 +-- PresentT 14 +-- +-- >>> pl @Abs 0 +-- Present 0 +-- PresentT 0 +-- +data Abs + +instance (Show a, Num a) => P Abs a where + type PP Abs a = a + eval _ opts a = + let d = abs a + in pure $ mkNode opts (PresentT d) ["Abs" <> show0 opts " " d <> showA opts " | " a] [] + +-- | similar to 'signum' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @Signum (-14) +-- Present -1 +-- PresentT (-1) +-- +-- >>> pl @Signum 14 +-- Present 1 +-- PresentT 1 +-- +-- >>> pl @Signum 0 +-- Present 0 +-- PresentT 0 +-- +data Signum + +instance (Show a, Num a) => P Signum a where + type PP Signum a = a + eval _ opts a = + let d = signum a + in pure $ mkNode opts (PresentT d) ["Signum" <> show0 opts " " d <> showA opts " | " a] [] + +-- | unwraps a value (see 'Control.Lens.Unwrapped') +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @Unwrap (SG.Sum (-13)) +-- Present -13 +-- PresentT (-13) +-- +data Unwrap + +instance (Show s + , Show (Unwrapped s) + , Wrapped s + ) => P Unwrap s where + type PP Unwrap s = Unwrapped s + eval _ opts as = + let d = as ^. _Wrapped' + in pure $ mkNode opts (PresentT d) ["Unwrap" <> show0 opts " " d <> showA opts " | " as] [] + +-- | wraps a value (see 'Control.Lens.Wrapped' and 'Control.Lens.Unwrapped') +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> :m + Data.List.NonEmpty +-- >>> pl @(Wrap (SG.Sum _) Id) (-13) +-- Present Sum {getSum = -13} +-- PresentT (Sum {getSum = -13}) +-- +-- >>> pl @(Wrap SG.Any (Ge 4)) 13 +-- Present Any {getAny = True} +-- PresentT (Any {getAny = True}) +-- +-- >>> pl @(Wrap (NonEmpty _) (Uncons >> 'Just Id)) "abcd" +-- Present 'a' :| "bcd" +-- PresentT ('a' :| "bcd") +-- +data Wrap' t p +type Wrap (t :: Type) p = Wrap' (Hole t) p + +instance (Show (PP p x) + , P p x + , Unwrapped (PP s x) ~ PP p x + , Wrapped (PP s x) + , Show (PP s x) + ) => P (Wrap' s p) x where + type PP (Wrap' s p) x = PP s x + eval _ opts x = do + let msg0 = "Wrap" + pp <- eval (Proxy @p) opts x + pure $ case getValueLR opts msg0 pp [] of + Left e -> e + Right p -> + let d = p ^. _Unwrapped' + in mkNode opts (PresentT d) ["Wrap" <> show0 opts " " d <> showA opts " | " p] [hh pp] + +-- | similar to 'coerce' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Coerce (SG.Sum Integer)) (Identity (-13)) +-- Present Sum {getSum = -13} +-- PresentT (Sum {getSum = -13}) +-- +data Coerce (t :: k) + +instance (Show a + , Show t + , Coercible t a + ) => P (Coerce t) a where + type PP (Coerce t) a = t + eval _ opts a = + let d = a ^. coerced + in pure $ mkNode opts (PresentT d) ["Coerce" <> show0 opts " " d <> showA opts " | " a] [] + +-- can coerce over a functor: but need to provide type of 'a' and 't' explicitly + +-- | see 'Coerce': coerce over a functor +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Coerce2 (SG.Sum Integer)) [Identity (-13), Identity 4, Identity 99] +-- Present [Sum {getSum = -13},Sum {getSum = 4},Sum {getSum = 99}] +-- PresentT [Sum {getSum = -13},Sum {getSum = 4},Sum {getSum = 99}] +-- +-- >>> pl @(Coerce2 (SG.Sum Integer)) (Just (Identity (-13))) +-- Present Just (Sum {getSum = -13}) +-- PresentT (Just (Sum {getSum = -13})) +-- +-- >>> pl @(Coerce2 (SG.Sum Int)) (Nothing @(Identity Int)) +-- Present Nothing +-- PresentT Nothing +-- +data Coerce2 (t :: k) +instance (Show (f a) + , Show (f t) + , Coercible t a + , Functor f + ) => P (Coerce2 t) (f a) where + type PP (Coerce2 t) (f a) = f t + eval _ opts fa = + let d = view coerced <$> fa + in pure $ mkNode opts (PresentT d) ["Coerce2" <> show0 opts " " d <> showA opts " | " fa] [] + +-- | lift mempty over a Functor +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(MemptyT2 (SG.Product Int)) [Identity (-13), Identity 4, Identity 99] +-- Present [Product {getProduct = 1},Product {getProduct = 1},Product {getProduct = 1}] +-- PresentT [Product {getProduct = 1},Product {getProduct = 1},Product {getProduct = 1}] +-- +data MemptyT2' t +type MemptyT2 t = MemptyT2' (Hole t) + +instance (Show (f a) + , Show (f (PP t (f a))) + , Functor f + , Monoid (PP t (f a)) + ) => P (MemptyT2' t) (f a) where + type PP (MemptyT2' t) (f a) = f (PP t (f a)) + eval _ opts fa = + let b = mempty <$> fa + in pure $ mkNode opts (PresentT b) ["MemptyT2" <> show0 opts " " b <> showA opts " | " fa] [] + +-- | lift pure over a Functor +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Pure2 (Either String)) [1,2,4] +-- Present [Right 1,Right 2,Right 4] +-- PresentT [Right 1,Right 2,Right 4] +-- +data Pure2 (t :: Type -> Type) +type Right t = Pure (Either t) Id +type Left t = Right t >> Swap + +instance (Show (f (t a)) + , Show (f a) + , Applicative t + , Functor f + ) => P (Pure2 t) (f a) where + type PP (Pure2 t) (f a) = f (t a) + eval _ opts fa = + let b = fmap pure fa + in pure $ mkNode opts (PresentT b) ["Pure2" <> show0 opts " " b <> showA opts " | " fa] [] + +-- | similar to 'reverse' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @Reverse [1,2,4] +-- Present [4,2,1] +-- PresentT [4,2,1] +-- +-- >>> pl @Reverse "AbcDeF" +-- Present "FeDcbA" +-- PresentT "FeDcbA" +-- +data Reverse + +instance (Show a, as ~ [a]) => P Reverse as where + type PP Reverse as = as + eval _ opts as = + let d = reverse as + in pure $ mkNode opts (PresentT d) ["Reverse" <> show0 opts " " d <> showA opts " | " as] [] + +-- | reverses using 'reversing' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> import Data.Text (Text) +-- >>> pl @ReverseL ("AbcDeF" :: Text) +-- Present "FeDcbA" +-- PresentT "FeDcbA" +-- +data ReverseL + +instance (Show t, Reversing t) => P ReverseL t where + type PP ReverseL t = t + eval _ opts as = + let d = as ^. reversed + in pure $ mkNode opts (PresentT d) ["ReverseL" <> show0 opts " " d <> showA opts " | " as] [] + +-- | swaps using 'swapped' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @Swap (Left 123) +-- Present Right 123 +-- PresentT (Right 123) +-- +-- >>> pl @Swap (Right 123) +-- Present Left 123 +-- PresentT (Left 123) +-- +-- >>> pl @Swap (These 'x' 123) +-- Present These 123 'x' +-- PresentT (These 123 'x') +-- +-- >>> pl @Swap (This 'x') +-- Present That 'x' +-- PresentT (That 'x') +-- +-- >>> pl @Swap (That 123) +-- Present This 123 +-- PresentT (This 123) +-- +-- >>> pl @Swap (123,'x') +-- Present ('x',123) +-- PresentT ('x',123) +-- +data Swap + +instance (Show (p a b) + , Swapped p + , Show (p b a) + ) => P Swap (p a b) where + type PP Swap (p a b) = p b a + eval _ opts pab = + let d = pab ^. swapped + in pure $ mkNode opts (PresentT d) ["Swap" <> show0 opts " " d <> showA opts " | " pab] [] + +-- | bounded 'succ' function +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @SuccB' (13 :: Int) +-- Present 14 +-- PresentT 14 +-- +-- >>> pl @SuccB' LT +-- Present EQ +-- PresentT EQ +-- +-- >>> pl @(SuccB 'LT) GT +-- Present LT +-- PresentT LT +-- +-- >>> pl @SuccB' GT +-- Error Succ bounded failed +-- FailT "Succ bounded failed" +-- +data SuccB def +type SuccB' = SuccB (Failp "Succ bounded failed") + +instance (P def (Proxy a) + , PP def (Proxy a) ~ a + , Show a + , Eq a + , Bounded a + , Enum a + ) => P (SuccB def) a where + type PP (SuccB def) a = a + eval _ opts a = do + let msg0 = "SuccB" + case succMay a of + Nothing -> do + let msg1 = msg0 <> " out of range" + pp <- eval (Proxy @def) opts (Proxy @a) + pure $ case getValueLR opts msg1 pp [] of + Left e -> e + Right _ -> mkNode opts (_tBool pp) [msg1] [hh pp] + Just n -> pure $ mkNode opts (PresentT n) [msg0 <> show0 opts " " n <> showA opts " | " a] [] + +-- | bounded 'pred' function +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @PredB' (13 :: Int) +-- Present 12 +-- PresentT 12 +data PredB def +type PredB' = PredB (Failp "Pred bounded failed") + +instance (P def (Proxy a) + , PP def (Proxy a) ~ a + , Show a + , Eq a + , Bounded a + , Enum a + ) => P (PredB def) a where + type PP (PredB def) a = a + eval _ opts a = do + let msg0 = "PredB" + case predMay a of + Nothing -> do + let msg1 = msg0 <> " out of range" + pp <- eval (Proxy @def) opts (Proxy @a) + pure $ case getValueLR opts msg1 pp [] of + Left e -> e + Right _ -> mkNode opts (_tBool pp) [msg1] [hh pp] + Just n -> pure $ mkNode opts (PresentT n) [msg0 <> show0 opts " " n <> showA opts " | " a] [] + +-- | unbounded 'succ' function +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @Succ 13 +-- Present 14 +-- PresentT 14 +-- +-- >>> pl @Succ LT +-- Present EQ +-- PresentT EQ +-- +-- >>> pl @Succ GT +-- Error Succ IO e=Prelude.Enum.Ordering.succ: bad argument +-- FailT "Succ IO e=Prelude.Enum.Ordering.succ: bad argument" +-- +data Succ +instance (Show a, Enum a) => P Succ a where + type PP Succ a = a + eval _ opts a = do + let msg = "Succ" + lr <- catchit @_ @E.SomeException (succ a) + pure $ case lr of + Left e -> mkNode opts (FailT (msg <> " " <> e)) [msg <> show0 opts " " a] [] + Right n -> mkNode opts (PresentT n) [msg <> show0 opts " " n <> showA opts " | " a] [] + + +-- | unbounded 'pred' function +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @Pred 13 +-- Present 12 +-- PresentT 12 +data Pred +instance (Show a, Enum a) => P Pred a where + type PP Pred a = a + eval _ opts a = do + let msg = "Pred" + lr <- catchit @_ @E.SomeException (pred a) + pure $ case lr of + Left e -> mkNode opts (FailT (msg <> " " <> e)) [msg <> show0 opts " " a] [] + Right n -> mkNode opts (PresentT n) [msg <> show0 opts " " n <> showA opts " | " a] [] + + +-- | 'fromEnum' function +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @FromEnum 'x' +-- Present 120 +-- PresentT 120 +data FromEnum +instance (Show a, Enum a) => P FromEnum a where + type PP FromEnum a = Int + eval _ opts a = + let n = fromEnum a + in pure $ mkNode opts (PresentT n) ["FromEnum" <> show0 opts " " n <> showA opts " | " a] [] + +-- | unsafe 'toEnum' function +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(ToEnum Char) 120 +-- Present 'x' +-- PresentT 'x' +data ToEnum' t +type ToEnum (t :: Type) = ToEnum' (Hole t) + +instance (Show a + , Enum (PP t a) + , Show (PP t a) + , Integral a + ) => P (ToEnum' t) a where + type PP (ToEnum' t) a = PP t a + eval _ opts a = do + let msg = "ToEnum" + lr <- catchit @_ @E.SomeException (toEnum $! fromIntegral a) + pure $ case lr of + Left e -> mkNode opts (FailT (msg <> " " <> e)) [msg <> show0 opts " " a] [] + Right n -> mkNode opts (PresentT n) [msg <> show0 opts " " n <> showA opts " | " a] [] + +-- | bounded 'toEnum' function +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(ToEnumB Ordering LT) 2 +-- Present GT +-- PresentT GT +-- +-- >>> pl @(ToEnumB Ordering LT) 6 +-- Present LT +-- PresentT LT +-- +-- >>> pl @(ToEnumBF Ordering) 6 +-- Error ToEnum bounded failed +-- FailT "ToEnum bounded failed" +-- +data ToEnumB' t def +type ToEnumB (t :: Type) def = ToEnumB' (Hole t) def +type ToEnumBF (t :: Type) = ToEnumB' (Hole t) (Failp "ToEnum bounded failed") + +instance (P def (Proxy (PP t a)) + , PP def (Proxy (PP t a)) ~ (PP t a) + , Show a + , Show (PP t a) + , Bounded (PP t a) + , Enum (PP t a) + , Integral a + ) => P (ToEnumB' t def) a where + type PP (ToEnumB' t def) a = PP t a + eval _ opts a = do + let msg0 = "ToEnumB" + case toEnumMay $ fromIntegral a of + Nothing -> do + let msg1 = msg0 <> " out of range" + pp <- eval (Proxy @def) opts (Proxy @(PP t a)) + pure $ case getValueLR opts msg1 pp [] of + Left e -> e + Right _ -> mkNode opts (_tBool pp) [msg1] [hh pp] + Just n -> pure $ mkNode opts (PresentT n) [msg0 <> show0 opts " " n <> showA opts " | " a] [] + +-- | a predicate on prime numbers +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @Prime 2 +-- True +-- TrueT +-- +-- >>> pl @(Map '(Id,Prime) Id) [0..12] +-- Present [(0,False),(1,False),(2,True),(3,True),(4,False),(5,True),(6,False),(7,True),(8,False),(9,False),(10,False),(11,True),(12,False)] +-- PresentT [(0,False),(1,False),(2,True),(3,True),(4,False),(5,True),(6,False),(7,True),(8,False),(9,False),(10,False),(11,True),(12,False)] +-- +data Prime + +instance (Show a, Integral a) => P Prime a where + type PP Prime a = Bool + eval _ opts a = + let b = isPrime $ fromIntegral a + in pure $ mkNodeB opts b ["Prime" <> showA opts " | " a] [] + + +-- | 'not' function +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @Not False +-- True +-- TrueT +-- +-- >>> pl @Not True +-- False +-- FalseT +-- +data Not +instance a ~ Bool => P Not a where + type PP Not a = Bool + eval _ opts a = + let b = not a + in pure $ mkNodeB opts b ["Not"] [] -- already has FalseT TrueT so no need to add confusing data + + +-- empty lists at the type level wont work here + +-- | filters a list \'q\' keeping or removing those elements in \'p\' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Keep '[5] '[1,5,5,2,5,2]) () +-- Present [5,5,5] +-- PresentT [5,5,5] +-- +-- >>> pl @(Keep '[0,1,1,5] '[1,5,5,2,5,2]) () +-- Present [1,5,5,5] +-- PresentT [1,5,5,5] +-- +-- >>> pl @(Remove '[5] '[1,5,5,2,5,2]) () +-- Present [1,2,2] +-- PresentT [1,2,2] +-- +-- >>> pl @(Remove '[0,1,1,5] '[1,5,5,2,5,2]) () +-- Present [2,2] +-- PresentT [2,2] +-- +-- >>> pl @(Remove '[99] '[1,5,5,2,5,2]) () +-- Present [1,5,5,2,5,2] +-- PresentT [1,5,5,2,5,2] +-- +-- >>> pl @(Remove '[99,91] '[1,5,5,2,5,2]) () +-- Present [1,5,5,2,5,2] +-- PresentT [1,5,5,2,5,2] +-- +-- >>> pl @(Remove Id '[1,5,5,2,5,2]) [] +-- Present [1,5,5,2,5,2] +-- PresentT [1,5,5,2,5,2] +-- +-- >>> pl @(Remove '[] '[1,5,5,2,5,2]) 44 -- works if you make this a number! +-- Present [1,5,5,2,5,2] +-- PresentT [1,5,5,2,5,2] +-- +data KeepImpl (keep :: Bool) p q +type Remove p q = KeepImpl 'False p q +type Keep p q = KeepImpl 'True p q + +instance (GetBool keep + , Eq a + , Show a + , P p x + , P q x + , PP p x ~ PP q x + , PP q x ~ [a] + ) => P (KeepImpl keep p q) x where + type PP (KeepImpl keep p q) x = PP q x + eval _ opts x = do + let msg0 = if keep then "Keep" else "Remove" + keep = getBool @keep + lr <- runPQ msg0 (Proxy @p) (Proxy @q) opts x + pure $ case lr of + Left e -> e + Right (p,q,pp,qq) -> + let ret = filter (bool not id keep . (`elem` p)) q + in mkNode opts (PresentT ret) [msg0 <> show0 opts " " ret <> showA opts " | p=" p <> showA opts " | q=" q] [hh pp, hh qq] + +-- | 'elem' function +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Elem Fst Snd) ('x',"abcdxy") +-- True +-- TrueT +-- +-- >>> pl @(Elem Fst Snd) ('z',"abcdxy") +-- False +-- FalseT +-- +data Elem p q +type ElemAll p q = Any (Elem I q) + +instance ([PP p a] ~ PP q a + , P p a + , P q a + , Show (PP p a) + , Eq (PP p a) + ) => P (Elem p q) a where + type PP (Elem p q) a = Bool + eval _ opts a = do + let msg0 = "Elem" + lr <- runPQ msg0 (Proxy @p) (Proxy @q) opts a + pure $ case lr of + Left e -> e + Right (p,q,pp,qq) -> + let b = p `elem` q + in mkNodeB opts b [show p <> " `elem` " <> show q] [hh pp, hh qq] + +-- | 'const' () function +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @() "Asf" +-- Present () +-- PresentT () +-- +instance Show a => P () a where + type PP () a = () + eval _ opts a = pure $ mkNode opts (PresentT ()) ["()" <> show0 opts " " a] [] + +type Head' p = HeadFail "Head(empty)" p +type Tail' p = TailFail "Tail(empty)" p +type Last' p = LastFail "Last(empty)" p +type Init' p = InitFail "Init(empty)" p + +-- | similar to fmap fst +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @Fmap_1 (Just (13,"Asf")) +-- Present Just 13 +-- PresentT (Just 13) +-- +-- to make this work we grab the fst or snd out of the Maybe so it is a head or not/ is a tail or not etc! +-- we still have access to the whole original list so we dont lose anything! +data Fmap_1 +instance Functor f => P Fmap_1 (f (a,x)) where + type PP Fmap_1 (f (a,x)) = f a + eval _ opts mb = pure $ mkNode opts (PresentT (fst <$> mb)) ["Fmap_1"] [] + +-- | similar to fmap snd +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @Fmap_2 (Just ("asf",13)) +-- Present Just 13 +-- PresentT (Just 13) +-- +data Fmap_2 +instance Functor f => P Fmap_2 (f (x,a)) where + type PP Fmap_2 (f (x,a)) = f a + eval _ opts mb = pure $ mkNode opts (PresentT (snd <$> mb)) ["Fmap_2"] [] + +type HeadDef p q = GDef (Uncons >> Fmap_1) p q +type HeadP q = GProxy (Uncons >> Fmap_1) q +type HeadFail msg q = GFail (Uncons >> Fmap_1) msg q + +type TailDef p q = GDef (Uncons >> Fmap_2) p q +type TailP q = GProxy (Uncons >> Fmap_2) q +type TailFail msg q = GFail (Uncons >> Fmap_2) msg q + +type LastDef p q = GDef (Unsnoc >> Fmap_2) p q +type LastP q = GProxy (Unsnoc >> Fmap_2) q +type LastFail msg q = GFail (Unsnoc >> Fmap_2) msg q + +type InitDef p q = GDef (Unsnoc >> Fmap_1) p q +type InitP q = GProxy (Unsnoc >> Fmap_1) q +type InitFail msg q = GFail (Unsnoc >> Fmap_1) msg q + +-- 'x' and 'a' for Just condition +-- 'x' for Nothing condition +-- Snd at the end says we only want to process the Maybe which is the rhs of &&& ie Snd +type GDef' z p q r = '(I, r >> z) >> MaybeXP (X >> p) q Snd +type JustDef' p q r = GDef' I p q r + +-- access everything ie 'x' and Proxy a for Nothing condition +-- 'x' and 'a' for Just condition +type GDef'' z p q r = '(I, r >> z) >> MaybeXP p q Snd +type JustDef'' p q r = GDef'' I p q r + +type PA = Snd -- 'Proxy a' -- to distinguish from A +type A = Snd -- 'a' +type X = Fst >> Fst -- 'x' ie the whole original environment +type XA = I -- ie noop +type XPA = I -- ie noop + +-- Nothing has access to 'x' only +-- Just has access to (x,a) +--type GDef_X z p q r = (I &&& (r >> z)) >> MaybeXP (Fst >> Fst >> p) ((Fst *** I) >> q) Snd +type GDef_X z p q r = '(I, r >> z) >> MaybeXP (X >> p) ('(X,A) >> q) A +type JustDef''' p q r = GDef_X I p q r + +-- Nothing has access to 'Proxy a' only +-- Just has access to (x,a) +type GDef_PA z p q r = Hide % '(I, r >> z) >> MaybeXP (PA >> p) ('(X,A) >> q) A + +-- Nothing case sees ((I,qz), Proxy a) -- hence the Fst >> Fst +-- Just case sees (I,qz), a) -- hence the Snd to get the 'a' only -- if you want the 'x' then Fst >> Fst +-- we have lost 'x' on the rhs: use GDef_X to access 'x' and 'a' for the Just condition +type GDef z p q = '(I, q >> z) >> MaybeXP (X >> p) A A -- Hide % immediately before MaybeXP +type GProxy z q = '(I, q >> z) >> MaybeXP (PA >> MemptyP) A A +type GFail z msg q = '(I, q >> z) >> MaybeXP (Fail (PA >> Unproxy) (X >> msg)) A A + +-- use these! +type LookupDef' x y p q = GDef (Lookup x y) p q +type LookupP' x y q = GProxy (Lookup x y) q +type LookupFail' msg x y q = GFail (Lookup x y) msg q + +type LookupDef x y p = LookupDef' x y p I +type LookupP x y = LookupP' x y I +type LookupFail msg x y = LookupFail' msg x y I + +type Just' p = JustFail "expected Just" p +type Left' p = LeftFail "expected Left" p +type Right' p = RightFail "expected Right" p +type This' p = ThisFail "expected This" p +type That' p = ThatFail "expected That" p +type TheseIn' p = TheseFail "expected These" p + +type JustDef p q = GDef I p q +type JustP q = GProxy I q +type JustFail msg q = GFail I msg q + +type LeftDef p q = GDef LeftToMaybe p q +type LeftP q = GProxy LeftToMaybe q +type LeftFail msg q = GFail LeftToMaybe msg q + +type RightDef p q = GDef RightToMaybe p q +type RightP q = GProxy RightToMaybe q +type RightFail msg q = GFail RightToMaybe msg q + +type ThisDef p q = GDef ThisToMaybe p q +type ThisP q = GProxy ThisToMaybe q +type ThisFail msg q = GFail ThisToMaybe msg q + +type ThatDef p q = GDef ThatToMaybe p q +type ThatP q = GProxy ThatToMaybe q +type ThatFail msg q = GFail ThatToMaybe msg q + +type TheseDef p q = GDef TheseToMaybe p q +type TheseP q = GProxy TheseToMaybe q +type TheseFail msg q = GFail TheseToMaybe msg q + +-- tacks on a Proxy to Nothing side! but a Proxy a not Proxy of the final result +-- this is for default use cases for either/these/head/tail/last/init etc +data MaybeXP p q r +type MaybeX p q r = MaybeXP (Fst >> p) q r + +instance (P r x + , P p (x, Proxy a) + , P q (x,a) + , PP r x ~ Maybe a + , PP p (x, Proxy a) ~ b + , PP q (x,a) ~ b + ) => P (MaybeXP p q r) x where + type PP (MaybeXP p q r) x = MaybeXPT (PP r x) x q + eval _ opts x = do + let msg0 = "MaybeXP" + rr <- eval (Proxy @r) opts x + case getValueLR opts msg0 rr [] of + Left e -> pure e + Right Nothing -> do + let msg1 = msg0 <> "(Nothing)" + pp <- eval (Proxy @p) opts (x, Proxy @a) + pure $ case getValueLR opts msg1 pp [hh rr] of + Left e -> e + Right _ -> mkNode opts (_tBool pp) [msg1] [hh rr, hh pp] + Right (Just a) -> do + let msg1 = msg0 <> "(Just)" + qq <- eval (Proxy @q) opts (x,a) + pure $ case getValueLR opts msg1 qq [hh rr] of + Left e -> e + Right _ -> mkNode opts (_tBool qq) [msg1] [hh rr, hh qq] + +type family MaybeXPT lr x q where + MaybeXPT (Maybe a) x q = PP q (x,a) + + +-- | similar to either Just (const Nothing) +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @LeftToMaybe (Left 13) +-- Present Just 13 +-- PresentT (Just 13) +-- +-- >>> pl @LeftToMaybe (Right 13) +-- Present Nothing +-- PresentT Nothing +-- +data LeftToMaybe +instance P LeftToMaybe (Either a x) where + type PP LeftToMaybe (Either a x) = Maybe a + eval _ opts lr = pure $ mkNode opts (PresentT (either Just (const Nothing) lr)) ["LeftToMaybe"] [] + + +-- | similar to either (const Nothing) Just +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @RightToMaybe (Right 13) +-- Present Just 13 +-- PresentT (Just 13) +-- +-- >>> pl @RightToMaybe (Left 13) +-- Present Nothing +-- PresentT Nothing +-- +data RightToMaybe +instance P RightToMaybe (Either x a) where + type PP RightToMaybe (Either x a) = Maybe a + eval _ opts lr = pure $ mkNode opts (PresentT (either (const Nothing) Just lr)) ["RightToMaybe"] [] + +data ThisToMaybe + +instance P ThisToMaybe (These a x) where + type PP ThisToMaybe (These a x) = Maybe a + eval _ opts th = pure $ mkNode opts (PresentT (these Just (const Nothing) (const . const Nothing) th)) ["ThisToMaybe"] [] + +data ThatToMaybe + +instance P ThatToMaybe (These x a) where + type PP ThatToMaybe (These x a) = Maybe a + eval _ opts th = pure $ mkNode opts (PresentT (these (const Nothing) Just (const . const Nothing) th)) ["ThatToMaybe"] [] + +data TheseToMaybe + +instance P TheseToMaybe (These a b) where + type PP TheseToMaybe (These a b) = Maybe (a,b) + eval _ opts th = pure $ mkNode opts (PresentT (these (const Nothing) (const Nothing) ((Just .) . (,)) th)) ["TheseToMaybe"] [] + +-- | similar to 'Control.Arrow.|||' but additionally gives \'p\' and \'q\' the original input +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(EitherX (((Fst >> Fst) + Snd) >> ShowP) ShowP Snd) (9,Left 123) +-- Present "132" +-- PresentT "132" +-- +-- >>> pl @(EitherX (((Fst >> Fst) + Snd) >> ShowP) ShowP Snd) (9,Right 'x') +-- Present "((9,Right 'x'),'x')" +-- PresentT "((9,Right 'x'),'x')" +-- +-- >>> pl @(EitherX ShowP (Second Succ >> ShowP) Snd) (9,Right 'x') +-- Present "((9,Right 'x'),'y')" +-- PresentT "((9,Right 'x'),'y')" +-- +data EitherX p q r +instance (P r x + , P p (x,a) + , P q (x,b) + , PP r x ~ Either a b + , PP p (x,a) ~ c + , PP q (x,b) ~ c + ) => P (EitherX p q r) x where + type PP (EitherX p q r) x = EitherXT (PP r x) x p + eval _ opts x = do + let msg0 = "EitherX" + rr <- eval (Proxy @r) opts x + case getValueLR opts msg0 rr [] of + Left e -> pure e + Right (Left a) -> do + let msg1 = msg0 <> "(Left)" + pp <- eval (Proxy @p) opts (x,a) + pure $ case getValueLR opts msg1 pp [hh rr] of + Left e -> e + Right _ -> mkNode opts (_tBool pp) [msg1] [hh rr, hh pp] + Right (Right b) -> do + let msg1 = msg0 <> "(Right)" + qq <- eval (Proxy @q) opts (x,b) + pure $ case getValueLR opts msg1 qq [hh rr] of + Left e -> e + Right _ -> mkNode opts (_tBool qq) [msg1] [hh rr, hh qq] + +type family EitherXT lr x p where + EitherXT (Either a b) x p = PP p (x,a) + +-- | similar to 'Data.These.mergeTheseWith' but additionally provides \'p\', '\q'\ and \'r\' the original input as the first element in the tuple +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(TheseX (((Fst >> Fst) + Snd) >> ShowP) ShowP (Snd >> Snd) Snd) (9,This 123) +-- Present "132" +-- PresentT "132" +-- +-- >>> pl @(TheseX '(Snd,"fromthis") '(99 >> Negate,Snd) Snd Id) (This 123) +-- Present (123,"fromthis") +-- PresentT (123,"fromthis") +-- +-- >>> pl @(TheseX '(Snd,"fromthis") '(99 >> Negate,Snd) Snd Id) (That "fromthat") +-- Present (-99,"fromthat") +-- PresentT (-99,"fromthat") +-- +-- >>> pl @(TheseX '(Snd,"fromthis") '(99 >> Negate,Snd) Snd Id) (These 123 "fromthese") +-- Present (123,"fromthese") +-- PresentT (123,"fromthese") +-- +data TheseX p q r s + +instance (P s x + , P p (x,a) + , P q (x,b) + , P r (x,(a,b)) + , PP s x ~ These a b + , PP p (x,a) ~ c + , PP q (x,b) ~ c + , PP r (x,(a,b)) ~ c + ) => P (TheseX p q r s) x where + type PP (TheseX p q r s) x = TheseXT (PP s x) x p + eval _ opts x = do + let msg0 = "TheseX" + ss <- eval (Proxy @s) opts x + case getValueLR opts msg0 ss [] of + Left e -> pure e + Right (This a) -> do + let msg1 = msg0 <> "(This)" + pp <- eval (Proxy @p) opts (x,a) + pure $ case getValueLR opts msg1 pp [hh ss] of + Left e -> e + Right _ -> mkNode opts (_tBool pp) [msg1] [hh ss, hh pp] + Right (That b) -> do + let msg1 = msg0 <> "(That)" + qq <- eval (Proxy @q) opts (x,b) + pure $ case getValueLR opts msg1 qq [hh ss] of + Left e -> e + Right _ -> mkNode opts (_tBool qq) [msg1] [hh ss, hh qq] + Right (These a b) -> do + let msg1 = msg0 <> "(These)" + rr <- eval (Proxy @r) opts (x,(a,b)) + pure $ case getValueLR opts msg1 rr [hh ss] of + Left e -> e + Right _ -> mkNode opts (_tBool rr) [msg1] [hh ss, hh rr] + +type family TheseXT lr x p where + TheseXT (These a b) x p = PP p (x,a) + +-- | similar to 'maybe' +-- +-- similar to 'MaybeX' but provides a Proxy to the result of \'q\' and does not provide the surrounding context +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(MaybeIn "foundnothing" (Pred >> ShowP)) (Just 20) +-- Present "19" +-- PresentT "19" +-- +-- >>> pl @(MaybeIn "found nothing" (Pred >> ShowP)) Nothing +-- Present "found nothing" +-- PresentT "found nothing" +-- +data MaybeIn p q +type IsNothing = MaybeIn 'True 'False +type IsJust = MaybeIn 'False 'True + +-- tricky: the nothing case is the proxy of PP q a: ie proxy of the final result!! +-- this is different from MaybeXP which gives you a proxy of 'a' [you need both!] +instance (P q a + , Show a + , Show (PP q a) + , PP p (Proxy (PP q a)) ~ PP q a + , P p (Proxy (PP q a)) + ) => P (MaybeIn p q) (Maybe a) where + type PP (MaybeIn p q) (Maybe a) = PP q a + eval _ opts ma = do + let msg0 = "MaybeIn" + case ma of + Nothing -> do + let msg1 = msg0 <> "(Nothing)" + pp <- eval (Proxy @p) opts (Proxy @(PP q a)) + pure $ case getValueLR opts msg1 pp [] of + Left e -> e + Right b -> mkNode opts (_tBool pp) [msg1 <> show0 opts " " b <> " | Proxy"] [hh pp] + Just a -> do + let msg1 = msg0 <> "(Nothing)" + qq <- eval (Proxy @q) opts a + pure $ case getValueLR opts msg1 qq [] of + Left e -> e + Right b -> mkNode opts (_tBool qq) [msg1 <> show0 opts " " b <> showA opts " | " a] [hh qq] + + +-- | similar to 'SG.stimes' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(STimes 4 Id) (SG.Sum 3) +-- Present Sum {getSum = 12} +-- PresentT (Sum {getSum = 12}) +-- +-- >>> pl @(STimes 4 Id) "ab" +-- Present "abababab" +-- PresentT "abababab" +-- +data STimes n p +instance (P n a + , Integral (PP n a) + , Semigroup (PP p a) + , P p a + , Show (PP p a) + ) => P (STimes n p) a where + type PP (STimes n p) a = PP p a + eval _ opts a = do + let msg0 = "STimes" + lr <- runPQ msg0 (Proxy @n) (Proxy @p) opts a + pure $ case lr of + Left e -> e + Right (fromIntegral -> (n::Int),p,pp,qq) -> + let msg1 = msg0 <> show0 opts " " n <> " p=" <> show p + b = SG.stimes n p + in mkNode opts (PresentT b) [msg1 <> show0 opts " " b <> showA opts " | n=" n <> showA opts " | " p] [hh pp, hh qq] + + +-- | similar to 'pure' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Pure Maybe Id) 4 +-- Present Just 4 +-- PresentT (Just 4) +-- +-- >>> pl @(Pure [] Id) 4 +-- Present [4] +-- PresentT [4] +-- +data Pure (t :: Type -> Type) p +instance (P p x + , Show (PP p x) + , Show (t (PP p x)) + , Applicative t + ) => P (Pure t p) x where + type PP (Pure t p) x = t (PP p x) + eval _ opts x = do + let msg0 = "Pure" + pp <- eval (Proxy @p) opts x + pure $ case getValueLR opts msg0 pp [] of + Left e -> e + Right a -> + let b = pure a + in mkNode opts (PresentT b) [msg0 <> show0 opts " " b <> showA opts " | " a] [hh pp] + +type PMempty = MemptyT' 'Proxy -- lifts 'a' to 'Proxy a' then we can use it with MemptyP + +-- | similar to 'mempty' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(MemptyT (SG.Sum Int)) () +-- Present Sum {getSum = 0} +-- PresentT (Sum {getSum = 0}) +-- +-- no Monoid for Maybe a unless a is also a monoid but can use empty! +data MemptyT' t +type MemptyT (t :: Type) = MemptyT' (Hole t) +type MemptyP = MemptyT' Unproxy -- expects a proxy: so only some things work with this: eg Pad MaybeIn etc + +instance (Show (PP t a), Monoid (PP t a)) => P (MemptyT' t) a where + type PP (MemptyT' t) a = PP t a + eval _ opts _ = + let b = mempty @(PP t a) + in pure $ mkNode opts (PresentT b) ["MemptyT" <> show0 opts " " b] [] + +data MemptyProxy +instance Monoid a => P MemptyProxy (Proxy (a :: Type)) where + type PP MemptyProxy (Proxy a) = a + eval _ opts _pa = + let b = mempty @a + in pure $ mkNode opts (PresentT b) ["MemptyProxy"] [] + +-- | similar to 'empty' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(EmptyT Maybe) () +-- Present Nothing +-- PresentT Nothing +-- +data EmptyT (t :: Type -> Type) +type MkNothing'' = EmptyT Maybe -- works if a->Maybe a: MkNothing will always work cos can override the output type + +instance (Show (t a), Alternative t) => P (EmptyT t) a where + type PP (EmptyT t) a = t a + eval _ opts _ = + let b = empty @t + in pure $ mkNode opts (PresentT b) ["EmptyT" <> show0 opts " " b] [] + +data MkNothing' t -- works always! MaybeB is a good alternative and then dont need the extra 't' +type MkNothing (t :: Type) = MkNothing' (Hole t) + +-- for this to be useful has to have 't' else we end up with tons of problems +instance P (MkNothing' t) a where + type PP (MkNothing' t) a = Maybe (PP t a) + eval _ opts _ = + let msg = "MkNothing" + in pure $ mkNode opts (PresentT Nothing) [msg] [] + +-- | 'Just' constructor +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @MkJust 44 +-- Present Just 44 +-- PresentT (Just 44) +-- +data MkJust +instance Show a => P MkJust a where + type PP MkJust a = Maybe a + eval _ opts a = + let msg0 = "MkJust" + d = Just a + in pure $ mkNode opts (PresentT d) [msg0 <> show0 opts " Just " a] [] + +-- | 'Data.Either.Left' constructor +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(MkLeft _ Id) 44 +-- Present Left 44 +-- PresentT (Left 44) +-- +data MkLeft' t p +type MkLeft (t :: Type) p = MkLeft' (Hole t) p + +instance (Show (PP p x), P p x) => P (MkLeft' t p) x where + type PP (MkLeft' t p) x = Either (PP p x) (PP t x) + eval _ opts x = do + let msg0 = "MkLeft" + pp <- eval (Proxy @p) opts x + pure $ case getValueLR opts msg0 pp [] of + Left e -> e + Right p -> + let d = Left p + in mkNode opts (PresentT d) [msg0 <> show0 opts " Left " p] [] + +-- | 'Data.Either.Right' constructor +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(MkRight _ Id) 44 +-- Present Right 44 +-- PresentT (Right 44) +-- +data MkRight' t p +type MkRight (t :: Type) p = MkRight' (Hole t) p + +instance (Show (PP p x), P p x) => P (MkRight' t p) x where + type PP (MkRight' t p) x = Either (PP t x) (PP p x) + eval _ opts x = do + let msg0 = "MkRight" + pp <- eval (Proxy @p) opts x + pure $ case getValueLR opts msg0 pp [] of + Left e -> e + Right p -> + let d = Right p + in mkNode opts (PresentT d) [msg0 <> show0 opts " Right " p] [] + +-- | 'Data.These.This' constructor +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(MkThis _ Id) 44 +-- Present This 44 +-- PresentT (This 44) +-- +data MkThis' t p +type MkThis (t :: Type) p = MkThis' (Hole t) p + +instance (Show (PP p x), P p x) => P (MkThis' t p) x where + type PP (MkThis' t p) x = These (PP p x) (PP t x) + eval _ opts x = do + let msg0 = "MkThis" + pp <- eval (Proxy @p) opts x + pure $ case getValueLR opts msg0 pp [] of + Left e -> e + Right p -> + let d = This p + in mkNode opts (PresentT d) [msg0 <> show0 opts " This " p] [] + +-- | 'Data.These.That' constructor +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(MkThat _ Id) 44 +-- Present That 44 +-- PresentT (That 44) +-- +data MkThat' t p +type MkThat (t :: Type) p = MkThat' (Hole t) p + +instance (Show (PP p x), P p x) => P (MkThat' t p) x where + type PP (MkThat' t p) x = These (PP t x) (PP p x) + eval _ opts x = do + let msg0 = "MkThat" + pp <- eval (Proxy @p) opts x + pure $ case getValueLR opts msg0 pp [] of + Left e -> e + Right p -> + let d = That p + in mkNode opts (PresentT d) [msg0 <> show0 opts " That " p] [] + +--type MkThat t p = MkThis t p >> Swap +-- type MkThat' (t :: Type) = Pure (These t) Id -- t has to be a semigroup + +-- | 'Data.These.These' constructor +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(MkThese Fst Snd) (44,'x') +-- Present These 44 'x' +-- PresentT (These 44 'x') +-- +data MkThese p q +instance (P p a + , P q a + , Show (PP p a) + , Show (PP q a) + ) => P (MkThese p q) a where + type PP (MkThese p q) a = These (PP p a) (PP q a) + eval _ opts a = do + let msg0 = "MkThese" + lr <- runPQ msg0 (Proxy @p) (Proxy @q) opts a + pure $ case lr of + Left e -> e + Right (p,q,pp,qq) -> + let d = These p q + in mkNode opts (PresentT d) [msg0 <> show0 opts " " d] [hh pp, hh qq] + +-- | similar to 'mconcat' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @MConcat [SG.Sum 44, SG.Sum 12, SG.Sum 3] +-- Present Sum {getSum = 59} +-- PresentT (Sum {getSum = 59}) +-- +data MConcat +type FoldMap (t :: Type) p = Map (Wrap t Id) p >> MConcat >> Unwrap + +type Sum (t :: Type) = FoldMap (SG.Sum t) Id +type Min' (t :: Type) = FoldMap (SG.Min t) Id -- requires t be Bounded for monoid instance + +instance (Show a, Monoid a) => P MConcat [a] where + type PP MConcat [a] = a + eval _ opts a = + let b = mconcat a + in pure $ mkNode opts (PresentT b) ["MConcat" <> show0 opts " " b <> showA opts " | " a] [] + +-- | similar to 'concat' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Concat Id) ["abc","D","eF","","G"] +-- Present "abcDeFG" +-- PresentT "abcDeFG" +-- +-- >>> pl @(Concat Snd) ('x',["abc","D","eF","","G"]) +-- Present "abcDeFG" +-- PresentT "abcDeFG" +-- +data Concat p + +instance (Show a + , Show (t [a]) + , PP p x ~ (t [a]) + , P p x + , Foldable t + ) => P (Concat p) x where + type PP (Concat p) x = MapTX (PP p x) + eval _ opts x = do + let msg0 = "Concat" + pp <- eval (Proxy @p) opts x + pure $ case getValueLR opts msg0 pp [] of + Left e -> e + Right as -> + let b = concat as + in mkNode opts (PresentT b) ["Concat" <> show0 opts " " b <> showA opts " | " as] [hh pp] + +instance P (Proxy t) a where + type PP (Proxy t) a = Proxy t + eval _ opts _ = + pure $ mkNode opts (PresentT Proxy) ["Proxy"] [] + +data ProxyT' t +type ProxyT (t :: Type) = ProxyT' (Hole t) + +instance Typeable t => P (ProxyT' (t :: Type)) a where + type PP (ProxyT' t) a = Proxy (PP t a) + eval _ opts _ = + let t = showT @t + in pure $ mkNode opts (PresentT Proxy) ["ProxyT(" <> show t ++ ")"] [] + +-- | similar to 'Data.List.!!' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Ix 4 "not found") ["abc","D","eF","","G"] +-- Present "G" +-- PresentT "G" +-- +-- >>> pl @(Ix 40 "not found") ["abc","D","eF","","G"] +-- Present "not found" +-- PresentT "not found" +-- +data Ix (n :: Nat) def +type Ix' (n :: Nat) = Ix n (Failp "Ix index not found") + +instance (P def (Proxy a) + , PP def (Proxy a) ~ a + , KnownNat n + , Show a + ) => P (Ix n def) [a] where + type PP (Ix n def) [a] = a + eval _ opts as = do + let n = nat @n + msg0 = "Ix " <> show n + case as ^? ix n of + Nothing -> do + let msg1 = msg0 <> " not found" + pp <- eval (Proxy @def) opts (Proxy @a) + pure $ case getValueLR opts msg1 pp [] of + Left e -> e + Right _ -> mkNode opts (_tBool pp) [msg1] [hh pp] + Just a -> pure $ mkNode opts (PresentT a) [msg0 <> show0 opts " " a] [] + +-- | similar to 'Data.List.!!' leveraging 'Ixed' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> import qualified Data.Map.Strict as M +-- >>> pl @(Id !! 2) ["abc","D","eF","","G"] +-- Present "eF" +-- PresentT "eF" +-- +-- >>> pl @(Id !! 20) ["abc","D","eF","","G"] +-- Error (!!) index not found +-- FailT "(!!) index not found" +-- +-- >>> pl @(Id !! "eF") (M.fromList (flip zip [0..] ["abc","D","eF","","G"])) +-- Present 2 +-- PresentT 2 +-- +data IxL p q def -- p is the big value and q is the index and def is the default +type p !! q = IxL p q (Failp "(!!) index not found") +instance (P q a + , P p a + , Show (PP p a) + , Ixed (PP p a) + , PP q a ~ Index (PP p a) + , Show (Index (PP p a)) + , Show (IxValue (PP p a)) + , P r (Proxy (IxValue (PP p a))) + , PP r (Proxy (IxValue (PP p a))) ~ IxValue (PP p a) + ) + => P (IxL p q r) a where + type PP (IxL p q r) a = IxValue (PP p a) + eval _ opts a = do + let msg0 = "IxL" + lr <- runPQ msg0 (Proxy @p) (Proxy @q) opts a + case lr of + Left e -> pure e + Right (p,q,pp,qq) -> + let msg1 = msg0 <> "(" <> show q <> ")" + in case p ^? ix q of + Nothing -> do + rr <- eval (Proxy @r) opts (Proxy @(IxValue (PP p a))) + pure $ case getValueLR opts msg1 rr [hh pp, hh qq] of + Left e -> e + Right _ -> mkNode opts (_tBool rr) [msg1 <> " index not found"] [hh pp, hh qq] + Just ret -> pure $ mkNode opts (PresentT ret) [msg1 <> show0 opts " " ret <> showA opts " | p=" p <> showA opts " | q=" q] [hh pp, hh qq] + +-- | 'lookup' leveraging 'Ixed' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> import qualified Data.Map.Strict as M +-- >>> pl @(Id !!! 2) ["abc","D","eF","","G"] +-- Present "eF" +-- PresentT "eF" +-- +-- >>> pl @(Id !!! 20) ["abc","D","eF","","G"] +-- Error index not found +-- FailT "index not found" +-- +-- >>> pl @(Id !!! "eF") (M.fromList (flip zip [0..] ["abc","D","eF","","G"])) +-- Present 2 +-- PresentT 2 +-- +-- >>> pl @(Lookup Id 2) ["abc","D","eF","","G"] +-- Present Just "eF" +-- PresentT (Just "eF") +-- +-- >>> pl @(Lookup Id 20) ["abc","D","eF","","G"] +-- Present Nothing +-- PresentT Nothing +-- +data Lookup p q +type p !!! q = Lookup p q >> MaybeIn (Failp "index not found") Id -- use !! +-- Lookup' is interesting but just use Lookup or !! +type Lookup' (t :: Type) p q = q &&& Lookup p q >> If (Snd >> IsNothing) (Fst >> ShowP >> Fail (Hole t) (Printf "index(%s) not found" Id)) (Snd >> 'Just Id) + + +instance (P q a + , P p a + , Show (PP p a) + , Ixed (PP p a) + , PP q a ~ Index (PP p a) + , Show (Index (PP p a)) + , Show (IxValue (PP p a)) + ) + => P (Lookup p q) a where + type PP (Lookup p q) a = Maybe (IxValue (PP p a)) + eval _ opts a = do + let msg0 = "Lookup" + lr <- runPQ msg0 (Proxy @p) (Proxy @q) opts a + pure $ case lr of + Left e -> e + Right (p,q,pp,qq) -> + let msg1 = msg0 <> "(" <> show q <> ")" + hhs = [hh pp, hh qq] + in case p ^? ix q of + Nothing -> mkNode opts (PresentT Nothing) [msg1 <> " not found"] hhs + Just ret -> mkNode opts (PresentT (Just ret)) [msg1 <> show0 opts " " ret <> showA opts " | p=" p <> showA opts " | q=" q] hhs + +-- | 'Data.List.ands' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @Ands [True,True,True] +-- True +-- TrueT +-- +-- >>> pl @Ands [True,True,True,False] +-- False +-- FalseT +-- +-- >>> pl @Ands [] +-- True +-- TrueT +-- +data Ands +type Ands' = FoldMap SG.All Id + +instance (as ~ t a + , Show (t a) + , Foldable t + , a ~ Bool + ) => P Ands as where + type PP Ands as = Bool + eval _ opts as = + let b = and as + in pure $ mkNodeB opts b ["Ands" <> showA opts " | " as] [] + +-- | 'Data.List.ors' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @Ors [False,False,False] +-- False +-- FalseT +-- +-- >>> pl @Ors [True,True,True,False] +-- True +-- TrueT +-- +-- >>> pl @Ors [] +-- False +-- FalseT +-- +data Ors +type Ors' = FoldMap SG.Any Id + +instance (as ~ t a + , Show (t a) + , Foldable t + , a ~ Bool + ) => P Ors as where + type PP Ors as = Bool + eval _ opts as = + let b = or as + in pure $ mkNodeB opts b ["Ors" <> showA opts " | " as] [] + +-- | similar to cons +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Fst :+ Snd) (99,[1,2,3,4]) +-- Present [99,1,2,3,4] +-- PresentT [99,1,2,3,4] +-- +data p :+ q +infixr 5 :+ +instance (P p x + , P q x + , Show (PP p x) + , Show (PP q x) + , Cons (PP q x) (PP q x) (PP p x) (PP p x) + ) => P (p :+ q) x where + type PP (p :+ q) x = PP q x + eval _ opts z = do + let msg0 = "(:+)" + lr <- runPQ msg0 (Proxy @p) (Proxy @q) opts z + pure $ case lr of + Left e -> e + Right (p,q,pp,qq) -> + let b = p `cons` q + in mkNode opts (PresentT b) [msg0 <> show0 opts " " b <> showA opts " | p=" p <> showA opts " | q=" q] [hh pp, hh qq] + +-- | similar to snoc +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Snd +: Fst) (99,[1,2,3,4]) +-- Present [1,2,3,4,99] +-- PresentT [1,2,3,4,99] +-- +data p +: q +infixl 5 +: + +instance (P p x + , P q x + , Show (PP q x) + , Show (PP p x) + , Snoc (PP p x) (PP p x) (PP q x) (PP q x) + ) => P (p +: q) x where + type PP (p +: q) x = PP p x + eval _ opts z = do + let msg0 = "(+:)" + lr <- runPQ msg0 (Proxy @p) (Proxy @q) opts z + pure $ case lr of + Left e -> e + Right (p,q,pp,qq) -> + let b = p `snoc` q + in mkNode opts (PresentT b) [msg0 <> show0 opts " " b <> showA opts " | p=" p <> showA opts " | q=" q] [hh pp, hh qq] + +-- | 'Control.Lens.uncons' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @Uncons [1,2,3,4] +-- Present Just (1,[2,3,4]) +-- PresentT (Just (1,[2,3,4])) +-- +-- >>> pl @Uncons [] +-- Present Nothing +-- PresentT Nothing +-- +data Uncons + +instance (Show (ConsT s) + , Show s + , Cons s s (ConsT s) (ConsT s) + ) => P Uncons s where + type PP Uncons s = Maybe (ConsT s,s) + eval _ opts as = + let b = as ^? _Cons + in pure $ mkNode opts (PresentT b) ["Uncons" <> show0 opts " " b <> showA opts " | " as] [] + +-- | 'Control.Lens.unsnoc' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @Unsnoc [1,2,3,4] +-- Present Just ([1,2,3],4) +-- PresentT (Just ([1,2,3],4)) +-- +-- >>> pl @Unsnoc [] +-- Present Nothing +-- PresentT Nothing +-- +data Unsnoc + +instance (Show (ConsT s) + , Show s + , Snoc s s (ConsT s) (ConsT s) + ) => P Unsnoc s where + type PP Unsnoc s = Maybe (s,ConsT s) + eval _ opts as = + let b = as ^? _Snoc + in pure $ mkNode opts (PresentT b) ["Unsnoc" <> show0 opts " " b <> showA opts " | " as] [] + +-- | similar to 'null' using 'AsEmpty' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @IsEmpty [1,2,3,4] +-- False +-- FalseT +-- +-- >>> pl @IsEmpty [] +-- True +-- TrueT +-- +-- >>> pl @IsEmpty LT +-- False +-- FalseT +-- +-- >>> pl @IsEmpty EQ +-- True +-- TrueT +-- +data IsEmpty + +instance (Show as, AsEmpty as) => P IsEmpty as where + type PP IsEmpty as = Bool + eval _ opts as = + let b = has _Empty as + in pure $ mkNodeB opts b ["IsEmpty" <> showA opts " | " as] [] + +-- | similar to 'null' using 'Foldable' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @Null [1,2,3,4] +-- False +-- FalseT +-- +-- >>> pl @Null [] +-- True +-- TrueT +-- +data Null + +instance (Show (t a) + , Foldable t + , t a ~ as + ) => P Null as where + type PP Null as = Bool + eval _ opts as = + let b = null as + in pure $ mkNodeB opts b ["Null" <> showA opts " | " as] [] + +-- | similar to 'enumFromTo' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(EnumFromTo 2 5) () +-- Present [2,3,4,5] +-- PresentT [2,3,4,5] +-- +-- >>> pl @(EnumFromTo LT GT) () +-- Present [LT,EQ,GT] +-- PresentT [LT,EQ,GT] +-- + +data EnumFromTo p q +instance (P p x + , P q x + , PP p x ~ a + , Show a + , PP q x ~ a + , Enum a + ) => P (EnumFromTo p q) x where + type PP (EnumFromTo p q) x = [PP p x] + eval _ opts z = do + let msg0 = "EnumFromTo" + lr <- runPQ msg0 (Proxy @p) (Proxy @q) opts z + pure $ case lr of + Left e -> e + Right (p,q,pp,qq) -> mkNode opts (PresentT (enumFromTo p q)) [msg0 <> " [" <> show p <> " .. " <> show q <> "]"] [hh pp, hh qq] + +type MapMaybe p q = ConcatMap (p >> MaybeIn MemptyP '[Id]) q +type CatMaybes q = MapMaybe Id q + +-- | similar to 'partitionEithers' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @PartitionEithers [Left 'a',Right 2,Left 'c',Right 4,Right 99] +-- Present ("ac",[2,4,99]) +-- PresentT ("ac",[2,4,99]) +-- +data PartitionEithers + +instance (Show a, Show b) => P PartitionEithers [Either a b] where + type PP PartitionEithers [Either a b] = ([a], [b]) + eval _ opts as = + let b = partitionEithers as + in pure $ mkNode opts (PresentT b) ["PartitionEithers" <> show0 opts " " b <> showA opts " | " as] [] + +-- | similar to 'partitionThese' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @PartitionThese [This 'a', That 2, This 'c', These 'z' 1, That 4, These 'a' 2, That 99] +-- Present (("ac",[2,4,99]),[('z',1),('a',2)]) +-- PresentT (("ac",[2,4,99]),[('z',1),('a',2)]) +-- +data PartitionThese +instance (Show a, Show b) => P PartitionThese [These a b] where + type PP PartitionThese [These a b] = (([a], [b]), [(a, b)]) + eval _ opts as = + let (x,y,z) = partitionThese as + b = ((x,y),z) + in pure $ mkNode opts (PresentT b) ["PartitionThese" <> show0 opts " " b <> showA opts " | " as] [] + +type Thiss = PartitionThese >> Fst >> Fst +type Thats = PartitionThese >> Fst >> Snd +type Theses = PartitionThese >> Snd + +type CatMaybesa t = Foldl (Fst <> (Snd >> MaybeIn MemptyP '[Id])) (MemptyT t) Id +type CatMaybesx t = Foldl (JustDef' Fst ((Fst >> Fst >> Fst) +: Snd) Snd) (MemptyT [t]) Id +type CatMaybesy t = Foldl (JustDef'' (Fst >> Fst >> Fst) ((Fst >> Fst >> Fst) +: Snd) Snd) (MemptyT [t]) Id +type CatMaybesz t = Foldl (JustDef''' Fst ((Fst >> Fst) +: Snd) Snd) (MemptyT [t]) Id + +-- want to pass Proxy b to q but then we have no way to calculate 'b' + +-- | similar to 'scanl' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Scanl (Snd :+ Fst) Fst Snd) ([99],[1..5]) +-- Present [[99],[1,99],[2,1,99],[3,2,1,99],[4,3,2,1,99],[5,4,3,2,1,99]] +-- PresentT [[99],[1,99],[2,1,99],[3,2,1,99],[4,3,2,1,99],[5,4,3,2,1,99]] +-- +data Scanl p q r +-- scanr :: (a -> b -> b) -> b -> [a] -> [b] +-- result is scanl but signature is flipped ((a,b) -> b) -> b -> [a] -> [b] + +type ScanN n p q = Scanl (Fst >> q) p (EnumFromTo 1 n) -- n times using q then run p +type ScanNA q = ScanN Fst Snd q +type Repeat n p q = Last' (ScanN n p q) +type Foldl p q r = Last' (Scanl p q r) + +instance (PP p (b,a) ~ b + , PP q x ~ b + , PP r x ~ [a] + , P p (b,a) + , P q x + , P r x + , Show b + , Show a + ) + => P (Scanl p q r) x where + type PP (Scanl p q r) x = [PP q x] + eval _ opts z = do + let msg0 = "Scanl" + lr <- runPQ msg0 (Proxy @q) (Proxy @r) opts z + case lr of + Left e -> pure e + Right (q,r,qq,rr) -> do + let msg1 = msg0 -- <> show0 opts " " q <> show0 opts " " r + ff i b as' rs + | i >= _MX = pure (rs, Left $ mkNode opts (FailT (msg1 <> ":failed at i=" <> show i)) [msg1 <> " i=" <> show i <> " (b,as')=" <> show (b,as')] []) + | otherwise = + case as' of + [] -> pure (rs, Right ()) -- ++ [((i,q), mkNode opts (PresentT q) [msg1 <> "(done)"] [])], Right ()) + a:as -> do + pp :: TT b <- eval (Proxy @p) opts (b,a) + case getValueLR opts (msg1 <> " i=" <> show i <> " a=" <> show a) pp [] of + Left e -> pure (rs,Left e) + Right b' -> ff (i+1) b' as (rs ++ [((i,b), pp)]) + (ts,lrx) :: ([((Int, b), TT b)], Either (TT [b]) ()) <- ff 1 q r [] + pure $ case splitAndAlign opts [msg1] (((0,q), mkNode opts (PresentT q) [msg1 <> "(initial)"] []) : ts) of + Left _e -> error "cant happen!" + Right (vals,itts) -> + case lrx of + Left e -> mkNode opts (_tBool e) [msg1] (hh qq : hh rr : map (hh . fixit) itts ++ [hh e]) + Right () -> mkNode opts (PresentT vals) [msg1 <> show0 opts " " vals <> showA opts " | b=" q <> showA opts " | as=" r] (hh qq : hh rr : map (hh . fixit) itts) + +type family UnfoldT mbs where + UnfoldT (Maybe (b,s)) = b + +-- | similar to 'unfoldr' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Unfoldr (MaybeB (Null >> Not) (SplitAt 2 Id)) Id) [1..5] +-- Present [[1,2],[3,4],[5]] +-- PresentT [[1,2],[3,4],[5]] +-- +data Unfoldr p q +--type Iteraten (t :: Type) n f = Unfoldr (If (Fst == 0) (MkNothing t) (Snd &&& (Pred *** f) >> MkJust)) '(n, Id) +type IterateN n f = Unfoldr (MaybeB (Fst > 0) '(Snd, Pred *** f)) '(n, Id) +type IterateUntil p f = IterateWhile (p >> Not) f +type IterateWhile p f = Unfoldr (MaybeB p '(Id, f)) Id +type IterateNWhile n p f = '(n, Id) >> IterateWhile (Fst > 0 && (Snd >> p)) (Pred *** f) >> Map Snd Id +type IterateNUntil n p f = IterateNWhile n (p >> Not) f + +instance (PP q a ~ s + , PP p s ~ Maybe (b,s) + , P q a + , P p s + , Show s + , Show b + ) + => P (Unfoldr p q) a where + type PP (Unfoldr p q) a = [UnfoldT (PP p (PP q a))] + eval _ opts z = do + let msg0 = "Unfoldr" + qq <- eval (Proxy @q) opts z + case getValueLR opts msg0 qq [] of + Left e -> pure e + Right q -> do + let msg1 = msg0 <> show0 opts " " q + ff i s rs | i >= _MX = pure (rs, Left $ mkNode opts (FailT (msg1 <> ":failed at i=" <> show i)) [msg1 <> " i=" <> show i <> " s=" <> show s] []) + | otherwise = do + pp :: TT (PP p s) <- eval (Proxy @p) opts s + case getValueLR opts (msg1 <> " i=" <> show i <> " s=" <> show s) pp [] of + Left e -> pure (rs, Left e) + Right Nothing -> pure (rs, Right ()) + Right w@(Just (_b,s')) -> ff (i+1) s' (rs ++ [((i,w), pp)]) + (ts,lr) :: ([((Int, PP p s), TT (PP p s))], Either (TT [b]) ()) <- ff 1 q [] + pure $ case splitAndAlign opts [msg1] ts of + Left _e -> error "cant happen" + Right (vals, itts) -> + case lr of + Left e -> mkNode opts (_tBool e) [msg1] (hh qq : map (hh . fixit) itts ++ [hh e]) + Right () -> + let ret = fst <$> catMaybes vals + in mkNode opts (PresentT ret) [msg1 <> show0 opts " " ret <> showA opts " | s=" q ] (hh qq : map (hh . fixit) itts) + +-- | similar to 'map' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Map Pred Id) [1..5] +-- Present [0,1,2,3,4] +-- PresentT [0,1,2,3,4] +-- +data Map p q +type ConcatMap p q = Concat (Map p q) + +instance (Show (PP p a) + , P p a + , PP q x ~ f a + , P q x + , Show a + , Show (f a) + , Foldable f + ) => P (Map p q) x where + type PP (Map p q) x = [PP p (MapTX (PP q x))] + eval _ opts x = do + let msg0 = "Map" + qq <- eval (Proxy @q) opts x + case getValueLR opts msg0 qq [] of + Left e -> pure e + Right as -> do + ts <- zipWithM (\i a -> ((i, a),) <$> eval (Proxy @p) opts a) [0::Int ..] (toList as) + pure $ case splitAndAlign opts [msg0] ts of + Left e -> e + Right (vals, _) -> mkNode opts (PresentT vals) [msg0 <> show0 opts " " vals <> showA opts " | " as] (hh qq : map (hh . fixit) ts) + +type family MapTX ta where + MapTX (t a) = a + +-- | if p then run q else run r +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(If (Gt 4) "greater than 4" "less than or equal to 4" ) 10 +-- Present "greater than 4" +-- PresentT "greater than 4" +-- +-- >>> pl @(If (Gt 4) "greater than 4" "less than or equal to 4") 0 +-- Present "less than or equal to 4" +-- PresentT "less than or equal to 4" +data If p q r + +instance (Show (PP r a) + , P p a + , PP p a ~ Bool + , P q a + , P r a + , PP q a ~ PP r a + ) => P (If p q r) a where + type PP (If p q r) a = PP q a + eval _ opts a = do + let msg0 = "If" + pp <- evalBool (Proxy @p) opts a + case getValueLR opts (msg0 <> " condition failed") pp [] of + Left e -> pure e + Right b -> do + qqrr <- if b + then eval (Proxy @q) opts a + else eval (Proxy @r) opts a + pure $ case getValueLR opts (msg0 <> " [" <> show b <> "]") qqrr [hh pp, hh qqrr] of + Left e -> e + Right ret -> mkNode opts (_tBool qqrr) [msg0 <> " " <> if b then "(true cond)" else "(false cond)" <> show0 opts " " ret] [hh pp, hh qqrr] + +-- | creates a list of overlapping pairs of elements +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @Pairs [1,2,3,4] +-- Present [(1,2),(2,3),(3,4)] +-- PresentT [(1,2),(2,3),(3,4)] +-- +-- >>> pl @Pairs [] +-- Error Pairs no data found +-- FailT "Pairs no data found" +-- +-- >>> pl @Pairs [1] +-- Error Pairs only one element found +-- FailT "Pairs only one element found" +-- +data Pairs +instance Show a => P Pairs [a] where + type PP Pairs [a] = [(a,a)] + eval _ opts as = + let msg0 = "Pairs" + lr = case as of + [] -> Left (msg0 <> " no data found") + [_] -> Left (msg0 <> " only one element found") + _:bs@(_:_) -> Right (zip as bs) + in pure $ case lr of + Left e -> mkNode opts (FailT e) [e] [] + Right zs -> mkNode opts (PresentT zs) [msg0 <> show0 opts " " zs <> showA opts " | " as ] [] + + +-- | similar to 'partition' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Partition (Ge 3) Id) [10,4,1,7,3,1,3,5] +-- Present ([10,4,7,3,3,5],[1,1]) +-- PresentT ([10,4,7,3,3,5],[1,1]) +-- +data Partition p q +type FilterBy p q = Partition p q >> Fst + +instance (P p x + , Show x + , PP q a ~ [x] + , PP p x ~ Bool + , P q a + ) => P (Partition p q) a where + type PP (Partition p q) a = (PP q a, PP q a) + eval _ opts a' = do + let msg0 = "Partition" + qq <- eval (Proxy @q) opts a' + case getValueLR opts msg0 qq [] of + Left e -> pure e + Right as -> do + ts <- zipWithM (\i a -> ((i, a),) <$> evalBool (Proxy @p) opts a) [0::Int ..] as + pure $ case splitAndAlign opts [msg0] ts of + Left e -> e + Right (vals, tfs) -> + let w0 = partition fst $ zip vals tfs + zz1 = (map (snd . fst . snd) *** map (snd . fst . snd)) w0 + in mkNode opts (PresentT zz1) [msg0 <> show0 opts " " zz1 <> showA opts " | s=" as] (hh qq : map (hh . fixit) tfs) + + +-- | similar to 'break' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Break (Ge 3) Id) [10,4,1,7,3,1,3,5] +-- Present ([],[10,4,1,7,3,1,3,5]) +-- PresentT ([],[10,4,1,7,3,1,3,5]) +-- +-- >>> pl @(Break (Lt 3) Id) [10,4,1,7,3,1,3,5] +-- Present ([10,4],[1,7,3,1,3,5]) +-- PresentT ([10,4],[1,7,3,1,3,5]) +-- +data Break p q +type Span p q = Break (p >> Not) q +-- only process up to the pivot! only process while Right False +-- a predicate can return PresentP not just TrueP +instance (P p x + , PP q a ~ [x] + , PP p x ~ Bool + , P q a + ) => P (Break p q) a where + type PP (Break p q) a = (PP q a, PP q a) + eval _ opts a' = do + let msg0 = "Break" + qq <- eval (Proxy @q) opts a' + case getValueLR opts msg0 qq [] of + Left e -> pure e + Right as -> do + let ff [] zs = pure (zs, [], Nothing) -- [(ia,qq)] extras | the rest of the data | optional last pivot or error + ff ((i,a):ias) zs = do + pp <- evalBool (Proxy @p) opts a + let v = ((i,a), pp) + case getValueLR opts msg0 pp [hh qq] of + Right False -> ff ias (zs :> v) + Right True -> pure (zs,map snd ias,Just v) + Left _ -> pure (zs,map snd ias,Just v) + (ialls,rhs,mpivot) <- ff (zip [0::Int ..] as) Seq.empty + pure $ case mpivot of + Nothing -> + mkNode opts (PresentT (map (snd . fst) (toList ialls), rhs)) + ([msg0] <> ["cnt=" <> show (length ialls, length rhs)]) + (map (hh . fixit) (toList ialls)) + Just iall@(ia, tt) -> + case getValueLR opts (msg0 <> " predicate failed") tt (hh qq : map (hh . fixit) (toList (ialls :> iall))) of + Right True -> + mkNode opts (PresentT (map (snd . fst) (toList ialls), snd ia : rhs)) + ([msg0] <> ["cnt=" <> show (length ialls, 1+length rhs)]) + (hh qq : hh tt : map (hh . fixit) (toList (ialls :> iall))) + + Right False -> error "shouldnt happen" + Left e -> e + +-- | Fails the computation with a message +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Failt Int (Printf "value=%03d" Id)) 99 +-- Error value=099 +-- FailT "value=099" +-- +-- >>> pl @(FailS (Printf2 "value=%03d string=%s")) (99,"somedata") +-- Error value=099 string=somedata +-- FailT "value=099 string=somedata" +-- +data Fail t prt -- t=output type prt=msg +type Failp s = Fail Unproxy s +type Failt (t :: Type) prt = Fail (Hole t) prt +type FailS s = Fail I s +type FailPrt (t :: Type) prt = Fail (Hole t)(Printf prt) +type FailPrt2 (t :: Type) prt = Fail (Hole t)(Printf2 prt) + +instance (P prt a + , PP prt a ~ String + ) => P (Fail t prt) a where + type PP (Fail t prt) a = PP t a + eval _ opts a = do + let msg = "Fail" + pp <- eval (Proxy @prt) opts a + pure $ case getValueLR opts msg pp [] of + Left e -> e + Right s -> mkNode opts (FailT s) [msg <> " " <> s] [hh pp] + +data Hole (t :: Type) +type T (t :: Type) = Hole t -- easier to type + +-- | Acts as a proxy in this dsl where you can explicitly set the Type. +-- +-- It is passed around as an argument to help the type checker when needed. +-- see 'ReadP', 'ParseTimeP', 'ShowP' +-- +instance Typeable t => P (Hole t) a where + type PP (Hole t) a = t -- can only be Type not Type -> Type (can use Proxy but then we go down the rabbithole) + eval _ opts _a = + let msg = "Hole(" <> showT @t <> ")" + in pure $ mkNode opts (FailT msg) [msg <> " you probably meant to get access to the type of PP only and not evaluate"] [] + +data Unproxy + +instance Typeable a => P Unproxy (Proxy (a :: Type)) where + type PP Unproxy (Proxy a) = a + eval _ opts _a = + let msg = "Unproxy(" <> showT @a <> ")" + in pure $ mkNode opts (FailT msg) [msg <> " you probably meant to get access to the type of PP only and not evaluate"] [] + +-- | transparent predicate wrapper to make k of kind Type so it can be in a promoted list (cant mix kinds) see 'Do' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Do '[W 123, W "xyz", Len &&& Id, Pred *** Id<>Id]) () +-- Present (2,"xyzxyz") +-- PresentT (2,"xyzxyz") +-- +-- +-- >>> pl @(TupleI '[W 999,W "somestring",W 'True, Id, Pred >> ShowP]) 23 +-- Present (999,("somestring",(True,(23,("22",()))))) +-- PresentT (999,("somestring",(True,(23,("22",()))))) +-- +data W (p :: k) +instance P p a => P (W p) a where + type PP (W p) a = PP p a + eval _ = eval (Proxy @(Msg "W" p)) + +-- | catch a failure +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Catch Succ (Fst >> Second ShowP >> Printf2 "%s %s" >> 'LT)) GT +-- Present LT +-- PresentT LT +-- +-- >>> pl @(Catch' Succ (Second ShowP >> Printf2 "%s %s")) GT +-- Error Succ IO e=Prelude.Enum.Ordering.succ: bad argument GT +-- FailT "Succ IO e=Prelude.Enum.Ordering.succ: bad argument GT" +-- +-- >>> pl @(Catch' Succ (Second ShowP >> Printf2 "%s %s")) LT +-- Present EQ +-- PresentT EQ +-- +-- more flexible: takes a (String,x) and a proxy so we can still call 'False 'True +-- now takes the FailT string and x so you can print more detail if you want +-- need the proxy so we can fail without having to explicitly specify a type +data Catch p q -- catch p and if fails runs q only on failt +type Catch' p s = Catch p (FailCatch s) -- eg set eg s=Printf "%d" Id or Printf "%s" ShowP +type FailCatch s = Fail (Snd >> Unproxy) (Fst >> s) + +instance (P p x + , P q ((String, x) + , Proxy (PP p x)) + , PP p x ~ PP q ((String, x), Proxy (PP p x)) + ) => P (Catch p q) x where + type PP (Catch p q) x = PP p x + eval _ opts x = do + let msg0 = "Catch" + pp <- eval (Proxy @p) opts x + case getValueLR opts msg0 pp [] of + Left e -> do + let emsg = e ^?! tBool . _FailT -- extract the failt string a push back into the fail case + qq <- eval (Proxy @q) opts ((emsg, x), Proxy @(PP p x)) + pure $ case getValueLR opts (msg0 <> " default condition failed") qq [hh pp] of + Left e1 -> e1 + Right _ -> mkNode opts (_tBool qq) [msg0 <> " caught exception[" <> emsg <> "]"] [hh pp, hh qq] + Right _ -> pure $ mkNode opts (_tBool pp) [msg0 <> " did not fire"] [hh pp] + +type Even = Mod I 2 >> Same 0 +type Odd = Mod I 2 >> Same 1 +type Div' p q = DivMod p q >> Fst +type Mod' p q = DivMod p q >> Snd + +-- | similar to 'div' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Div Fst Snd) (10,4) +-- Present 2 +-- PresentT 2 +-- +-- >>> pl @(Div Fst Snd) (10,0) +-- Error Div zero denominator +-- FailT "Div zero denominator" +-- +data Div p q +instance (PP p a ~ PP q a + , P p a + , P q a + , Show (PP p a) + , Integral (PP p a) + ) => P (Div p q) a where + type PP (Div p q) a = PP p a + eval _ opts a = do + let msg = "Div" + lr <- runPQ msg (Proxy @p) (Proxy @q) opts a + pure $ case lr of + Left e -> e + Right (p,q,pp,qq) -> + let hhs = [hh pp, hh qq] + in case q of + 0 -> mkNode opts (FailT (msg <> " zero denominator")) [msg <> " zero denominator"] hhs + _ -> let d = p `div` q + in mkNode opts (PresentT d) [show p <> " `div` " <> show q <> " = " <> show d] hhs + + +-- | similar to 'mod' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Mod Fst Snd) (10,3) +-- Present 1 +-- PresentT 1 +-- +-- >>> pl @(Mod Fst Snd) (10,0) +-- Error Mod zero denominator +-- FailT "Mod zero denominator" +-- +data Mod p q +instance (PP p a ~ PP q a + , P p a + , P q a + , Show (PP p a) + , Integral (PP p a) + ) => P (Mod p q) a where + type PP (Mod p q) a = PP p a + eval _ opts a = do + let msg = "Mod" + lr <- runPQ msg (Proxy @p) (Proxy @q) opts a + pure $ case lr of + Left e -> e + Right (p,q,pp,qq) -> + let hhs = [hh pp, hh qq] + in case q of + 0 -> mkNode opts (FailT (msg <> " zero denominator")) [msg <> " zero denominator"] hhs + _ -> let d = p `mod` q + in mkNode opts (PresentT d) [show p <> " `mod` " <> show q <> " = " <> show d] hhs + +-- | similar to 'divMod' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(DivMod Fst Snd) (10,3) +-- Present (3,1) +-- PresentT (3,1) +-- +-- >>> pl @(DivMod Fst Snd) (10,-3) +-- Present (-4,-2) +-- PresentT (-4,-2) +-- +-- >>> pl @(DivMod Fst Snd) (-10,3) +-- Present (-4,2) +-- PresentT (-4,2) +-- +-- >>> pl @(DivMod Fst Snd) (-10,-3) +-- Present (3,-1) +-- PresentT (3,-1) +-- +-- >>> pl @(DivMod Fst Snd) (10,0) +-- Error DivMod zero denominator +-- FailT "DivMod zero denominator" +-- +data DivMod p q + +instance (PP p a ~ PP q a + , P p a + , P q a + , Show (PP p a) + , Integral (PP p a) + ) => P (DivMod p q) a where + type PP (DivMod p q) a = (PP p a, PP p a) + eval _ opts a = do + let msg = "DivMod" + lr <- runPQ msg (Proxy @p) (Proxy @q) opts a + pure $ case lr of + Left e -> e + Right (p,q,pp,qq) -> + let hhs = [hh pp, hh qq] + in case q of + 0 -> mkNode opts (FailT (msg <> " zero denominator")) [msg <> " zero denominator"] hhs + _ -> let d = p `divMod` q + in mkNode opts (PresentT d) [show p <> " `divMod` " <> show q <> " = " <> show d] hhs + +-- | similar to 'quotRem' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(QuotRem Fst Snd) (10,3) +-- Present (3,1) +-- PresentT (3,1) +-- +-- >>> pl @(QuotRem Fst Snd) (10,-3) +-- Present (-3,1) +-- PresentT (-3,1) +-- +-- >>> pl @(QuotRem Fst Snd) (-10,-3) +-- Present (3,-1) +-- PresentT (3,-1) +-- +-- >>> pl @(QuotRem Fst Snd) (-10,3) +-- Present (-3,-1) +-- PresentT (-3,-1) +-- +-- >>> pl @(QuotRem Fst Snd) (10,0) +-- Error QuotRem zero denominator +-- FailT "QuotRem zero denominator" +-- +data QuotRem p q + +instance (PP p a ~ PP q a + , P p a + , P q a + , Show (PP p a) + , Integral (PP p a) + ) => P (QuotRem p q) a where + type PP (QuotRem p q) a = (PP p a, PP p a) + eval _ opts a = do + let msg = "QuotRem" + lr <- runPQ msg (Proxy @p) (Proxy @q) opts a + pure $ case lr of + Left e -> e + Right (p,q,pp,qq) -> + let hhs = [hh pp, hh qq] + in case q of + 0 -> mkNode opts (FailT (msg <> " zero denominator")) [msg <> " zero denominator"] hhs + _ -> let d = p `quotRem` q + in mkNode opts (PresentT d) [show p <> " `quotRem` " <> show q <> " = " <> show d] hhs + +type Quot p q = QuotRem p q >> Fst +type Rem p q = QuotRem p q >> Snd + +--type OneP = Guard "expected list of length 1" (Len >> Same 1) >> Head' +type OneP = Guard (Printf "expected list of length 1 but found length=%d" Len) (Len >> Same 1) >> Head + +strictmsg :: forall strict . GetBool strict => String +strictmsg = if getBool @strict then "" else "Lax" + +-- k or prt has access to (Int,a) where Int is the current guard position: hence need to use Printf2 +-- todo: better explanation of how this works +-- passthru but adds the length of ps (replaces LenT in the type synonym to avoid type synonyms being expanded out) + +-- | Guards contain a type level list of tuples the action to run on failure of the predicate and the predicate itself +-- Each tuple validating against the corresponding value in a value list +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Guards '[ '("arg1 failed",Gt 4), '("arg2 failed", Same 4)]) [17,4] +-- Present [17,4] +-- PresentT [17,4] +-- +-- >>> pl @(Guards '[ '("arg1 failed",Gt 4), '("arg2 failed", Same 5)]) [17,4] +-- Error arg2 failed +-- FailT "arg2 failed" +-- +-- >>> pl @(Guards '[ '("arg1 failed",Gt 99), '("arg2 failed", Same 4)]) [17,4] +-- Error arg1 failed +-- FailT "arg1 failed" +-- +-- >>> pl @(Guards '[ '(Printf2 "arg %d failed with value %d",Gt 4), '(Printf2 "%d %d", Same 4)]) [17,3] +-- Error 1 3 +-- FailT "1 3" +-- +-- >>> pl @(GuardsQuick (Printf2 "arg %d failed with value %d") '[Gt 4, Ge 3, Same 4]) [17,3,5] +-- Error arg 2 failed with value 5 +-- FailT "arg 2 failed with value 5" +-- +-- >>> pl @(GuardsQuick (Printf2 "arg %d failed with value %d") '[Gt 4, Ge 3, Same 4]) [17,3,5,99] +-- Error Guards: data elements(4) /= predicates(3) +-- FailT "Guards: data elements(4) /= predicates(3)" +-- +data GuardsImpl (n :: Nat) (strict :: Bool) (os :: [(k,k1)]) +type Guards (os :: [(k,k1)]) = GuardsImplW 'True os +type GuardsLax (os :: [(k,k1)]) = GuardsImplW 'False os +type GuardsQuick (prt :: k) (os :: [k1]) = Guards (ToGuardsT prt os) + +data GuardsImplW (strict :: Bool) (ps :: [(k,k1)]) +instance (GetBool strict, GetLen ps, P (GuardsImpl (LenT ps) strict ps) [a]) => P (GuardsImplW strict ps) [a] where + type PP (GuardsImplW strict ps) [a] = PP (GuardsImpl (LenT ps) strict ps) [a] + eval _ opts as = do + let strict = getBool @strict + msgbase0 = "Guards" <> strictmsg @strict + n = getLen @ps + if strict && n /= length as then + let xx = msgbase0 <> ": data elements(" <> show (length as) <> ") /= predicates(" <> show n <> ")" + in pure $ mkNode opts (FailT xx) [xx] [] + else eval (Proxy @(GuardsImpl (LenT ps) strict ps)) opts as + +instance (KnownNat n + , GetBool strict + , Show a + ) => P (GuardsImpl n strict ('[] :: [(k,k1)])) [a] where + type PP (GuardsImpl n strict ('[] :: [(k,k1)])) [a] = [a] + eval _ opts as = + let msg = "Guards" <> strictmsg @strict <> "(" <> show n <> ")" + n :: Int = nat @n + in pure $ mkNode opts (PresentT as) [msg <> " done!" <> if null as then "" else showA opts " | leftovers=" as] [] + +instance (PP prt (Int, a) ~ String + , P prt (Int, a) + , KnownNat n + , GetBool strict + , GetLen ps + , P p a + , PP p a ~ Bool + , P (GuardsImpl n strict ps) [a] + , PP (GuardsImpl n strict ps) [a] ~ [a] + , Show a + ) => P (GuardsImpl n strict ('(prt,p) ': ps)) [a] where + type PP (GuardsImpl n strict ('(prt,p) ': ps)) [a] = [a] + eval _ opts as' = do + let msgbase0 = "Guards" <> strictmsg @strict <> "(" <> show (n-pos) <> ":" <> show n <> ")" + msgbase1 = "Guard" <> strictmsg @strict <> "(" <> show (n-pos) <> ")" + msgbase2 = "Guards" <> strictmsg @strict + n :: Int = nat @n + pos = getLen @ps + case as' of + [] -> pure $ mkNode opts mempty [msgbase0 <> " (ran out of data!!)"] [] + a:as -> do + pp <- evalBool (Proxy @p) opts a + case getValueLR opts (msgbase1 <> " p failed") pp [] of + Left e -> pure e + Right False -> do + qq <- eval (Proxy @prt) opts (n-pos-1,a) -- only run prt when predicate is False + pure $ case getValueLR opts (msgbase2 <> " False predicate and prt failed") qq [hh pp] of + Left e -> e + Right msgx -> mkNode opts (FailT msgx) [msgbase1 <> " failed [" <> msgx <> "]" <> show0 opts " " a] [hh pp, hh qq] + Right True -> do + ss <- eval (Proxy @(GuardsImpl n strict ps)) opts as + pure $ case getValueLRHide opts (msgbase1 <> " ok | rhs failed") ss [hh pp] of + Left e -> e -- shortcut else we get too compounding errors with the pp tree being added each time! + Right zs -> mkNode opts (PresentT (a:zs)) [msgbase1 <> show0 opts " " a] [hh pp, hh ss] + +-- | \'p\' is the predicate and on failure of the predicate runs \'prt\' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Guard "expected > 3" (Gt 3)) 17 +-- Present 17 +-- PresentT 17 +-- +-- >>> pl @(Guard "expected > 3" (Gt 3)) 1 +-- Error expected > 3 +-- FailT "expected > 3" +-- +-- >>> pl @(Guard (Printf "%d not > 3" Id) (Gt 3)) (-99) +-- Error -99 not > 3 +-- FailT "-99 not > 3" +-- +data Guard prt p +type Guard' p = Guard "Guard" p + +type ExitWhen prt p = Guard prt (p >> Not) +type ExitWhen' p = ExitWhen "ExitWhen" p + +instance (Show a + , P prt a + , PP prt a ~ String + , P p a + , PP p a ~ Bool + ) => P (Guard prt p) a where + type PP (Guard prt p) a = a + eval _ opts a = do + let msg0 = "Guard" + pp <- evalBool (Proxy @p) opts a + case getValueLR opts (msg0 <> " p failed") pp [] of + Left e -> pure e + Right False -> do + qq <- eval (Proxy @prt) opts a + pure $ case getValueLR opts (msg0 <> " Msg") qq [hh pp] of + Left e -> e + Right msgx -> mkNode opts (FailT msgx) [msg0 <> "(failed) [" <> msgx <> "]" <> show0 opts " | " a] [hh pp, hh qq] + Right True -> pure $ mkNode opts (PresentT a) [msg0 <> "(ok)" <> show0 opts " | " a] [hh pp] -- dont show the guard message if successful + +-- just run the effect but skip the value: eg use for Stdout so doesnt interfere with 'a' unless there is an error +data Skip p +type p |> q = Skip p >> q +infixr 1 |> +type p >| q = p >> Skip q +infixr 1 >| + +instance (Show (PP p a), P p a) => P (Skip p) a where + type PP (Skip p) a = a + eval _ opts a = do + let msg0 = "Skip" + pp <- eval (Proxy @p) opts a + pure $ case getValueLR opts (msg0 <> " p failed") pp [] of + Left e -> e + Right p -> mkNode opts (PresentT a) [msg0 <> show0 opts " " p] [hh pp] + +-- advantage of (>>) over 'Do [k] is we can use different kinds for (>>) without having to wrap in W + +-- | This is composition for predicates +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Fst >> Id !! 0 >> Succ) ([11,12],'x') +-- Present 12 +-- PresentT 12 +-- +-- >>> pl @(Len *** Succ >> First Pred >> ShowP) ([11,12],'x') +-- Present "(1,'y')" +-- PresentT "(1,'y')" +-- + +data (p :: k) >> (q :: k1) +infixr 1 >> + +type (<<) p q = q >> p +infixl 1 << + +instance (Show (PP p a) + , Show (PP q (PP p a)) + , P p a + , P q (PP p a) + ) => P (p >> q) a where + type PP (p >> q) a = PP q (PP p a) + eval _ opts a = do + let msg = ">>" + pp <- eval (Proxy @p) opts a + case getValueLRHide opts "lhs failed >>" pp [] of + Left e -> pure e + Right p -> do + qq <- eval (Proxy @q) opts p + pure $ case getValueLRHide opts (show p <> " >> rhs failed") qq [hh pp] of + Left e -> e + Right q -> mkNode opts (_tBool qq) [msg <> show0 opts " " q <> showA opts " | " p] [hh pp, hh qq] + +-- | similar to 'Prelude.&&' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Fst && (Snd >> Len >> Ge 4)) (True,[11,12,13,14]) +-- True +-- TrueT +-- +-- >>> pl @(Fst && (Snd >> Len >> Same 4)) (True,[12,11,12,13,14]) +-- False +-- FalseT +-- +data (&&) (p :: k) (q :: k1) +type And p q = p && q +infixr 3 && + +instance (P p a + , P q a + , PP p a ~ Bool + , PP q a ~ Bool + ) => P (p && q) a where + type PP (p && q) a = Bool + eval _ opts a = do + pp <- evalBool (Proxy @p) opts a + qq <- evalBool (Proxy @q) opts a + pure $ evalBinStrict opts "&&" (&&) pp qq + +-- | similar to 'Prelude.||' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Fst || (Snd >> Len >> Ge 4)) (False,[11,12,13,14]) +-- True +-- TrueT +-- +-- >>> pl @((Fst >> Not) || (Snd >> Len >> Same 4)) (True,[12,11,12,13,14]) +-- False +-- FalseT +-- +data (||) (p :: k) (q :: k1) +type OR p q = p || q +infixr 2 || + +instance (P p a + , P q a + , PP p a ~ Bool + , PP q a ~ Bool + ) => P (p || q) a where + type PP (p || q) a = Bool + eval _ opts a = do + pp <- evalBool (Proxy @p) opts a + qq <- evalBool (Proxy @q) opts a + pure $ evalBinStrict opts "||" (||) pp qq + +-- | implication +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Fst ~> (Snd >> Len >> Ge 4)) (True,[11,12,13,14]) +-- True +-- TrueT +-- +-- >>> pl @(Fst ~> (Snd >> Len >> Same 4)) (True,[12,11,12,13,14]) +-- False +-- FalseT +-- +-- >>> pl @(Fst ~> (Snd >> Len >> Same 4)) (False,[12,11,12,13,14]) +-- True +-- TrueT +-- +-- >>> pl @(Fst ~> (Snd >> Len >> Ge 4)) (False,[11,12,13,14]) +-- True +-- TrueT +-- +data (~>) (p :: k) (q :: k1) +type Imply p q = p ~> q +infixr 1 ~> + +instance (P p a + , P q a + , PP p a ~ Bool + , PP q a ~ Bool + ) => P (p ~> q) a where + type PP (p ~> q) a = Bool + eval _ opts a = do + pp <- evalBool (Proxy @p) opts a + qq <- evalBool (Proxy @q) opts a + pure $ evalBinStrict opts "~>" imply pp qq + +data OrdP p q +type p === q = OrdP p q +infix 4 === + +-- | similar to 'compare' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(OrdP Fst Snd) (10,9) +-- Present GT +-- PresentT GT +-- +-- >>> pl @(OrdP Fst Snd) (10,10) +-- Present EQ +-- PresentT EQ +-- +-- >>> pl @(OrdP Fst Snd) (10,11) +-- Present LT +-- PresentT LT +-- +type OrdA' p q = OrdP (Fst >> p) (Snd >> q) +type OrdA p = OrdA' p p + +instance (Ord (PP p a) + , PP p a ~ PP q a + , P p a + , Show (PP q a) + , P q a + ) => P (OrdP p q) a where + type PP (OrdP p q) a = Ordering + eval _ opts a = do + let msg0 = "OrdP" + lr <- runPQ msg0 (Proxy @p) (Proxy @q) opts a + pure $ case lr of + Left e -> e + Right (p,q,pp,qq) -> + let d = compare p q + in mkNode opts (PresentT d) [msg0 <> " " <> show p <> " " <> prettyOrd d <> show0 opts " " q] [hh pp, hh qq] + +-- | compare two strings ignoring case +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Fst ===? Snd) ("abC","aBc") +-- Present EQ +-- PresentT EQ +-- +-- >>> pl @(Fst ===? Snd) ("abC","DaBc") +-- Present LT +-- PresentT LT +-- +data OrdI p q +type p ===? q = OrdI p q +infix 4 ===? + +instance (PP p a ~ String + , PP p a ~ PP q a + , P p a + , P q a + ) => P (OrdI p q) a where + type PP (OrdI p q) a = Ordering + eval _ opts a = do + let msg0 = "OrdI" + lr <- runPQ msg0 (Proxy @p) (Proxy @q) opts a + pure $ case lr of + Left e -> e + Right (p,q,pp,qq) -> + let d = on compare (map toLower) p q + in mkNode opts (PresentT d) [msg0 <> " " <> p <> " " <> prettyOrd d <> " " <> q] [hh pp, hh qq] + +data Cmp (o :: OrderingP) p q + +instance (GetOrd o + , Ord (PP p a) + , Show (PP p a) + , PP p a ~ PP q a + , P p a + , P q a + ) => P (Cmp o p q) a where + type PP (Cmp o p q) a = Bool + eval _ opts a = do + let (sfn, fn) = getOrd @o + lr <- runPQ sfn (Proxy @p) (Proxy @q) opts a + pure $ case lr of + Left e -> e + Right (p,q,pp,qq) -> + let b = fn p q + in mkNodeB opts b [show p <> " " <> sfn <> show0 opts " " q] [hh pp, hh qq] + +-- for strings +data CmpI (o :: OrderingP) p q + +instance (PP p a ~ String + , GetOrd o + , PP p a ~ PP q a + , P p a + , P q a + ) => P (CmpI o p q) a where + type PP (CmpI o p q) a = Bool + eval _ opts a = do + let (sfn, fn) = getOrd @o + lr <- runPQ sfn (Proxy @p) (Proxy @q) opts a + pure $ case lr of + Left e -> e + Right (p,q,pp,qq) -> + let b = on fn (map toLower) p q + in mkNodeB opts b ["CmpI " <> p <> " " <> sfn <> " " <> q] [hh pp, hh qq] + +type Gt n = Cmp 'Cgt I n +type Ge n = Cmp 'Cge I n +type Same n = Cmp 'Ceq I n +type Le n = Cmp 'Cle I n +type Lt n = Cmp 'Clt I n +type Ne n = Cmp 'Cne I n + +-- | similar to 'Control.Lens.itoList' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(IToList _) ("aBc" :: String) +-- Present [(0,'a'),(1,'B'),(2,'c')] +-- PresentT [(0,'a'),(1,'B'),(2,'c')] +-- +data IToList' t p +type IToList (t :: Type) = IToList' (Hole t) Id + +instance (Show x + , P p x + , Typeable (PP t (PP p x)) + , Show (PP t (PP p x)) + , FoldableWithIndex (PP t (PP p x)) f + , PP p x ~ f a + , Show a + ) => P (IToList' t p) x where + type PP (IToList' t p) x = [(PP t (PP p x), UnIToListT (PP p x))] + eval _ opts x = do + let msg0 = "IToList" + pp <- eval (Proxy @p) opts x + pure $ case getValueLR opts (msg0 <> " p failed") pp [] of + Left e -> e + Right p -> + let b = itoList p + t = showT @(PP t (PP p x)) + in mkNode opts (PresentT b) [msg0 <> "(" <> t <> ")" <> show0 opts " " b <> showA opts " | " x] [hh pp] + +type family UnIToListT fa where + UnIToListT (f a) = a + +-- | similar to 'toList' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @ToList "aBc" +-- Present "aBc" +-- PresentT "aBc" +-- +data ToList +instance (Show (t a) + , Foldable t + , Show a + ) => P ToList (t a) where + type PP ToList (t a) = [a] + eval _ opts as = + let z = toList as + in pure $ mkNode opts (PresentT z) ["ToList" <> show0 opts " " z <> showA opts " | " as] [] + +data ToListExt + +instance (Show l + , Ge.IsList l + , Show (Ge.Item l) + ) => P ToListExt l where + type PP ToListExt l = [Ge.Item l] + eval _ opts as = + let z = Ge.toList as + in pure $ mkNode opts (PresentT z) ["ToListExt" <> show0 opts " " z <> showA opts " | " as] [] + +data FromList (t :: Type) -- doesnt work with OverloadedLists unless you cast to [a] explicitly + +instance (a ~ Ge.Item t + , Show t + , Ge.IsList t + ) => P (FromList t) [a] where + type PP (FromList t) [a] = t + eval _ opts as = + let z = Ge.fromList (as :: [Ge.Item t]) :: t + in pure $ mkNode opts (PresentT z) ["FromList" <> show0 opts " " z] [] + +data FromListF (t :: Type) -- works only with overloadedlists +-- l ~ l' is key +instance (Show l + , Ge.IsList l + , l ~ l' + ) => P (FromListF l') l where + type PP (FromListF l') l = l' + eval _ opts as = + let z = Ge.fromList (Ge.toList @l as) + in pure $ mkNode opts (PresentT z) ["FromListF" <> show0 opts " " z] [] + +-- | predicate on 'These' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @IsThis (This "aBc") +-- True +-- TrueT +-- +-- >>> pl @IsThis (These 1 'a') +-- False +-- FalseT +-- +-- >>> pl @IsThese (These 1 'a') +-- True +-- TrueT +-- +data IsTh (th :: These x y) -- x y can be anything + +type IsThis = IsTh ('This '()) +type IsThat = IsTh ('That '()) +type IsThese = IsTh ('These '() '()) + +-- trying to avoid show instance cos of ambiguities +instance (Show a + , Show b + , GetThese th + ) => P (IsTh (th :: These x y)) (These a b) where + type PP (IsTh th) (These a b) = Bool + eval _ opts th = + let (t,f) = getThese (Proxy @th) + b = f th + in pure $ mkNodeB opts b ["IsTh '" <> t <> showA opts " | " th] [] + +-- | similar to 'these' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(TheseIn Id Len (Fst + Length Snd)) (This 13) +-- Present 13 +-- PresentT 13 +-- +-- >>> pl @(TheseIn Id Len (Fst + Length Snd)) (That "this is a long string") +-- Present 21 +-- PresentT 21 +-- +-- >>> pl @(TheseIn Id Len (Fst + Length Snd)) (These 20 "somedata") +-- Present 28 +-- PresentT 28 +-- +-- >>> pl @(TheseIn (Left _) (Right _) (If (Fst > Length Snd) (MkLeft _ Fst) (MkRight _ Snd))) (That "this is a long string") +-- Present Right "this is a long string" +-- PresentT (Right "this is a long string") +-- +-- >>> pl @(TheseIn (Left _) (Right _) (If (Fst > Length Snd) (MkLeft _ Fst) (MkRight _ Snd))) (These 1 "this is a long string") +-- Present Right "this is a long string" +-- PresentT (Right "this is a long string") +-- +-- >>> pl @(TheseIn (Left _) (Right _) (If (Fst > Length Snd) (MkLeft _ Fst) (MkRight _ Snd))) (These 100 "this is a long string") +-- Present Left 100 +-- PresentT (Left 100) +-- +data TheseIn p q r +type Theseid p q = TheseIn '(I, p) '(q, I) I + +instance (Show a + , Show b + , Show (PP p a) + , P p a + , P q b + , P r (a,b) + , PP p a ~ PP q b + , PP p a ~ PP r (a,b) + , PP q b ~ PP r (a,b) + ) => P (TheseIn p q r) (These a b) where + type PP (TheseIn p q r) (These a b) = PP p a + eval _ opts = + \case + This a -> do + let msg = "This" + pp <- eval (Proxy @p) opts a + pure $ case getValueLR opts (msg <> " p failed") pp [] of + Left e -> e + Right c -> mkNode opts (PresentT c) [msg <> show0 opts " " c <> showA opts " | This " a] [hh pp] + That b -> do + let msg = "That" + qq <- eval (Proxy @q) opts b + pure $ case getValueLR opts (msg <> " q failed") qq [] of + Left e -> e + Right c -> mkNode opts (PresentT c) [msg <> show0 opts " " c <> showA opts " | That " b] [hh qq] + These a b -> do + let msg = "TheseIn" + rr <- eval (Proxy @r) opts (a,b) + pure $ case getValueLR opts (msg <> " r failed") rr [] of + Left e -> e + Right c -> mkNode opts (PresentT c) [msg <> show0 opts " " c <> showA opts " | " (These a b)] [hh rr] + +-- | extracts the first character from a non empty 'Symbol' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Char1 "aBc") () +-- Present 'a' +-- PresentT 'a' +-- +data Char1 (s :: Symbol) -- gets the first char from the Symbol [requires that Symbol is not empty] +instance (KnownSymbol s, NullT s ~ 'False) => P (Char1 s) a where + type PP (Char1 s) a = Char + eval _ opts _ = + let c = head $ symb @s + in pure $ mkNode opts (PresentT c) ["Char1" <> show0 opts " " c] [] + +-- | similar to 'Data.Align.align' thats pads with 'Data.These.This' or 'Data.These.That' if one list is shorter than the other +-- +-- the key is that all information about both lists are preserved +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(ZipThese Fst Snd) ("aBc", [1..5]) +-- Present [These 'a' 1,These 'B' 2,These 'c' 3,That 4,That 5] +-- PresentT [These 'a' 1,These 'B' 2,These 'c' 3,That 4,That 5] +-- +-- >>> pl @(ZipThese Fst Snd) ("aBcDeF", [1..3]) +-- Present [These 'a' 1,These 'B' 2,These 'c' 3,This 'D',This 'e',This 'F'] +-- PresentT [These 'a' 1,These 'B' 2,These 'c' 3,This 'D',This 'e',This 'F'] +-- +data ZipThese p q + +instance (PP p a ~ [x] + , PP q a ~ [y] + , P p a + , P q a + , Show x + , Show y + ) => P (ZipThese p q) a where + type PP (ZipThese p q) a = [These (ArrT (PP p a)) (ArrT (PP q a))] + eval _ opts a = do + let msg0 = "ZipThese" + lr <- runPQ msg0 (Proxy @p) (Proxy @q) opts a + pure $ case lr of + Left e -> e + Right (p,q,pp,qq) -> + let d = TA.align p q + in mkNode opts (PresentT d) [msg0 <> show0 opts " " d <> showA opts " | p=" p <> showA opts " | q=" q] [hh pp, hh qq] + +data ZipTheseF p q + +instance (Show (f y) + , PP p a ~ f x + , PP q a ~ f y + , ExtractT (f x) ~ x + , ExtractT (f y) ~ y + , Show (f x) + , TA.Align f + , Show (f (These x y)) + , P p a + , P q a) + => P (ZipTheseF p q) a where + type PP (ZipTheseF p q) a = ApplyConstT (PP p a) (These (ExtractT (PP p a)) (ExtractT (PP q a))) + eval _ opts a = do + let msg0 = "ZipTheseF" + lr <- runPQ msg0 (Proxy @p) (Proxy @q) opts a + pure $ case lr of + Left e -> e + Right (p,q,pp,qq) -> + let d = TA.align p q + in mkNode opts (PresentT d) [msg0 <> show0 opts " " d <> showA opts " | p=" p <> showA opts " | q=" q] [hh pp, hh qq] + +type family ExtractT (ta :: Type) :: Type where + ExtractT (t a) = a + ExtractT ta = GL.TypeError ( + 'GL.Text "ExtractT: expected (t a) but found something else" + ':$$: 'GL.Text "t a = " + ':<>: 'GL.ShowType ta) + +-- todo: get ArrT error to fire if wrong Type + +-- | Zip two lists optionally cycling the one of the lists to match the size +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Ziplc Fst Snd) ("abc", [1..5]) +-- Present [('a',1),('b',2),('c',3),('a',4),('b',5)] +-- PresentT [('a',1),('b',2),('c',3),('a',4),('b',5)] +-- +-- >>> pl @(Ziplc Fst Snd) ("abcdefg", [1..5]) +-- Present [('a',1),('b',2),('c',3),('d',4),('e',5)] +-- PresentT [('a',1),('b',2),('c',3),('d',4),('e',5)] +-- +-- >>> pl @(Ziprc Fst Snd) ("abcdefg", [1..5]) +-- Present [('a',1),('b',2),('c',3),('d',4),('e',5),('f',1),('g',2)] +-- PresentT [('a',1),('b',2),('c',3),('d',4),('e',5),('f',1),('g',2)] +-- +data Zip (lc :: Bool) (rc :: Bool) p q +type Ziplc p q = Zip 'True 'False p q +type Ziprc p q = Zip 'False 'True p q +type Zipn p q = Zip 'False 'False p q + +instance (GetBool lc + , GetBool rc + , PP p a ~ [x] + , PP q a ~ [y] + , P p a + , P q a + , Show x + , Show y + ) => P (Zip lc rc p q) a where + type PP (Zip lc rc p q) a = [(ArrT (PP p a), ArrT (PP q a))] + eval _ opts a = do + let msg0 = "Zip" <> cyc + lc = getBool @lc + rc = getBool @rc + cyc = case (lc,rc) of + (True,False) -> "LC" + (False,True) -> "RC" + _ -> "" + lr <- runPQ msg0 (Proxy @p) (Proxy @q) opts a + pure $ case lr of + Left e -> e + Right (p,q,pp,qq) -> + let d = case (lc,rc) of + (True,False) -> zip (take (length q) (cycle p)) q + (False,True) -> zip p (take (length p) (cycle q)) + _ -> zip p q + in mkNode opts (PresentT d) [msg0 <> show0 opts " " d <> showA opts " | p=" p <> showA opts " | q=" q] [hh pp, hh qq] + +-- | Luhn predicate check on last digit +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @Luhn [1,2,3,0] +-- True +-- TrueT +-- +-- >>> pl @Luhn [1,2,3,4] +-- False +-- FalseT +data Luhn + + +instance a ~ [Int] => P Luhn a where + type PP Luhn a = Bool + eval _ opts as = + let xs = zipWith (*) (reverse as) (cycle [1,2]) + ys = map (\x -> if x>=10 then x-9 else x) xs + z = sum ys + ret = z `mod` 10 + msg = "Luhn" + in pure $ if ret == 0 then mkNode opts TrueT [msg <> show0 opts " | " as] [] + else mkNode opts FalseT [msg <> " map=" <> show ys <> " sum=" <> show z <> " ret=" <> show ret <> showA opts " | as=" as] [] + +pe0, pe, pe1, pe2, pex, pe3, pl, plc :: forall p a . (Show (PP p a), P p a) => a -> IO (BoolT (PP p a)) +pe0 = peWith @p o0 +pe = peWith @p o02 +pex = peWith @p o03 +pe1 = peWith @p o1 +pe2 = peWith @p o2 +pe3 = peWith @p o3 +pl = peWith @p ol +plc = peWith @p olc + +peWith :: forall p a . (Show (PP p a), P p a) => -- Typeable (Proxy p), + POpts -> a -> IO (BoolT (PP p a)) +peWith opts a = do + pp <- eval (Proxy @p) opts a + let r = pp ^. tBool + if oLite opts then + let f = colorMe opts (r ^. boolT2P) + in putStrLn $ case r of + FailT e -> f "Error" <> " " <> e + TrueT -> f "True" + FalseT -> f "False" + PresentT x -> f "Present" <> " " <> show x + else prtTree opts (fromTT pp) + return r + +-- could get n::Nat as a predicate but it is fine as is! +-- | Read a number base 2 via 36 +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(ReadBase Int 16) "00feD" +-- Present 4077 +-- PresentT 4077 +-- +-- >>> pl @(ReadBase Int 16) "-ff" +-- Present -255 +-- PresentT (-255) +-- +-- >>> pl @(ReadBase Int 2) "10010011" +-- Present 147 +-- PresentT 147 +-- +-- supports negative numbers unlike readInt +data ReadBase' t (n :: Nat) p +type ReadBase (t :: Type) (n :: Nat) = ReadBase' (Hole t) n Id +type ReadBaseInt (n :: Nat) = ReadBase' (Hole Int) n Id + + +instance (Typeable (PP t x) + , BetweenT 2 36 n + , Show (PP t x) + , Num (PP t x) + , KnownNat n + , PP p x ~ String + , P p x + ) => P (ReadBase' t n p) x where + type PP (ReadBase' t n p) x = PP t x + eval _ opts x = do + let n = nat @n + xs = getValidBase n + msg0 = "ReadBase(" <> t <> "," <> show n <> ")" + t = showT @(PP t x) + pp <- eval (Proxy @p) opts x + pure $ case getValueLR opts (msg0 <> " p failed") pp [] of + Left e -> e + Right p -> + let (ff,p1) = case p of + '-':q -> (negate,q) + _ -> (id,p) + in case readInt (fromIntegral n) + ((`elem` xs) . toLower) + (fromJust . (`elemIndex` xs) . toLower) + p1 of + [(b,"")] -> mkNode opts (PresentT (ff b)) [msg0 <> show0 opts " " (ff b) <> showA opts " | " p] [hh pp] + o -> mkNode opts (FailT ("invalid base " <> show n)) [msg0 <> " as=" <> p <> " err=" <> show o] [hh pp] + +getValidBase :: Int -> String +getValidBase n = take n (['0'..'9'] <> ['a'..'z']) + +-- | Display a number at base 2 to 36 +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(ShowBase 16) 4077 +-- Present "fed" +-- PresentT "fed" +-- +-- >>> pl @(ShowBase 16) (-255) +-- Present "-ff" +-- PresentT "-ff" +-- +-- >>> pl @(ShowBase 2) 147 +-- Present "10010011" +-- PresentT "10010011" +-- +-- supports negative numbers unlike showIntAtBase +data ShowBase (n :: Nat) + +instance (Show a + , 2 GL.<= n + , n GL.<= 36 + , KnownNat n + , Integral a + ) => P (ShowBase n) a where + type PP (ShowBase n) a = String + eval _ opts a = + let n = nat @n + xs = getValidBase n + msg = "ShowBase " <> show n + (ff,a') = if a < 0 then (('-':), abs a) else (id,a) + b = showIntAtBase (fromIntegral n) (xs !!) a' "" + in pure $ mkNode opts (PresentT (ff b)) [msg <> showLit0 opts " " (ff b) <> showA opts " | " a] [] + +type Assocl = '(I *** Fst, Snd >> Snd) +type Assocr = '(Fst >> Fst, Snd *** I) +--type Assocl = (I *** Fst) &&& (Snd >> Snd) +--type Assocr = (Fst >> Fst) &&& (Snd *** I) + +-- | Intercalate +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Intercalate '["aB"] '["xxxx","yz","z","www","xyz"]) () +-- Present ["xxxx","aB","yz","aB","z","aB","www","aB","xyz"] +-- PresentT ["xxxx","aB","yz","aB","z","aB","www","aB","xyz"] +-- +data Intercalate p q + +instance (PP p a ~ [x] + , PP q a ~ PP p a + , P p a + , P q a + , Show x + ) => P (Intercalate p q) a where + type PP (Intercalate p q) a = PP p a + eval _ opts a = do + let msg0 = "Intercalate" + lr <- runPQ msg0 (Proxy @p) (Proxy @q) opts a + pure $ case lr of + Left e -> e + Right (p,q,pp,qq) -> + let d = intercalate p (map (:[]) q) + in mkNode opts (PresentT d) [msg0 <> show0 opts " " d <> showA opts " | " p <> showA opts " | " q] [hh pp, hh qq] + +getStringPrefix :: String -> (String,String) +getStringPrefix = fix (\k z -> \case + [] -> (z,[]) + '%':x:xs | x == '%' -> k (z <> ['%']) xs + | otherwise -> (z,'%':x:xs) + x:xs -> k (z <> [x]) xs + ) [] + +-- | uses Printf to format output +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Printf "value=%03d" Id) 12 +-- Present "value=012" +-- PresentT "value=012" +-- +-- splits string into pieces before "%" that way we have a chance of catching any errors +data Printf s p + +instance (PrintfArg (PP p x) + , Show (PP p x) + , PP s x ~ String + , P s x + , P p x + ) => P (Printf s p) x where + type PP (Printf s p) x = String + eval _ opts x = do + let msg0 = "Printf" + lrx <- runPQ msg0 (Proxy @s) (Proxy @p) opts x + case lrx of + Left e -> pure e + Right (s,p,ss,pp) -> do + let msg1 = msg0 + lr <- catchitNF @_ @E.SomeException (printf s p) + pure $ case lr of + Left e -> mkNode opts (FailT (msg1 <> " (" <> e <> ")")) [msg1 <> show0 opts " " p <> " s=" <> s] [hh ss, hh pp] + Right ret -> mkNode opts (PresentT ret) [msg1 <> " [" <> showLit0 opts "" ret <> "]" <> showA opts " | p=" p <> showLit opts " | s=" s] [hh ss, hh pp] + +type family GuardsT (ps :: [k]) where + GuardsT '[] = '[] + GuardsT (p ': ps) = Guard' p ': GuardsT ps + +type Guards' (ps :: [k]) = Para (GuardsT ps) + +type ToPara (os :: [k]) = Proxy (ParaImplW 'True os) + +type ToGuards (prt :: k) (os :: [k1]) = Proxy (Guards (ToGuardsT prt os)) + +type family ToGuardsT (prt :: k) (os :: [k1]) :: [(k,k1)] where +-- ToGuardsT prt '[] = '[] -- error condition + ToGuardsT prt '[p] = '(prt,p) : '[] + ToGuardsT prt (p ': ps) = '(prt,p) ': ToGuardsT prt ps + +-- | runs values in parallel unlike 'Do' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Para '[Id,Id + 1,Id * 4]) [10,20,30] +-- Present [10,21,120] +-- PresentT [10,21,120] +-- +data ParaImpl (n :: Nat) (strict :: Bool) (os :: [k]) +type Para (os :: [k]) = ParaImplW 'True os +type ParaLax (os :: [k]) = ParaImplW 'False os + +data ParaImplW (strict :: Bool) (ps :: [k]) + +type family GuardsViaParaT prt ps where + GuardsViaParaT prt '[] = '[] + GuardsViaParaT prt (p ': ps) = Guard prt p ': GuardsViaParaT prt ps + +type GuardsViaPara prt ps = Para (GuardsViaParaT prt ps) + +-- passthru but adds the length of ps (replaces LenT in the type synonym to avoid type synonyms being expanded out +instance (GetBool strict, GetLen ps, P (ParaImpl (LenT ps) strict ps) [a]) => P (ParaImplW strict ps) [a] where + type PP (ParaImplW strict ps) [a] = PP (ParaImpl (LenT ps) strict ps) [a] + eval _ opts as = do + let strict = getBool @strict + msgbase0 = "Para" <> strictmsg @strict + n = getLen @ps + if strict && n /= length as then + let xx = msgbase0 <> ": data elements(" <> show (length as) <> ") /= predicates(" <> show n <> ")" + in pure $ mkNode opts (FailT xx) [xx] [] + else eval (Proxy @(ParaImpl (LenT ps) strict ps)) opts as + +-- only allow non empty lists! +instance GL.TypeError ('GL.Text "ParaImpl '[] invalid: requires at least one value in the list") + => P (ParaImpl n strict ('[] :: [k])) [a] where + type PP (ParaImpl n strict ('[] :: [k])) [a] = Void + eval _ _ _ = error "should not get this far" + +-- forall k (p :: k) (n :: Nat) (strict :: Bool) a . +instance (Show (PP p a) + , KnownNat n + , GetBool strict + , Show a + , P p a + ) => P (ParaImpl n strict '[p]) [a] where + type PP (ParaImpl n strict '[p]) [a] = [PP p a] + eval _ opts as' = do + let strict = getBool @strict + msgbase0 = "Para" <> strictmsg @strict + msgbase1 = msgbase0 <> "(" <> show n <> ")" + n :: Int + n = nat @n + case as' of + [] -> pure $ mkNode opts mempty [msgbase1 <> " (ran out of data!!)"] [] + a:as -> do + pp <- eval (Proxy @p) opts a + pure $ case getValueLR opts msgbase1 pp [] of + Left e -> e + -- showA opts " " [b] fails but using 'b' is ok and (b : []) also works! + -- Ge.List error + Right b -> mkNode opts (PresentT [b]) [msgbase1 <> (if null as then " done!" else " Truncated") <> show0 opts " " (b : []) <> showA opts " | " a <> (if strict then "" else showA opts " | leftovers=" as)] [hh pp] + +instance (KnownNat n + , GetBool strict + , GetLen ps + , P p a + , P (ParaImpl n strict (p1 ': ps)) [a] + , PP (ParaImpl n strict (p1 ': ps)) [a] ~ [PP p a] + , Show a + , Show (PP p a) + ) + => P (ParaImpl n strict (p ': p1 ': ps)) [a] where + type PP (ParaImpl n strict (p ': p1 ': ps)) [a] = [PP p a] + eval _ opts as' = do + let msgbase0 = msgbase2 <> "(" <> show (n-pos) <> " of " <> show n <> ")" + msgbase1 = msgbase2 <> "(" <> show (n-pos) <> ")" + msgbase2 = "Para" <> strictmsg @strict + n = nat @n + pos = 1 + getLen @ps -- cos p1! + case as' of + [] -> pure $ mkNode opts mempty [msgbase0 <> " (ran out of data!!)"] [] + a:as -> do + pp <- eval (Proxy @p) opts a + case getValueLR opts msgbase0 pp [] of + Left e -> pure e + Right b -> do + qq <- eval (Proxy @(ParaImpl n strict (p1 ': ps))) opts as + pure $ case getValueLRHide opts (msgbase1 <> " rhs failed " <> show b) qq [hh pp] of + Left e -> e + Right bs -> mkNode opts (PresentT (b:bs)) [msgbase1 <> show0 opts " " (b:bs) <> showA opts " | " as'] [hh pp, hh qq] + +-- | tries each predicate ps and on the first match runs the corresponding qs but if there is no match on ps then runs the fail case e +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Case (FailS "asdf" >> Snd >> Unproxy ) '[Lt 4,Lt 10,Same 50] '[Printf "%d is lt4" Id, Printf "%d is lt10" Id, Printf "%d is same50" Id] Id) 50 +-- Present "50 is same50" +-- PresentT "50 is same50" +-- +-- >>> pl @(Case (FailS "asdf" >> Snd >> Unproxy ) '[Lt 4,Lt 10,Same 50] '[Printf "%d is lt4" Id, Printf "%d is lt10" Id, Printf "%d is same50" Id] Id) 9 +-- Present "9 is lt10" +-- PresentT "9 is lt10" +-- +-- >>> pl @(Case (FailS "asdf" >> Snd >> Unproxy ) '[Lt 4,Lt 10,Same 50] '[Printf "%d is lt4" Id, Printf "%d is lt10" Id, Printf "%d is same50" Id] Id) 3 +-- Present "3 is lt4" +-- PresentT "3 is lt4" +-- +-- >>> pl @(Case (FailS "asdf" >> Snd >> Unproxy ) '[Lt 4,Lt 10,Same 50] '[Printf "%d is lt4" Id, Printf "%d is lt10" Id, Printf "%d is same50" Id] Id) 99 +-- Error asdf +-- FailT "asdf" +-- +data CaseImpl (n :: Nat) (e :: k0) (ps :: [k]) (qs :: [k1]) (r :: k2) +-- ps = conditions +-- qs = what to do [one to one +-- r = the value +-- e = otherwise -- leave til later +data Case (e :: k0) (ps :: [k]) (qs :: [k1]) (r :: k2) +type Case' (ps :: [k]) (qs :: [k1]) (r :: k2) = Case (Snd >> Failp "Case:no match") ps qs r +type Case'' s (ps :: [k]) (qs :: [k1]) (r :: k2) = Case (FailCase s) ps qs r -- eg s= Printf "%s" ShowP + +type FailCase p = Fail (Snd >> Unproxy) (Fst >> p) + + +-- passthru but adds the length of ps (replaces LenT in the type synonym to avoid type synonyms being expanded out +instance (FailIfT (NotT (LenT ps DE.== LenT qs)) + ('GL.Text "lengths are not the same " + ':<>: 'GL.ShowType (LenT ps) + ':<>: 'GL.Text " vs " + ':<>: 'GL.ShowType (LenT qs)) + , P (CaseImpl (LenT ps) e ps qs r) x + ) => P (Case e ps qs r) x where + type PP (Case e ps qs r) x = PP (CaseImpl (LenT ps) e ps qs r) x + eval _ = eval (Proxy @(CaseImpl (LenT ps) e ps qs r)) + +-- only allow non empty lists! +instance (GL.TypeError ('GL.Text "CaseImpl '[] invalid: lhs requires at least one value in the list")) + => P (CaseImpl n e ('[] :: [k]) (q ': qs) r) x where + type PP (CaseImpl n e ('[] :: [k]) (q ': qs) r) x = Void + eval _ _ _ = error "should not get this far" + +instance (GL.TypeError ('GL.Text "CaseImpl '[] invalid: rhs requires at least one value in the list")) + => P (CaseImpl n e (p ': ps) ('[] :: [k1]) r) x where + type PP (CaseImpl n e (p ': ps) ('[] :: [k1]) r) x = Void + eval _ _ _ = error "should not get this far" + +instance (GL.TypeError ('GL.Text "CaseImpl '[] invalid: lists are both empty")) + => P (CaseImpl n e ('[] :: [k]) ('[] :: [k1]) r) x where + type PP (CaseImpl n e ('[] :: [k]) ('[] :: [k1]) r) x = Void + eval _ _ _ = error "should not get this far" + +instance (P r x + , P q (PP r x) + , Show (PP q (PP r x)) + , P p (PP r x) + , PP p (PP r x) ~ Bool + , KnownNat n + , Show (PP r x) + , P e (PP r x, Proxy (PP q (PP r x))) + , PP e (PP r x, Proxy (PP q (PP r x))) ~ PP q (PP r x) + ) => P (CaseImpl n e '[p] '[q] r) x where + type PP (CaseImpl n e '[p] '[q] r) x = PP q (PP r x) + eval _ opts z = do + let msgbase0 = "Case" <> "(" <> show n <> ")" + n :: Int = nat @n + rr <- eval (Proxy @r) opts z + case getValueLR opts msgbase0 rr [] of + Left e -> pure e + Right a -> do + pp <- evalBool (Proxy @p) opts a + case getValueLR opts msgbase0 pp [hh rr] of + Left e -> pure e + Right True -> do + qq <- eval (Proxy @q) opts a + pure $ case getValueLR opts msgbase0 qq [hh rr, hh pp] of + Left e -> e + Right b -> mkNode opts (PresentT b) [msgbase0 <> show0 opts " " b <> showA opts " | " a] [hh rr, hh pp, hh qq] + Right False -> do + ee <- eval (Proxy @e) opts (a, Proxy @(PP q (PP r x))) + pure $ case getValueLR opts (msgbase0 <> " otherwise failed") ee [hh rr, hh pp] of + Left e -> e + Right b -> mkNode opts (PresentT b) [msgbase0 <> show0 opts " " b <> showA opts " | " a] [hh rr, hh pp, hh ee] + +instance (KnownNat n + , GetLen ps + , P r x + , P p (PP r x) + , P q (PP r x) + , PP p (PP r x) ~ Bool + , Show (PP q (PP r x)) + , Show (PP r x) + , P (CaseImpl n e (p1 ': ps) (q1 ': qs) r) x + , PP (CaseImpl n e (p1 ': ps) (q1 ': qs) r) x ~ PP q (PP r x) + ) + => P (CaseImpl n e (p ': p1 ': ps) (q ': q1 ': qs) r) x where + type PP (CaseImpl n e (p ': p1 ': ps) (q ': q1 ': qs) r) x = PP q (PP r x) + eval _ opts z = do + let msgbase0 = msgbase2 <> "(" <> show (n-pos) <> " of " <> show n <> ")" + msgbase1 = msgbase2 <> "(" <> show (n-pos) <> ")" + msgbase2 = "Case" + n = nat @n + pos = 1 + getLen @ps -- cos p1! + rr <- eval (Proxy @r) opts z + case getValueLR opts msgbase0 rr [] of + Left e -> pure e + Right a -> do + pp <- evalBool (Proxy @p) opts a + case getValueLR opts msgbase0 pp [hh rr] of + Left e -> pure e + Right True -> do + qq <- eval (Proxy @q) opts a + pure $ case getValueLR opts msgbase0 qq [hh rr] of + Left e -> e + Right b -> mkNode opts (PresentT b) [msgbase0 <> show0 opts " " b <> showA opts " | " a] [hh rr, hh pp, hh qq] + Right False -> do + ww <- eval (Proxy @(CaseImpl n e (p1 ': ps) (q1 ': qs) r)) opts z + pure $ case getValueLR opts (msgbase1 <> " failed rhs") ww [hh rr, hh pp] of + Left e -> e + Right b -> mkNode opts (PresentT b) [msgbase1 <> show0 opts " " b <> showA opts " | " a] [hh rr, hh pp, hh ww] + +-- | similar to 'sequenceA' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @Sequence [Just 10, Just 20, Just 30] +-- Present Just [10,20,30] +-- PresentT (Just [10,20,30]) +-- +-- >>> pl @Sequence [Just 10, Just 20, Just 30, Nothing, Just 40] +-- Present Nothing +-- PresentT Nothing +-- +data Sequence +type Traverse p q = Map p q >> Sequence + + +instance (Show (f (t a)) + , Show (t (f a)) + , Traversable t + , Applicative f + ) => P Sequence (t (f a)) where + type PP Sequence (t (f a)) = f (t a) + eval _ opts tfa = + let d = sequenceA tfa + in pure $ mkNode opts (PresentT d) ["Sequence" <> show0 opts " " d <> showA opts " | " tfa] [] + +data Hide p +type H = Hide +-- type H p = Hide p -- doesnt work with % -- unsaturated! + +instance P p a => P (Hide p) a where + type PP (Hide p) a = PP p a + eval _ opts a = do + tt <- eval (Proxy @(Msg "!" p)) opts a + pure $ tt & tForest .~ [] + +-- | similar to 'readFile' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(ReadFile ".ghci" >> 'Just Id >> Len >> Gt 0) () +-- True +-- TrueT +-- +-- >>> pl @(FileExists "xyzzy") () +-- False +-- FalseT +-- +data ReadFile p +type FileExists p = ReadFile p >> IsJust + +instance (PP p x ~ String, P p x) => P (ReadFile p) x where + type PP (ReadFile p) x = Maybe String + eval _ opts x = do + let msg0 = "ReadFile" + pp <- eval (Proxy @p) opts x + case getValueLR opts msg0 pp [] of + Left e -> pure e + Right p -> do + let msg1 = msg0 <> "[" <> p <> "]" + mb <- runIO $ do + b <- doesFileExist p + if b then Just <$> readFile p + else pure Nothing + pure $ case mb of + Nothing -> mkNode opts (FailT (msg1 <> " must run in IO")) [msg1 <> " must run in IO"] [] + Just Nothing -> mkNode opts (PresentT Nothing) [msg1 <> " does not exist"] [] + Just (Just b) -> mkNode opts (PresentT (Just b)) [msg1 <> " len=" <> show (length b) <> showLit0 opts " Just " b] [] + +-- | does the directory exists +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(DirExists ".") () +-- True +-- TrueT +-- +data ReadDir p +type DirExists p = ReadDir p >> IsJust + +instance (PP p x ~ String, P p x) => P (ReadDir p) x where + type PP (ReadDir p) x = Maybe [FilePath] + eval _ opts x = do + let msg0 = "ReadDir" + pp <- eval (Proxy @p) opts x + case getValueLR opts msg0 pp [] of + Left e -> pure e + Right p -> do + let msg1 = msg0 <> "[" <> p <> "]" + mb <- runIO $ do + b <- doesDirectoryExist p + if b then Just <$> listDirectory p + else pure Nothing + pure $ case mb of + Nothing -> mkNode opts (FailT (msg1 <> " must run in IO")) [msg1 <> " must run in IO"] [] + Just Nothing -> mkNode opts (PresentT Nothing) [msg1 <> " does not exist"] [] + Just (Just b) -> mkNode opts (PresentT (Just b)) [msg1 <> " len=" <> show (length b) <> show0 opts " Just " b] [] + +-- | does the directory exists +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(DirExists ".") () +-- True +-- TrueT +-- +data ReadEnv p + +instance (PP p x ~ String, P p x) => P (ReadEnv p) x where + type PP (ReadEnv p) x = Maybe String + eval _ opts x = do + let msg0 = "ReadEnv" + pp <- eval (Proxy @p) opts x + case getValueLR opts msg0 pp [] of + Left e -> pure e + Right p -> do + let msg1 = msg0 <> "[" <> p <> "]" + mb <- runIO $ lookupEnv p + pure $ case mb of + Nothing -> mkNode opts (FailT (msg1 <> " must run in IO")) [msg1 <> " must run in IO"] [] + Just Nothing -> mkNode opts (PresentT Nothing) [msg1 <> " does not exist"] [] + Just (Just v) -> mkNode opts (PresentT (Just v)) [msg1 <> showLit0 opts " " v] [] + +data ReadEnvAll + +instance P ReadEnvAll a where + type PP ReadEnvAll a = [(String,String)] + eval _ opts _ = do + let msg0 = "ReadEnvAll" + mb <- runIO $ getEnvironment + pure $ case mb of + Nothing -> mkNode opts (FailT (msg0 <> " must run in IO")) [msg0 <> " must run in IO"] [] + Just v -> mkNode opts (PresentT v) [msg0 <> " count=" <> show (length v)] [] + +data TimeU + +instance P TimeU a where + type PP TimeU a = UTCTime + eval _ opts _a = do + let msg = "TimeU" + mb <- runIO $ getCurrentTime + pure $ case mb of + Nothing -> mkNode opts (FailT (msg <> " must run in IO")) [msg <> " must run in IO"] [] + Just v -> mkNode opts (PresentT v) [msg <> show0 opts " " v] [] + +data TimeZ + +instance P TimeZ a where + type PP TimeZ a = ZonedTime + eval _ opts _a = do + let msg = "TimeZ" + mb <- runIO $ getZonedTime + pure $ case mb of + Nothing -> mkNode opts (FailT (msg <> " must run in IO")) [msg <> " must run in IO"] [] + Just v -> mkNode opts (PresentT v) [msg <> show0 opts " " v] [] + +data FHandle s = FStdout | FStderr | FOther s WFMode deriving Show + +class GetFHandle (x :: FHandle Symbol) where getFHandle :: FHandle String +instance GetFHandle 'FStdout where getFHandle = FStdout +instance GetFHandle 'FStderr where getFHandle = FStderr +instance (GetMode w, KnownSymbol s) => GetFHandle ('FOther s w) where getFHandle = FOther (symb @s) (getMode @w) + +data WFMode = WFAppend | WFWrite | WFWriteForce deriving (Show,Eq) + +class GetMode (x :: WFMode) where getMode :: WFMode +instance GetMode 'WFAppend where getMode = WFAppend +instance GetMode 'WFWriteForce where getMode = WFWriteForce +instance GetMode 'WFWrite where getMode = WFWrite + +data WritefileImpl (hh :: FHandle Symbol) p +type Appendfile (s :: Symbol) p = WritefileImpl ('FOther s 'WFAppend) p +type Writefile' (s :: Symbol) p = WritefileImpl ('FOther s 'WFWriteForce) p +type Writefile (s :: Symbol) p = WritefileImpl ('FOther s 'WFWrite) p +type Stdout p = WritefileImpl 'FStdout p +type Stderr p = WritefileImpl 'FStderr p + +instance (GetFHandle fh + , P p a + , PP p a ~ String + ) => P (WritefileImpl fh p) a where + type PP (WritefileImpl fh p) a = () + eval _ opts a = do + let fh = getFHandle @fh + msg = case fh of + FStdout -> "Stdout" + FStderr -> "Stderr" + FOther s w -> (<>("[" <> s <> "]")) $ case w of + WFAppend -> "Appendfile" + WFWrite -> "Writefile" + WFWriteForce -> "Writefile'" + pp <- eval (Proxy @p) opts a + case getValueLR opts msg pp [] of + Left e -> pure e + Right ss -> do + mb <- runIO $ do + case fh of + FStdout -> fmap (left show) $ E.try @E.SomeException $ hPutStr stdout ss + FStderr -> fmap (left show) $ E.try @E.SomeException $ hPutStr stderr ss + FOther s w -> do + b <- doesFileExist s + if b && w == WFWrite then pure $ Left $ "file [" <> s <> "] already exists" + else do + let md = case w of + WFAppend -> AppendMode + _ -> WriteMode + fmap (left show) $ E.try @E.SomeException $ withFile s md (flip hPutStr ss) + pure $ case mb of + Nothing -> mkNode opts (FailT (msg <> " must run in IO")) [msg <> " must run in IO"] [hh pp] + Just (Left e) -> mkNode opts (FailT e) [msg <> " " <> e] [hh pp] + Just (Right ()) -> mkNode opts (PresentT ()) [msg] [hh pp] + +data Stdin + +instance P Stdin a where + type PP Stdin a = String + eval _ opts _a = do + let msg = "Stdin" + mb <- runIO $ do + lr <- E.try $ hGetContents stdin + pure $ case lr of + Left (e :: E.SomeException) -> Left $ show e + Right ss -> Right ss + pure $ case mb of + Nothing -> mkNode opts (FailT (msg <> " must run in IO")) [msg <> " must run in IO"] [] + Just (Left e) -> mkNode opts (FailT e) [msg <> " " <> e] [] + Just (Right ss) -> mkNode opts (PresentT ss) [msg <> "[" <> showLit opts "" ss <> "]"] [] + +--type Just' = JustFail "expected Just" Id +type Nothing' = Guard "expected Nothing" IsNothing + +-- | 'isInfixOf' 'isPrefixOf' 'isSuffixOf' equivalents +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(IsInfixI "abc" "axAbCd") () +-- True +-- TrueT +-- +-- >>> pl @(IsPrefixI "abc" "aBcbCd") () +-- True +-- TrueT +-- +-- >>> pl @(IsPrefix "abc" "aBcbCd") () +-- False +-- FalseT +-- +-- prefix infix suffix for strings +data IsFixImpl (cmp :: Ordering) (ignore :: Bool) p q + +type IsPrefix p q = IsFixImpl 'LT 'False p q +type IsInfix p q = IsFixImpl 'EQ 'False p q +type IsSuffix p q = IsFixImpl 'GT 'False p q + +type IsPrefixI p q = IsFixImpl 'LT 'True p q +type IsInfixI p q = IsFixImpl 'EQ 'True p q +type IsSuffixI p q = IsFixImpl 'GT 'True p q + +instance (GetBool ignore + , P p a + , P q a + , PP p a ~ String + , PP q a ~ String + , GetOrdering cmp + ) => P (IsFixImpl cmp ignore p q) a where + type PP (IsFixImpl cmp ignore p q) a = Bool + eval _ opts a = do + let cmp = getOrdering @cmp + ignore = getBool @ignore + lwr = if ignore then map toLower else id + (ff,msg0) = case cmp of + LT -> (isPrefixOf, "IsPrefix") + EQ -> (isInfixOf, "IsInfix") + GT -> (isSuffixOf, "IsSuffix") + pp <- eval (Proxy @p) opts a + case getValueLR opts (msg0 <> " p failed") pp [] of + Left e -> pure e + Right s0 -> do + let msg1 = msg0 <> (if ignore then "I" else "") <> "(" <> s0 <> ")" + qq <- eval (Proxy @q) opts a + pure $ case getValueLR opts (msg1 <> " q failed") qq [hh pp] of + Left e -> e + Right s1 -> mkNodeB opts (on ff lwr s0 s1) [msg1 <> showLit0 opts " " s1] [hh pp, hh qq] + +-- | similar to 'SG.<>' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Fst <> Snd) ("abc","def") +-- Present "abcdef" +-- PresentT "abcdef" +-- +data p <> q +infixr 6 <> +type Sapa' (t :: Type) = Wrap t Fst <> Wrap t Snd +type Sapa = Fst <> Snd + +instance (Semigroup (PP p a) + , PP p a ~ PP q a + , P p a + , Show (PP q a) + ,P q a + ) => P (p <> q) a where + type PP (p <> q) a = PP p a + eval _ opts a = do + let msg0 = "<>" + lr <- runPQ msg0 (Proxy @p) (Proxy @q) opts a + pure $ case lr of + Left e -> e + Right (p,q,pp,qq) -> + let d = p <> q + in mkNode opts (PresentT d) [show p <> " <> " <> show q <> " = " <> show d] [hh pp, hh qq] + +runPQ :: (P p a, P q a, MonadEval m) + => String + -> Proxy p + -> Proxy q + -> POpts + -> a + -> m (Either (TT x) (PP p a, PP q a, TT (PP p a), TT (PP q a))) +runPQ msg0 proxyp proxyq opts a = do + pp <- eval proxyp opts a + case getValueLR opts msg0 pp [] of + Left e -> pure $ Left e + Right p -> do + qq <- eval proxyq opts a + pure $ case getValueLR opts msg0 qq [hh pp] of + Left e -> Left e + Right q -> Right (p, q, pp, qq) + + +-- have to reverse the inductive tuples cos cant figure out how to reverse generically +-- uses inductive tuples to replace variable args +class PrintC x where + prtC :: (PrintfArg a, PrintfType r) => String -> (a,x) -> r +instance PrintC () where + prtC s (a,()) = printf s a +instance (PrintfArg a, PrintC rs) => PrintC (a,rs) where + prtC s (a,rs) = prtC s rs a + +data TupleListImpl (strict :: Bool) (n :: Nat) +type TupleList (n :: Nat) = TupleListImpl 'True n +type TupleListLax (n :: Nat) = TupleListImpl 'False n + +instance (Show a + , KnownNat n + , GetBool strict + , TupleListD (ToN n) a + , Show (TupleListT (ToN n) a) + ) => P (TupleListImpl strict n) [a] where + type PP (TupleListImpl strict n) [a] = TupleListT (ToN n) a + eval _ opts as = do + let strict = getBool @strict + n :: Int = nat @n + msg = "TupleList" <> (if strict then "" else "Lax") <> "(" <> show n <> ")" + pure $ case tupleListD @(ToN n) @a strict as of + Left e -> mkNode opts (FailT (msg <> " " <> e)) [msg <> " " <> e] [] + Right ret -> mkNode opts (PresentT ret) [msg <> show0 opts " " ret <> showA opts " | " as] [] + +data ReverseTupleN + +instance (ReverseTupleC tp + , Show (ReverseTupleP tp) + , Show tp + ) => P ReverseTupleN tp where + type PP ReverseTupleN tp = ReverseTupleP tp + eval _ opts tp = + let ret = reverseTupleC tp + in pure $ mkNode opts (PresentT ret) ["ReverseTupleN" <> show0 opts " " ret <> showA opts " | " tp] [] + +-- | Printfn prints +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Printfn "%s %s" Id) ("123",("def",())) +-- Present "123 def" +-- PresentT "123 def" +-- +-- >>> pl @(Printfn "s=%s d=%03d" Id) ("ab",(123,())) +-- Present "s=ab d=123" +-- PresentT "s=ab d=123" +-- +data Printfn s p +type Printfnt (n :: Nat) s = Printfn s (TupleList n) +type PrintfntLax (n :: Nat) s = Printfn s (TupleListLax n) + +type Printf2 (s :: Symbol) = Printfn s '(Fst,'(Snd, '())) +-- Printf3/Printf3' expected format is (a, (b,c)) : we dont support (a,b,c) ever! +type Printf3 (s :: Symbol) = Printfn s '(Fst, '(Snd >> Fst, '(Snd >> Snd, '()))) +type Printf3' (s :: Symbol) = Printfn s (TupleI '[Fst, Snd >> Fst, Snd >> Snd]) + +instance (KnownNat (TupleLenT as) + , PrintC bs + , (b,bs) ~ ReverseTupleP (a,as) + , ReverseTupleC (a,as) + , Show a + , Show as + , PrintfArg b + , PP s x ~ String + , PP p x ~ (a,as) + , P s x + , P p x + , CheckT (PP p x) ~ 'True + ) => P (Printfn s p) x where + type PP (Printfn s p) x = String + eval _ opts x = do + let msg0 = "Printfn" + lrx <- runPQ msg0 (Proxy @s) (Proxy @p) opts x + case lrx of + Left e -> pure e + Right (s,(a,as),ss,pp) -> do + let len :: Int = 1 + nat @(TupleLenT as) + msg1 = msg0 <> "(" <> show len <> ")" + hhs = [hh ss, hh pp] + lr <- catchitNF @_ @E.SomeException (prtC @bs s (reverseTupleC (a,as))) + pure $ case lr of + Left e -> mkNode opts (FailT (msg1 <> "(" <> e <> ")")) [msg1 <> show0 opts " " a <> " s=" <> s] hhs + Right ret -> mkNode opts (PresentT ret) [msg1 <> " [" <> showLit0 opts "" ret <> "]" <> showA opts " | (a,as)=" (a,as) <> showLit0 opts " | s=" s] hhs + +type family CheckT (tp :: Type) :: Bool where + CheckT () = GL.TypeError ('GL.Text "Printfn: inductive tuple cannot be empty") + CheckT o = 'True + +type family ApplyConstT (ta :: Type) (b :: Type) :: Type where +--type family ApplyConstT ta b where -- less restrictive so allows ('Just Int) Bool through! + ApplyConstT (t a) b = t b + ApplyConstT ta b = GL.TypeError ( + 'GL.Text "ApplyConstT: (t a) b but found something else" + ':$$: 'GL.Text "t a = " + ':<>: 'GL.ShowType ta + ':$$: 'GL.Text "b = " + ':<>: 'GL.ShowType b) + +-- | similar to 'Control.Applicative.<$' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Fst <$ Snd) ("abc",Just 20) +-- Present Just "abc" +-- PresentT (Just "abc") +-- +data p <$ q +infixl 4 <$ + +instance (P p a + , P q a + , Show (PP p a) + , Functor t + , PP q a ~ t c + , ApplyConstT (PP q a) (PP p a) ~ t (PP p a) + ) => P (p <$ q) a where + type PP (p <$ q) a = ApplyConstT (PP q a) (PP p a) + eval _ opts a = do + let msg0 = "(<$)" + lr <- runPQ msg0 (Proxy @p) (Proxy @q) opts a + pure $ case lr of + Left e -> e + Right (p,q,pp,qq) -> + let d = p <$ q + in mkNode opts (PresentT d) [msg0 <> show0 opts " " p] [hh pp, hh qq] + +data p <* q +infixl 4 <* + +-- | similar to 'Control.Applicative.<*' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Fst <* Snd) (Just "abc",Just 20) +-- Present Just "abc" +-- PresentT (Just "abc") +-- +type p *> q = q <* p +infixl 4 *> + +instance (Show (t c) + , P p a + , P q a + , Show (t b) + , Applicative t + , t b ~ PP p a + , PP q a ~ t c + ) => P (p <* q) a where + type PP (p <* q) a = PP p a + eval _ opts a = do + let msg0 = "(<*)" + lr <- runPQ msg0 (Proxy @p) (Proxy @q) opts a + pure $ case lr of + Left e -> e + Right (p,q,pp,qq) -> + let d = p <* q + in mkNode opts (PresentT d) [msg0 <> show0 opts " " p <> showA opts " | p=" p <> showA opts " | q=" q] [hh pp, hh qq] + +-- | similar to 'Control.Applicative.<|>' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Fst <|> Snd) (Nothing,Just 20) +-- Present Just 20 +-- PresentT (Just 20) +-- +-- >>> pl @(Fst <|> Snd) (Just 10,Just 20) +-- Present Just 10 +-- PresentT (Just 10) +-- +-- >>> pl @(Fst <|> Snd) (Nothing,Nothing) +-- Present Nothing +-- PresentT Nothing +-- +data p <|> q +infixl 3 <|> + +instance (P p a + , P q a + , Show (t b) + , Alternative t + , t b ~ PP p a + , PP q a ~ t b + ) => P (p <|> q) a where + type PP (p <|> q) a = PP p a + eval _ opts a = do + let msg0 = "(<|>)" + lr <- runPQ msg0 (Proxy @p) (Proxy @q) opts a + pure $ case lr of + Left e -> e + Right (p,q,pp,qq) -> + let d = p <|> q + in mkNode opts (PresentT d) [msg0 <> show0 opts " " d <> showA opts " | p=" p <> showA opts " | q=" q] [hh pp, hh qq] + + +-- | similar to 'Control.Comonad.extract' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @Extract (Nothing,Just 20) +-- Present Just 20 +-- PresentT (Just 20) +-- +-- >>> pl @Extract (Identity 20) +-- Present 20 +-- PresentT 20 +-- +data Extract +instance (Show (t a) + , Show a + , Comonad t + ) => P Extract (t a) where + type PP Extract (t a) = a + eval _ opts ta = + let msg0 = "Extract" + d = extract ta + in pure $ mkNode opts (PresentT d) [msg0 <> show0 opts " " d <> showA opts " | " ta] [] + +-- | similar to 'Control.Comonad.duplicate' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @Duplicate (20,"abc") +-- Present (20,(20,"abc")) +-- PresentT (20,(20,"abc")) +-- +data Duplicate + +instance (Show (t a) + , Show (t (t a)) + , Comonad t + ) => P Duplicate (t a) where + type PP Duplicate (t a) = t (t a) + eval _ opts ta = + let msg0 = "Duplicate" + d = duplicate ta + in pure $ mkNode opts (PresentT d) [msg0 <> show0 opts " " d <> showA opts " | " ta] [] + +-- | similar to 'Control.Monad.join' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @Join (Just (Just 20)) +-- Present Just 20 +-- PresentT (Just 20) +-- +data Join + + +instance (Show (t (t a)) + , Show (t a) + , Monad t + ) => P Join (t (t a)) where + type PP Join (t (t a)) = t a + eval _ opts tta = + let msg0 = "Join" + d = join tta + in pure $ mkNode opts (PresentT d) [msg0 <> show0 opts " " d <> showA opts " | " tta] [] + +-- same as $ but shows 'a' and 'b' +data p $ q +infixl 0 $ + +type p & q = q $ p -- flips the args eg a & b & (,) = (b,a) +infixr 1 & + +instance (P p x + , P q x + , PP p x ~ (a -> b) + , FnT (PP p x) ~ b + , PP q x ~ a + , Show a + , Show b + ) => P (p $ q) x where + type PP (p $ q) x = FnT (PP p x) + eval _ opts x = do + let msg0 = "($)" + lr <- runPQ msg0 (Proxy @p) (Proxy @q) opts x + pure $ case lr of + Left e -> e + Right (p,q,pp,qq) -> + let d = p q + in mkNode opts (PresentT d) ["fn $ " <> show q <> " = " <> show d] [hh pp, hh qq] + +type family FnT ab :: Type where + FnT (a -> b) = b + FnT ab = GL.TypeError ( + 'GL.Text "FnT: expected Type -> Type but found a simple Type?" + ':$$: 'GL.Text "ab = " + ':<>: 'GL.ShowType ab) + +evalQuick :: forall p i . P p i => i -> Either String (PP p i) +evalQuick i = getValLRFromTT (runIdentity (eval (Proxy @p) o0 i)) + +-- | similar to 'T.strip' 'T.stripStart' 'T.stripEnd' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(Trim Snd) (20," abc " :: String) +-- Present "abc" +-- PresentT "abc" +-- +-- >>> import Data.Text (Text) +-- >>> pl @(Trim Snd) (20," abc " :: Text) +-- Present "abc" +-- PresentT "abc" +-- +-- >>> pl @(TrimStart Snd) (20," abc ") +-- Present "abc " +-- PresentT "abc " +-- +-- >>> pl @(TrimEnd Snd) (20," abc ") +-- Present " abc" +-- PresentT " abc" +-- +-- todo: make it work for 'Data.Text.Lens.IsText' +data Trim' (left :: Bool) (right :: Bool) p +type Trim p = Trim' 'True 'True p +type TrimStart p = Trim' 'True 'False p +type TrimEnd p = Trim' 'False 'True p + +instance (FailIfT (NotT (OrT l r)) + ('GL.Text "Trim': left and right cannot both be False") + , GetBool l + , GetBool r + , IsText (PP p x) + , P p x + ) => P (Trim' l r p) x where + type PP (Trim' l r p) x = PP p x + eval _ opts x = do + let msg0 = "Trim" ++ (if l && r then "" else if l then "Start" else "End") + l = getBool @l + r = getBool @r + pp <- eval (Proxy @p) opts x + pure $ case getValueLR opts msg0 pp [] of + Left e -> e + Right (view unpacked -> p) -> + let fl = if l then dropWhile isSpace else id + fr = if r then dropWhileEnd isSpace else id + b = (fl . fr) p + in mkNode opts (PresentT (b ^. packed)) [msg0 <> showLit0 opts "" b <> showLit opts " | " p] [hh pp] + +-- | similar to 'T.stripLeft' 'T.stripRight' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> pl @(StripLeft "xyz" Id) ("xyzHello" :: String) +-- Present Just "Hello" +-- PresentT (Just "Hello") +-- +-- >>> import Data.Text (Text) +-- >>> pl @(StripLeft "xyz" Id) ("xyzHello" :: Text) +-- Present Just "Hello" +-- PresentT (Just "Hello") +-- +-- >>> pl @(StripLeft "xyz" Id) "xywHello" +-- Present Nothing +-- PresentT Nothing +-- +-- >>> pl @(StripRight "xyz" Id) "Hello xyz" +-- Present Just "Hello " +-- PresentT (Just "Hello ") +-- +-- >>> pl @(StripRight "xyz" Id) "xyzHelloxyw" +-- Present Nothing +-- PresentT Nothing +-- +data StripLR (right :: Bool) p q +type StripRight p q = StripLR 'True p q +type StripLeft p q = StripLR 'False p q + +instance (GetBool r + , PP p x ~ String + , P p x + , IsText (PP q x) + , P q x + ) => P (StripLR r p q) x where + type PP (StripLR r p q) x = Maybe (PP q x) + eval _ opts x = do + let msg0 = "Strip" ++ (if r then "Right" else "Left") + r = getBool @r + lr <- runPQ msg0 (Proxy @p) (Proxy @q) opts x + pure $ case lr of + Left e -> e + Right (p,view unpacked -> q,pp,qq) -> + let b = if r then + let (before,after) = splitAt (length q - length p) q + in if after == p then Just before else Nothing + else + let (before,after) = splitAt (length p) q + in if before == p then Just after else Nothing + in mkNode opts (PresentT (fmap (view packed) b)) [msg0 <> show0 opts "" b <> showLit opts " | p=" p <> showLit opts " | q=" q] [hh pp, hh qq] + + +-- PP is k -> * -> * but we need [k]: need to unwrap into [k] +data RepeatP (n :: Nat) p +instance (P (RepeatT n p) x) => P (RepeatP n p) x where + type PP (RepeatP n p) x = PP (RepeatT n p) x + eval _ opts x = + eval (Proxy @(RepeatT n p)) opts x +
+ src/Refined.hs view
@@ -0,0 +1,427 @@+{-# OPTIONS -Wall #-} +{-# OPTIONS -Wcompat #-} +{-# OPTIONS -Wincomplete-record-updates #-} +{-# OPTIONS -Wincomplete-uni-patterns #-} +{-# OPTIONS -Wredundant-constraints #-} +{-# LANGUAGE TypeOperators #-} +{-# LANGUAGE UndecidableInstances #-} +{-# LANGUAGE FlexibleContexts #-} +{-# LANGUAGE AllowAmbiguousTypes #-} +{-# LANGUAGE FlexibleInstances #-} +{-# LANGUAGE MultiParamTypeClasses #-} +{-# LANGUAGE TypeApplications #-} +{-# LANGUAGE DataKinds #-} +{-# LANGUAGE GADTs #-} +{-# LANGUAGE TypeFamilies #-} +{-# LANGUAGE PolyKinds #-} +{-# LANGUAGE ScopedTypeVariables #-} +{-# LANGUAGE RankNTypes #-} +{-# LANGUAGE OverloadedStrings #-} +{-# LANGUAGE ConstraintKinds #-} +{-# LANGUAGE GeneralizedNewtypeDeriving #-} +{-# LANGUAGE DeriveGeneric #-} +{-# LANGUAGE NoStarIsType #-} +{-# LANGUAGE DerivingVia #-} +{-# LANGUAGE ViewPatterns #-} +{-# LANGUAGE TupleSections #-} +{-# LANGUAGE TemplateHaskell #-} +{-# LANGUAGE DeriveLift #-} +{-# LANGUAGE StandaloneDeriving #-} +{-# LANGUAGE LambdaCase #-} +-- | +-- Module : Refined +-- Description : Traditional refinement type with only one type +-- Copyright : (c) Grant Weyburne, 2019 +-- License : BSD-3 +-- Maintainer : gbwey9@gmail.com +-- +module Refined ( + Refined(unRefined) + , RefinedC + , arbRefined + , rapply + , rapply0 + , rapply1 + , convertRefinedT + , withRefinedT + , withRefinedTIO + , newRefined + , prtRefinedIO + , newRefinedT + , newRefinedTIO + , RefinedT(..) + , prtRefinedTIO + , prtRefinedT + , unRavelT + , unRavelTIO + , unRavelTI + , unsafeRefined + , unsafeRefined' + ) where +import Predicate +import UtilP +import Control.Lens hiding (strict,iall) +import Data.Proxy +import Control.Monad.Except +import Control.Monad.Writer (WriterT(..), runWriterT, MonadWriter, tell) +import Control.Monad.Cont +import Data.Aeson +import GHC.Generics (Generic) +import qualified Language.Haskell.TH.Syntax as TH +import System.Console.Pretty +import Test.QuickCheck +import qualified GHC.Read as GR +import qualified Text.ParserCombinators.ReadPrec as PCR +import qualified Text.Read.Lex as RL +import qualified Data.Binary as B +import Data.Binary (Binary) +-- | a simple refinement type that ensures the predicate \'p\' holds for the type \'a\' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> :m + Data.Time.Calendar.WeekDate +-- >>> prtRefinedIO @(Between 10 14) ol 13 +-- Right (Refined {unRefined = 13}) +-- +-- >>> prtRefinedIO @(Between 10 14) ol 99 +-- Left FalseP +-- +-- >>> :set -XTypeOperators +-- >>> prtRefinedIO @(Last >> Len == 4) ol ["one","two","three","four"] +-- Right (Refined {unRefined = ["one","two","three","four"]}) +-- +-- >>> prtRefinedIO @(Re "^\\d{1,3}\\.\\d{1,3}\\.\\d{1,3}\\.\\d{1,3}$" Id) ol "141.213.1.99" +-- Right (Refined {unRefined = "141.213.1.99"}) +-- +-- >>> prtRefinedIO @(Re "^\\d{1,3}\\.\\d{1,3}\\.\\d{1,3}\\.\\d{1,3}$" Id) ol "141.213.1" +-- Left FalseP +-- +-- >>> prtRefinedIO @(Map (ReadP Int) (Resplit "\\." Id) >> Guard (Printf "bad length: found %d" Len) (Len == 4) >> 'True) ol "141.213.1" +-- Left (FailP "bad length: found 3") +-- +-- >>> prtRefinedIO @(Map (ReadP Int) (Resplit "\\." Id) >> Guard (Printf "bad length: found %d" Len) (Len == 4) >> GuardsQuick (Printf2 "octet %d out of range %d") (RepeatT 4 (Between 0 255)) >> 'True) ol "141.213.1.444" +-- Left (FailP "octet 3 out of range 444") +-- +-- >>> prtRefinedIO @(Map (ReadP Int) (Resplit "\\." Id) >> Guard (Printf "bad length: found %d" Len) (Len == 4) >> GuardsQuick (Printf2 "octet %d out of range %d") (RepeatT 4 (Between 0 255)) >> 'True) ol "141.213.1x34.444" +-- Left (FailP "ReadP Int (1x34) failed") +-- +-- >>> prtRefinedIO @(Map ('[Id] >> ReadP Int) Id >> Luhn) ol "12344" +-- Right (Refined {unRefined = "12344"}) +-- +-- >>> prtRefinedIO @(Map ('[Id] >> ReadP Int) Id >> Luhn) ol "12340" +-- Left FalseP +-- +-- >>> prtRefinedIO @(Any Prime) ol [11,13,17,18] +-- Right (Refined {unRefined = [11,13,17,18]}) +-- +-- >>> prtRefinedIO @(All Prime) ol [11,13,17,18] +-- Left FalseP +-- +-- >>> prtRefinedIO @(Snd !! Fst >> Len > 5) ol (2,["abc","defghij","xyzxyazsfd"]) +-- Right (Refined {unRefined = (2,["abc","defghij","xyzxyazsfd"])}) +-- +-- >>> prtRefinedIO @(Snd !! Fst >> Len > 5) ol (27,["abc","defghij","xyzxyazsfd"]) +-- Left (FailP "(!!) index not found") +-- +-- >>> prtRefinedIO @(Snd !! Fst >> Len <= 5) ol (2,["abc","defghij","xyzxyazsfd"]) +-- Left FalseP +newtype Refined p a = Refined { unRefined :: a } deriving (Show, Eq, Generic, TH.Lift) + +-- | 'Read' instance for 'Refined' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> :set -XOverloadedStrings +-- >>> reads @(Refined (Between 0 255) Int) "Refined {unRefined = 254}" +-- [(Refined {unRefined = 254},"")] +-- +-- >>> reads @(Refined (Between 0 255) Int) "Refined {unRefined = 300}" +-- [] +-- + +instance (RefinedC p a, Read a) => Read (Refined p a) where + readPrec + = GR.parens + (PCR.prec + 11 + (do GR.expectP (RL.Ident "Refined") + GR.expectP (RL.Punc "{") + fld0 <- GR.readField + "unRefined" + (PCR.reset GR.readPrec) + GR.expectP (RL.Punc "}") + let (_,mr) = runIdentity $ newRefined @p ol fld0 + case mr of + Nothing -> fail "" + Just _r -> pure (Refined fld0) + )) + readList = GR.readListDefault + readListPrec = GR.readListPrecDefault + +-- | the constraints that 'Refined' must adhere to +type RefinedC p a = (PP p a ~ Bool, P p a) + +-- | 'ToJSON' instance for 'Refined' +instance ToJSON a => ToJSON (Refined p a) where + toJSON = toJSON . unRefined + +-- | 'FromJSON' instance for 'Refined' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> :set -XOverloadedStrings +-- >>> eitherDecode' @(Refined (Between 10 14) Int) "13" +-- Right (Refined {unRefined = 13}) +-- +-- >>> removeAnsiForDocTest $ eitherDecode' @(Refined (Between 10 14) Int) "16" +-- Error in $: Refined:FalseP +-- False True && False +-- | +-- +- True 16 >= 10 +-- | | +-- | +- P I +-- | | +-- | `- P '10 +-- | +-- `- False 16 <= 14 +-- | +-- +- P I +-- | +-- `- P '14 +-- <BLANKLINE> +-- +instance (RefinedC p a, FromJSON a) => FromJSON (Refined p a) where + parseJSON z = do + a <- parseJSON z + let ((bp,e),mr) = runIdentity $ newRefined @p o2 a + case mr of + Nothing -> fail $ "Refined:" ++ show bp ++ "\n" ++ e + Just r -> return r + +-- | 'Binary' instance for 'Refined' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> import Data.Time +-- >>> import Control.Arrow ((+++)) +-- >>> type K1 = Refined (ReadP Day >> 'True) String +-- >>> type K2 = Refined (ReadP Day >> Between (ReadP' Day "2019-03-30") (ReadP' Day "2019-06-01")) String +-- >>> type K3 = Refined (ReadP Day >> Between (ReadP' Day "2019-05-30") (ReadP' Day "2019-06-01")) String +-- >>> r = unsafeRefined' ol "2019-04-23" :: K1 +-- >>> removeAnsiForDocTest $ (view _3 +++ view _3) $ B.decodeOrFail @K1 (B.encode r) +-- Refined {unRefined = "2019-04-23"} +-- +-- >>> removeAnsiForDocTest $ (view _3 +++ view _3) $ B.decodeOrFail @K2 (B.encode r) +-- Refined {unRefined = "2019-04-23"} +-- +-- >>> removeAnsiForDocTest $ (view _3 +++ view _3) $ B.decodeOrFail @K3 (B.encode r) +-- Refined:FalseP +-- False >> False | 2019-04-23 +-- | +-- +- P ReadP Day (2019-04-23) 2019-04-23 | 2019-04-23 +-- | | +-- | `- P Id "2019-04-23" +-- | +-- `- False False && True +-- | +-- +- False 2019-04-23 >= 2019-05-30 +-- | | +-- | +- P I +-- | | +-- | `- P ReadP Day (2019-05-30) 2019-05-30 | 2019-05-30 +-- | | +-- | `- P '2019-05-30 +-- | +-- `- True 2019-04-23 <= 2019-06-01 +-- | +-- +- P I +-- | +-- `- P ReadP Day (2019-06-01) 2019-06-01 | 2019-06-01 +-- | +-- `- P '2019-06-01 +-- <BLANKLINE> +-- +instance (RefinedC p a, Binary a) => Binary (Refined p a) where + get = do + fld0 <- B.get @a + let ((bp,e),mr) = runIdentity $ newRefined @p o2 fld0 + case mr of + Nothing -> fail $ "Refined:" ++ show bp ++ "\n" ++ e + Just r -> return r + put (Refined r) = B.put @a r + +-- | 'arbitrary' value for 'Refined' +arbRefined :: forall p a. + ( Arbitrary a + , RefinedC p a + ) => POpts -> Gen (Refined p a) +arbRefined opts = suchThatMap (arbitrary @a) (snd . runIdentity . newRefined @p opts) + +-- | binary operation applied to two 'RefinedT' values +rapply :: forall m p a . (RefinedC p a, Monad m) + => POpts + -> (a -> a -> a) + -> RefinedT m (Refined p a) + -> RefinedT m (Refined p a) + -> RefinedT m (Refined p a) +rapply opts f ma mb = do + tell [bgColor Blue "=== a ==="] + Refined x <- ma + tell [bgColor Blue "=== b ==="] + Refined y <- mb + tell [bgColor Blue "=== a `op` b ==="] + newRefinedT opts (f x y) + +-- | takes two values and lifts them into 'RefinedT' and then applies the binary operation +rapply0 :: forall p a m . (RefinedC p a, Monad m) + => POpts + -> (a -> a -> a) + -> a + -> a + -> RefinedT m (Refined p a) +rapply0 opts f a b = rapply opts f (newRefinedT opts a) (newRefinedT opts b) + +-- | same as 'rapply' except we already have valid 'Refined' values as input +rapply1 :: forall m p a . (RefinedC p a, Monad m) + => POpts + -> (a -> a -> a) + -> Refined p a + -> Refined p a + -> RefinedT m (Refined p a) +rapply1 opts f (Refined a) (Refined b) = newRefinedT opts (f a b) + +-- | attempts to lift a refinement type to another refinement type by way of transformation function +-- you can control both the predicate and the type +convertRefinedT :: forall m p p1 a a1 + . ( RefinedC p1 a1 + , Monad m) + => POpts + -> (a -> a1) + -> RefinedT m (Refined p a) + -> RefinedT m (Refined p1 a1) +convertRefinedT opts f ma = do + Refined a <- ma -- you already got a refined in there so no need to check RefinedC + newRefinedT @p1 opts (f a) + +-- | invokes the callback with the 'Refined' value if \'a\' is valid for the predicate \'p\' +withRefinedT :: forall p m a b + . (Monad m, RefinedC p a) + => POpts + -> a + -> (Refined p a -> RefinedT m b) + -> RefinedT m b +withRefinedT opts a k = newRefinedT @p opts a >>= k + +withRefinedTIO :: forall p m a b + . (MonadIO m, RefinedC p a) + => POpts + -> a + -> (Refined p a -> RefinedT m b) + -> RefinedT m b +withRefinedTIO opts a k = newRefinedTIO @p opts a >>= k + +-- | same as 'newRefined' but prints the results +prtRefinedIO :: forall p a + . RefinedC p a + => POpts + -> a + -> IO (Either BoolP (Refined p a)) +prtRefinedIO opts a = do + tt <- evalBool (Proxy @p) opts a + let msg = (_tBool tt ^. boolT2P, prtTreePure opts (fromTT tt)) + unless (oLite opts) $ putStrLn $ snd msg + pure $ case getValueLR opts "" tt [] of + Right True -> Right (Refined a) + _ -> Left (fst msg) + +-- | returns a 'Refined' value if \'a\' is valid for the predicate \'p\' +newRefined :: forall p a m . (MonadEval m, RefinedC p a) + => POpts + -> a + -> m ((BoolP, String), Maybe (Refined p a)) +newRefined opts a = do + tt <- evalBool (Proxy @p) opts a + let msg = (_tBool tt ^. boolT2P, prtTreePure opts (fromTT tt)) + pure $ (msg,) $ case getValueLR opts "" tt [] of + Right True -> Just (Refined a) + _ -> Nothing + +newRefinedTImpl :: forall p a n m . (RefinedC p a, Monad m, MonadEval n) + => (forall x . n x -> RefinedT m x) + -> POpts + -> a + -> RefinedT m (Refined p a) +newRefinedTImpl f opts a = do + tt <- f $ evalBool (Proxy @p) opts a + let msg = prtTreePure opts (fromTT tt) + tell [msg] + case getValueLR opts "" tt [] of + Right True -> return (Refined a) -- FalseP is also a failure! + _ -> let rc = show (_tBool tt ^. boolT2P) + in throwError rc -- RefinedT $ ExceptT $ WriterT $ return (Left rc, []) + +-- | returns a wrapper 'RefinedT' around a possible 'Refined' value if \'a\' is valid for the predicate \'p\' +newRefinedT :: forall p a m + . ( RefinedC p a + , Monad m) + => POpts + -> a + -> RefinedT m (Refined p a) +newRefinedT = newRefinedTImpl (return . runIdentity) + +newRefinedTIO :: forall p a m + . ( RefinedC p a + , MonadIO m) + => POpts + -> a + -> RefinedT m (Refined p a) +newRefinedTIO = newRefinedTImpl liftIO + +newtype RefinedT m a = RefinedT { unRefinedT :: ExceptT String (WriterT [String] m) a } + deriving (Functor, Applicative, Monad, MonadCont, MonadWriter [String], Show, MonadIO) + deriving MonadTrans via RefinedT + +instance Monad m => MonadError String (RefinedT m) where + throwError e = RefinedT $ ExceptT $ WriterT $ return (Left e,[]) + catchError (RefinedT (ExceptT (WriterT ma))) ema = + RefinedT $ ExceptT $ WriterT $ do + (lr,ss) <- ma + case lr of + Left e -> unRavelT (tell ss >> ema e) -- keep the old messages?? + Right _ -> ma + +unRavelT :: RefinedT m a -> m (Either String a, [String]) +unRavelT = runWriterT . runExceptT . unRefinedT + +unRavelTIO :: RefinedT IO a -> IO (Either String a, [String]) +unRavelTIO = runWriterT . runExceptT . unRefinedT + +unRavelTI :: RefinedT Identity a -> (Either String a, [String]) +unRavelTI = runIdentity . runWriterT . runExceptT . unRefinedT + +prtRefinedTImpl :: forall n m a . (MonadIO n, Show a) => (forall x . m x -> n x) -> RefinedT m a -> n () +prtRefinedTImpl f rt = do + (lr,ws) <- f $ unRavelT rt + liftIO $ do + forM_ (zip [1::Int ..] ws) $ \(_,y) -> putStrLn y + case lr of + Left e -> putStrLn $ "failure msg[" <> e <> "]" + Right a -> print a + +prtRefinedTIO :: (MonadIO m, Show a) => RefinedT m a -> m () +prtRefinedTIO = prtRefinedTImpl id + +prtRefinedT :: (MonadIO m, Show a) => RefinedT Identity a -> m () +prtRefinedT = prtRefinedTImpl (return . runIdentity) + +-- | a way to unsafely create a 'Refined' value +unsafeRefined :: forall p a . a -> Refined p a +unsafeRefined = Refined + +-- | a way to unsafely create a 'Refined' value but run the predicate +unsafeRefined' :: forall p a . RefinedC p a => POpts -> a -> Refined p a +unsafeRefined' opts a = + let tt = runIdentity $ evalBool (Proxy @p) opts a + in case getValueLR opts "" tt [] of + Right True -> Refined a + _ -> error $ prtTreePure opts (fromTT tt)
+ src/Refined3.hs view
@@ -0,0 +1,742 @@+{-# OPTIONS -Wall #-} +{-# OPTIONS -Wcompat #-} +{-# OPTIONS -Wincomplete-record-updates #-} +{-# OPTIONS -Wincomplete-uni-patterns #-} +{-# OPTIONS -Wredundant-constraints #-} +{-# LANGUAGE TypeOperators #-} +{-# LANGUAGE UndecidableInstances #-} +{-# LANGUAGE FlexibleContexts #-} +{-# LANGUAGE AllowAmbiguousTypes #-} +{-# LANGUAGE FlexibleInstances #-} +{-# LANGUAGE MultiParamTypeClasses #-} +{-# LANGUAGE TypeApplications #-} +{-# LANGUAGE DataKinds #-} +{-# LANGUAGE GADTs #-} +{-# LANGUAGE TypeFamilies #-} +{-# LANGUAGE PolyKinds #-} +{-# LANGUAGE ScopedTypeVariables #-} +{-# LANGUAGE LambdaCase #-} +{-# LANGUAGE RankNTypes #-} +{-# LANGUAGE OverloadedStrings #-} +{-# LANGUAGE ConstraintKinds #-} +{-# LANGUAGE StandaloneDeriving #-} +{-# LANGUAGE ViewPatterns #-} +{-# LANGUAGE NoStarIsType #-} +{-# LANGUAGE TemplateHaskell #-} +{-# LANGUAGE DeriveLift #-} +{- | +Module : Refined3 +Description : Refinement type allowing the external type to differ from the internal type +Copyright : (c) Grant Weyburne, 2019 +License : BSD-3 +Maintainer : gbwey9@gmail.com + +see 'Refined3' +contains Json and Read instances and arbitrary functions +-} +module Refined3 ( + Refined3(r3In,r3Out) + , Refined3C + , mkProxy3 + , mkProxy3P + , MkProxy3T + , withRefined3TIO + , withRefined3T + , withRefined3TP + , newRefined3T + , newRefined3TP + , newRefined3TPIO + , convertRefined3T + , convertRefined3TP + , rapply3 + , rapply3P + , prtEval3P + , prtEval3PIO + , prtEval3 + , eval3P + , eval3 + , eval3M + , eval3PX + , eval3X + , prt3IO + , prt3 + , arbRefined3 + , arbRefined3With + , Msg3 (..) + , prt3Impl + , MakeR3 + , Results (..) + , RResults (..) + , unsafeRefined3 + , unsafeRefined3' + ) where +import Refined +import Predicate +import UtilP +import Control.Lens hiding (strict,iall) +import Data.Tree +import Data.Proxy +import Control.Monad.Except +import Control.Monad.Writer (tell) +import Data.Aeson +import qualified Language.Haskell.TH.Syntax as TH +import System.Console.Pretty +import Test.QuickCheck +import qualified GHC.Read as GR +import qualified Text.ParserCombinators.ReadPrec as PCR +import qualified Text.Read.Lex as RL +import qualified Data.Binary as B +import Data.Binary (Binary) +-- | Refinement type that differentiates the input type from output type +-- +-- @ +-- \'i\' is the input type +-- \'ip\' converts i to PP ip i which is the internal type +-- \'op\' validates that internal type using PP op (PP ip i) ~ Bool +-- \'fmt\' outputs the internal type PP fmt (PP ip i) ~ i +-- PP fmt (PP ip i) should be valid input to Refined3 +-- @ +-- +-- If we fix the input type to String then it looks similar to: +-- 1. read into an internal type +-- 2. validate internal type with a predicate function +-- 3. show/format the internal type +-- +-- Although the most common scenario is String as input, you are free to choose any input type you like +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> :set -XTypeOperators +-- >>> :m + Data.Time.Calendar.WeekDate +-- >>> prtEval3 @(ReadBase Int 16) @(Lt 255) @(Printf "%x" Id) ol "00fe" +-- Right (Refined3 {r3In = 254, r3Out = "fe"}) +-- +-- >>> prtEval3 @(ReadBase Int 16) @(Lt 253) @(Printf "%x" Id) ol "00fe" +-- Left Step 2. False Boolean Check(op) | FalseP +-- +-- >>> prtEval3 @(ReadBase Int 16) @(Lt 255) @(Printf "%x" Id) ol "00fg" +-- Left Step 1. Initial Conversion(ip) Failed | invalid base 16 +-- +-- >>> prtEval3 @(Map (ReadP Int) (Resplit "\\." Id)) @(Guard (Printf "found length=%d" Len) (Len >> Id == 4) >> 'True) @(Printfnt 4 "%03d.%03d.%03d.%03d") ol "198.162.3.1.5" +-- Left Step 2. Failed Boolean Check(op) | found length=5 +-- +-- >>> prtEval3 @(Map (ReadP Int) (Resplit "\\." Id)) @(Guard (Printf "found length=%d" Len) (Len >> Id == 4) >> 'True) @(Printfnt 4 "%03d.%03d.%03d.%03d") ol "198.162.3.1" +-- Right (Refined3 {r3In = [198,162,3,1], r3Out = "198.162.003.001"}) +-- +-- >>> prtEval3 @(MkDay Fst (Snd >> Fst) (Snd >> Snd) >> 'Just Id) @(Guard "expected a Sunday" (Snd >> Snd == 7) >> 'True) @(Fst >> UnMkDay) ol (2019,(10,13)) +-- Right (Refined3 {r3In = (2019-10-13,(41,7)), r3Out = (2019,(10,13))}) +-- +-- >>> prtEval3 @(MkDay Fst (Snd >> Fst) (Snd >> Snd) >> 'Just Id) @(Guard "expected a Sunday" (Snd >> Snd == 7) >> 'True) @(Fst >> UnMkDay) ol (2019,(10,12)) +-- Left Step 2. Failed Boolean Check(op) | expected a Sunday +-- +-- >>> type T4 k = '(MkDay Fst (Snd >> Fst) (Snd >> Snd) >> 'Just Id, Guard "expected a Sunday" (Snd >> Snd == 7) >> 'True, Fst >> UnMkDay, k) +-- >>> prtEval3P (Proxy @(T4 _)) ol (2019,(10,12)) +-- Left Step 2. Failed Boolean Check(op) | expected a Sunday +-- +-- >>> prtEval3P (Proxy @(T4 _)) ol (2019,(10,13)) +-- Right (Refined3 {r3In = (2019-10-13,(41,7)), r3Out = (2019,(10,13))}) +-- +data Refined3 ip op fmt i = Refined3 { r3In :: PP ip i, r3Out :: PP fmt (PP ip i) } + +-- | directly load values into 'Refined3'. It still checks to see that those values are valid +unsafeRefined3' :: forall ip op fmt i + . (Show i, Show (PP ip i), Refined3C ip op fmt i) + => POpts + -> i + -> Refined3 ip op fmt i +unsafeRefined3' opts i = + let (ret,mr) = eval3 @ip @op @fmt opts i + in case mr of + Nothing -> error $ show (prt3Impl opts ret) + Just r -> r +-- | directly load values into 'Refined3' without any checking +unsafeRefined3 :: forall ip op fmt i . PP ip i -> PP fmt (PP ip i) -> Refined3 ip op fmt i +unsafeRefined3 = Refined3 + + +-- | Provides the constraints on Refined3 +type Refined3C ip op fmt i = + ( P ip i + , P op (PP ip i) + , PP op (PP ip i) ~ Bool -- the internal value needs to pass the predicate check + , P fmt (PP ip i) + , PP fmt (PP ip i) ~ i -- the output type must match the original input type + ) + +deriving instance (Show i, Show (PP ip i), Show (PP fmt (PP ip i))) => Show (Refined3 ip op fmt i) +deriving instance (Eq i, Eq (PP ip i), Eq (PP fmt (PP ip i))) => Eq (Refined3 ip op fmt i) +deriving instance (TH.Lift (PP ip i), TH.Lift (PP fmt (PP ip i))) => TH.Lift (Refined3 ip op fmt i) + +-- read instance from -ddump-deriv +-- | 'Read' instance for 'Refined3' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> :set -XOverloadedStrings +-- >>> reads @(Refined3 (ReadBase Int 16) (Between 0 255) (ShowBase 16) String) "Refined3 {r3In = 254, r3Out = \"fe\"}" +-- [(Refined3 {r3In = 254, r3Out = "fe"},"")] +-- +-- >>> reads @(Refined3 (ReadBase Int 16) (Between 0 255) (ShowBase 16) String) "Refined3 {r3In = 300, r3Out = \"12c\"}" +-- [] +-- +-- >>> reads @(Refined3 (ReadBase Int 16) (Id < 0) (ShowBase 16) String) "Refined3 {r3In = -1234, r3Out = \"-4d2\"}" +-- [(Refined3 {r3In = -1234, r3Out = "-4d2"},"")] +-- +-- >>> reads @(Refined3 (Map (ReadP Int) (Resplit "\\." Id)) (Guard "len/=4" (Len == 4) >> 'True) (Printfnt 4 "%d.%d.%d.%d") String) "Refined3 {r3In = [192,168,0,1], r3Out = \"192.168.0.1\"}" +-- [(Refined3 {r3In = [192,168,0,1], r3Out = "192.168.0.1"},"")] +-- +instance (Eq i, Show i, Show (PP ip i), Refined3C ip op fmt i, Read (PP ip i), Read (PP fmt (PP ip i))) => Read (Refined3 ip op fmt i) where + readPrec + = GR.parens + (PCR.prec + 11 + (do GR.expectP (RL.Ident "Refined3") + GR.expectP (RL.Punc "{") + fld1 <- GR.readField + "r3In" (PCR.reset GR.readPrec) + GR.expectP (RL.Punc ",") + fld2 <- GR.readField + "r3Out" (PCR.reset GR.readPrec) + GR.expectP (RL.Punc "}") + + let (_ret,mr) = runIdentity $ eval3MSkip @_ @ip @op @fmt o2 fld1 + case mr of + Nothing -> fail "" -- show (prt3Impl o2 _ret) + Just (Refined3 _r1 r2) | r2 == fld2 -> pure (Refined3 fld1 fld2) + | otherwise -> fail "" -- "mismatch on r3Out fmt: found (" ++ show fld2 ++ ") but expected(" ++ show r2 ++ ")" + )) + readList = GR.readListDefault + readListPrec = GR.readListPrecDefault + +-- | 'ToJSON' instance for 'Refined3' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> :set -XOverloadedStrings +-- >>> encode (unsafeRefined3 @(ReadBase Int 16) @(Between 0 255) @(ShowBase 16) 254 "fe") +-- "\"fe\"" +-- +-- >>> encode (unsafeRefined3 @Id @'True @Id 123 123) +-- "123" +-- +instance ToJSON (PP fmt (PP ip i)) => ToJSON (Refined3 ip op fmt i) where + toJSON = toJSON . r3Out + + +-- | 'FromJSON' instance for 'Refined3' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> :set -XOverloadedStrings +-- >>> eitherDecode' @(Refined3 (ReadBase Int 16) (Id > 10 && Id < 256) (ShowBase 16) String) "\"00fe\"" +-- Right (Refined3 {r3In = 254, r3Out = "fe"}) +-- +-- >>> removeAnsiForDocTest $ eitherDecode' @(Refined3 (ReadBase Int 16) (Id > 10 && Id < 256) (ShowBase 16) String) "\"00fe443a\"" +-- Error in $: Refined3:Step 2. False Boolean Check(op) | FalseP +-- <BLANKLINE> +-- *** Step 1. Success Initial Conversion(ip) [16663610] *** +-- <BLANKLINE> +-- P ReadBase(Int,16) 16663610 | "00fe443a" +-- | +-- `- P Id "00fe443a" +-- <BLANKLINE> +-- *** Step 2. False Boolean Check(op) *** +-- <BLANKLINE> +-- False True && False +-- | +-- +- True 16663610 > 10 +-- | | +-- | +- P Id 16663610 +-- | | +-- | `- P '10 +-- | +-- `- False 16663610 < 256 +-- | +-- +- P Id 16663610 +-- | +-- `- P '256 +-- <BLANKLINE> +-- +instance (Show (PP fmt (PP ip i)), Show (PP ip i), Refined3C ip op fmt i, FromJSON i) => FromJSON (Refined3 ip op fmt i) where + parseJSON z = do + i <- parseJSON @i z + let (ret,mr) = eval3 @ip @op @fmt o2 i + case mr of + Nothing -> fail $ "Refined3:" ++ show (prt3Impl o2 ret) + Just r -> return r + +{- +instance (Arbitrary (PP ip i) + , Show (PP ip i) + , Show i + , Refined3C ip op fmt i + ) => Arbitrary (Refined3 ip op fmt i) where + arbitrary = suchThatMap (arbitrary @(PP ip i)) $ eval3MQuickIdentity @ip @op @fmt o2 +-} +arbRefined3 :: forall ip op fmt i . + ( Arbitrary (PP ip i) + , Refined3C ip op fmt i + ) => Proxy '(ip,op,fmt,i) -> POpts -> Gen (Refined3 ip op fmt i) +arbRefined3 _ = suchThatMap (arbitrary @(PP ip i)) . eval3MQuickIdentity @ip @op @fmt + +-- help things along a little +arbRefined3With :: + forall ip op fmt i + . (Arbitrary (PP ip i) + , Refined3C ip op fmt i) + => Proxy '(ip,op,fmt,i) + -> POpts + -> (PP ip i -> PP ip i) + -> Gen (Refined3 ip op fmt i) +arbRefined3With _ opts f = + suchThatMap (f <$> arbitrary @(PP ip i)) $ eval3MQuickIdentity @ip @op @fmt opts + +-- | 'Binary' instance for 'Refined3' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> import Control.Arrow ((+++)) +-- >>> import Data.Time +-- >>> type K1 = MakeR3 '(ReadP Day, 'True, ShowP, String) +-- >>> type K2 = MakeR3 '(ReadP Day, Between (ReadP' Day "2019-03-30") (ReadP' Day "2019-06-01"), ShowP, String) +-- >>> type K3 = MakeR3 '(ReadP Day, Between (ReadP' Day "2019-05-30") (ReadP' Day "2019-06-01"), ShowP, String) +-- >>> r = unsafeRefined3' ol "2019-04-23" :: K1 +-- >>> removeAnsiForDocTest $ (view _3 +++ view _3) $ B.decodeOrFail @K1 (B.encode r) +-- Refined3 {r3In = 2019-04-23, r3Out = "2019-04-23"} +-- +-- >>> removeAnsiForDocTest $ (view _3 +++ view _3) $ B.decodeOrFail @K2 (B.encode r) +-- Refined3 {r3In = 2019-04-23, r3Out = "2019-04-23"} +-- +-- >>> removeAnsiForDocTest $ (view _3 +++ view _3) $ B.decodeOrFail @K3 (B.encode r) +-- Refined3:Step 2. False Boolean Check(op) | FalseP +-- <BLANKLINE> +-- *** Step 1. Success Initial Conversion(ip) [2019-04-23] *** +-- <BLANKLINE> +-- P ReadP Day (2019-04-23) 2019-04-23 | 2019-04-23 +-- | +-- `- P Id "2019-04-23" +-- <BLANKLINE> +-- *** Step 2. False Boolean Check(op) *** +-- <BLANKLINE> +-- False False && True +-- | +-- +- False 2019-04-23 >= 2019-05-30 +-- | | +-- | +- P I +-- | | +-- | `- P ReadP Day (2019-05-30) 2019-05-30 | 2019-05-30 +-- | | +-- | `- P '2019-05-30 +-- | +-- `- True 2019-04-23 <= 2019-06-01 +-- | +-- +- P I +-- | +-- `- P ReadP Day (2019-06-01) 2019-06-01 | 2019-06-01 +-- | +-- `- P '2019-06-01 +-- <BLANKLINE> +-- +instance (Show (PP fmt (PP ip i)), Show (PP ip i), Refined3C ip op fmt i, Binary i) => Binary (Refined3 ip op fmt i) where + get = do + i <- B.get @i + let (ret,mr) = eval3 @ip @op @fmt o2 i + case mr of + Nothing -> fail $ "Refined3:" ++ show (prt3Impl o2 ret) + Just r -> return r + put (Refined3 _ r) = B.put @i r + +-- | wraps the parameters for 'Refined3' in a 4-tuple for use with methods such as 'withRefined3TP' and 'newRefined3TP' +mkProxy3 :: forall ip op fmt i . Refined3C ip op fmt i => Proxy '(ip,op,fmt,i) +mkProxy3 = Proxy + +-- | use type application to set the parameters then it will be wrapped into a 4-tuple +-- checks to make sure the proxy is consistent with Refined3C +-- use for passing into eval3P you can pass in a promoted 4 tuple to other methods +mkProxy3P :: forall z ip op fmt i . (z ~ '(ip,op,fmt,i), Refined3C ip op fmt i) => Proxy '(ip,op,fmt,i) +mkProxy3P = Proxy + +-- | convenience type family for converting from a 4-tuple '(ip,op,fmt,i) to a 'Refined3' signature +type family MakeR3 p where + MakeR3 '(ip,op,fmt,i) = Refined3 ip op fmt i + +-- | convenience type family for converting from a 4-tuple '(ip,op,fmt,i) to a Proxy +type family MkProxy3T p where + MkProxy3T '(ip,op,fmt,i) = Proxy '(ip,op,fmt,i) + +withRefined3TIO :: forall ip op fmt i m b + . (MonadIO m, Refined3C ip op fmt i, Show (PP ip i), Show i) + => POpts + -> i + -> (Refined3 ip op fmt i -> RefinedT m b) + -> RefinedT m b +withRefined3TIO opts = (>>=) . newRefined3TPIO (Proxy @'(ip,op,fmt,i)) opts + +withRefined3T :: forall ip op fmt i m b + . (Monad m, Refined3C ip op fmt i, Show (PP ip i), Show i) + => POpts + -> i + -> (Refined3 ip op fmt i -> RefinedT m b) + -> RefinedT m b +withRefined3T opts = (>>=) . newRefined3TP (Proxy @'(ip,op,fmt,i)) opts + +withRefined3TP :: forall ip op fmt i m b proxy + . (Monad m, Refined3C ip op fmt i, Show (PP ip i), Show i) + => proxy '(ip,op,fmt,i) + -> POpts + -> i + -> (Refined3 ip op fmt i -> RefinedT m b) + -> RefinedT m b +withRefined3TP p opts = (>>=) . newRefined3TP p opts + +newRefined3T :: forall m ip op fmt i . (Refined3C ip op fmt i, Monad m, Show (PP ip i), Show i) + => POpts + -> i + -> RefinedT m (Refined3 ip op fmt i) +newRefined3T = newRefined3TP (Proxy @'(ip,op,fmt,i)) + +newRefined3TP :: forall m ip op fmt i proxy + . (Refined3C ip op fmt i, Monad m, Show (PP ip i), Show i) + => proxy '(ip,op,fmt,i) + -> POpts + -> i + -> RefinedT m (Refined3 ip op fmt i) +newRefined3TP = newRefined3TPImpl (return . runIdentity) + +newRefined3TPIO :: forall m ip op fmt i proxy + . (Refined3C ip op fmt i, MonadIO m, Show (PP ip i), Show i) + => proxy '(ip,op,fmt,i) + -> POpts + -> i + -> RefinedT m (Refined3 ip op fmt i) +newRefined3TPIO = newRefined3TPImpl liftIO + +-- we call this cos we need to do the bool check and get fmt value +-- eval (PP op i) ~ True and eval (PP fmt i) to get the other value +-- input is set to @Id meaning that PP fmt (PP ip i) /= i doesnt hold +newRefined3TPImpl :: forall n m ip op fmt i proxy + . (Refined3C ip op fmt i, Monad m, MonadEval n, Show (PP ip i), Show (PP fmt (PP ip i))) + => (forall x . n x -> RefinedT m x) + -> proxy '(ip,op,fmt,i) + -> POpts + -> i + -> RefinedT m (Refined3 ip op fmt i) +newRefined3TPImpl f _ opts i = do + (ret,mr) <- f $ eval3M opts i + let m3 = prt3Impl opts ret + tell [m3Long m3] + case mr of + Nothing -> throwError $ m3Desc m3 <> " | " <> m3Short m3 + Just r -> return r + +newRefined3TPSkipIPImpl :: forall n m ip op fmt i proxy + . (Refined3C ip op fmt i + , Monad m + , MonadEval n + , Show (PP ip i) + , Show (PP fmt (PP ip i))) + => (forall x . n x -> RefinedT m x) + -> proxy '(ip,op,fmt,i) + -> POpts + -> PP ip i + -> RefinedT m (Refined3 ip op fmt i) +newRefined3TPSkipIPImpl f _ opts a = do + (ret,mr) <- f $ eval3MSkip opts a + let m3 = prt3Impl opts ret + tell [m3Long m3] + case mr of + Nothing -> throwError $ m3Desc m3 <> " | " <> m3Short m3 + Just r -> return r + +-- optional Refined3C ip op fmt i [not required!] +convertRefined3T :: forall m ip op fmt i ip1 op1 fmt1 i1 . + ( Refined3C ip1 op1 fmt1 i1 + , Monad m + , Show (PP ip i) + , PP ip i ~ PP ip1 i1 + , Show i1) + => POpts + -> RefinedT m (Refined3 ip op fmt i) + -> RefinedT m (Refined3 ip1 op1 fmt1 i1) +convertRefined3T opts ma = do + Refined3 x _ <- ma + -- we skip the input value @Id and go straight to the internal value so PP fmt (PP ip i) /= i for this call + Refined3 a b <- newRefined3TPSkipIPImpl (return . runIdentity) (Proxy @'(ip1, op1, fmt1, i1)) opts x + return (Refined3 a b) + +convertRefined3TP :: forall m ip op fmt i ip1 op1 fmt1 i1 . + ( Refined3C ip1 op1 fmt1 i1 + , Monad m + , Show (PP ip i) + , PP ip i ~ PP ip1 i1 + , Show i1) + => Proxy '(ip, op, fmt, i) + -> Proxy '(ip1, op1, fmt1, i1) + -> POpts + -> RefinedT m (Refined3 ip op fmt i) + -> RefinedT m (Refined3 ip1 op1 fmt1 i1) +convertRefined3TP _ _ opts ma = do + Refined3 x _ <- ma + -- we skip the input value @Id and go straight to the internal value so PP fmt (PP ip i) /= i for this call + Refined3 a b <- newRefined3TPSkipIPImpl (return . runIdentity) (Proxy @'(ip1, op1, fmt1, i1)) opts x + return (Refined3 a b) + +rapply3 :: forall m ip op fmt i . + ( Refined3C ip op fmt i + , Monad m + , Show (PP ip i) + , Show i) + => POpts + -> (PP ip i -> PP ip i -> PP ip i) + -> RefinedT m (Refined3 ip op fmt i) + -> RefinedT m (Refined3 ip op fmt i) + -> RefinedT m (Refined3 ip op fmt i) +rapply3 = rapply3P (Proxy @'(ip,op,fmt,i)) + +-- this is the most generic +-- prtRefinedT $ rapply3P base16 (+) (newRefined3TP Proxy "ff") (newRefined3TP Proxy "22") +rapply3P :: forall m ip op fmt i proxy . + ( Refined3C ip op fmt i + , Monad m + , Show (PP ip i) + , Show i) + => proxy '(ip,op,fmt,i) + -> POpts + -> (PP ip i -> PP ip i -> PP ip i) + -> RefinedT m (Refined3 ip op fmt i) + -> RefinedT m (Refined3 ip op fmt i) + -> RefinedT m (Refined3 ip op fmt i) +rapply3P p opts f ma mb = do + tell [bgColor Blue "=== a ==="] + Refined3 x _ <- ma + tell [bgColor Blue "=== b ==="] + Refined3 y _ <- mb + -- we skip the input value @Id and go straight to the internal value so PP fmt (PP ip i) /= i for this call + tell [bgColor Blue "=== a `op` b ==="] + Refined3 a b <- newRefined3TPSkipIPImpl (return . runIdentity) p opts (f x y) + return (Refined3 a b) + +data Results a b = + XF String -- Left e + | XTF a String -- Right a + Left e + | XTFalse a -- Right a + Right False + | XTTrueF a String -- Right a + Right True + Left e + | XTTrueT a b -- Right a + Right True + Right b + deriving (Show,Eq) + +data RResults a b = + RF String (Tree PE) -- Left e + | RTF a (Tree PE) String (Tree PE) -- Right a + Left e + | RTFalse a (Tree PE) (Tree PE) -- Right a + Right False + | RTTrueF a (Tree PE) (Tree PE) String (Tree PE) -- Right a + Right True + Left e + | RTTrueT a (Tree PE) (Tree PE) b (Tree PE) -- Right a + Right True + Right b + deriving Show + +prtEval3PIO :: forall ip op fmt i proxy + . ( Refined3C ip op fmt i + , Show (PP ip i) + , Show i) + => proxy '(ip,op,fmt,i) + -> POpts + -> i + -> IO (Either String (Refined3 ip op fmt i)) +prtEval3PIO _ opts i = do + x <- eval3M opts i + prt3IO opts x + +prtEval3 :: forall ip op fmt i + . ( Refined3C ip op fmt i + , Show (PP ip i) + , Show i) + => POpts + -> i + -> Either Msg3 (Refined3 ip op fmt i) +prtEval3 opts i = + let x = eval3 opts i + in prt3 opts x + +prtEval3P :: forall ip op fmt i proxy + . ( Refined3C ip op fmt i + , Show (PP ip i) + , Show i) + => proxy '(ip,op,fmt,i) + -> POpts + -> i + -> Either Msg3 (Refined3 ip op fmt i) +prtEval3P _ opts i = + let x = eval3 opts i + in prt3 opts x + + +-- pass in a proxy (use mkProxy to package all the types together as a 4-tuple) +-- ip converts input 'i' to format used for op and fmt +-- op is a boolean predicate [has to be True to continue] (uses P ip i as input) +-- fmt formats the output (can be anything ie not just String) (uses P ip i as input) +eval3P :: forall ip op fmt i proxy . Refined3C ip op fmt i + => proxy '(ip,op,fmt,i) + -> POpts + -> i + -> (RResults (PP ip i) (PP fmt (PP ip i)), Maybe (Refined3 ip op fmt i)) +eval3P _ opts = runIdentity . eval3M opts + +-- same as eval3P but can just pass in ip op fmt separately +-- with eval3P we have to specify 'i' but in eval3 we dont cos gets it from context +eval3 :: forall ip op fmt i . Refined3C ip op fmt i + => POpts + -> i + -> (RResults (PP ip i) (PP fmt (PP ip i)), Maybe (Refined3 ip op fmt i)) +eval3 opts = runIdentity . eval3M opts + +eval3M :: forall m ip op fmt i . (MonadEval m, Refined3C ip op fmt i) + => POpts + -> i + -> m (RResults (PP ip i) (PP fmt (PP ip i)), Maybe (Refined3 ip op fmt i)) +eval3M opts i = do + ll@(fromTT -> t1) <- eval (Proxy @ip) opts i + case getValLR (_tBool ll) of + Right a -> do + rr@(fromTT -> t2) <- evalBool (Proxy @op) opts a + case getValLR (_tBool rr) of + Right True -> do + ss@(fromTT -> t3) <- eval (Proxy @fmt) opts a + pure $ case getValLR (_tBool ss) of + Right b -> (RTTrueT a t1 t2 b t3, Just (Refined3 a b)) + Left e -> (RTTrueF a t1 t2 e t3, Nothing) + Right False -> pure (RTFalse a t1 t2, Nothing) + Left e -> pure (RTF a t1 e t2, Nothing) + Left e -> pure (RF e t1, Nothing) + +-- skip ip conversion: ie uses internal value +eval3MSkip :: forall m ip op fmt i . (MonadEval m, Refined3C ip op fmt i) + => POpts + -> PP ip i + -> m (RResults (PP ip i) (PP fmt (PP ip i)), Maybe (Refined3 ip op fmt i)) +eval3MSkip opts a = do + let t1 = Node (PE (TrueP) ["skipped PP ip i = Id"]) [] + rr@(fromTT -> t2) <- evalBool (Proxy @op) opts a + case getValLR (_tBool rr) of + Right True -> do + ss@(fromTT -> t3) <- eval (Proxy @fmt) opts a + pure $ case getValLR (_tBool ss) of + Right b -> (RTTrueT a t1 t2 b t3, Just (Refined3 a b)) + Left e -> (RTTrueF a t1 t2 e t3, Nothing) + Right False -> pure (RTFalse a t1 t2, Nothing) + Left e -> pure (RTF a t1 e t2, Nothing) + +-- calculates from internal value +eval3MQuickIdentity :: forall ip op fmt i . Refined3C ip op fmt i + => POpts + -> PP ip i + -> Maybe (Refined3 ip op fmt i) +eval3MQuickIdentity opts = runIdentity . eval3MQuick opts + +-- from PP ip i +eval3MQuick :: forall m ip op fmt i . (MonadEval m, Refined3C ip op fmt i) + => POpts + -> PP ip i + -> m (Maybe (Refined3 ip op fmt i)) +eval3MQuick opts a = do + rr <- evalBool (Proxy @op) opts a + case getValLR (_tBool rr) of + Right True -> do + ss <- eval (Proxy @fmt) opts a + pure $ case getValLR (_tBool ss) of + Right b -> Just (Refined3 a b) + _ -> Nothing + _ -> pure Nothing + +prt3IO :: (Show a, Show b) => POpts -> (RResults a b, Maybe r) -> IO (Either String r) +prt3IO opts (ret,mr) = do + let m3 = prt3Impl opts ret + unless (oLite opts) $ putStrLn $ m3Long m3 + return $ maybe (Left (m3Desc m3 <> " | " <> m3Short m3)) Right mr + +prt3 :: (Show a, Show b) => POpts -> (RResults a b, Maybe r) -> Either Msg3 r +prt3 opts (ret,mr) = maybe (Left $ prt3Impl opts ret) Right mr + +data Msg3 = Msg3 { m3Desc :: String, m3Short :: String, m3Long :: String } deriving Eq +instance Show Msg3 where + show (Msg3 a b c) = a <> " | " <> b <> (if null c then "" else "\n" <> c) + +prt3Impl :: (Show a, Show b) + => POpts + -> RResults a b + -> Msg3 +prt3Impl opts v = + let outmsg msg = "\n*** " <> msg <> " ***\n\n" + msg1 a = outmsg ("Step 1. Success Initial Conversion(ip) [" ++ show a ++ "]") + mkMsg3 m n r | oLite opts = Msg3 m n "" + | otherwise = Msg3 m n r + in case v of + RF e t1 -> + let (m,n) = ("Step 1. Initial Conversion(ip) Failed", e) + r = outmsg m + <> prtTreePure opts t1 + in mkMsg3 m n r + RTF a t1 e t2 -> + let (m,n) = ("Step 2. Failed Boolean Check(op)", e) + r = msg1 a + <> fixLite opts a t1 + <> outmsg m + <> prtTreePure opts t2 + in mkMsg3 m n r + RTFalse a t1 t2 -> + let (m,n) = ("Step 2. False Boolean Check(op)", "FalseP") + r = msg1 a + <> fixLite opts a t1 + <> outmsg m + <> prtTreePure opts t2 + in mkMsg3 m n r + RTTrueF a t1 t2 e t3 -> + let (m,n) = ("Step 3. Failed Output Conversion(fmt)", e) + r = msg1 a + <> fixLite opts a t1 + <> outmsg "Step 2. Success Boolean Check(op)" + <> prtTreePure opts t2 + <> outmsg m + <> prtTreePure opts t3 + in mkMsg3 m n r + RTTrueT a t1 t2 b t3 -> + let (m,n) = ("Step 3. Success Output Conversion(fmt)", show b) + r = msg1 a + <> fixLite opts a t1 + <> outmsg "Step 2. Success Boolean Check(op)" + <> prtTreePure opts t2 + <> outmsg m + <> fixLite opts b t3 + in mkMsg3 m n r + + +-- | emulates 'Refined3' but uses 'Refined' +-- reuses the mkProxy3 but returns Refined vs Refined3 +-- using plain Refined to emulate Refined3 sort of +-- we just output fmt instead of embedding it in Refined3 +-- so \'ip\' predicate gets us started: we store that \'PP ip i\' in Refined +-- then we run the boolean predicate \'op\' which is successful if true +-- then we run \'fmt\' against '\PP ip i\' and output both the Refined (PP p i) and the PP fmt (PP (ip i)) ie \'fmt\' run against PP ip i +-- if any of the three steps fails the process stops immediately and dumps out RResults +eval3PX :: forall ip op fmt i proxy . Refined3C ip op fmt i + => proxy '(ip,op,fmt,i) + -> POpts + -> i + -> (RResults (PP ip i) (PP fmt (PP ip i)), Maybe (Refined op (PP ip i), PP fmt (PP ip i))) +eval3PX _ = eval3X @ip @op @fmt + +eval3X :: forall ip op fmt i . Refined3C ip op fmt i + => POpts + -> i + -> (RResults (PP ip i) (PP fmt (PP ip i)), Maybe (Refined op (PP ip i), PP fmt (PP ip i))) +eval3X opts i = runIdentity $ do + ll@(fromTT -> t1) <- eval (Proxy @ip) opts i + case getValLR (_tBool ll) of + Right a -> do + rr@(fromTT -> t2) <- evalBool (Proxy @op) opts a + case getValLR (_tBool rr) of + Right True -> do + ss@(fromTT -> t3) <- eval (Proxy @fmt) opts a + pure $ case getValLR (_tBool ss) of + Right b -> (RTTrueT a t1 t2 b t3, Just (unsafeRefined a, b)) + Left e -> (RTTrueF a t1 t2 e t3, Nothing) + Right False -> pure (RTFalse a t1 t2, Nothing) + Left e -> pure (RTF a t1 e t2, Nothing) + Left e -> pure (RF e t1, Nothing)
+ src/Refined3Helper.hs view
@@ -0,0 +1,334 @@+{-# OPTIONS -Wall #-} +{-# OPTIONS -Wcompat #-} +{-# OPTIONS -Wincomplete-record-updates #-} +{-# OPTIONS -Wincomplete-uni-patterns #-} +{-# OPTIONS -Wredundant-constraints #-} +{-# LANGUAGE TypeOperators #-} +{-# LANGUAGE UndecidableInstances #-} +{-# LANGUAGE FlexibleContexts #-} +{-# LANGUAGE AllowAmbiguousTypes #-} +{-# LANGUAGE FlexibleInstances #-} +{-# LANGUAGE MultiParamTypeClasses #-} +{-# LANGUAGE TypeApplications #-} +{-# LANGUAGE DataKinds #-} +{-# LANGUAGE GADTs #-} +{-# LANGUAGE TypeFamilies #-} +{-# LANGUAGE PolyKinds #-} +{-# LANGUAGE ScopedTypeVariables #-} +{-# LANGUAGE LambdaCase #-} +{-# LANGUAGE RankNTypes #-} +{-# LANGUAGE OverloadedStrings #-} +{-# LANGUAGE ConstraintKinds #-} +{-# LANGUAGE StandaloneDeriving #-} +{-# LANGUAGE ViewPatterns #-} +{-# LANGUAGE NoStarIsType #-} +{-# LANGUAGE TemplateHaskell #-} +{-# LANGUAGE DeriveLift #-} +-- | +-- Module : Refined3Helper +-- Description : Contains convenient prepackaged 4-tuples to use with Refined3 +-- Copyright : (c) Grant Weyburne, 2019 +-- License : BSD-3 +-- Maintainer : gbwey9@gmail.com +-- +-- Prepackaged proxies for use with 'Refined3.Refined3' +-- +module Refined3Helper where +import Refined3 +import Predicate +import UtilP +import Data.Proxy +import GHC.TypeLits (AppendSymbol,Nat,KnownNat) +import Data.Kind (Type) +import Data.Time +import qualified Data.Semigroup as SG + +-- | credit card with luhn algorithm +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> prtEval3P cc11 ol "1234-5678-901" +-- Left Step 2. False Boolean Check(op) | FalseP +-- +-- >>> prtEval3P cc11 ol "1234-5678-903" +-- Right (Refined3 {r3In = [1,2,3,4,5,6,7,8,9,0,3], r3Out = "1234-5678-903"}) +-- +type Ccip = Map (ReadP Int) (Remove "-" Id >> Ones) +type Ccop (n :: Nat) = Guard ('(n,Len) >> Printf2 "expected %d digits but found %d") (Len >> Same n) >> Luhn +type Ccfmt (ns :: [Nat]) = ConcatMap ShowP Id >> SplitAts ns Id >> Concat (Intercalate '["-"] Id) + +type Ccn (ns :: [Nat]) = '(Ccip, Ccop (SumT ns), Ccfmt ns, String) + +type CC11 = Ccn '[4,4,3] + +-- not great for the general case: but specific case is easier +ccn :: forall (ns :: [Nat]) . (KnownNat (SumT ns), P ns String, PP ns String ~ [Integer]) => Proxy (Ccn ns) +ccn = mkProxy3 + +cc11 :: Proxy (Ccn '[4,4,3]) -- or Proxy CC11 +cc11 = mkProxy3P + +-- | read in a datetime +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> prtEval3P (Proxy @(DateTime1 UTCTime)) ol "2018-09-14 02:57:04" +-- Right (Refined3 {r3In = 2018-09-14 02:57:04 UTC, r3Out = "2018-09-14 02:57:04"}) +-- +type DateTime1 (t :: Type) = '(Dtip1 t, Dtop1, Dtfmt1, String) +type Dtip1 t = ParseTimeP t "%F %T" Id + +-- extra check to validate the time as parseTime doesnt validate the time component +type Dtop1 = + Map (ReadP Int) (FormatTimeP "%H %M %S" Id >> Resplit "\\s+" Id) + >> Guards '[ '(Printf2 "guard %d invalid hours %d", Between 0 23) + , '(Printf2 "guard %d invalid minutes %d", Between 0 59) + , '(Printf2 "guard %d invalid seconds %d", Between 0 59) + ] >> 'True +type Dtfmt1 = FormatTimeP "%F %T" Id + +ssn :: Proxy Ssn +ssn = mkProxy3 + +-- | read in an ssn +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> prtEval3P ssn ol "134-01-2211" +-- Right (Refined3 {r3In = [134,1,2211], r3Out = "134-01-2211"}) +-- +-- >>> prtEval3P ssn ol "666-01-2211" +-- Left Step 2. Failed Boolean Check(op) | number for group 0 invalid: found 666 +-- +-- >>> prtEval3P ssn ol "666-01-2211" +-- Left Step 2. Failed Boolean Check(op) | number for group 0 invalid: found 666 +-- +-- >>> prtEval3P ssn ol "667-00-2211" +-- Left Step 2. Failed Boolean Check(op) | number for group 1 invalid: found 0 +-- +-- >>> prtEval3P ssn ol "666-01-2211" +-- Left Step 2. Failed Boolean Check(op) | number for group 0 invalid: found 666 +-- +-- >>> prtEval3P ssn ol "991-22-9999" +-- Left Step 2. Failed Boolean Check(op) | number for group 0 invalid: found 991 +-- +type Ssn = '(Ssnip, Ssnop, Ssnfmt, String) + +type Ssnip = Map (ReadP Int) (Rescan "^(\\d{3})-(\\d{2})-(\\d{4})$" Id >> OneP >> Snd) +type Ssnop = GuardsQuick (Printf2 "number for group %d invalid: found %d") + '[Between 1 899 && Id /= 666, Between 1 99, Between 1 9999] + >> 'True +type Ssnop' = Guards '[ '(Printf2 "guard %d invalid: found %d", Between 1 899 && Id /= 666) + , '(Printf2 "group %d invalid: found %d", Between 1 99) + , '(Printf2 "group %d invalid: found %d", Between 1 9999) + ] >> 'True +type Ssnfmt = Printfnt 3 "%03d-%02d-%04d" + +-- | read in a time and validate it +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> prtEval3P hms ol "23:13:59" +-- Right (Refined3 {r3In = [23,13,59], r3Out = "23:13:59"}) +-- +-- >>> prtEval3P hms ol "23:13:60" +-- Left Step 2. Failed Boolean Check(op) | guard(2) 60 secs is out of range +-- +-- >>> prtEval3P hms ol "26:13:59" +-- Left Step 2. Failed Boolean Check(op) | guard(0) 26 hours is out of range +-- +hms :: Proxy Hms +hms = mkProxy3 + +type Hms = '(Hmsip, Hmsop >> 'True, Hmsfmt, String) + +type Hmsip = Map (ReadP Int) (Resplit ":" Id) +type Hmsop = Guard (Printf "expected len 3 but found %d" Len) (Len >> Same 3) + >> Guards '[ '(Printf2 "guard(%d) %d hours is out of range", Between 0 23) + , '(Printf2 "guard(%d) %d mins is out of range", Between 0 59) + , '(Printf2 "guard(%d) %d secs is out of range", Between 0 59)] +type Hmsfmt = Printfnt 3 "%02d:%02d:%02d" + +-- | read in an ipv4 address and validate it +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> prtEval3P ip ol "001.223.14.1" +-- Right (Refined3 {r3In = [1,223,14,1], r3Out = "001.223.014.001"}) +-- +-- >>> prtEval3P ip ol "001.223.14.999" +-- Left Step 2. Failed Boolean Check(op) | guard(3) octet out of range 0-255 found 999 +-- +-- >>> prtEval3P ip ol "001.223.14.999.1" +-- Left Step 1. Initial Conversion(ip) Failed | Regex no results +-- +-- >>> prtEval3P ip ol "001.257.14.1" +-- Left Step 2. Failed Boolean Check(op) | guard(1) octet out of range 0-255 found 257 +-- +type Ip = '(Ipip, Ipop, Ipfmt, String) +type Ip1 = '(Ipip, Ipop', Ipfmt, String) + +ip :: Proxy Ip +ip = mkProxy3 + +ip1 :: Proxy Ip1 +ip1 = mkProxy3 + +type Ipip = Map (ReadP Int) (Rescan "^(\\d{1,3}).(\\d{1,3}).(\\d{1,3}).(\\d{1,3})$" Id >> OneP >> Snd) +-- RepeatT is a type family so it expands everything! replace RepeatT with a type class +type Ipop = GuardsQuick (Printf2 "guard(%d) octet out of range 0-255 found %d") (RepeatT 4 (Between 0 255)) >> 'True +type Ipop' = Guards '[ + '(Printf2 "octet %d out of range 0-255 found %d", Between 0 255) + , '(Printf2 "octet %d out of range 0-255 found %d", Between 0 255) + , '(Printf2 "octet %d out of range 0-255 found %d", Between 0 255) + , '(Printf2 "octet %d out of range 0-255 found %d", Between 0 255) + ] >> 'True +type Ipfmt = Printfnt 4 "%03d.%03d.%03d.%03d" + +type HmsRE = "^([0-1][0-9]|2[0-3]):([0-5][0-9]):([0-5][0-9])$" -- padded only -- dumb because strict validation should not be done twice: ie in ip and op! +type Hmsconv = Do '[Rescan HmsRE Id, Head, Snd, Map (ReadBaseInt 10) Id] +type Hmsval = GuardsQuick (Printf2 "guard(%d) %d is out of range") '[Between 0 23, Between 0 59, Between 0 59] + +type Hms4 = '(Hmsconv, Hmsval >> 'True, Hmsfmt, String) + +hms4 :: Proxy Hms4 +hms4 = mkProxy3 + +type OctetRE = "(25[0-5]|2[0..4][0-9]|1[0-9][0-9]|[1-9][0-9]|[0-9])" -- no padded numbers allowed +--type Ip4strictRE = "^" `AppendSymbol` OctetRE `AppendSymbol` "\\." `AppendSymbol` OctetRE `AppendSymbol` "\\." `AppendSymbol` OctetRE `AppendSymbol` "\\." `AppendSymbol` OctetRE `AppendSymbol` "$" +type Ip4strictRE = "^" `AppendSymbol` IntersperseT "\\." (RepeatT 4 OctetRE) `AppendSymbol` "$" + +-- valid dates for for DateFmts are "2001-01-01" "Jan 24 2009" and "03/29/07" +type DateFmts = '["%Y-%m-%d", "%m/%d/%y", "%B %d %Y"] +type DateN = '(ParseTimes Day DateFmts Id, 'True, FormatTimeP "%Y-%m-%d" Id, String) + +type DateTimeFmts = '["%Y-%m-%d %H:%M:%S", "%m/%d/%y %H:%M:%S", "%B %d %Y %H:%M:%S", "%Y-%m-%dT%H:%M:%S"] +type DateTimeN = '(ParseTimes UTCTime DateTimeFmts Id, 'True, FormatTimeP "%Y-%m-%d %H:%M:%S" Id, String) + +-- | convert a string from the given base \'i\' but stores it internally as a string of base \'j\' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> prtEval3P (Proxy @(BaseN 16)) ol "00fe" +-- Right (Refined3 {r3In = 254, r3Out = "fe"}) +-- +type BaseN (n :: Nat) = '(ReadBase Integer n, 'True, ShowBase n, String) + +base16 :: Proxy (BaseN 16) +base16 = mkProxy3 + +daten :: Proxy DateN +daten = mkProxy3 + +datetimen :: Proxy DateTimeN +datetimen = mkProxy3 + +type BetweenR m n = Refined3 Id (Between m n) Id Int + +type LuhnR (n :: Nat) = MakeR3 (LuhnY n) +type LuhnR' (n :: Nat) = MakeR3 (LuhnX n) + +-- uses builtin Luhn vs long winded version LuhnX +type LuhnY (n :: Nat) = + '(Map (ReadP Int) Ones + , Guard (Printfn "incorrect number of digits found %d but expected %d in [%s]" (TupleI '[Len, W n, ShowP])) + (Len >> Same n) + >> Guard ("luhn check failed") Luhn >> 'True + , ConcatMap ShowP Id + , String) + +type LuhnX (n :: Nat) = + '(Map (ReadP Int) Ones + , Luhn'' n >> 'True + , ConcatMap ShowP Id + , String) + +type Luhn'' (n :: Nat) = + Guard (Printfn "incorrect number of digits found %d but expected %d in [%s]" (TupleI '[Len, W n, ShowP])) (Len >> Same n) + >> Do '[ + Reverse + ,Ziplc [1,2] Id + ,Map (Fst * Snd >> If (Id >= 10) (Id - 9) Id) Id + ,FoldMap (SG.Sum Int) Id + ] + >> Guard (Printfn "expected %d mod 10 = 0 but found %d" (TupleI '[Id, Id `Mod` 10])) (Mod Id 10 >> Same 0) + +type Luhn' (n :: Nat) = + Msg "Luhn'" (Do + '[Guard (Printfn "incorrect number of digits found %d but expected %d in [%s]" (TupleI '[Len, W n, Id])) (Len >> Same n) + ,Do + '[Ones + ,Map (ReadP Int) Id + ,Reverse + ,Ziplc [1,2] Id + ,Map (Fst * Snd >> If (Id >= 10) (Id - 9) Id) Id + ,FoldMap (SG.Sum Int) Id + ] + ,Guard (Printfn "expected %d mod 10 = 0 but found %d" (TupleI '[Id, Id `Mod` 10])) (Mod Id 10 >> Same 0) + ]) + +-- noop true +type Ok (t :: Type) = '(Id, 'True, Id, t) +type OkR (t :: Type) = MakeR3 (Ok t) + +-- noop false +type OkNot (t :: Type) = '(Id, 'False, Id, t) +type OkNotR (t :: Type) = MakeR3 (OkNot t) + +-- | convert a string from the given base \'i\' but stores it internally as a string of base \'j\' +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> prtEval3P (Proxy @(BaseIJ 16 2)) ol "fe" +-- Right (Refined3 {r3In = "11111110", r3Out = "fe"}) +-- +-- >>> prtEval3P (Proxy @(BaseIJ 16 2)) ol "fge" +-- Left Step 1. Initial Conversion(ip) Failed | invalid base 16 +-- +type BaseIJ (i :: Nat) (j :: Nat) = BaseIJ' i j 'True +type BaseIJ' (i :: Nat) (j :: Nat) p = '(ReadBase Int i >> ShowBase j, p, ReadBase Int j >> ShowBase i, String) + +-- | take any valid Read/Show instance and turn it into a valid Refined3 +-- +-- >>> :set -XTypeApplications +-- >>> :set -XDataKinds +-- >>> :set -XTypeOperators +-- >>> :m + Data.Ratio +-- >>> prtEval3P (Proxy @(ReadShow Rational)) ol "13 % 3" +-- Right (Refined3 {r3In = 13 % 3, r3Out = "13 % 3"}) +-- +-- >>> prtEval3P (Proxy @(ReadShow Rational)) ol "13x % 3" +-- Left Step 1. Initial Conversion(ip) Failed | ReadP Ratio Integer (13x % 3) failed +-- +-- >>> prtEval3P (Proxy @(ReadShow' Rational (Between (Pos 3) (Pos 5)))) ol "13 % 3" +-- Right (Refined3 {r3In = 13 % 3, r3Out = "13 % 3"}) +-- +-- >>> prtEval3P (Proxy @(ReadShow' Rational (Between (NegR 11 2) (Neg 3)))) ol "-13 % 3" +-- Right (Refined3 {r3In = (-13) % 3, r3Out = "(-13) % 3"}) +-- +-- >>> prtEval3P (Proxy @(ReadShow' Rational (Id > Pos 15))) ol "13 % 3" +-- Left Step 2. False Boolean Check(op) | FalseP +-- +-- >>> prtEval3P (Proxy @(ReadShow' Rational (Guard (Printf "invalid=%3.2f" (FromRational Double Id)) (Id > Pos 15) >> 'True))) ol "13 % 3" +-- Left Step 2. Failed Boolean Check(op) | invalid=4.33 +-- +-- >>> prtEval3P (Proxy @(ReadShow' Rational (Id > Pos 11))) ol "13 % 3" +-- Left Step 2. False Boolean Check(op) | FalseP +-- +-- >>> let tmString = "2018-10-19 14:53:11.5121359 UTC" +-- >>> let tm = read tmString :: UTCTime +-- >>> prtEval3P (Proxy @(ReadShow UTCTime)) ol tmString +-- Right (Refined3 {r3In = 2018-10-19 14:53:11.5121359 UTC, r3Out = "2018-10-19 14:53:11.5121359 UTC"}) +-- +-- >>> :m + Data.Aeson +-- >>> prtEval3P (Proxy @(ReadShow Value)) ol "String \"jsonstring\"" +-- Right (Refined3 {r3In = String "jsonstring", r3Out = "String \"jsonstring\""}) +-- +-- >>> prtEval3P (Proxy @(ReadShow Value)) ol "Number 123.4" +-- Right (Refined3 {r3In = Number 123.4, r3Out = "Number 123.4"}) +-- +type ReadShow (t :: Type) = '(ReadP t, 'True, ShowP, String) +type ReadShowR (t :: Type) = MakeR3 (ReadShow t) + +type ReadShow' (t :: Type) p = '(ReadP t, p, ShowP, String) +type ReadShowR' (t :: Type) p = MakeR3 (ReadShow' t p)
+ src/TH_Orphans.hs view
@@ -0,0 +1,28 @@+{-# OPTIONS -Wall #-} +{-# OPTIONS -Wno-orphans #-} +{-# LANGUAGE DeriveLift #-} +{-# LANGUAGE StandaloneDeriving #-} +{-# LANGUAGE TemplateHaskell #-} +-- | +-- Module : TH_Orphans +-- Description : Mainly contains useful Template Haskell Lift instances for Date Time +-- Copyright : (c) Grant Weyburne, 2019 +-- License : BSD-3 +-- Maintainer : gbwey9@gmail.com +-- +module TH_Orphans where +import qualified Language.Haskell.TH.Syntax as TH +import qualified Language.Haskell.TH.Instances () -- other orphans +import qualified Language.Haskell.TH.Lift as TL +import Data.Time +import Data.Fixed + +deriving instance TH.Lift Day +deriving instance TH.Lift LocalTime +deriving instance TH.Lift TimeOfDay +deriving instance TH.Lift (Fixed a) + +deriving instance TH.Lift UTCTime + +$(TL.deriveLift ''DiffTime) +
+ src/UtilP.hs view
@@ -0,0 +1,994 @@+{-# OPTIONS -Wall #-} +{-# OPTIONS -Wcompat #-} +{-# OPTIONS -Wincomplete-record-updates #-} +{-# OPTIONS -Wincomplete-uni-patterns #-} +{-# OPTIONS -Wredundant-constraints #-} +{-# LANGUAGE TypeOperators #-} +{-# LANGUAGE UndecidableInstances #-} +{-# LANGUAGE FlexibleContexts #-} +{-# LANGUAGE AllowAmbiguousTypes #-} +{-# LANGUAGE FlexibleInstances #-} +{-# LANGUAGE MultiParamTypeClasses #-} +{-# LANGUAGE TypeApplications #-} +{-# LANGUAGE DataKinds #-} +{-# LANGUAGE GADTs #-} +{-# LANGUAGE TypeFamilies #-} +{-# LANGUAGE PolyKinds #-} +{-# LANGUAGE ScopedTypeVariables #-} +{-# LANGUAGE LambdaCase #-} +{-# LANGUAGE RankNTypes #-} +{-# LANGUAGE TupleSections #-} +{-# LANGUAGE OverloadedStrings #-} +{-# LANGUAGE ViewPatterns #-} +{-# LANGUAGE ConstraintKinds #-} +{-# LANGUAGE TemplateHaskell #-} +{-# LANGUAGE MultiWayIf #-} +{-# LANGUAGE DefaultSignatures #-} +{-# LANGUAGE GeneralizedNewtypeDeriving #-} +{-# LANGUAGE StandaloneDeriving #-} +-- | +-- Module : UtilP +-- Description : Utility methods for Predicate / methods for displaying the evaluation tree ... +-- Copyright : (c) Grant Weyburne, 2019 +-- License : BSD-3 +-- Maintainer : gbwey9@gmail.com +-- +module UtilP where +import qualified GHC.TypeNats as GN +import Data.Ratio +import GHC.TypeLits (Symbol,Nat,KnownSymbol,KnownNat,ErrorMessage((:$$:),(:<>:))) +import qualified GHC.TypeLits as GL +import Control.Lens +import Control.Arrow +import Data.List +import qualified Data.Tree.View as TV +import Data.Tree +import Data.Tree.Lens +import Data.Tree.Pretty +import Data.Proxy +import Data.List.NonEmpty (NonEmpty(..)) +import Data.Char +import Data.Data +import System.Console.Pretty +import qualified Text.PrettyPrint as PP +import qualified Data.Type.Equality as DE +import GHC.Exts (Constraint) +import qualified Text.Regex.PCRE.Heavy as RH +import qualified Text.Regex.PCRE.Light as RL +import qualified Data.ByteString.Char8 as B8 +import qualified Data.Discrimination as D +import qualified Data.Text as T +import Data.ByteString (ByteString) +import GHC.Word (Word8) +import Data.Sequence (Seq) +import Control.Applicative (ZipList) +import Data.Kind (Type) +import Data.Either +import Data.These +import Data.These.Combinators +import qualified Control.Exception as E +import Control.DeepSeq +import System.IO.Unsafe (unsafePerformIO) +import Data.Bool +import Data.Foldable + +-- | describes the evaluation tree for predicates +data TT a = TT { _tBool :: BoolT a -- ^ the value at this root node + , _tStrings :: [String] -- ^ detailed information eg input and output and text + , _tForest :: Forest PE } -- ^ the child nodes + deriving Show + +-- | contains the typed result from evaluating the expression tree to this point +data BoolT a where + FailT :: String -> BoolT a -- ^ failure with string + FalseT :: BoolT Bool -- ^ false predicate + TrueT :: BoolT Bool -- ^ true predicate + PresentT :: a -> BoolT a -- ^ non predicate value + +deriving instance Show a => Show (BoolT a) +deriving instance Eq a => Eq (BoolT a) + +tBool :: Lens (TT a) (TT b) (BoolT a) (BoolT b) +tBool afb s = (\b -> s { _tBool = b }) <$> afb (_tBool s) + +tStrings :: Lens' (TT a) [String] +tStrings afb s = (\b -> s { _tStrings = b }) <$> afb (_tStrings s) + +tForest :: Lens' (TT a) (Forest PE) +tForest afb s = (\b -> s { _tForest = b }) <$> afb (_tForest s) + +pStrings :: Lens' PE [String] +pStrings afb s = (\b -> s { _pStrings = b }) <$> afb (_pStrings s) + +-- | a lens from typed BoolT to the untyped BoolP +boolT2P :: Lens' (BoolT a) BoolP +boolT2P afb = \case + FailT e -> FailT e <$ afb (FailP e) + TrueT -> TrueT <$ afb TrueP + FalseT -> FalseT <$ afb FalseP + PresentT a -> PresentT a <$ afb PresentP + +-- | contains the untyped result from evaluating the expression tree to this point +data BoolP = + FailP String + | FalseP + | TrueP + | PresentP + deriving (Show, Eq) + +-- need a semigroup constraint else we have to throw away one of the PresentT a ie First or Last +instance Semigroup a => Semigroup (BoolT a) where + FailT e <> FailT e1 = FailT (e <> e1) + o@FailT {} <> _ = o + _ <> o@FailT {} = o + o@TrueT <> TrueT = o + o@FalseT <> _ = o + _ <> o@FalseT = o + -- cant pattern match on PresentT True on lhs (hence PresentT a) but can use 'a' as a Bool on rhs! + PresentT a <> TrueT = review _boolT a + TrueT <> PresentT a = review _boolT a + PresentT a <> PresentT a1 = PresentT (a <> a1) + +instance Monoid a => Monoid (BoolT a) where + mempty = PresentT mempty + +data PE = PE { _pBool :: BoolP -- ^ holds the result of running the predicate + , _pStrings :: [String] -- ^ optional strings to include in the results + } deriving Show + +pBool :: Lens' PE BoolP +pBool afb (PE x y) = flip PE y <$> afb x + +-- | creates a Node for the evaluation tree +mkNode :: POpts -> BoolT a -> [String] -> [Holder] -> TT a +mkNode opts bt ss hs + | oLite opts = TT bt [] [] + | otherwise = TT bt ss (map fromTTH hs) + +-- | creates a Boolean node for a predicate type +mkNodeB :: POpts -> Bool -> [String] -> [Holder] -> TT Bool +mkNodeB opts b = mkNode opts (bool FalseT TrueT b) + +-- | partition a tree into failures and non failures +partitionTTs :: [TT a] -> ([TT x], [TT a]) +partitionTTs = partitionEithers . map getTTLR + +getTTLR :: TT a -> Either (TT x) (TT a) +getTTLR t = + case _tBool t of + FailT e -> Left $ t & tBool .~ FailT e + _ -> Right t + +partitionTTExtended :: (w, TT a) -> ([((w, TT x), String)], [(w, TT a)]) +partitionTTExtended z@(_, t) = + case _tBool t of + FailT e -> ([(z & _2 . tBool .~ FailT e, e)], []) + _ -> ([], [z]) + +getValLRFromTT :: TT a -> Either String a +getValLRFromTT = getValLR . _tBool + +-- | get the value from BoolT or fail +getValLR :: BoolT a -> Either String a +getValLR = \case + FailT e -> Left e + TrueT -> Right True + FalseT -> Right False + PresentT a -> Right a + +shortTT :: BoolT Bool -> Either String String +shortTT z = case z of + FailT e -> Left $ "FailT " <> e + TrueT -> Right $ show z + FalseT -> Left $ show z + PresentT True -> Right $ show z + PresentT False -> Left $ show z + +-- | converts a typed tree to an untyped on for display +fromTT :: TT a -> Tree PE +fromTT (TT bt ss tt) = Node (PE (bt ^. boolT2P) ss) tt + +-- | a monomorphic container of trees +data Holder = forall w . Holder { unHolder :: TT w } + +-- | converts a typed tree into an untyped one +fromTTH :: Holder -> Tree PE +fromTTH (Holder x) = fromTT x + +-- | convenience method to wrap a typed tree +hh :: TT w -> Holder +hh = Holder + +-- | see 'getValueLRImpl' : add more detail to the tree if there are errors +getValueLR :: POpts -> String -> TT a -> [Holder] -> Either (TT x) a +getValueLR = getValueLRImpl True + +-- | see 'getValueLRImpl' : add less detail to the tree if there are errors +getValueLRHide :: POpts -> String -> TT a -> [Holder] -> Either (TT x) a +getValueLRHide = getValueLRImpl False + +-- elide FailT msg in tStrings[0] if showError is False +-- | a helper method to add extra context on failure to the tree or extract the value at the root of the tree +getValueLRImpl :: Bool -> POpts -> String -> TT a -> [Holder] -> Either (TT x) a +getValueLRImpl showError opts msg0 tt hs = + let tt' = hs ++ [hh tt] + in left (\e -> mkNode + opts + (FailT e) + [msg0 <> if showError then (if null msg0 then "" else " ") <> "[" <> e <> "]" + else ""] + tt' + ) + (getValLRFromTT tt) + +-- | the color palette for displaying the expression tree +newtype PColor = PColor { unPColor :: BoolP -> String -> String } + +-- | customizable options +data POpts = POpts { oShowA :: Maybe Int -- ^ length of data to display for 'showA' + , oDebug :: !Int -- ^ debug level + , oDisp :: Disp -- ^ how to display the tree orientation and unicode etc + , oHide :: !Int -- ^ hides one layer of a tree + , oColor :: !(String, PColor) -- ^ color palette used + , oLite :: !Bool + } + +-- | display format for the tree +data Disp = NormalDisp -- ^ draw horizontal tree + | Vertical !Int -- ^ draw vertical tree + | Unicode -- ^ use unicode + | PPTree -- ^ pretty printer tree + deriving (Show, Eq) + +instance Show POpts where + show opts = + "POpts: showA=" <> show (oShowA opts) + <> " debug=" <> show (oDebug opts) + <> " disp=" <> show (oDisp opts) + <> " hide=" <> show (oHide opts) + <> " color=" <> show (fst (oColor opts)) + <> " lite=" <> show (oLite opts) + +defOpts :: POpts +defOpts = POpts + { oShowA = Just 110 + , oDebug = 0 + , oDisp = NormalDisp + , oHide = 0 + , oColor = color1 + , oLite = False + } + +-- | skip colors and just return the summary +ol :: POpts +ol = o0 { oColor = color0, oLite = True } + +-- | skip the detail and just return the summary but keep the colors +olc :: POpts +olc = ol { oColor = color1 } + +o0 :: POpts +o0 = defOpts { oColor = color0 } + +-- | skip colors +o02 :: POpts +o02 = o2 { oColor = color0 } + +o03 :: POpts +o03 = o3 { oColor = color0 } + + +o1 :: POpts +o1 = defOpts { oDebug = 1, oShowA = Just 120 } + +-- | colors and details +o2 :: POpts +o2 = defOpts { oDebug = 2, oShowA = Just 200 } + +o3 :: POpts +o3 = defOpts { oDebug = 3, oShowA = Just 400 } + + +-- | helper method to set the width of data to be shown in the tree +seta :: Int -> POpts -> POpts +seta w o = o { oShowA = Just w } + +-- | helper method to display the tree vertically +setv :: Int -> POpts -> POpts +setv w o = o { oDisp = Vertical w } + +-- | helper method to set the debug level +setd :: Int -> POpts -> POpts +setd v o = o { oDebug = v } + +setu :: POpts -> POpts +setu o = o { oDisp = Unicode } + +setc :: (String, PColor) -> POpts -> POpts +setc pc o = o { oColor = pc } + +setc0, setc1, setc2, setc3, setc4 :: POpts -> POpts +setc0 o = o { oColor = color0 } +setc1 o = o { oColor = color1 } +setc2 o = o { oColor = color2 } +setc3 o = o { oColor = color3 } +setc4 o = o { oColor = color4 } + +-- | italics dont work but underline does +-- | color palettes +color0, color1, color2, color3, color4 :: (String, PColor) +color0 = ("color0", PColor $ flip const) + +color1 = + ("color1",) $ PColor $ \case + FailP {} -> bgColor Magenta + FalseP -> bgColor Red + TrueP -> bgColor Green + PresentP -> bgColor Yellow + +color2 = + ("color2",) $ PColor $ \case + FailP {} -> bgColor Magenta + FalseP -> bgColor Red + TrueP -> bgColor White + PresentP -> bgColor Yellow + +color3 = + ("color3",) $ PColor $ \case + FailP {} -> bgColor Blue + FalseP -> color Red + TrueP -> color White + PresentP -> bgColor Yellow + +color4 = + ("color4",) $ PColor $ \case + FailP {} -> bgColor Cyan + FalseP -> color Red + TrueP -> color Green + PresentP -> bgColor Yellow + +defh, defv, defu :: POpts +defh = o1 +defv = defv' defaultGap +defu = setu o1 + +defv' :: Width -> POpts +defv' w = setv w o1 + +-- | fix PresentT Bool to TrueT or FalseT +fixBoolT :: TT Bool -> TT Bool +fixBoolT t = + case t ^? tBool . _PresentT of + Nothing -> t + Just b -> t & tBool .~ _boolT # b + +showLit0 :: POpts -> String -> String -> String +showLit0 o s a = showLit' o 0 s a + +showLit3 :: POpts -> String -> String -> String +showLit3 o s a = showLit' o 3 s a + +showLit :: POpts -> String -> String -> String +showLit o s a = showLit' o 1 s a + +showLit' :: POpts -> Int -> String -> String -> String +showLit' o i s a = + if oDebug o >= i then + let f n = let ss = take n a + in ss <> (if length ss==n then " ..." else "") + in maybe "" (\n -> s <> f n) (oShowA o) + else "" + +show0 :: Show a => POpts -> String -> a -> String +show0 o s a = showA' o 0 s a + +show3 :: Show a => POpts -> String -> a -> String +show3 o s a = showA' o 3 s a + +showA :: Show a => POpts -> String -> a -> String +showA o s a = showA' o 1 s a + +showA' :: Show a => POpts -> Int -> String -> a -> String +showA' o i s a = showLit' o i s (show a) + +-- | Regex options for Rescan Resplit Re etc +data ROpt = + Anchored -- ^ Force pattern anchoring + | Auto_callout -- ^ Compile automatic callouts +-- | Bsr_anycrlf -- \R matches only CR, LF, or CrlF +-- | Bsr_unicode -- ^ \R matches all Unicode line endings + | Caseless -- ^ Do caseless matching + | Dollar_endonly -- ^ dollar not to match newline at end + | Dotall -- ^ matches anything including NL + | Dupnames -- ^ Allow duplicate names for subpatterns + | Extended -- ^ Ignore whitespace and # comments + | Extra -- ^ PCRE extra features (not much use currently) + | Firstline -- ^ Force matching to be before newline + | Multiline -- ^ caret and dollar match newlines within data +-- | Newline_any -- ^ Recognize any Unicode newline sequence +-- | Newline_anycrlf -- ^ Recognize CR, LF, and CrlF as newline sequences + | Newline_cr -- ^ Set CR as the newline sequence + | Newline_crlf -- ^ Set CrlF as the newline sequence + | Newline_lf -- ^ Set LF as the newline sequence + | No_auto_capture -- ^ Disable numbered capturing parentheses (named ones available) + | Ungreedy -- ^ Invert greediness of quantifiers + | Utf8 -- ^ Run in UTF--8 mode + | No_utf8_check -- ^ Do not check the pattern for UTF-8 validity + deriving (Show,Eq,Ord,Enum,Bounded) + +-- | compile a regex using the type level symbol +compileRegex :: forall rs a . GetROpts rs + => POpts -> String -> String -> [Holder] -> Either (TT a) RH.Regex +compileRegex opts nm s hhs = + let rs = getROpts @rs + mm = nm <> " " <> show rs + in flip left (RH.compileM (B8.pack s) rs) + $ \e -> mkNode opts (FailT "Regex failed to compile") [mm <> " compile failed with regex msg[" <> e <> "]"] hhs + +-- | extract the regex options from the type level list +class GetROpts (os :: [ROpt]) where + getROpts :: [RL.PCREOption] +instance GetROpts '[] where + getROpts = [] +instance (GetROpt r, GetROpts rs) => GetROpts (r ': rs) where + getROpts = getROpt @r : getROpts @rs + +-- | convert type level regex option to the value level +class GetROpt (o :: ROpt) where + getROpt :: RL.PCREOption +instance GetROpt 'Anchored where getROpt = RL.anchored +instance GetROpt 'Auto_callout where getROpt = RL.auto_callout +--instance GetROpt 'Bsr_anycrlf where getROpt = RL.bsr_anycrlf +--instance GetROpt 'Bsr_unicode where getROpt = RL.bsr_unicode +instance GetROpt 'Caseless where getROpt = RL.caseless +instance GetROpt 'Dollar_endonly where getROpt = RL.dollar_endonly +instance GetROpt 'Dotall where getROpt = RL.dotall +instance GetROpt 'Dupnames where getROpt = RL.dupnames +instance GetROpt 'Extended where getROpt = RL.extended +instance GetROpt 'Extra where getROpt = RL.extra +instance GetROpt 'Firstline where getROpt = RL.firstline +instance GetROpt 'Multiline where getROpt = RL.multiline +--instance GetROpt 'Newline_any where getROpt = RL.newline_any +--instance GetROpt 'Newline_anycrlf where getROpt = RL.newline_anycrlf +instance GetROpt 'Newline_cr where getROpt = RL.newline_cr +instance GetROpt 'Newline_crlf where getROpt = RL.newline_crlf +instance GetROpt 'Newline_lf where getROpt = RL.newline_lf +instance GetROpt 'No_auto_capture where getROpt = RL.no_auto_capture +instance GetROpt 'Ungreedy where getROpt = RL.ungreedy +instance GetROpt 'Utf8 where getROpt = RL.utf8 +instance GetROpt 'No_utf8_check where getROpt = RL.no_utf8_check + +-- | used by 'Predicate.ReplaceImpl' and 'RH.sub' and 'RH.gsub' to allow more flexible replacement +-- These parallel the RegexReplacement (not exported) class in "Text.Regex.PCRE.Heavy" but have overlappable instances which is problematic for this code so I use 'RR' +data RR = + RR String + | RR1 (String -> [String] -> String) + | RR2 (String -> String) + | RR3 ([String] -> String) + +-- | extract values from the trees or if there are errors returned a tree with added context +splitAndAlign :: Show x => + POpts + -> [String] + -> [((Int, x), TT a)] + -> Either (TT w) + ([a] + ,[((Int, x), TT a)] + ) +splitAndAlign opts msgs ts = + case mconcat $ map partitionTTExtended ts of + (excs@(e:_), _) -> + Left $ mkNode opts + (FailT (groupErrors (map snd excs))) + (msgs <> ["excs=" <> show (length excs) <> " " <> formatList opts [fst e]]) + (map (hh . snd) ts) + ([], tfs) -> Right (valsFromTTs (map snd ts), tfs) + +formatList :: forall x z . Show x => POpts -> [((Int, x), z)] -> String +formatList opts = unwords . map (\((i, a), _) -> "(i=" <> show i <> showA' opts 0 ", a=" a <> ")") + +-- | extract all root values from a list of trees +valsFromTTs :: [TT a] -> [a] +valsFromTTs = concatMap toList + +instance Foldable TT where + foldMap am = foldMap am . _tBool + +instance Foldable BoolT where + foldMap am = either (const mempty) am . getValLR + +isTrue :: BoolT Bool -> Bool +isTrue = and +--isTrue = or + +-- cant use: is / isn't / has cos only FailT will be False: use Fold +-- this is more specific to TrueP FalseP +-- | prism from BoolT to Bool +_boolT :: Prism' (BoolT Bool) Bool +_boolT = prism' (bool FalseT TrueT) + $ \case + PresentT a -> Just a + TrueT -> Just True + FalseT -> Just False + FailT {} -> Nothing + +groupErrors :: [String] -> String +groupErrors = intercalate " | " . map (\xs -> head xs <> (if length xs > 1 then "(" <> show (length xs) <> ")" else "")) . D.group + +_FailT :: Prism' (BoolT a) String +_FailT = prism' FailT $ \case + FailT s -> Just s + _ -> Nothing + +_PresentT :: Prism' (BoolT a) a +_PresentT = prism' PresentT $ \case + PresentT a -> Just a + _ -> Nothing + +_FalseT :: Prism' (BoolT Bool) () +_FalseT = prism' (const FalseT) $ + \case + FalseT -> Just () + _ -> Nothing + +_TrueT :: Prism' (BoolT Bool) () +_TrueT = prism' (const TrueT) $ + \case + TrueT -> Just () + _ -> Nothing + +imply :: Bool -> Bool -> Bool +imply p q = not p || q + +-- msg is only used for an exception: up to the calling programs to deal with ading msg to good and bad +-- | applies a boolean binary operation against the values from two boolean trees +evalBinStrict :: POpts + -> String + -> (Bool -> Bool -> Bool) + -> TT Bool + -> TT Bool + -> TT Bool +evalBinStrict opts s fn ll rr = + case getValueLR opts (s <> " p") ll [Holder rr] of + Left e -> e + Right a -> + case getValueLR opts (s <> " q") rr [hh ll] of + Left e -> e + Right b -> + let z = fn a b + in mkNodeB opts z [show a <> " " <> s <> " " <> show b] [hh ll, hh rr] + +-- | type level Between +type family BetweenT (a :: Nat) (b :: Nat) (v :: Nat) :: Constraint where + BetweenT m n v = + FailIfT (NotT (AndT (m GL.<=? v) (v GL.<=? n))) + ('GL.Text "BetweenT failure" + ':$$: 'GL.ShowType v + ':$$: 'GL.Text " is outside of " + ':$$: 'GL.ShowType m + ':<>: 'GL.Text " and " + ':<>: 'GL.ShowType n) + +-- | makes zero invalid at the type level +type NotZeroT v = FailIfT (v DE.== 0) ('GL.Text "found zero value") + +-- | typelevel Null on Symbol +type family NullT (x :: Symbol) :: Bool where + NullT ("" :: Symbol) = 'True + NullT _ = 'False + +-- | helper method to fail with an error if the True +type family FailIfT (b :: Bool) (msg :: GL.ErrorMessage) :: Constraint where + FailIfT 'False _ = () + FailIfT 'True e = GL.TypeError e + +-- | typelevel And +type family AndT (b :: Bool) (b1 :: Bool) :: Bool where + AndT 'False _ = 'False + AndT 'True b1 = b1 + +-- | typelevel Or +type family OrT (b :: Bool) (b1 :: Bool) :: Bool where + OrT 'True _ = 'True + OrT 'False b1 = b1 + +-- | typelevel Not +type family NotT (b :: Bool) :: Bool where + NotT 'True = 'False + NotT 'False = 'True + +-- | get a Nat from the typelevel +nat :: forall n a . (KnownNat n, Num a) => a +nat = fromIntegral (GL.natVal (Proxy @n)) + +-- | gets the Symbol from the typelevel +symb :: forall s . KnownSymbol s => String +symb = GL.symbolVal (Proxy @s) + +-- | get a list of Nats from the typelevel +class GetNats as where + getNats :: [Int] +instance GetNats '[] where + getNats = [] +instance (KnownNat n, GetNats ns) => GetNats (n ': ns) where + getNats = nat @n : getNats @ns + +-- | get a list of Symbols from the typelevel +class GetSymbs ns where + getSymbs :: [String] +instance GetSymbs '[] where + getSymbs = [] +instance (KnownSymbol s, GetSymbs ss) => GetSymbs (s ': ss) where + getSymbs = symb @s : getSymbs @ss + +getLen :: forall xs . GetLen xs => Int +getLen = getLenP (Proxy @xs) + +-- really need a proxy for this to work +-- | gets length of a typelevel list +class GetLen (xs :: [k]) where -- defaults to xs :: k (how to make it [k]) cos is not free + getLenP :: Proxy (xs :: [k]) -> Int +instance GetLen '[] where + getLenP _ = 0 +instance GetLen xs => GetLen (x ': xs) where + getLenP _ = 1 + getLenP (Proxy @xs) + +showThese :: These a b -> String +showThese = these (const "This") (const "That") (const (const "These")) + +-- hard without a Proxy +class GetThese (th :: These x y) where + getThese :: Proxy th -> (String, These w v -> Bool) +instance GetThese ('This x) where + getThese _ = ("This", isThis) +instance GetThese ('That y) where + getThese _ = ("That", isThat) +instance GetThese ('These x y) where + getThese _ = ("These", isThese) + +-- | get ordering from the typelevel +class GetOrdering (cmp :: Ordering) where + getOrdering :: Ordering +instance GetOrdering 'LT where + getOrdering = LT +instance GetOrdering 'EQ where + getOrdering = EQ +instance GetOrdering 'GT where + getOrdering = GT + +-- | get bool from the typelevel +class GetBool (a :: Bool) where + getBool :: Bool +instance GetBool 'True where + getBool = True +instance GetBool 'False where + getBool = False + +data N = S N | Z + +-- a shim for TupleListImpl used mainly by Printfn +-- | inductive numbers +type family ToN (n :: Nat) :: N where + ToN 0 = 'Z + ToN n = 'S (ToN (n GL.- 1)) + +-- | converts an inductive number to Nat +type family FromN (n :: N) :: Nat where + FromN 'Z = 0 + FromN ('S n) = 1 GL.+ FromN n + +-- | extract N from the type level to Int +class GetNatN (n :: N) where + getNatN :: Int +instance GetNatN 'Z where + getNatN = 0 +instance GetNatN n => GetNatN ('S n) where + getNatN = 1 + getNatN @n + +getN :: Typeable t => Proxy (t :: N) -> Int +getN p = length (show (typeRep p)) `div` 5 + +data OrderingP = Cgt | Cge | Ceq | Cle | Clt | Cne deriving (Show, Eq, Enum, Bounded) + +class GetOrd (k :: OrderingP) where + getOrd :: Ord a => (String, a -> a -> Bool) + +instance GetOrd 'Cgt where getOrd = (">", (>)) +instance GetOrd 'Cge where getOrd = (">=",(>=)) +instance GetOrd 'Ceq where getOrd = ("==",(==)) +instance GetOrd 'Cle where getOrd = ("<=",(<=)) +instance GetOrd 'Clt where getOrd = ("<", (<)) +instance GetOrd 'Cne where getOrd = ("/=",(/=)) + +-- only hides BoolP part! not sure of the point +toNodeString :: POpts -> PE -> String +toNodeString opts bpe + | oLite opts = error $ "shouldnt be calling this if we are going lite: toNodeString oLite " ++ show bpe + | otherwise = showBoolP opts (_pBool bpe) <> " " <> displayMessages (_pStrings bpe) + +nullSpace :: String -> String +nullSpace s | null s = "" + | otherwise = " " <> s + +showBoolP :: POpts -> BoolP -> String +showBoolP o = + \case + b@(FailP e) -> "[" <> colorMe o b "Error" <> nullSpace e <> "]" + b@PresentP -> colorMe o b "P" + b@TrueP -> colorMe o b "True " + b@FalseP -> colorMe o b "False" + +displayMessages :: [String] -> String +displayMessages es = + case filter (not . all isSpace) es of + [] -> "" + z -> intercalate " | " z + +-- | colors the result of the predicate based on the current color palette +colorMe :: POpts -> BoolP -> String -> String +colorMe o b s = + let (_, PColor f) = oColor o + in f b s + +prtTTIO :: POpts -> IO (TT a) -> IO () +prtTTIO = prtTT' + +prtTT :: POpts -> Identity (TT a) -> IO () +prtTT = prtTT' + +prtTT' :: MonadEval m => POpts -> m (TT a) -> IO () +prtTT' o y = liftEval y >>= prtTree o . fromTT + +prtTree :: POpts -> Tree PE -> IO () +prtTree o = prtImpl o . fmap (toNodeString o) + +prtImpl :: POpts -> Tree String -> IO () +prtImpl = (putStr .) . showImpl + +fixLite :: forall a . Show a => POpts -> a -> Tree PE -> String +fixLite opts a t + | oLite opts = fixPresentP opts (t ^. root . pBool) a <> "\n" + | otherwise = prtTreePure opts t + +fixPresentP :: Show a => POpts -> BoolP -> a -> String +fixPresentP opts bp a = + case bp of + PresentP -> colorMe opts PresentP "Present" <> " " <> show a + _ -> showBoolP opts bp + +prtTreePure :: POpts -> Tree PE -> String +prtTreePure opts t + | oLite opts = showBoolP opts (t ^. root . pBool) + | otherwise = showImpl opts $ fmap (toNodeString opts) t + +showImpl :: POpts -> Tree String -> String +showImpl o = + case oDisp o of + Unicode -> TV.showTree + NormalDisp -> drawTree -- to drop the last newline else we have to make sure that everywhere else has that newline: eg fixLite + Vertical w -> drawVerticalTreeWith w + PPTree -> (<>"\n") . PP.render . ppTree PP.text -- no newlines! + +lite :: POpts -> POpts +lite o = o { oLite = True } + +unicode :: POpts -> POpts +unicode o = o { oDisp = Unicode } + +horizontal :: POpts -> POpts +horizontal o = o { oDisp = NormalDisp } + +vertical :: POpts -> POpts +vertical = vertical' defaultGap + +vertical' :: Width -> POpts -> POpts +vertical' w o = o { oDisp = Vertical w } + + +-- | display in document in tree format +ppTree :: (a -> PP.Doc) -> Tree a -> PP.Doc +ppTree pp = ppT + where + ppT (Node x []) = pp x + ppT (Node x xs) = PP.parens $ PP.hang (pp x) 2 $ + PP.sep $ map ppT xs + + +prettyRational :: Rational -> String +prettyRational (numerator &&& denominator -> (n,d)) = + if | n == 0 -> "0" + | d == 1 -> show n + | otherwise -> show n <> " / " <> show d + +fixit :: ((Int, x), TT a) -> TT a +fixit ((i, _), t) = prefixMsg ("i=" <> show i <> ":") t + +prefixMsg :: String -> TT a -> TT a +prefixMsg msg t = + t & tStrings .ix 0 %~ (msg <>) + +showNat :: forall n . KnownNat n => String +showNat = show (nat @n :: Integer) + +showT :: forall (t :: Type) . Typeable t => String +showT = show (typeRep (Proxy @t)) + +showTProxy :: forall p . Typeable (Proxy p) => String +showTProxy = drop 8 $ show (typeOf (Proxy @p)) + +prettyOrd :: Ordering -> String +prettyOrd = \case + LT -> "<" + EQ -> "=" + GT -> ">" + +type family RepeatT (n :: Nat) (p :: k) :: [k] where + RepeatT 0 p = GL.TypeError ('GL.Text "RepeatT is not defined for zero") + RepeatT 1 p = p ': '[] + RepeatT n p = p ': RepeatT (n GN.- 1) p + +type family IntersperseT (s :: Symbol) (xs :: [Symbol]) :: Symbol where + IntersperseT s '[] = "" + IntersperseT s '[x] = x + IntersperseT s (x ': y ': xs) = x `GL.AppendSymbol` s `GL.AppendSymbol` IntersperseT s (y ': xs) + +type family LenT (xs :: [k]) :: Nat where + LenT '[] = 0 + LenT (x ': xs) = 1 GN.+ LenT xs + + +type NEmptyT k = ('[] ':| '[] :: NonEmpty [k]) + +isPrime :: Integer -> Bool +isPrime n = n==2 || n>2 && all ((> 0).rem n) (2:[3,5 .. floor . sqrt @Double . fromIntegral $ n+1]) + +type family TupleListT (n :: N) a where + TupleListT 'Z a = () + TupleListT ('S n) a = (a, TupleListT n a) + +class TupleListD (n :: N) a where + tupleListD :: Bool -> [a] -> Either String (TupleListT n a) + +instance TupleListD 'Z a where + tupleListD isStrict = \case + z@(_:_) | isStrict -> + let len = length z + in Left $ "is strict and has " <> show len <> " extra element" <> (if len == 1 then "" else "s") + _ -> Right () + +instance (TupleListD n a) => TupleListD ('S n) a where + tupleListD isStrict = \case + [] -> Left "no data left" -- nothing i can do here even if not strict + a:as -> (a,) <$> tupleListD @n @a isStrict as + +-- up to 12 +class ReverseTupleC x where + type ReverseTupleP x + reverseTupleC :: x -> ReverseTupleP x +instance (GL.TypeError ('GL.Text "ReverseTupleC: inductive tuple cannot be empty")) => ReverseTupleC () where + type ReverseTupleP () = () + reverseTupleC () = () +instance ReverseTupleC (a,()) where + type ReverseTupleP (a,()) = (a,()) + reverseTupleC (a,()) = (a,()) +instance ReverseTupleC (a,(b,())) where + type ReverseTupleP (a,(b,())) = (b,(a,())) + reverseTupleC (a,(b,())) = (b,(a,())) +instance ReverseTupleC (a,(b,(c,()))) where + type ReverseTupleP (a,(b,(c,()))) = (c,(b,(a,()))) + reverseTupleC (a,(b,(c,()))) = (c,(b,(a,()))) +instance ReverseTupleC (a,(b,(c,(d,())))) where + type ReverseTupleP (a,(b,(c,(d,())))) = (d,(c,(b,(a,())))) + reverseTupleC (a,(b,(c,(d,())))) = (d,(c,(b,(a,())))) +instance ReverseTupleC (a,(b,(c,(d,(e,()))))) where + type ReverseTupleP (a,(b,(c,(d,(e,()))))) = (e,(d,(c,(b,(a,()))))) + reverseTupleC (a,(b,(c,(d,(e,()))))) = (e,(d,(c,(b,(a,()))))) +instance ReverseTupleC (a,(b,(c,(d,(e,(f,())))))) where + type ReverseTupleP (a,(b,(c,(d,(e,(f,())))))) = (f,(e,(d,(c,(b,(a,())))))) + reverseTupleC (a,(b,(c,(d,(e,(f,())))))) = (f,(e,(d,(c,(b,(a,())))))) +instance ReverseTupleC (a,(b,(c,(d,(e,(f,(g,()))))))) where + type ReverseTupleP (a,(b,(c,(d,(e,(f,(g,()))))))) = (g,(f,(e,(d,(c,(b,(a,()))))))) + reverseTupleC (a,(b,(c,(d,(e,(f,(g,()))))))) = (g,(f,(e,(d,(c,(b,(a,()))))))) +instance ReverseTupleC (a,(b,(c,(d,(e,(f,(g,(h,())))))))) where + type ReverseTupleP (a,(b,(c,(d,(e,(f,(g,(h,())))))))) = (h,(g,(f,(e,(d,(c,(b,(a,())))))))) + reverseTupleC (a,(b,(c,(d,(e,(f,(g,(h,())))))))) = (h,(g,(f,(e,(d,(c,(b,(a,())))))))) +instance ReverseTupleC (a,(b,(c,(d,(e,(f,(g,(h,(i,()))))))))) where + type ReverseTupleP (a,(b,(c,(d,(e,(f,(g,(h,(i,()))))))))) = (i,(h,(g,(f,(e,(d,(c,(b,(a,()))))))))) + reverseTupleC (a,(b,(c,(d,(e,(f,(g,(h,(i,()))))))))) = (i,(h,(g,(f,(e,(d,(c,(b,(a,()))))))))) +instance ReverseTupleC (a,(b,(c,(d,(e,(f,(g,(h,(i,(j,())))))))))) where + type ReverseTupleP (a,(b,(c,(d,(e,(f,(g,(h,(i,(j,())))))))))) = (j,(i,(h,(g,(f,(e,(d,(c,(b,(a,())))))))))) + reverseTupleC (a,(b,(c,(d,(e,(f,(g,(h,(i,(j,())))))))))) = (j,(i,(h,(g,(f,(e,(d,(c,(b,(a,())))))))))) +instance ReverseTupleC (a,(b,(c,(d,(e,(f,(g,(h,(i,(j,(k,()))))))))))) where + type ReverseTupleP (a,(b,(c,(d,(e,(f,(g,(h,(i,(j,(k,()))))))))))) = (k,(j,(i,(h,(g,(f,(e,(d,(c,(b,(a,()))))))))))) + reverseTupleC (a,(b,(c,(d,(e,(f,(g,(h,(i,(j,(k,()))))))))))) = (k,(j,(i,(h,(g,(f,(e,(d,(c,(b,(a,()))))))))))) +instance ReverseTupleC (a,(b,(c,(d,(e,(f,(g,(h,(i,(j,(k,(l,())))))))))))) where + type ReverseTupleP (a,(b,(c,(d,(e,(f,(g,(h,(i,(j,(k,(l,())))))))))))) = (l,(k,(j,(i,(h,(g,(f,(e,(d,(c,(b,(a,())))))))))))) + reverseTupleC (a,(b,(c,(d,(e,(f,(g,(h,(i,(j,(k,(l,())))))))))))) = (l,(k,(j,(i,(h,(g,(f,(e,(d,(c,(b,(a,())))))))))))) + +-- a hack to get 'a' from '[a]' which I need for type PP +type family ArrT (as :: Type) :: Type where + ArrT [a] = a + ArrT as = GL.TypeError ( + 'GL.Text "ArrT: expected [a] but found something else" + ':$$: 'GL.Text "as = " + ':<>: 'GL.ShowType as) + +type family TupleLenT (t :: Type) :: Nat where + TupleLenT () = 0 + TupleLenT (_,ts) = 1 GN.+ TupleLenT ts + TupleLenT t = GL.TypeError ( + 'GL.Text "TupleLenT: expected a valid inductive tuple" + ':$$: 'GL.Text "t = " + ':<>: 'GL.ShowType t) + +-- partially apply the 2nd arg to an ADT -- $ and & work with functions only +-- doesnt apply more than once because we need to eval it +type family (p :: k -> k1) % (q :: k) :: k1 where + p % q = p q + +infixl 9 % + +type family (p :: k) %& (q :: k -> k1) :: k1 where + p %& q = q p + +infixr 9 %& + +type family FlipT (d :: k1 -> k -> k2) (p :: k) (q :: k1) :: k2 where + FlipT d p q = d q p + +type family IfT (b :: Bool) (t :: k) (f :: k) :: k where + IfT 'True t f = t + IfT 'False t f = f + +type family SumT (ns :: [Nat]) :: Nat where + SumT '[] = 0 + SumT (n ': ns) = n GL.+ SumT ns + +-- only works if you use ADTs not type synonyms +type family MapT (f :: k -> k1) (xs :: [k]) :: [k1] where + MapT f '[] = '[] + MapT f (x ': xs) = f x ': MapT f xs + +type family ConsT s where + ConsT [a] = a + ConsT (ZipList a) = a + ConsT T.Text = Char + ConsT ByteString = Word8 + ConsT (Seq a) = a + ConsT s = GL.TypeError ( + 'GL.Text "ConsT: not a valid ConsT" + ':$$: 'GL.Text "s = " + ':<>: 'GL.ShowType s) + +class Monad m => MonadEval m where + runIO :: IO a -> m (Maybe a) + catchit :: E.Exception e => a -> m (Either String a) + catchitNF :: (E.Exception e, NFData a) => a -> m (Either String a) + liftEval :: m a -> IO a + +instance MonadEval Identity where + runIO _ = Identity Nothing + catchit v = Identity $ unsafePerformIO $ catchit @IO @E.SomeException v + catchitNF v = Identity $ unsafePerformIO $ catchitNF @IO @E.SomeException v + liftEval = return . runIdentity + +instance MonadEval IO where + runIO ioa = Just <$> ioa + catchit v = E.evaluate (Right $! v) `E.catch` (\(E.SomeException e) -> pure $ Left ("IO e=" <> show e)) + catchitNF v = E.evaluate (Right $!! v) `E.catch` (\(E.SomeException e) -> pure $ Left ("IO e=" <> show e)) + liftEval = id + +removeAnsiForDocTest :: Show a => Either String a -> IO () +removeAnsiForDocTest = + \case + Left e -> let esc = '\x1b' + f :: String -> Maybe (String, String) + f = \case + [] -> Nothing + c:cs | c == esc -> case break (=='m') cs of + (_,'m':s) -> Just ("",s) + _ -> Nothing + | otherwise -> Just $ break (==esc) (c:cs) + in putStrLn $ concat $ unfoldr f e + Right a -> print a +
+ src/UtilP_TH.hs view
@@ -0,0 +1,83 @@+{-# OPTIONS -Wall #-} +{-# LANGUAGE ScopedTypeVariables #-} +{-# LANGUAGE FlexibleContexts #-} +{-# LANGUAGE OverloadedStrings #-} +{-# LANGUAGE TypeApplications #-} +{-# LANGUAGE GADTs #-} +{-# LANGUAGE TemplateHaskell #-} +{-# LANGUAGE KindSignatures #-} +{-# LANGUAGE AllowAmbiguousTypes #-} +{-# LANGUAGE PolyKinds #-} +-- | +-- Module : UtilP_TH +-- Description : Template Haskell methods for creating Refined and Refined3 refinement types +-- Copyright : (c) Grant Weyburne, 2019 +-- License : BSD-3 +-- Maintainer : gbwey9@gmail.com +-- +-- Template Haskell methods for creating Refined and Refined3 refinement types +-- +module UtilP_TH + ( refined3TH + , refined3TH' + , refinedTH + , refinedTH' + ) where +import Refined3 +import Refined +import Predicate +import qualified Language.Haskell.TH.Syntax as TH +import Data.Functor.Identity +import UtilP +import Control.Monad (unless) + +-- | creates a 'Refined3.Refined3' refinement type with terse output +refined3TH :: forall ip op fmt i + . (Show i, Show (PP ip i), TH.Lift i, TH.Lift (PP ip i), Refined3C ip op fmt i) + => i + -> TH.Q (TH.TExp (Refined3 ip op fmt i)) +refined3TH = refined3TH' ol + +-- | creates a 'Refined3.Refined3' refinement type and allows you to specify options (ol to just get a summary / o2 for full detail and colors) +refined3TH' :: forall ip op fmt i + . (Show i, Show (PP ip i), TH.Lift i, TH.Lift (PP ip i), Refined3C ip op fmt i) + => POpts + -> i + -> TH.Q (TH.TExp (Refined3 ip op fmt i)) +refined3TH' opts i = do + let msg0 = "refined3TH" + (ret,mr) = eval3 @ip @op @fmt opts i + m3 = prt3Impl opts ret + unless (oLite opts) $ + TH.runIO $ do + putStrLn $ "\n>>>>>>> Start " ++ msg0 ++ " " ++ show i + putStrLn $ m3Long m3 + putStrLn $ "<<<<<<< End "++ msg0 ++ " " ++ show i -- ++ "\n" + case mr of + Nothing -> fail $ msg0 ++ ": predicate failed with " ++ (m3Desc m3 <> " | " <> m3Short m3) + Just r -> TH.TExp <$> TH.lift r + +-- | creates a 'Refined.Refined' refinement type with terse output +refinedTH :: forall p i + . (Show i, TH.Lift i, RefinedC p i) + => i + -> TH.Q (TH.TExp (Refined p i)) +refinedTH = refinedTH' ol + +-- | creates a 'Refined.Refined' refinement type and allows you to specify options +refinedTH' :: forall p i + . (Show i, TH.Lift i, RefinedC p i) + => POpts + -> i + -> TH.Q (TH.TExp (Refined p i)) +refinedTH' opts i = do + let msg0 = "refinedTH" + let ((bp,e),mr) = runIdentity $ newRefined @p opts i + unless (oLite opts) $ + TH.runIO $ do + putStrLn $ "\n>>>>>>> Start " ++ msg0 ++ " " ++ show i + putStrLn $ "\n" ++ e + putStrLn $ "<<<<<<< End " ++ msg0 ++ " " ++ show i + case mr of + Nothing -> fail $ msg0 ++ ": predicate failed with " ++ show bp -- ++ "\n" ++ e + Just r -> TH.TExp <$> TH.lift r
+ test/TastyExtras.hs view
@@ -0,0 +1,78 @@+{-# OPTIONS -Wall #-} +{-# OPTIONS -Wcompat #-} +{-# OPTIONS -Wincomplete-record-updates #-} +{-# OPTIONS -Wincomplete-uni-patterns #-} +{-# OPTIONS -Wno-type-defaults #-} +{-# OPTIONS -Wno-redundant-constraints #-} +{-# LANGUAGE TypeOperators #-} +{-# LANGUAGE AllowAmbiguousTypes #-} +{-# LANGUAGE TypeApplications #-} +{-# LANGUAGE DataKinds #-} +{-# LANGUAGE GADTs #-} +{-# LANGUAGE PolyKinds #-} +{-# LANGUAGE ScopedTypeVariables #-} +{-# LANGUAGE OverloadedStrings #-} +{-# LANGUAGE NoOverloadedLists #-} -- overloaded lists breaks some predicates +{-# LANGUAGE ViewPatterns #-} +{-# LANGUAGE FlexibleContexts #-} +{-# LANGUAGE NoStarIsType #-} +{-# LANGUAGE DeriveGeneric #-} +{-# LANGUAGE TemplateHaskell #-} +{-# LANGUAGE LambdaCase #-} +module TastyExtras where +import Test.Tasty +import Test.Tasty.HUnit +import UtilP +import Data.Aeson +import Data.List +import Text.Show.Functions () + +expectIO :: (HasCallStack, Show a) => IO (Either String a) -> (Either String a -> Either String ()) -> IO () +expectIO iolr p = do + lr <- iolr + case p lr of + Left e -> assertFailure $ "expectIO: " <> e <> " lr=" <> show lr + Right () -> pure () + +expectLeftWith :: Show a => String -> Either String a -> Either String () +expectLeftWith _ (Right a) = Left $ "expected fail but was actually successful " ++ show a +expectLeftWith n (Left s) + | n `isInfixOf` s = Right () + | otherwise = Left $ "found fail but infix string did not match: actual[" ++ s ++ "] infix[" ++ n ++ "]" + +expectLeft :: Show b => Either a b -> IO () +expectLeft = \case + Left _ -> pure () + Right e -> assertFailure $ "expected Left but found Right " ++ show e + +expectRight :: Show a => Either a b -> IO () +expectRight = \case + Right _ -> pure () + Left e -> assertFailure $ "expected Right but found Left " ++ show e + +toFrom :: (FromJSON a1, ToJSON a2, a1 ~ a2) => a2 -> Either String a1 +toFrom = eitherDecode . encode + +orderTests :: [Assertion] -> [TestTree] +orderTests = zipWith (\i t -> testCase (show i) t) [1::Int ..] + +expectPE :: (Show a, Eq a, HasCallStack) => BoolT a -> IO (BoolT a) -> IO () +expectPE bp m = do + x <- m + print (x,bp) + bp @?= x + +expectJ :: (HasCallStack, Show a, Eq a) + => Either [String] a + -> Either String a + -> IO () +expectJ lhs rhs = + case (lhs,rhs) of + (Left _e,Right r) -> assertFailure $ "expected left but found right " <> show r + (Right r,Right r1) -> r @?= r1 + (Right _r,Left e) -> assertFailure $ "expected right but found left " <> e + (Left ss, Left e) + | all (`isInfixOf` e) ss -> pure () + | otherwise -> assertFailure $ "both left but expected " <> (show ss) <> " in " <> e + +
+ test/TestJson.hs view
@@ -0,0 +1,94 @@+{-# OPTIONS -Wall #-} +{-# OPTIONS -Wcompat #-} +{-# OPTIONS -Wincomplete-record-updates #-} +{-# OPTIONS -Wincomplete-uni-patterns #-} +{-# OPTIONS -Wno-type-defaults #-} +{-# OPTIONS -Wno-redundant-constraints #-} +{-# LANGUAGE TypeOperators #-} +{-# LANGUAGE AllowAmbiguousTypes #-} +{-# LANGUAGE TypeApplications #-} +{-# LANGUAGE DataKinds #-} +{-# LANGUAGE GADTs #-} +{-# LANGUAGE PolyKinds #-} +{-# LANGUAGE ScopedTypeVariables #-} +{-# LANGUAGE OverloadedStrings #-} +{-# LANGUAGE ViewPatterns #-} +{-# LANGUAGE FlexibleContexts #-} +{-# LANGUAGE NoStarIsType #-} +{-# LANGUAGE DeriveGeneric #-} +{-# LANGUAGE TemplateHaskell #-} +module TestJson where +import TastyExtras +import Test.Tasty +import Test.Tasty.HUnit +import Predicate +import Refined +import Refined3 +import Refined3Helper +import UtilP +import GHC.Generics (Generic) +import Data.Text (Text) +import Data.Aeson + +suite :: IO () +suite = defaultMain $ testGroup "testrefined" + [ testCase "testperson ok" $ expectIO testPerson (() <$) + , testCase "testperson1 ok" $ expectIO (testPerson1 2) (() <$) + , testCase "testperson1 bad ipaddress" $ expectIO (testPerson1 3) (expectLeftWith "expected between 0 and 255 found 260") + , testCase "testperson1 bad lastname lowercase first letter" $ expectIO (testPerson1 4) (expectLeftWith "invalid name(diaz)") + , testCase "testperson1 age 99 out of range" $ expectIO (testPerson1 5) (expectLeftWith "Error in $[0].age1") + ] + +testPerson :: IO (Either String [Person]) +testPerson = eitherDecodeFileStrict' "test1.json" + +testPerson1 :: Int -> IO (Either String [Person1]) +testPerson1 i = do + let fn = "test" ++ show i ++ ".json" + eitherDecodeFileStrict' fn + +data Person = Person { + firstName :: !Text + , lastName :: !Text + , age :: !Int + , likesPizza :: Bool + } deriving (Show,Generic,Eq) + +instance ToJSON Person +instance FromJSON Person + +data Person1 = Person1 { + firstName1 :: NameR + , lastName1 :: NameR + , age1 :: AgeR + , likesPizza1 :: Bool + , date1 :: DateTimeNR + , ipaddress1 :: Ip4R + } deriving (Show,Generic,Eq) + +instance ToJSON Person1 +instance FromJSON Person1 + +type ValidName = + Guard (Printf "invalid name(%s)" Id) + (Re "^[A-Z][a-z']+$" Id) >> 'True + +type NameR = Refined ValidName String + +type NameR1 = Refined (Name1 >> 'True) String +type Name1 = + Uncons + >> 'Just Id + >> Guard (Printf "not upper first(%c)" Fst) (Fst >> '[Id] >> IsCharSet 'CUpper) + >> Guard (Printf "not lower rest(%s)" Snd) (Snd >> IsCharSet 'CLower) + +type AgeR = Refined (Between 10 60) Int + +type Ip4R = MakeR3 '(Ip4ip, Ip4op >> 'True, Ip4fmt, String) + +type Ip4ip = Map (ReadP Int) (Resplit "\\." Id) +type Ip4op = Guard (Printf "expected length 4 found %d" Len) (Len >> Same 4) + >> GuardsQuick (Printf2 "guard(%d): expected between 0 and 255 found %d") (RepeatT 4 (Between 0 255)) +type Ip4fmt = Printfnt 4 "%03d.%03d.%03d.%03d" + +type DateTimeNR = MakeR3 DateTimeN
+ test/TestPredicate.hs view
@@ -0,0 +1,919 @@+{-# OPTIONS -Wall #-} +{-# OPTIONS -Wcompat #-} +{-# OPTIONS -Wincomplete-record-updates #-} +{-# OPTIONS -Wincomplete-uni-patterns #-} +{-# OPTIONS -Wno-type-defaults #-} +{-# OPTIONS -Wno-redundant-constraints #-} +{-# LANGUAGE TypeOperators #-} +{-# LANGUAGE AllowAmbiguousTypes #-} +{-# LANGUAGE TypeApplications #-} +{-# LANGUAGE DataKinds #-} +{-# LANGUAGE GADTs #-} +{-# LANGUAGE PolyKinds #-} +{-# LANGUAGE ScopedTypeVariables #-} +{-# LANGUAGE OverloadedStrings #-} +{-# LANGUAGE NoOverloadedLists #-} -- overloaded lists breaks some predicates +{-# LANGUAGE ViewPatterns #-} +{-# LANGUAGE FlexibleContexts #-} +{-# LANGUAGE NoStarIsType #-} +{-# LANGUAGE DeriveGeneric #-} +{-# LANGUAGE TemplateHaskell #-} +{-# LANGUAGE LambdaCase #-} +module TestPredicate where +import TastyExtras +import Test.Tasty +import Test.Tasty.HUnit +--import Test.Tasty.QuickCheck +import TestRefined +import TestRefined3 +import Predicate +import Refined +import Refined3 +import Refined3Helper +import UtilP +import Data.Ratio + +import Data.Typeable +import Control.Lens +import Data.Time +import Text.Show.Functions () +import Data.Functor.Compose +import qualified Data.Map.Strict as M +import qualified Data.Set as S +import qualified Data.Text as T +import qualified Data.Monoid as MM +import qualified Data.Semigroup as SG +import Data.These +import Data.These.Lens () + +suite :: IO () +suite = defaultMain $ testGroup "TestPredicate" (orderTests allTests) + +allTests :: [IO ()] +allTests = + [expectPE (PresentT LT) $ pl @("aa" === Id) "aaaa" + , expectPE FalseT $ pl @("aa" === Id >> FromEnum >> Same 1) "aaaa" + , expectPE (PresentT (Right 1)) $ pl @(HeadDef 'False Id +++ Id) (Right @[Bool] 1) -- need @[Bool] cos we said 'False! + , expectPE (PresentT (Left True)) $ pl @(HeadDef 'False Id +++ Id) (Left @_ @Int [True,False]) -- need @[Bool] cos we said 'False! + , expectPE (PresentT (Right True)) $ pl @(Not +++ Id) (Right True) + , expectPE (PresentT (4,4)) $ pl @(Dup >> Id) 4 + , expectPE (PresentT (Right 12)) $ pl @(Not +++ Id) (Right 12) + , expectPE (PresentT (Right 1)) $ pl @(HeadDef () Id +++ Id) (Right @[()] 1) -- breaks otherwise: Id says () -> () so has to be a list of [()] + , expectPE (PresentT (Right 1)) $ pl @(HeadDef () Id +++ Id) (Right @[()] 1) -- this breaks! cos Left doesnt have a type + , expectPE FalseT $ pl @(Fst >> Len >> Le 6 >> Not) ([2..7],True) + , expectPE TrueT $ pl @(Fst >> Len >> Le 6) ([2..7],True) + , expectPE TrueT $ pl @(Length Fst <= 6) ([2..7],True) + , expectPE TrueT $ pl @(Fst >> (Len >> Le 6)) ([2..7],True) + , expectPE FalseT $ pl @(HeadDef 12 Fst >> Le 6) ([],True) + , expectPE TrueT $ pl @(HeadDef 1 Fst >> Le 6) ([],True) + , expectPE (FailT "Head(empty)") $ pl @(Head' Fst >> Le 6) ([] @Int, True) + , expectPE FalseT $ pl @(HeadDef 10 Fst >> Le 6) ([],True) + , expectPE (FailT "zz") $ pl @(HeadFail "zz" Fst >> Le 6) ([],True) + , expectPE (FailT "failed1") $ pl @((HeadFail "failed1" Fst >> Le 6) `OR` 'False) ([],True) + , expectPE TrueT $ pl @((Fst >> HeadFail "failed2" Id >> Le (Neg 6)) `OR` 'False) ([-9],True) + , expectPE (FailT "failed3") $ pl @((Fst >> Failt _ "failed3" >> Le (Neg 6)) `OR` 'False) ([-5],True) + , expectPE TrueT $ pl @(MaybeIn 'True Id) (Nothing @Bool) -- need @() else breaks + , expectPE (PresentT 10) $ pl @(MaybeIn (Failt _ "failed4") Id) (Just 10) + , expectPE (PresentT 10) $ pl @(Just' Id) (Just 10) + , expectPE FalseT $ pl @(MaybeIn 'False Id) (Nothing @Bool) -- breaks otherwise + , expectPE FalseT $ pl @(Id > "xx") "abc" + , expectPE TrueT $ pl @(Id > "aa") "abc" + , expectPE TrueT $ pl @(Gt 4) 5 + , expectPE TrueT $ pl @(Map (Gt 3) Fst >> Ors) ([10,12,3,5],"ss") + , expectPE FalseT $ pl @(Map (Gt 3) Fst >> Ands) ([10,12,3,5],"ss") + , expectPE (PresentT [False,False,False,True]) $ pl @(Map (Mod Id 3) Fst >> Map (Gt 1) Id) ([10,12,3,5],"ss") + , expectPE (PresentT (12,5)) $ pl @(Fst >> Dup >> Star (Ix 1 (Failp "failed5")) (Ix 3 (Failp "failed5")) >> Id) ([10,12,3,5],"ss") + , expectPE FalseT $ pl @(Fst >> Dup >> Star (Ix 1 (Failp "failed5")) (Ix 3 (Failp "failed5")) >> Fst < Snd) ([10,12,3,5],"ss") + , expectPE TrueT $ pl @(Fst >> Dup >> Star (Ix 1 (Failp "failed5")) (Ix 3 (Failp "failed5")) >> Fst > Snd) ([10,12,3,5],"ss") + , expectPE TrueT $ pl @(Fst > Snd) (True,False) + , expectPE FalseT $ pl @(Fst == Snd) (True,False) + , expectPE TrueT $ pl @(Star Not Id >> Fst == Snd) (True,False) + , expectPE FalseT $ pl @(Snd >> Len &&& Ix 3 (Failp "someval1") >> Fst == Snd) ('x',[1..5]) + , expectPE FalseT $ pl @(Snd >> Len &&& Ix 3 (Failp "someval2") >> Fst < Snd) ('x',[1..5]) + , expectPE TrueT $ pl @(Snd >> Len &&& Ix 3 (Failp "someval3") >> Fst > Snd) ('x',[1..5]) + , expectPE FalseT $ pl @(Snd >> SplitAt 2 Id >> Star Len Len >> Fst > Snd) ('x',[1..5]) + , expectPE FalseT $ pl @(Map (Same 2) Id >> Ors) [1,4,5] + , expectPE TrueT $ pl @(Map (Same 2) Id >> Ors) [1,4,5,2,1] + , expectPE TrueT $ pl @(Elem Id '[2,3,4]) 2 + , expectPE FalseT $ pl @(Elem Id '[2,3,4]) 6 + , expectPE TrueT $ pl @(Elem Id '[PosR 13 2]) 6.5 + , expectPE TrueT $ pl @(Elem Id '[PosR 13 2, Pos 12]) 6.5 + , expectPE FalseT $ pl @(Elem Id '[PosR 13 2, Pos 12]) 6 + , expectPE (FailT "lhs") $ pl @(Map Len Id >> (Ix 3 (Failp "lhs")) &&& (Ix 0 5) >> Fst == Snd) [[1..4],[4..5]] + , expectPE FalseT $ pl @(Map Len Id >> (Ix 0 (Failp "lhs")) &&& (Ix 1 5) >> Fst == Snd) [[1..4],[4..5]] + , expectPE (FailT "rhs") $ pl @(Map Len Id >> (Ix 1 (Failp "lhs")) &&& (Ix 3 (Failp "rhs")) >> Fst == Snd) [[1..4],[4..5]] + , expectPE (FailT "lhs") $ pl @(Map Len Id >> (Ix 10 (Failp "lhs")) &&& (Ix 1 (Failp "rhs")) >> Fst == Snd) [[1..4],[4..5]] + , expectPE (FailT "rhs") $ pl @(Map Len Id >> (Ix 0 (Failp "lhs")) &&& (Ix 10 (Failp "rhs")) >> Fst == Snd) [[1..4],[4..5]] + , expectPE FalseT $ pl @(Map Len Id >> (Ix 10 3) &&& (Ix 1 (Failp "rhs")) >> Fst == Snd) [[1..4],[4..5]] + , expectPE FalseT $ pl @(Map Len Id >> (Ix 3 3) &&& (Ix 1 4) >> Fst == Snd) [[1..4],[4..5]] + , expectPE FalseT $ pl @(Map Len Id >> (Ix 10 3) &&& (Ix 1 4) >> Fst == Snd) [[1..4],[4..5]] + , expectPE FalseT $ pl @(Map Len Id >> (Ix 10 5) &&& (Ix 1 4) >> Fst == Snd) [[1..4],[4..5]] + , expectPE TrueT $ pl @(Map Len Id >> (Ix 10 2) &&& (Ix 1 4) >> Fst == Snd) [[1..4],[4..5]] + , expectPE (PresentT ([1],[2,3,4,5])) $ pl @(Partition (Lt 2) Id >> Id) [1,2,3,4,5] + , expectPE (PresentT [1,2,3]) $ pl @(MaybeIn MemptyP Id) (Just [1,2,3]) + , expectPE (PresentT []) $ pl @(MaybeIn MemptyP Id) (Nothing @[Int]) + , expectPE (FailT "'Just found Nothing") $ pl @('Just (FailS "someval")) (Nothing @()) -- breaks otherwise + , expectPE (PresentT (4,4)) $ pl @Dup 4 + , expectPE (PresentT 3) $ pl @(Last' Id) [1,2,3] + , expectPE (PresentT 123) $ pl @(Just' Id >> Id) (Just 123) + , expectPE (FailT "Asdf") $ pl @(HeadFail "Asdf" Id) ([] @()) -- breaks otherwise + , expectPE (FailT "Head(empty)") $ pl @(Head' Id) ([] @Int) + , expectPE (FailT "Head(empty)") $ pl @(Head' Id) ([] @Double) + , expectPE (FailT "Succ bounded failed") $ pl @SuccB' GT + , expectPE (PresentT LT) $ pl @(SuccB 'LT) GT + , expectPE (PresentT EQ) $ pl @(SuccB 'GT) LT + , expectPE (PresentT EQ) $ pl @SuccB' LT + , expectPE (FailT "Pred bounded failed") $ pl @PredB' LT + , expectPE (PresentT GT) $ pl @(PredB 'GT) LT + , expectPE (PresentT EQ) $ pl @(PredB 'LT) GT + , expectPE (PresentT EQ) $ pl @PredB' GT + , expectPE (FailT "ToEnum bounded failed") $ pl @(ToEnumBF Ordering) 44 + , expectPE (PresentT LT) $ pl @(ToEnumB Ordering 'LT) 123 + , expectPE (PresentT EQ) $ pl @(ToEnumB Ordering 'GT) 1 + , expectPE (PresentT EQ) $ pl @(ToEnumBF Ordering) 1 + , expectPE (PresentT 11) $ pl @Succ 10 + , expectPE (FailT "Succ IO e=Prelude.Enum.Bool.succ: bad argument") $ pl @Succ True -- captures the exception + , expectPE (PresentT ([4,5,6,7,8,9,10],[1,2,3])) $ pl @(Partition (Gt 3) Id) [1..10] + , expectPE (PresentT ([2,4,6],[1,3,5])) $ pl @(Partition Even Id) [1..6] + , expectPE TrueT $ pl @(Partition Even Id >> Star Null (Len >> Gt 4) >> Fst == Snd) [1..6] + , expectPE (PresentT 5) $ pl @(Snd >> Snd >> Snd >> Snd >> Id) (9,(1,(2,(3,5)))) + , expectPE (FailT "ExitWhen") $ pl @((HeadFail "failedn" Id) &&& (Len >> Same 1 >> ExitWhen' Id) >> Fst) [3] + , expectPE (PresentT 3) $ pl @((HeadFail "failedn" Id) &&& (Len >> Same 1 >> Not >> ExitWhen' Id) >> Fst) [3] + , expectPE (PresentT 3) $ pl @((HeadFail "failedn" Id) &&& (Len >> Same 1 >> ExitWhen' Not) >> Fst) [3] + , expectPE (FailT "ExitWhen") $ pl @(ExitWhen' (Len >> Ne 1) >> HeadFail "failedn" Id) [3,1] + , expectPE (PresentT 3) $ pl @(ExitWhen' (Len >> Ne 1) >> HeadFail "failedn" Id) [3] + , expectPE TrueT $ pl @(ExitWhen' (Len >> Ne 1) >> HeadFail "failedn" Id >> Gt (Neg 20)) [3] + , expectPE FalseT $ pl @(ExitWhen' (Len >> Ne 1) >> HeadFail "failedn" Id >> Gt (Neg 20)) [-23] + , expectPE (PresentT (-1.0)) $ pl @(Negate >> Dup >> First Succ >> Swap >> Fst - Snd) 4 + , expectPE (PresentT (Right 12)) $ pl @(Not +++ Id) (Right @Bool 12) + , expectPE (PresentT Cgt) $ pl @("aa" === Id >> FromEnum >> ToEnum OrderingP) "aaaa" + , expectPE (PresentT False) $ pl @(Msg "someval4" (Gt 4 >> Id)) 4 + , expectPE (PresentT ()) $ pl @(Snd >> Snd >> Snd >> Snd >> Id) (1,('a',(3,(True,())))) + , expectPE TrueT $ pl @(Re "\\d{4}-\\d{3}" Id) "1234-123" + , expectPE FalseT $ pl @(Re "\\d{4}-\\d{3}" Id) "1234-1x3" + , expectPE TrueT $ pl @(Re' '[ 'Caseless, 'Dotall ] "ab" Id) "aB" + , expectPE TrueT $ pl @(Re' '[ 'Caseless, 'Dotall ] "ab." Id) "aB\n" + , expectPE FalseT $ pl @(Re' '[ 'Caseless ] "ab." Id) "aB\n" + , expectPE TrueT $ pl @(Re "(?i)ab" Id) "aB" -- runtime [use 'Caseless instead] + , expectPE FalseT $ pl @(Re "ab" Id) "aB" + , expectPE (PresentT [("aB",["B"]),("cd",["d"])]) $ pl @(Rescan ".(.)" Id) "aBcd" + , expectPE (PresentT [14,12,10,4,2]) $ pl @(SortOnDesc Id Id) [10,4,2,12,14] + , expectPE (PresentT [2,4,10,12,14]) $ pl @(SortOn Id Id) [10,4,2,12,14] + , expectPE (PresentT [14,12,10,4,2]) $ pl @(SortOn Negate Id) [10,4,2,12,14] + , expectPE (PresentT [('a',4),('a',14),('b',2),('c',10),('d',12),('z',1)]) $ pl @(SortOn Fst Id) (zip "cabdaz" [10,4,2,12,14,1]) + , expectPE (FailT "asdf(4)") $ pl @(SortOn (FailS "asdf") Id) [10,4,2,12,14] + , expectPE TrueT $ pl @(Min &&& Max >> Id >> Fst < Snd) [10,4,2,12,14] + , expectPE (FailT "ExitWhen") $ pl @(Partition (ExitWhen' (Gt 10) >> Gt 2) Id) [1..11] + , expectPE (PresentT [False,False,True,True,True]) $ pl @(Map (ExitWhen' (Gt 10) >> Gt 2) Id) [1..5] + , expectPE (PresentT ([1,2],[3,4,5,6,7,8,9,10,11])) $ pl @(Break (Gt 2) Id) [1..11] + , expectPE (PresentT ([1,2,3],[4,5,6,7,8,9,10,11])) $ pl @(Span (Lt 4) Id) [1..11] + , expectPE (PresentT [GT,GT,LT,EQ]) $ pl @(Pairs >> Map (First (Succ >> Succ) >> Fst === Snd) Id) [1,2,3,6,8] + , expectPE TrueT $ pl @(Re "^\\d{1,3}(?:\\.\\d{1,3}){3}$" Id) "123.1.1.21" + , expectPE (PresentT [("123.",["."]),("8.",["."]),("99.",["."]),("21",[])]) $ pl @(Rescan "\\d{1,3}(\\.)?" Id) "123.8.99.21" + , expectPE (PresentT 117) $ pl @(MaybeIn (Failp "err") Succ) (Just 116) + , expectPE (PresentT 99) $ pl @(MaybeIn 99 Succ) (Nothing @Int) + , expectPE (FailT "someval") $ pl @(MaybeIn (Failp "someval") Succ) (Nothing @()) + , expectPE TrueT $ pl @(MaybeIn 'True 'False) (Nothing @()) + , expectPE FalseT $ pl @(MaybeIn 'True 'False) (Just "aa") + , expectPE (PresentT LT) $ pl @(MaybeIn MemptyP (Fst === Snd)) (Just ('x','z')) + , expectPE (PresentT EQ) $ pl @(MaybeIn MemptyP (Fst === Snd)) (Nothing @(Char,Char)) + , expectPE TrueT $ pl @('True ||| 'False) (Left @_ @() "someval") + , expectPE FalseT $ pl @('True ||| 'False) (Right @() "someval") + , expectPE (PresentT 123) $ pl @('Left Id) (Left 123) + , expectPE (FailT "'Left found Right") $ pl @('Left Id) (Right @() 123) + , expectPE (PresentT 123) $ pl @('Right Id) (Right 123) + , expectPE (FailT "'Right found Left") $ pl @('Right Id) (Left @_ @() 123) + , expectPE (PresentT ["1","2","3"]) $ pl @(MaybeIn MemptyP (ShowP >> Ones)) (Just 123) + , expectPE (PresentT []) $ pl @(MaybeIn MemptyP (ShowP >> Ones)) (Nothing @String) + , expectPE (PresentT "124") $ pl @((Succ >> ShowP) ||| ShowP ) (Left @_ @() 123) + , expectPE (PresentT "True") $ pl @((Succ >> ShowP) ||| ShowP ) (Right @Int True) + , expectPE (PresentT (123 % 4)) $ pl @(ReadP Rational) "123 % 4" + , expectPE (FailT "ReadP Ratio Integer (x123 % 4) failed") $ pl @(ReadP Rational) "x123 % 4" + , expectPE (PresentT "") $ pl @('Proxy >> MemptyP) "abc" + , expectPE (PresentT ["a","b","c"]) $ pl @(MemptyT _ ||| Ones) (Right @() "abc") + , expectPE (PresentT []) $ pl @(MemptyT _ ||| Ones) (Left @_ @[String] ["ab"]) + , expectPE (PresentT ["a","b"]) $ pl @(MaybeIn MemptyP Ones) (Just @String "ab") + , expectPE (PresentT []) $ pl @(MaybeIn MemptyP Ones) (Nothing @String) + , expectPE (PresentT (True, 13)) $ pl @((IsNothing >> Not) &&& (Just' Id >> Add Id 12)) (Just 1) + , expectPE (FailT "expected Just") $ pl @((IsNothing >> Not) &&& (Just' Id >> Add Id 12)) Nothing + , expectPE (PresentT True) $ pl @(ThdL _ Id >> FstL _ Id) (1,2,(True,4)) + , expectPE (PresentT True) $ pl @(FstL _ (ThdL _ Id)) (1,2,(True,4)) + , expectPE (PresentT 'd') $ pl @(Id !! 3) ("asfd" :: T.Text) + , expectPE (FailT "(!!) index not found") $ pl @(Id !! 4) ("asfd" :: T.Text) + , expectPE (PresentT "dfsa") $ pl @ReverseL ("asfd" :: T.Text) + , expectPE (PresentT (Left "asfd")) $ pl @Swap (Right @() "asfd") -- @() else breaks: ok in ghci + , expectPE (PresentT ("asfd",12)) $ pl @Swap (12,"asfd") + , expectPE (PresentT (Just ('a',"sfd"))) $ pl @Uncons ("asfd" :: T.Text) + , expectPE (PresentT Nothing) $ pl @Uncons ("" :: T.Text) + , expectPE (PresentT (Just ("asf",'d'))) $ pl @Unsnoc ("asfd" :: T.Text) + , expectPE (PresentT Nothing) $ pl @Unsnoc ("" :: T.Text) + , expectPE FalseT $ pl @IsEmpty ("failed11" :: T.Text) + , expectPE TrueT $ pl @IsEmpty ("" :: T.Text) + , expectPE (PresentT 14) $ pl @(Unwrap >> Succ) (SG.Sum 13) + , expectPE (PresentT 4) $ pl @(MemptyT (SG.Sum _) >> Unwrap >> Add Id 4) () + , expectPE (PresentT (SG.Sum 13)) $ pl @(Wrap (SG.Sum _) Id) 13 + , expectPE (PresentT "a") $ pl @(Id !! MemptyT _) (Just "a") + , expectPE (FailT "(!!) index not found") $ pl @(Id !! MemptyT _) (Nothing @()) -- had to add @() to keep this happy: ghci is fine + , expectPE (PresentT 'a') $ pl @(Id !! 0) ('a','b','c') + , expectPE (FailT "err") $ pl @(Id !! Failt _ "err") ('a','b','c') + , expectPE (PresentT 3) $ pl @(Id !! "d") (M.fromList $ zip (map (:[]) "abcd") [0 ..]) + , expectPE (PresentT 3) $ pl @(Id !! (HeadFail "failedn" "d")) (M.fromList $ zip "abcd" [0 ..]) -- had to String (instead of _) to keep this happy: ghci is fine + , expectPE (PresentT ()) $ pl @(Id !! (HeadFail "failedn" "d")) (S.fromList "abcd") -- had to String (instead of _) to keep this happy: ghci is fine + , expectPE (FailT "(!!) index not found") $ pl @(Id !! (HeadFail "failedn" "e")) (S.fromList "abcd") -- had to String (instead of _) to keep this happy: ghci is fine + , expectPE (PresentT 13.345) $ pl @(ExitWhen' (Re "^\\d+(?:\\.\\d+)?$" Id >> Not) >> ReadP Double) "13.345" + , expectPE (PresentT 13) $ pl @(ExitWhen' (Re "^\\d+(?:\\.\\d+)?$" Id >> Not) >> ReadP Double) "13" + , expectPE (FailT "regex failed") $ pl @(ExitWhen "regex failed" (Re "^\\d+(?:\\.\\d+)?$" Id >> Not) >> ReadP Double) "-13.4" + , expectPE (PresentT GT) $ pl @(Repeat 2 Id Succ) LT + , expectPE (FailT "Succ IO e=Prelude.Enum.Ordering.succ: bad argument") $ pl @(Repeat 30 Id Succ) LT + , expectPE (PresentT 'g') $ pl @(Repeat 6 Id Succ) 'a' + , expectPE (PresentT '[') $ pl @(Repeat 6 Id Pred) 'a' + , expectPE (FailT "Regex failed to compile") $ pl @(Re "\\d{4}\\" Id) "ayx" + , expectPE (PresentT LT) $ pl @(Repeat 0 Id Succ) LT + , expectPE (PresentT LT) $ pl @(Repeat 2 Id Succ >> Repeat 2 Id Pred) LT + , expectPE (PresentT ["2","2"]) $ pl @(Map Fst (ShowP >> Rescan "." Id) >> FilterBy (Same "2") Id) 12324 + , expectPE (PresentT [LT,LT,LT,GT,EQ,LT]) $ pl @((Ones << Id << ShowP) >> Map (Fst === Snd) Pairs) 1234223 + , expectPE (PresentT [(0,'a'),(1,'b'),(2,'c'),(3,'d')]) $ pl @(IToList _) ("abcd" :: String) + , expectPE (PresentT "abcd") $ pl @ToList (M.fromList $ zip [0..] "abcd") + , expectPE (PresentT [123]) $ pl @ToList (Just 123) + , expectPE (FailT "failed20") $ pl @(MaybeIn (Failp "failed20") 'False) (Nothing @Int) + , expectPE (FailT "failed21") $ pl @(MaybeIn ('False >> FailS "failed21") 'False) (Nothing @Double) + , expectPE (FailT "err") $ pl @(MaybeIn (Failp "err") Id) (Nothing @Int) + , expectPE (FailT "err") $ pl @(MaybeIn (Failp "err") Id) (Nothing @()) + , expectPE (PresentT [(0,'a'),(1,'b'),(2,'c'),(3,'d')]) $ pl @(IToList _) (M.fromList $ itoList ("abcd" :: String)) + , expectPE (PresentT [('a',1),('b',2),('c',3),('d',4),('a',5),('b',6),('c',7)]) $ pl @(Ziplc "abcd" Id) [1..7] + , expectPE (PresentT [('a',1),('b',2),('c',3),('d',4)]) $ pl @(Zipn "abcd" Id) [1..7] + , expectPE (PresentT []) $ pl @(Zipn "" Id) [1..7] + , expectPE (PresentT [(1 % 1,'a'),(2 % 1,'b'),(3 % 1,'c'),(1 % 1,'d')]) $ pl @(Ziplc '[Pos 1, Pos 2, Pos 3] Id) ("abcd" ) + , expectPE (PresentT []) $ pl @(Ziprc (EmptyT _) Id) "abcd" + , expectPE (PresentT [9,2,7,4]) $ pl @ToList (M.fromList (zip ['a'..] [9,2,7,4])) + , expectPE (PresentT [(0,9),(1,2),(2,7),(3,4)]) $ pl @(IToList _) [9,2,7,4] + , expectPE (PresentT [('a',9),('b',2),('c',7),('d',4)]) $ pl @(IToList _) (M.fromList (zip ['a'..] [9,2,7,4])) + , expectPE (PresentT [((),234)]) $ pl @(IToList _) (Just 234) + , expectPE (PresentT []) $ pl @(IToList _) (Nothing @Double) + , expectPE (PresentT (-4,5)) $ pl @(DivMod Negate 7) 23 + , expectPE (PresentT (-3,-2)) $ pl @(QuotRem Negate 7) 23 + , expectPE (PresentT (True,3.4)) $ pl @(ThdL _ Id >> SndL _ Id >> FstL _ Id) (1,'a',('x',((True,3.4),999))) + , expectPE (PresentT (True,3.4)) $ pl @(FstL _ (SndL _ (ThdL _ Id))) (1,'a',('x',((True,3.4),999))) + , expectPE (PresentT 7) $ pl @(FstL _ Id) (7,999.12) + , expectPE (PresentT (M.fromList [(1,'a')])) $ pl @(MaybeIn MemptyP Id) (Just (M.fromList [(1,'a')])) + , expectPE (PresentT (M.fromList [])) $ pl @(MaybeIn MemptyP Id) (Nothing @(M.Map () ())) + , expectPE (PresentT [("1",["1"]),("2",["2"]),("3",["3"]),("4",["4"])]) $ pl @(Rescan "(\\d)+?" Id) "1234" + , expectPE (PresentT [("1234",["4"])]) $ pl @(Rescan "(\\d)+" Id) "1234" + , expectPE (PresentT [("1.2",["1",".2","2"]),("3.4",["3",".4","4"])]) $ pl @(Rescan "(\\d{1,3})(\\.(\\d{1,3}))+?" Id) "1.2.3.4" -- overcapturing + , expectPE (PresentT [("1234",["4"])]) $ pl @(Rescan "^(\\d)+?$" Id) "1234" + , expectPE (PresentT [("1.2",["1",".2","2"]),("3.4",["3",".4","4"])]) $ pl @(Rescan "(\\d{1,3})(\\.(\\d{1,3}))+?" Id) "1.2.3.4" + , expectPE (PresentT ["123","2","3","5","6"]) $ pl @(Resplit "\\." Id) "123.2.3.5.6" + , expectPE (PresentT [("1.2",["1","2"]),("3.4",["3","4"])]) $ pl @(Rescan "(\\d{1,3})(?:\\.(\\d{1,3}))+?" Id) "1.2.3.4" -- bizzare! + , expectPE (PresentT [("1.2.3.4",["1","2","3","4"])]) $ pl @(Rescan "^(\\d{1,3})\\.(\\d{1,3})\\.(\\d{1,3})\\.(\\d{1,3})$" Id) "1.2.3.4" -- this is good! + , expectPE (PresentT [13,16,17]) $ pl @(Guard "err" (Len >> Gt 2) >> Map Succ Id) [12,15,16] + , expectPE (FailT "err found len=3") $ pl @(Guard (Printf "err found len=%d" Len) (Len >> Gt 5) >> Map Succ Id) [12,15,16] + , expectPE (FailT "Printf (IO e=printf: bad formatting char 'd')") $ pl @(Printf "someval %d" Id) ("!23"::String) + , expectPE (PresentT [12,0,1,13,0,1,14,0,1,15,0,1,16]) $ pl @(Intercalate Fst Snd) ([0,1], [12,13,14,15,16]) + , expectPE (PresentT [12,-5,13,-5,14,-5,15,-5,16]) $ pl @(((Pure [] (Len >> Negate)) &&& Id) >> Intercalate Fst Snd) [12,13,14,15,16] + , expectPE (PresentT [13,16,17]) $ pl @(If (Len >> Gt 2) (Map Succ Id) (FailS "someval")) [12,15,16] + , expectPE (PresentT [13,16,17]) $ pl @(Guard' (Len >> Gt 2) >> Map Succ Id) [12,15,16] + , expectPE (FailT "err") $ pl @(ExitWhen "err" (Len >> Gt 2) >> Map Succ Id) [12,15,16] + , expectPE (PresentT [13]) $ pl @(ExitWhen "err" (Len >> Gt 2) >> Map Succ Id) [12] + , expectPE (FailT "err") $ pl @(Guard "err" (Len >> Gt 2) >> Map Succ Id) [12] + , expectPE (PresentT 12) $ pl @OneP [12] + , expectPE (FailT "expected list of length 1 but found length=5") $ pl @OneP [1..5] + , expectPE (FailT "expected list of length 1 but found length=0") $ pl @OneP ([] @()) + , expectPE (FailT "err(8)") $ pl @(Map (If (Lt 3) 'True (Failt _ "err")) Id) [1..10] + , expectPE (FailT "someval(8)") $ pl @(Map (If (Lt 3) 'True (Failt _ "someval")) Id) [1..10] -- use Guard ie Guard + , expectPE (PresentT [True,True,False,False,False]) $ pl @(Map (If (Lt 3) 'True 'False) Id) [1..5] + , expectPE (PresentT ["a","b","c"]) $ pl @(MaybeIn MemptyP Ones) (Just @String "abc") + , expectPE (FailT "someval") $ pl @(Guard "someval" (Len == 2) >> (ShowP &&& Id)) ([] @Int) + , expectPE (PresentT ([2,3],"[2,3]")) $ pl @(Guard "someval" (Len == 2) >> (Id &&& ShowP)) [2,3] + , expectPE (FailT "someval") $ pl @(Guard "someval" (Len == 2) >> (ShowP &&& Id)) [2,3,4] + , expectPE (PresentT 55) $ pl @(Map (Wrap (SG.Sum _) Id) Id >> MConcat >> Unwrap) [1..10] + , expectPE (PresentT True) $ pl @(EitherIn Not Id) (Right @Bool True) + , expectPE FalseT $ pl @(EitherIn Not Id) (Left @_ @Bool True) + , expectPE FalseT $ pl @(Re "^\\d+$" Id) "123\nx" + , expectPE TrueT $ pl @(Re "(?m)^\\d+$" Id) "123\nx" -- (?m) anchors match beginning/end of line instead of whole string + , expectPE (PresentT (Just 'x')) $ pl @(Pure Maybe Id) 'x' + , expectPE (PresentT (Right @() 'x')) $ pl @(Pure (Either _) Id) 'x' + , expectPE (PresentT Nothing) $ pl @(MemptyT (Maybe ())) 'x' + , expectPE (PresentT (Left @_ @() 'x')) $ pl @(Pure (Either _) Id >> Swap) 'x' + , expectPE (PresentT (Left 'x')) $ pl @(Pure (Either ()) Id >> Swap) 'x' + , expectPE (PresentT (SG.Sum 52)) $ pl @(STimes 4 Id) (SG.Sum 13) + , expectPE (PresentT (SG.Sum 52)) $ pl @(Wrap (SG.Sum _) Id >> STimes 4 Id) 13 + , expectPE (PresentT 52) $ pl @(FoldMap (SG.Sum _) Id) [14,8,17,13] + , expectPE (PresentT 17) $ pl @(FoldMap (SG.Max _) Id) [14 :: Int,8,17,13] -- cos Bounded! + , expectPE FalseT $ pl @(Catch (Re "\\d+(" Id) 'False) "123" + , expectPE TrueT $ pl @(Catch (Re "\\d+" Id) 'False) "123" + , expectPE (PresentT 3) $ pl @(Id !! (Head' "d")) (M.fromList $ zip "abcd" [0 ..]) -- use Char1 "d" instead of "d" >> Head' + , expectPE (PresentT 10) $ pl @(Id !! MemptyT _) (Just 10) + , expectPE (FailT "(!!) index not found") $ pl @(Id !! MemptyT _) (Nothing @()) + , expectPE TrueT $ pl @((Len >> (Elem Id '[4,7,1] || (Mod Id 3 >> Same 0))) || (FoldMap (SG.Sum _) Id >> Gt 200)) [1..20] + , expectPE FalseT $ pl @((Len >> (Elem Id '[4,7,1] || (Mod Id 3 >> Same 0))) || (FoldMap (SG.Sum _) Id >> Gt 200)) [1..19] + , expectPE TrueT $ pl @((Len >> (Elem Id '[4,7,1] || (Mod Id 3 >> Same 0))) || (FoldMap (SG.Sum _) Id >> Gt 200)) [] + , expectPE (PresentT (False, 210)) $ pl @((Len >> (Elem Id '[4,7,1] || (Mod Id 3 >> Same 0))) &&& FoldMap (SG.Sum _) Id) [1..20] + , expectPE (PresentT 'g') $ pl @(Id !! 6) ['a'..'z'] + , expectPE (PresentT ([141,214,125,1,2,3333],(False,False))) $ pl @(Map (ReadP Int) (Resplit "\\." Id) >> '(Id, '(Len == 4, All (Between 0 255)))) "141.214.125.1.2.3333" + , expectPE (PresentT ([141,214,125,1,2,6],(False,True))) $ pl @(Map (ReadP Int) (Resplit "\\." Id) >> Id &&& ((Len == 4) &&& All (Between 0 255))) "141.214.125.1.2.6" + , expectPE (FailT "ReadP Int () failed") $ pl @(Resplit "\\." Id >> Map (ReadP Int) Id >> Id &&& ((Len >> Same 4) &&& All (Between 0 255))) "141.214.125." + , expectPE (PresentT 9) $ pl @((Wrap _ Id *** Wrap (SG.Sum _) Id) >> Sapa >> Unwrap) (4,5) + , expectPE (PresentT (SG.Sum 9)) $ pl @((Wrap _ Id *** Wrap _ Id) >> Sapa) (4,5) + , expectPE (PresentT 9) $ pl @(Sapa' (SG.Sum _) >> Unwrap) (4,5) + , expectPE (PresentT "abcde") $ pl @(ScanNA Succ) (4,'a') + , expectPE (PresentT ["abcd","bcd","cd","d",""]) $ pl @(ScanNA Tail) (4,"abcd" :: String) + , expectPE (PresentT ["abcd","bcd","cd","d",""]) $ pl @(Len &&& Id >> ScanNA Tail) "abcd" + , expectPE (PresentT ["abcd","bcd","cd","d",""]) $ pl @Tails ("abcd" :: String) + , expectPE (PresentT (-4,-2)) $ pl @(DivMod Fst Snd) (10,-3) + , expectPE (PresentT (-3,1)) $ pl @(QuotRem Fst Snd) (10,-3) + , expectPE (FailT "DivMod zero denominator") $ pl @(DivMod Fst Snd) (10,0) + , expectPE (PresentT 'd') $ pl @(Snd !! Fst) (3,"abcde" :: String) + , expectPE (FailT "(!!) index not found") $ pl @(Snd !! Fst) (4,[9,8]) + , expectPE (PresentT 'c') $ pl @(2 &&& Id >> Snd !! Fst) ("abcdef" :: String) + , expectPE (PresentT 'f') $ pl @((Len >> Pred) &&& Id >> Snd !! Fst) "abcdef" + , expectPE (FailT "len is bad") $ pl @Ip6 "FE80::203:Baff:FE77:326FF" + , expectPE (FailT "not a hex") $ pl @Ip6 "FE80::203:Baff:GE77:326F" + , expectPE (FailT "count is bad") $ pl @Ip6 "FE80::203:Baff:FE77:326F:::::" + , expectPE (PresentT 65504) $ pl @(ReadBaseInt 16) "fFe0" + , expectPE (PresentT "ffe0") $ pl @(ShowBase 16) 65504 + , expectPE (FailT "invalid base 22") $ pl @(ReadBaseInt 22) "zzz" + , expectPE (PresentT ("ffe0","fFe0")) $ pl @((ReadBaseInt 16 &&& Id) >> (First (ShowBase 16))) "fFe0" + , expectPE FalseT $ pl @(Id == "Abc") "abc" + , expectPE TrueT $ pl @("Abc" ==? Id) "abc" + , expectPE (PresentT LT) $ pl @("Abc" === Id) "abc" + , expectPE (PresentT EQ) $ pl @("Abc" ===? Id) "abc" + , expectPE (PresentT 'd') $ pl @(Id !! 3) ('a','b','c','d','e') + , expectPE (PresentT 99) $ pl @(Id !! "s") $ M.fromList [("t",1), ("s", 20), ("s", 99)] + , expectPE (PresentT 1) $ pl @(Head' Id) [1,2,3] + , expectPE (PresentT (Just (1,[2,3,4,5]))) $ pl @Uncons [1..5] -- with Typeable would need to specify the type of [1..5] + , expectPE (PresentT (Just ([1,2,3,4],5))) $ pl @Unsnoc [1..5] + , expectPE (PresentT [(0,1),(1,2),(2,3),(3,4),(4,5)]) $ pl @(IToList _) [1..5] + , expectPE (PresentT [(0,'a'),(1,'b'),(2,'c')]) $ pl @(IToList _) ['a','b','c'] + , expectPE (PresentT [(1,'a'),(2,'b'),(3,'c'),(4,'d'),(5,'e')]) $ pl @(Zipn Fst Snd) ([1..5],['a'..'z']) + , expectPE (PresentT [1,2,3,8,8]) $ pl @(PadR 5 8 Id) [1..3] + , expectPE (PresentT [1,2,3,4,5]) $ pl @(PadR 5 0 Id) [1..5] + , expectPE (PresentT [1,2,3,4,5,6]) $ pl @(PadR 5 0 Id) [1..6] + , expectPE (PresentT [0,0,1,2,3]) $ pl @(PadL 5 0 Id) [1..3] + , expectPE (PresentT []) $ pl @(Catch (Resplit "\\d+(" Id) (Snd >> MemptyP)) "123" + , expectPE (FailT "someval(8)") $ pl @(Map (Guard "someval" (Lt 3) >> 'True) Id) [1::Int ..10] + , expectPE (PresentT [True,True,True,True,True,True,True,True,True,True]) $ pl @(Map (Guard "someval" (Ge 1) >> 'True) Id) [1::Int ..10] + , expectPE (PresentT [4,5,6]) $ pl @(ScanN 2 Id Succ) 4 + , expectPE (PresentT [4,4,4,4,4,4]) $ pl @(ScanN 5 Id Id) 4 + , expectPE (PresentT [1,2,3,244]) $ pl @(Rescan Ip4RE Id >> OneP >> Map (ReadBaseInt 10) Snd >> Ip4guard) "1.2.3.244" + , expectPE (FailT "0-255") $ pl @(Rescan Ip4RE Id >> OneP >> Map (ReadBaseInt 10) Snd >> Ip4guard) "1.256.3.244" + , expectPE (FailT "0-255") $ pl @(Rescan "(\\d+)\\.?" Id >> ConcatMap Snd Id >> Map (ReadBaseInt 10) Id >> Ip4guard) "1.22.312.66" + , expectPE (FailT "4octets") $ pl @(Rescan "(\\d+)\\.?" Id >> ConcatMap Snd Id >> Map (ReadBaseInt 10) Id >> Ip4guard) "1.22.244.66.77" + , expectPE (PresentT [1,23,43,214]) $ pl @(Rescan "(\\d+)\\.?" Id >> ConcatMap Snd Id >> Map (ReadBaseInt 10) Id >> Ip4guard) "1.23.43.214" + , expectPE (PresentT (SG.Sum 123)) $ pl @(JustP Id) (Just (SG.Sum 123)) + , expectPE (PresentT (SG.Sum 0)) $ pl @(JustP Id) (Nothing @(SG.Sum _)) + , expectPE (PresentT (636 % 5)) $ pl @((ToRational 123 &&& Id) >> Fst + Snd) 4.2 + , expectPE (PresentT 127) $ pl @((123 &&& Id) >> Fst + Snd) 4 + , expectPE (PresentT 256) $ pl @(Rescan "(?i)^\\\\x([0-9a-f]{2})$" Id >> OneP >> Snd >> OneP >> ReadBaseInt 16 >> Succ) "\\xfF" + , expectPE (PresentT 256) $ pl @(Rescan "(?i)^\\\\x(.{2})$" Id >> OneP >> Snd >> OneP >> ReadBaseInt 16 >> Succ) "\\xfF" + , expectPE (PresentT (("fF",(255,"ff")),False)) $ pl @(Rescan "(?i)^\\\\x([0-9a-f]{2})$" Id >> OneP >> Snd >> OneP >> (Id &&& (ReadBaseInt 16 >> (Id &&& ShowBase 16))) >> (Id &&& ((Id *** Snd) >> Fst == Snd))) "\\xfF" + , expectPE (PresentT [1,2,4,0]) $ pl @(Do '[Succ,Id,ShowP,Ones,Map (ReadBaseInt 8) Id]) 1239 + , expectPE (FailT "invalid base 8") $ pl @(Do '[Pred,Id,ShowP,Ones,Map (ReadBaseInt 8) Id]) 1239 + , expectPE (PresentT 47) $ pl @(ReadBaseInt 2) "101111" + , expectPE (PresentT [LT,EQ,GT,EQ,EQ,EQ,EQ,EQ,EQ,EQ]) $ pl @(ScanN 2 Id Succ >> PadR 10 (MemptyT Ordering) Id) LT + , expectPE (PresentT 12) $ pl @('This Id) (This 12) + , expectPE (FailT "'This found That") $ pl @('This Id) (That @() 12) + , expectPE (PresentT (SG.Sum 12)) $ pl @(ThisP Id) (This @_ @() (SG.Sum 12)) + , expectPE (PresentT ()) $ pl @(ThisP Id) (That 12) + , expectPE (PresentT (SG.Sum 12)) $ pl @(ThisFail "sdf" Id) (This @_ @() (SG.Sum 12)) + , expectPE (FailT "sdf") $ pl @(ThisFail "sdf" Id) (That @() (SG.Sum 12)) + , expectPE (FailT "sdf") $ pl @(ThisFail "sdf" Id) (That @Int 12) + , expectPE (PresentT "this") $ pl @(TheseIn "this" "that" "these") (This @_ @() (SG.Sum 12)) + , expectPE FalseT $ pl @IsThese (That @() (SG.Sum 12)) + , expectPE TrueT $ pl @IsThese (These 1 (SG.Sum 12)) + , expectPE (PresentT ("Ab",13)) $ pl @(TheseIn (Id &&& 999) ("no value" &&& Id) Id) (These "Ab" 13) + , expectPE (PresentT ("Ab",999)) $ pl @(TheseIn (Id &&& 999) ("no value" &&& Id) Id) (This "Ab") + , expectPE (PresentT ("no value",13)) $ pl @(TheseIn (Id &&& 999) ("no value" &&& Id) Id) (That 13) + , expectPE (PresentT "wxydef") $ pl @(ZipThese Fst Snd >> Map (TheseIn Id Id Fst) Id) (['w'..'y'],['a'..'f']) + , expectPE (PresentT [("fe",["fe"]),("b1",["b1"]),("2a",["2a"])]) $ pl @(Rescan "([[:xdigit:]]{2})" Id) "wfeb12az" + -- anchored means it has to start at the beginning: can have junk on the end which we cant detect but at least we know it starts at beginning + , expectPE (FailT "Regex no results") $ pl @(Rescan' '[ 'Anchored ] "([[:xdigit:]]{2})" Id) "wfeb12az" + , expectPE (PresentT [('s',1),('d',2),('f',3),('x',4),('x',5)]) $ pl @(("sdf" &&& Id) >> ZipThese Fst Snd >> Map (TheseIn (Id &&& 0) ((Head' "x") &&& Id) Id) Id) [1..5] + , expectPE (PresentT "abc") $ pl @"abc" () + , expectPE FalseT $ pl @('True >> Not) () + , expectPE TrueT $ pl @'True () + , expectPE FalseT $ pl @'False () + , expectPE (PresentT LT) $ pl @'LT () + , expectPE (PresentT 123) $ pl @123 () + , expectPE (PresentT (4,("sadf",LT))) $ pl @(4 &&& "sadf" &&& 'LT) () + , expectPE (PresentT (4,("sadf",LT))) $ pl @(4 *** "sadf" *** 'LT) ('x',("abv",[1])) + , expectPE (PresentT 6) $ pl @(Do '[4,5,6]) () + , expectPE (PresentT "hhhhh") $ pl @(Do '["abc", "Def", "ggg", "hhhhh"]) () + , expectPE (PresentT GT) $ pl @(Do '[ 'LT, 'EQ, 'GT ]) () + , expectPE (PresentT (-3 % 1)) $ pl @(Do '[Rat 'True 4 4,Pos 22,NegR 12 4]) () + , expectPE (PresentT [10,2,5,8]) $ pl @(GuardsLax (ToGuardsT (Printf2 "guard(%d) %d is out of range") '[Between 0 11, Between 1 4,Between 3 5])) [10::Int,2,5,8] + , expectPE (PresentT [31,11,1999]) $ pl @(Rescan DdmmyyyyRE Id >> OneP >> Map (ReadBaseInt 10) Snd >> Ddmmyyyyval) "31-11-1999" + , expectPE (PresentT [31,11,1999]) $ pl @(Guards (ToGuardsT (Printf2 "guard(%d) %d is out of range") '[Between 1 31, Between 1 12, Between 1990 2050])) [31,11,1999::Int] + , expectPE (PresentT [31,11,1999,123,44]) $ pl @(GuardsLax (ToGuardsT (Printf2 "guard(%d) %d is out of range") '[Between 1 31, Between 1 12, Between 1990 2050])) [31,11,1999,123,44::Int] + , expectPE (FailT "Guards: data elements(2) /= predicates(3)") $ pl @(Guards (ToGuardsT (Printf2 "guard(%d) %d is out of range") '[Between 1 31, Between 1 12, Between 1990 2050])) [31,11::Int] + , expectPE (FailT "guard(1) 13 is out of range") $ pl @(Guards (ToGuardsT (Printf2 "guard(%d) %d is out of range") '[Between 1 31, Between 1 12, Between 1990 2050])) [31,13,1999::Int] + , expectPE (FailT "guard(0) 0 is out of range") $ pl @(Guards (ToGuardsT (Printf2 "guard(%d) %d is out of range") '[Between 1 31, Between 1 12, Between 1990 2050])) [0,44,1999::Int] + , expectPE (PresentT (fromGregorian 1999 11 30)) $ pl @(ReadP Day) "1999-11-30" + , expectPE (FailT "ReadP Day (1999-02-29) failed") $ pl @(ReadP Day) "1999-02-29" + , expectPE (PresentT (TimeOfDay 14 59 20)) $ pl @(ReadP TimeOfDay) "14:59:20" + , expectPE (PresentT (TimeOfDay 26 61 61)) $ pl @(ReadP TimeOfDay) "26:61:61" -- yep: this is valid! need to do your own validation + , expectPE (FailT "ParseTimeP TimeOfDay (%H:%M%S) failed to parse") $ pl @(ParseTimeP TimeOfDay "%H:%M%S" Id) "14:04:61" + , expectPE (PresentT (TimeOfDay 23 13 59)) $ pl @(Guard "hh:mm:ss regex failed" (Re HmsRE Id) >> ReadP TimeOfDay) "23:13:59" + , expectPE (FailT "hh:mm:ss regex failed") $ pl @(Guard "hh:mm:ss regex failed" (Re HmsRE Id) >> ReadP TimeOfDay) "23:13:60" + , expectPE (FailT "Guards: data elements(5) /= predicates(3)") $ pl @(Guards (ToGuardsT (Printf2 "guard(%d) %d is out of range") '[Between 1 31, Between 1 12, Between 1990 2050])) [31,11,2000,1,2::Int] + , expectPE (PresentT [31,11,2000,1,2]) $ pl @(GuardsLax (ToGuardsT (Printf2 "guard(%d) %d is out of range") '[Between 1 31, Between 1 12, Between 1990 2050])) [31,11,2000,1,2::Int] + , expectPE (PresentT [0,0,0,0,0,0,0,1,2,3]) $ pl @(PadL 10 0 Id) [1..3] + , expectPE (PresentT (124,["1","2","2"])) $ pl @('Left Id >> (Succ &&& (Pred >> ShowP >> Ones))) (Left 123) + , expectPE (PresentT [1,2,3,4]) $ pl @(Guards (ToGuardsT (Printf2 "guard(%d) %d is out of range") (RepeatT 4 (Between 0 255)))) [1,2,3,4::Int] + , expectPE (FailT "Guards: data elements(5) /= predicates(4)") $ pl @(Guards (ToGuardsT (Printf2 "guard(%d) %d is out of range") (RepeatT 4 (Between 0 255)))) [1,2,3,4,5::Int] + , expectPE (FailT "Guards: data elements(3) /= predicates(4)") $ pl @(Guards (ToGuardsT (Printf2 "guard(%d) %d is out of range") (RepeatT 4 (Between 0 255)))) [1,2,3::Int] + , expectPE (PresentT (read "1999-01-01 12:12:12 Utc")) $ pl @(ParseTimeP UTCTime "%F %T" Id) "1999-01-01 12:12:12" + , expectPE (PresentT 123) $ pl @(JustDef 0 Id) (Just 123) + , expectPE (PresentT 0) $ pl @(JustDef 0 Id) Nothing + , expectPE (PresentT 12) $ pl @(LastDef 0 Id) [1..12] + , expectPE (PresentT 0) $ pl @(LastDef 0 Id) [] + , expectPE (PresentT (1,("asdf",True))) $ pl @'(1,'("asdf",'True)) () + , expectPE (PresentT (1,("asdf",(True,())))) $ pl @(TupleI '[W 1,W "asdf",W 'True]) () + , expectPE (PresentT ("abc", True)) $ pl @(Theseid 'True "xyz") (This "abc") + , expectPE (PresentT ("xyz", False)) $ pl @(Theseid 'True "xyz") (That False) + , expectPE (PresentT ("abc", False)) $ pl @(Theseid 'True "xyz") (These "abc" False) + , expectPE (PresentT ("xyz", True)) $ pl @(TheseDef '("xyz",'True) Id) (This "abc") + , expectPE (PresentT ("xyz", True)) $ pl @(TheseDef '("xyz",'True) Id) (That False) + , expectPE (PresentT ("abc", False)) $ pl @(TheseDef '("xyz",'True) Id) (These "abc" False) + , expectPE (PresentT 3) $ pl @(Id !! Char1 "d") (M.fromList $ zip "abcd" [0 ..]) + , expectPE (PresentT (12, False)) $ pl @('These Id Not) (These 12 True) + , expectPE (PresentT (SG.Any True)) $ pl @(Coerce SG.Any) True + , expectPE (PresentT True) $ pl @(Coerce Bool) (SG.Any True) + , expectPE (PresentT (3, SG.Any True)) $ pl @(Id !! (FromStringP _ "d") &&& (Map (Snd >> Gt 3 >> Coerce SG.Any) (IToList _) >> MConcat ) ) (M.fromList $ zip (map T.singleton "abcdefgh") [0 ..]) + , expectPE (PresentT (3, True)) $ pl @(Id !! (FromStringP _ "d") &&& (Map (Snd >> Gt 3 >> Wrap SG.Any Id) (IToList _) >> MConcat >> Unwrap) ) (M.fromList $ zip (map T.singleton "abcdefgh") [0 ..]) + --- have to wrap with W cos different kinds +-- , expectPE TrueT $ pl @(Do '[ W ('PresentT I), W 'FalseT, Not]) False +-- , expectPE FalseT $ pl @(Do '[ W ('PresentT Id), W 'FalseT ]) True -- have to wrap them cos BoolT a vs BoolT Bool ie different types +-- , expectPE TrueT $ pl @('PresentT I >> 'FalseT >> Not) False + -- IxL "d" doesnt work cos is Text not String + , expectPE (PresentT 3) $ pl @(Id !! FromStringP _ "d") (M.fromList $ zip (map T.singleton "abcd") [0 ..]) + -- use Fromstring + , expectPE (PresentT 3) $ pl @(Id !! (FromStringP _ "d")) (M.fromList $ zip (map T.singleton "abcd") [0 ..]) + , expectPE (PresentT [7,9,9,2,7,3,9,8,7,1,3]) $ pl @(Map (ReadP Int) Ones >> Guard "checkdigit fail" Luhn) "79927398713" + , expectPE (FailT "checkdigit fail") $ pl @(Map (ReadP Int) Ones >> Guard "checkdigit fail" Luhn) "79927398714" + , expectPE TrueT $ pl @(Ccip >> Ccop 11) "79927398713" + , expectPE (FailT "expected 10 digits but found 11") $ pl @(Ccip >> Ccop 10) "79927398713" + , expectPE (PresentT [10,14,15,9]) $ pl @(MM1 16 >> MM2 16) "aef9" + , expectPE (FailT "invalid base 16") $ pl @(MM1 16 >> MM2 16) "aef9g" + , expectPE (FailT "found empty") $ pl @(MM1 16 >> MM2 16) "" + , expectPE (FailT "0<=x<n") $ pl @(MM2 16) [10,1,17,1,-3,7] + , expectPE (PresentT ((10,'c'),True)) $ pl @Assocl (10,('c',True)) + , expectPE (PresentT (10,('c',True))) $ pl @Assocr ((10,'c'),True) + , expectPE (PresentT 70) $ pl @(Luhn' 11) "79927398713" + , expectPE (FailT "expected 71 mod 10 = 0 but found 1") $ pl @(Luhn' 11) "79927398714" + +-- works but way to difficult: use Guard to do all the work +-- >pl @(((Rescan "([[:xdigit:]])" >> Map Snd >> (Id &&& Len)) &&& Len) >> Guard "notallmatched" ((Snd *** Id) >> Fst == Snd)) "134F" +-- have to check the length of the match vs input to see that are the same + , expectPE (PresentT [1,3,4,15]) $ pl @(((Rescan "([[:xdigit:]])" Id >> Map (Snd >> OneP >> ReadBase Int 16) Id) &&& Id) >> Guard "notallmatched" ((Len *** Len) >> Fst == Snd) >> Fst) "134F" + , expectPE (FailT "notallmatched") $ pl @(((Rescan "([[:xdigit:]])" Id >> Map (Snd >> OneP >> ReadBase Int 16) Id) &&& Id) >> Guard "notallmatched" ((Len *** Len) >> Fst == Snd) >> Fst) "134g" + , expectPE (PresentT True) $ pl @(FoldMap SG.Any Id) [False,False,True,False] + , expectPE (PresentT False) $ pl @(FoldMap SG.All Id) [False,False,True,False] + , expectPE TrueT $ pl @(Map (ReadP _) Ones >> Luhn) "12345678903" + , expectPE FalseT $ pl @(Map (ReadP _) Ones >> Luhn) "12345678904" + , expectPE (FailT "incorrect number of digits found 10 but expected 11 in [1234567890]") $ pl @(Luhn' 11) "1234567890" + , expectPE (PresentT ([1,2],[3,4,5,6,7,8])) $ pl @(Break (If (Gt 2) 'True (If (Gt 4) (Failt _ "ASfd") 'False)) Id) [1..8] + , expectPE (PresentT ([1,2],[3,4,5,6,7,8])) $ pl @(Break (Case 'False '[Gt 2,Gt 4] '[ W 'True, Failt _ "ASfd"] Id) Id) [1..8] -- case version + , expectPE (FailT "ASfd") $ pl @(Break (If (Gt 2) (Failt _ "ASfd") 'False) Id) [1..8] + , expectPE (PresentT ([(1,False),(2,False),(3,False)],[(4,True),(5,True),(6,False)])) $ pl @(Break Snd Id) (zip [1..] [False,False,False,True,True,False]) + , expectPE (PresentT ([(1,False),(2,False),(3,False),(4,False)],[])) $ pl @(Break Snd Id) (zip [1..] [False,False,False,False]) + , expectPE (PresentT ([],[(1,True),(2,True),(3,True),(4,True)])) $ pl @(Break Snd Id) (zip [1..] [True,True,True,True]) + , (@?=) (Just "abc") ((_FailT # "abc") ^? _FailT) + , (@?=) (Just ()) ((_TrueT # ()) ^? _TrueT) + , (@?=) (Just ()) ((_FalseT # ()) ^? _FalseT) + , (@?=) (Just 'x') ((_PresentT # 'x') ^? _PresentT) + , expectPE (PresentT (111,'b')) $ pl @('(123,Char1 "c") >> (Sub Id 12 *** Pred)) () + , expectPE (PresentT (SG.Min 19)) $ pl @((FromInteger _ 12 &&& Id) >> Fst + Snd) (SG.Min 7) + , expectPE (PresentT (SG.Product 84)) $ pl @((FromInteger _ 12 &&& Id) >> Sapa) (SG.Product 7) + , expectPE (PresentT (123,((),()))) $ pl @(TupleI '[W 123,()]) 99 + , expectPE (PresentT (4,(5,(6,(7,()))))) $ pl @(TupleI '[4,5,6,7]) 99 + , expectPE (PresentT ("ss",(4,(SG.Min 9223372036854775807,())))) $ pl @(TupleI '[W "ss",W 4,MemptyT (SG.Min Int)]) 99 + , expectPE (PresentT ("ss",(4,(SG.Sum 0,())))) $ pl @(TupleI '[W "ss",W 4,MemptyT (SG.Sum _)]) 99 + , expectPE (PresentT "xyxyxyxy") $ pl @(STimes Fst Snd) (4,['x','y']) + , expectPE (PresentT (concat (replicate 16 "abc"))) $ pl @(Repeat 4 Id ((Id &&& Id) >> Sapa)) "abc" + , expectPE (PresentT (concat (replicate 4 "abc"))) $ pl @(STimes Fst Snd) (4,"abc") + , expectPE (PresentT (concat (replicate 4 "abc"))) $ pl @(STimes 4 Id) "abc" + , expectPE (PresentT "abcd") $ pl @(Map FromEnum Id >> Map (ToEnum Char) Id) ("abcd" :: String) + , expectPE (FailT "ToEnum IO e=Prelude.Enum.Ordering.toEnum: bad argument(2)") $ pl @(Map FromEnum Id >> Map (Sub Id 97 >> ToEnum Ordering) Id) ("abcde" :: String) + , expectPE (PresentT ([2,3,5,7,11,13], [1,4,6,8,9,10,12,14,15])) $ pl @(Partition Prime Id) [1..15] + , expectPE (PresentT Nothing) $ pl @Nothing' (Nothing @Int) + , expectPE (FailT "expected Nothing") $ pl @Nothing' (Just 10) + , expectPE (FailT "'Nothing found Just") $ pl @'Nothing (Just 12) + , expectPE (PresentT (Just 10,((),()))) $ pl @(Id &&& '() &&& ()) (Just 10) + , expectPE (PresentT [(-999) % 1,10 % 1,20 % 1,(-999) % 1,30 % 1]) $ pl @(Map (Wrap (MM.First _) Id &&& (Pure Maybe (Neg 999) >> Wrap (MM.First _) Id)) Id >> Map Sapa Id >> Map (Just' Unwrap) Id) [Nothing,Just 10,Just 20,Nothing,Just 30] + , expectPE (PresentT 12) $ pl @(MaybeIn 99 Id) (Just 12) + , expectPE (PresentT 12) $ pl @(JustDef 99 Id) (Just 12) + , expectPE (PresentT 99) $ pl @(MaybeIn 99 Id) Nothing + , expectPE (PresentT 99) $ pl @(JustDef 99 Id) Nothing + , expectPE (PresentT (-99)) $ pl @(MaybeIn (Neg 99) Id) Nothing + , expectPE (PresentT (-99)) $ pl @(JustDef (Neg 99) Id) Nothing + , expectPE (PresentT [1,2,3,4,12]) $ pl @(Para (RepeatT 5 (Guard "0-255" (Between 0 255)))) [1,2,3,4,12] + , expectPE (FailT "0-255") $ pl @(Para (RepeatT 5 (Guard "0-255" (Between 0 255)))) [1,2,3,400,12] + , expectPE (PresentT ["141","021","003","000"]) $ pl @(Para (RepeatT 4 (Printf "%03d" Id))) [141,21,3,0::Int] + , expect3 (Right (unsafeRefined3 [1,2,3,4] "001.002.003.004")) $ eval3 @Ip4A @Ip4B @(Para (RepeatT 4 (Printf "%03d" Id)) >> Concat (Intercalate '["."] Id)) ol "1.2.3.4" + , expect3 (Right (unsafeRefined3 [1,2,3,4] "abc__002__3__zzz")) $ eval3 @Ip4A @Ip4B @(Para '[W "abc",Printf "%03d" Id,Printf "%d" Id,W "zzz"] >> Concat (Intercalate '["__"] Id)) ol "1.2.3.4" + , expect3 (Right (unsafeRefined [1,2,3,4], "001.002.003.004")) $ eval3PX (Proxy @'(Ip4A, Ip4B, Para (RepeatT 4 (Printf "%03d" Id)) >> Concat (Intercalate '["."] Id), _)) ol "1.2.3.4" + , expect3 (Right (unsafeRefined [1,2,3,4], "001.002.003.004")) $ eval3PX (mkProxy3 @Ip4A @Ip4B @(Para (RepeatT 4 (Printf "%03d" Id)) >> Concat (Intercalate '["."] Id))) ol "1.2.3.4" + + -- keep the original value + , expect3 (Right $ unsafeRefined3 ("1.2.3.4", [1,2,3,4]) "001.002.003.004") $ eval3 @(Id &&& Ip4A) @(Snd >> Ip4B) @(Snd >> Para (RepeatT 4 (Printf "%03d" Id)) >> Concat (Intercalate '["."] Id)) ol "1.2.3.4" + + -- need to fill in the types for both even in ghci + , expectPE (PresentT (Just (SG.Sum 10))) $ pl @(Coerce2 (SG.Sum Int)) (Just (10 :: Int)) + , expectPE (PresentT (Just (SG.Sum 0))) $ pl @(MemptyT2 (SG.Sum _)) (Just ()) + , expectPE (PresentT 13) $ pl @(FoldMap (SG.Sum _) Id) (Just 13) + , expectPE (PresentT 55) $ pl @(FoldMap (SG.Sum _) Id) [1..10] + , expectPE (PresentT [Just 1,Just 2,Just 3,Just 4]) $ pl @Sequence (Just [1..4]) + , expectPE (PresentT (Just (SG.Sum 20))) $ pl @(Pure2 SG.Sum) (Just 20) + , expectPE (PresentT Nothing) $ pl @(Traverse (If (Gt 3) (Pure Maybe Id) (EmptyT Maybe)) Id) [1..5] + , expectPE (PresentT Nothing) $ pl @(Traverse (MaybeB (Le 3) Id) Id) [1..5] + , expectPE (PresentT (Just [1,2,3,4,5])) $ pl @(Traverse (If (Gt 0) (Pure Maybe Id) (EmptyT Maybe)) Id) [1..5] + , expectPE (PresentT (Just [1,2,3,4,5])) $ pl @(Traverse (If (Gt 0) (Pure Maybe Id) (MkNothing _)) Id) [1..5] + , expectPE (PresentT (Just [1,2,3,4,5])) $ pl @(Traverse (MaybeB (Id >= 0) Id) Id) [1..5] + , expectPE (PresentT Nothing) $ pl @(Traverse (MaybeB (Id <= 3) Id) Id) [1..5] + + , expectPE (FailT "Printf (IO e=printf: bad formatting char 's')") $ pl @(Printf "%-6s" Id) (1234 :: Int) + , expectPE (PresentT "0004d2") $ pl @(Printf "%06x" Id) (1234 :: Int) + , expectPE (PresentT (Left 123)) $ pl @(Pure (Either String) Id >> Swap) 123 + , expectPE (PresentT [13,2,1999]) $ pl @(Rescan DdmmyyyyRE Id >> OneP >> Map (ReadP Int) Snd) "13-02-1999" + , expectPE (PresentT [3,2,1999]) $ pl @(Rescan DdmmyyyyRE Id >> OneP >> Map (ReadP Int) Snd >> Ddmmyyyyval) "03-02-1999" + , expectPE (FailT "guard(1) month 13 is out of range") $ pl @(Rescan DdmmyyyyRE Id >> OneP >> Map (ReadP Int) Snd >> Ddmmyyyyval) "12-13-1999" + , expectPE (PresentT [[1],[2,3,4],[5,6,7,8],[9,10,11,12]]) $ pl @(SplitAts '[1,3,4] Id) [1..12] + , expectPE (PresentT [[1,2,3],[4]]) $ pl @(SplitAts '[3,1,1,1] Id >> FilterBy (Null >> Not) Id) [1..4] + , expectPE (PresentT 1) $ pl @(Msg (Printf "digits=%d" Len) (Head' Id)) [1..4] + , expectPE (PresentT 10) $ pl @(Luhn' 4) "1230" + , expectPE (FailT "expected 14 mod 10 = 0 but found 4") $ pl @(Luhn' 4) "1234" + , expectPE (PresentT "lhs = 123 rhs = asdf") $ pl @(Printf2 "lhs = %d rhs = %s") (123::Int,"asdf"::String) + , expectPE TrueT $ pl @(DirExists ".") () + , expectPE FalseT $ pl @(DirExists "xxy") () + , expectPE TrueT $ pl @(FileExists ".ghci") () + , expectPE FalseT $ pl @(FileExists "xxy") () + , expectPE TrueT $ pl @(IsInfix "ab" Id) "xyzabw" + , expectPE FalseT $ pl @(IsInfix "aB" Id) "xyzAbw" + , expectPE TrueT $ pl @(IsInfixI "aB" Id) "xyzAbw" + , expectPE FalseT $ pl @(IsInfix "ab" Id) "xyzbaw" + , expectPE TrueT $ pl @(IsPrefix "xy" Id) "xyzabw" + , expectPE FalseT $ pl @(IsPrefix "ab" Id) "xyzbaw" + , expectPE TrueT $ pl @(IsSuffix "bw" Id) "xyzabw" + , expectPE FalseT $ pl @(IsSuffix "bw" Id) "xyzbaw" + , expectPE TrueT $ pl @(IsInfix Fst Snd) ("ab","xyzabw") + , expectPE (PresentT [1 % 1,(-3) % 2,(-3) % 1]) $ pl @'[Pos 1,NegR 3 2,Neg 3] () + , expectPE (PresentT [4, 7, 8, 9]) $ pl @'[4,7,8,9] () + , expectPE (PresentT ["aa","b","","ddd"]) $ pl @'["aa","b","","ddd"] () + , expectPE (PresentT 17) $ pl @(Do (RepeatT 4 (Add Id 4))) 1 + , expectPE (PresentT 24) $ pl @((Id <> Id) >> Unwrap) (SG.Sum 12) + , expectPE (PresentT "abcdef") $ pl @(Fst <> (Snd >> Fst)) ("abc",("def",12)) + , expectPE (PresentT (SG.Sum 25)) $ pl @(Wrap _ 13 <> Id) (SG.Sum @Int 12) + , expectPE (PresentT 23) $ pl @(Add Fst (Last' Snd)) (10,[12,13]) + , expectPE (PresentT (-1,12)) $ pl @(DivMod (Sub 9 Fst) (Last' Snd)) (10,[12,13]) + , expectPE (PresentT [True,False,False,True]) $ pl @(Para '[ W 'True, Ge 12, W 'False, Lt 2 ]) [1,2,-99,-999] + , expectPE (FailT "Para: data elements(3) /= predicates(4)") $ pl @(Para '[ W 'True, Ge 12, W 'False, Lt 2 ]) [1,2,-99] + , expectPE (PresentT [True, False, False]) $ pl @(ParaLax '[ W 'True, Ge 12, W 'False, Lt 2 ]) [1,2,-99] + , expectPE (FailT "Para: data elements(7) /= predicates(4)") $ pl @(Para '[ W 'True, Ge 12, W 'False, Lt 2 ]) [1,2,-99,-999,1,1,2] + , expectPE (FailT "guard(1) err 002") $ pl @(Guards (ToGuardsT (Printf2 "guard(%d) err %03d") '[ W 'True, Ge 12, W 'False, Lt 2 ])) [1,2,-99,-999] + , expectPE (FailT "Guards: data elements(3) /= predicates(4)") $ pl @(Guards (ToGuardsT (Printf2 "guard(%d) err %03d") '[ W 'True, Ge 12, W 'False, Lt 2 ])) [1,2,-99] + , expectPE (FailT "Guards: data elements(7) /= predicates(4)") $ pl @(Guards (ToGuardsT (Printf2 "guard(%d) err %03d") '[ W 'True, Ge 12, W 'True, Lt 2 ])) [1,22,-99,-999,1,1,2] + , expectPE (PresentT [1,22,-99,-999,1,1,2]) $ pl @(GuardsLax (ToGuardsT (Printf2 "guard(%d) err %03d") '[ W 'True, Ge 12, W 'True, Lt 2 ])) [1,22,-99,-999,1,1,2] + , expectPE (PresentT [1,22]) $ pl @(GuardsLax (ToGuardsT (Printf2 "guard(%d) err %03d") '[ W 'True, Ge 12, W 'True, Lt 2 ])) [1,22] + , expectPE (PresentT [1,22,-99,-999]) $ pl @(Guards (ToGuardsT (Printf2 "guard(%d) err %03d") '[ W 'True, Ge 12, W 'True, Lt 2 ])) [1,22,-99,-999] + , expectPE TrueT $ pl @(Fst /= Snd) ("ab","xyzabw") + , expectPE FalseT $ pl @(Fst == Snd) ("ab","xyzabw") + , expectPE (PresentT 157) $ pl @(Fst * (Snd >> Fst) + (Snd >> Snd) `Div` 2) (12,(13,3)) + , expectPE TrueT $ pl @(Fst >= Snd || Snd > 23 || NegR 12 5 <= ToRational Fst) (12,13) + , expectPE (PresentT LT) $ pl @(Fst === Snd) (3,12) + , expectPE TrueT $ pl @(Fst ==? Snd) ("aBc","AbC") + , expectPE (PresentT EQ) $ pl @(Fst ===? Snd) ("aBc","AbC") + , expectPE FalseT $ pl @(Fst == Snd) ("aBc","AbC") + , expectPE (PresentT GT) $ pl @(Fst === Snd) ("aBc","AbC") + , expectPE (PresentT LT) $ pl @(Snd === Fst) ("aBc","AbC") + , expectPE TrueT $ pl @(Fst ==? Snd && Fst == Snd) ("Abc","Abc") + , expectPE (PresentT (EQ,EQ)) $ pl @(Fst ===? Snd &&& Fst === Snd) ("abc","abc") + , expectPE (PresentT "ask%dfas%kef00035 hey %") $ pl @(Printf "ask%%dfas%%kef%05d hey %%" Id) (35 :: Int) + , expectPE (PresentT 100) $ pl @(Id !! 2 !! 0) [[1..5],[10..14],[100..110]] + , expectPE (FailT "(!!) index not found") $ pl @(Id !! 1 !! 7) [[1..5],[10..14],[100..110]] + , expectPE (PresentT '2') $ pl @(IxL Id 1 (Char1 "x")) ("123" :: T.Text) + , expectPE (PresentT 'x') $ pl @(IxL Id 15 (Char1 "x")) ("123" :: T.Text) + , expectPE (FailT "someval int=45") $ pl @(Fail () (Printf "someval int=%d" Id)) (45 :: Int) + , expectPE (FailT "failing with 45") $ pl @(If (Gt 4) (Fail (T _) (Printf "failing with %d" Id)) ()) 45 + , expectPE (PresentT 21) $ pl @(If (Gt 4) (Fail (T _) (Printf "failing with %d" Id)) (Id * 7)) 3 + , expectPE (PresentT ["2","1"]) $ pl @(If (Gt 4) (Fail (T _) (Printf "failing with %d" Id)) (Id * 7 >> ShowP >> Ones)) 3 + , expectPE (FailT "failing with 19") $ pl @(If (Gt 4) (Fail (T _) (Printf "failing with %d" Id)) (Id * 7 >> ShowP >> Ones)) 19 + , expectPE (PresentT 31) $ pl @(Do (RepeatT 4 (Id + 7))) 3 + , expectPE (PresentT 9) $ pl @(Do (RepeatT 4 9)) () + , expectPE (PresentT 3) $ pl @(Do '[1,2,3]) () + , expectPE (PresentT "xy") $ pl @(Do (RepeatT 4 "xy")) 3 + , expectPE (PresentT ["xy","xy","xy","xy"]) $ pl @(RepeatT 4 "xy") 3 + , expectPE (PresentT (Proxy @'["xy","xy","xy","xy"])) $ pl @(Proxy (RepeatT 4 "xy")) 3 + , expectPE (PresentT (This @_ @() 'x')) $ pl @(MkThis () Id) 'x' + , expectPE (PresentT (This @_ @() 'x')) $ pl @(MkThis () Fst) ('x',True) + , expectPE (PresentT (That 'x')) $ pl @(MkThat () Id) 'x' + , expectPE (PresentT (These 'x' True)) $ pl @(MkThese Id 'True) 'x' + , expectPE (PresentT 123) $ pl @(MaybeIn 123 Id) (Nothing @Int) + , expectPE (PresentT 9) $ pl @(MaybeIn 123 Id) (Just 9) + , expectPE (PresentT [1,2,3]) $ pl @(Just' Id) (Just [1,2,3]) + , expectPE (FailT "expected Just") $ pl @(Just' Id) (Nothing @[Int]) + , expectPE (PresentT (66788,26232)) $ pl @(Last' Id >> Id * 123 >> Dup >> (Pred *** (ShowP >> Rescan "(\\d{2})" Id >> Concat (ConcatMap Snd Id) >> ReadBase Int 16))) [12,13,543::Int] + , expectPE (PresentT "d=009 s=ab") $ pl @(Printfn "d=%03d s=%s" Id) (9::Int,("ab"::String,())) + , expectPE (PresentT "d=009 s=ab c=x f=1.54") $ pl @(Printfn "d=%03d s=%s c=%c f=%4.2f" Id) (9::Int,("ab"::String,('x',(1.54::Float,())))) + , expectPE (FailT "Printfn(4)(IO e=printf: formatting string ended prematurely)") $ pl @(Printfn "d=%03d s=%s" Id) (9::Int,("ab"::String,('x',(1.54::Float,())))) + , expectPE (PresentT "lhs = 123 rhs = asdf c=120") $ pl @(Printf3 "lhs = %d rhs = %s c=%d") (123::Int,("asdf"::String,'x')) + , expectPE (PresentT (1,('x',(True,())))) $ pl @(FstL Int Id &&& SndL Char Id &&& ThdL Bool Id &&& ()) (1,'x',True) + , expectPE (PresentT (1,('x',(True,())))) $ pl @(FstL _ Id &&& SndL _ Id &&& ThdL _ Id &&& ()) (1,'x',True) + , expectPE (PresentT (1,(1.4,("aaa",())))) $ pl @(FstL _ Id &&& SndL _ Id &&& ThdL _ Id &&& ()) (1,1.4,"aaa") + , (@?=) ("xx",(True,('x',(1,())))) (reverseTupleC (1,('x',(True,("xx",()))))) + , expectPE (PresentT "hello d=12 z someval") $ pl @(TupleI '[W 12, Char1 "z", W "someval"] >> Printfn "hello d=%d %c %s" Id) () + , expectPE (PresentT "ipaddress 001.002.003.004") $ pl @(TupleI '[1,2,3,4] >> Printfn "ipaddress %03d.%03d.%03d.%03d" Id) () + , expectPE (PresentT (1,(2,(3,(4,()))))) $ pl @(TupleI '[1,2,3,4]) 4 + , expectPE (PresentT (4,(3,(2,(1,()))))) $ pl @(TupleI '[1,2,3,4] >> ReverseTupleN) 4 + , expectPE (PresentT (1,(2,(3,(4,()))))) $ pl @(TupleI '[1,2,3,4] >> ReverseTupleN >> ReverseTupleN) 4 + + , expectPE (PresentT "001.002.003.004") $ pl @(Printfnt 4 "%03d.%03d.%03d.%03d") [1,2,3,4::Int] + , expectPE (FailT "TupleList(4) is strict and has 1 extra element") $ pl @(Printfnt 4 "%03d.%03d.%03d.%03d") [1,2,3,4,5::Int] + , expectPE (FailT "TupleList(4) no data left") $ pl @(Printfnt 4 "%03d.%03d.%03d.%03d") [1,2,3::Int] + + , expectPE (PresentT "001.002.003.004") $ pl @(PrintfntLax 4 "%03d.%03d.%03d.%03d") [1,2,3,4::Int] + , expectPE (PresentT "001.002.003.004") $ pl @(PrintfntLax 4 "%03d.%03d.%03d.%03d") [1,2,3,4,5::Int] + , expectPE (FailT "TupleListLax(4) no data left") $ pl @(PrintfntLax 4 "%03d.%03d.%03d.%03d") [1,2,3::Int] + , expectPE (FailT "Pairs no data found") $ pl @Pairs ([] @()) + , expectPE (FailT "Pairs only one element found") $ pl @Pairs [1] + , expectPE (PresentT [(1,2)]) $ pl @Pairs [1,2] + , expectPE (PresentT [(1,2),(2,3)]) $ pl @Pairs [1,2,3] + , expectPE (PresentT [(1,2),(2,3),(3,4)]) $ pl @Pairs [1,2,3,4] + , expectPE (PresentT "1 2 3 004") $ pl @(PrintfntLax 4 "%d %4d %-d %03d") [1..10::Int] + , expectPE (PresentT "2019-08-17") $ pl @(FormatTimeP "%Y-%m-%d" Id) (read "2019-08-17" :: Day) + , expectPE (PresentT (20,20)) $ pl @(Dup << Fst * Snd) (4,5) + , expectPE (PresentT (20,20)) $ pl @(Fst * Snd >> Dup) (4,5) + , expectPE (PresentT (These "xxx" 4)) $ pl @(Fst <$ Snd) (4,These "xxx" 'a') + , expectPE (PresentT (This 'a')) $ pl @(Fst <$ Snd) (4,This @_ @String 'a') + , expectPE (PresentT (Just 4)) $ pl @(Fst <$ Snd) (4,Just 'a') + , expectPE (PresentT Nothing) $ pl @(Fst <$ Snd) (4,Nothing @Int) + , expectPE (PresentT (Just 4)) $ pl @(Fst <* Snd) (Just 4,Just 'a') + , expectPE (PresentT (Just 'a')) $ pl @(Fst *> Snd) (Just 4,Just 'a') + , expectPE (PresentT ('x',('x',"someval"))) $ pl @Duplicate ('x',"someval") + , expectPE (PresentT "someval") $ pl @Extract ('x',"someval") + , expectPE (PresentT (Just "cdef")) $ pl @(Fst <|> Snd) (Just "cdef",Just "ab") + , expectPE (PresentT "cdefab") $ pl @(Fst <|> Snd) ("cdef","ab"::String) + , expectPE (PresentT (9,"abc")) $ pl @(I $ 9 $ "abc") (,) + , expectPE (PresentT ("abc",9)) $ pl @(9 & "abc" & I) (,) + , expectPE (PresentT "28") $ pl @(Fst $ Snd) (show . (7*),4) + , expectPE (PresentT (12,"12")) $ pl @(Fst $ Snd $ (Snd >> ShowP)) ((,),12) + , expectPE (PresentT (Just (This [1,2,3,4]))) $ pl @(ZipTheseF Fst Snd) (Just [1..4],Nothing @()) + , expectPE (PresentT [These 1 'a',These 2 'b',These 3 'c',This 4]) $ pl @(ZipTheseF Fst Snd) ([1..4],['a'..'c']) + , expectPE (PresentT [True,True,True,True]) $ pl @('True <$ Id) [1..4] + , expectPE (PresentT (Compose (Just "aaaa"))) $ pl @(Char1 "ab" <$ Id) (Compose $ Just [1..4]) + , expectPE (PresentT (4,("aa",'x'))) $ pl @'(4,'(Fst,Snd)) ("aa",'x') + , expectPE (PresentT (4,"aa",'x')) $ pl @'(4,Fst,Snd) ("aa",'x') + , expectPE (PresentT (Just [10])) $ pl @(Pure2 []) (Just 10) + , expectPE (PresentT "hello") $ pl @Extract (10,"hello") + , expectPE (PresentT (M.fromList [(4,"x"),(5,"dd")])) $ pl @(FromList (M.Map _ _)) [(4,"x"),(5,"dd")] + , expectPE (PresentT False) $ pl @(FromList (M.Map _ _) >> I !! Char1 "y") [('x',True),('y',False)] + -- FromListF works only if OverloadedLists is on +-- , expectPE (PresentT (M.fromList [(4,"x"),(5,"dd")])) $ pl @(FromListF (M.Map _ _)) [(4,"x"),(5,"dd")] + , expectPE (PresentT (Just False)) $ pl @(FromList (M.Map _ _) >> Lookup Id (Char1 "y")) [('x',True),('y',False)] + , expectPE (PresentT Nothing) $ pl @(FromList (M.Map _ _) >> Lookup Id (Char1 "z")) [('x',True),('y',False)] + , expectPE (FailT "index('z') not found") $ pl @(FromList (M.Map _ _) >> (Char1 "z" &&& Lookup Id (Char1 "z")) >> If (Snd >> IsNothing) (Fst >> ShowP >> Fail I (Printf "index(%s) not found" Id) >> 'False) (Snd >> 'Just Id)) [('x',True),('y',False)] + , expectPE (PresentT True) $ pl @(FromList (M.Map _ _) >> (Char1 "z" &&& Lookup Id (Char1 "x")) >> If (Snd >> IsNothing) (Fst >> ShowP >> Fail I (Printf "index(%s) not found" Id) >> 'False) (Snd >> 'Just Id)) [('x',True),('y',False)] + , expectPE (FailT "index('z') not found") $ pl @(FromList (M.Map _ _) >> Lookup' _ Id (Char1 "z")) [('x',True),('y',False)] + , expectPE (PresentT ["abc","bcd","cde","def","efg","fgh","ghi","hi","i"]) $ pl @(Unfoldr (If Null (MkNothing _) ('(Take 3 Id, Drop 1 Id) >> MkJust)) Id) "abcdefghi" + , expectPE (PresentT [[1,2],[3,4],[5]]) $ pl @(Unfoldr (If Null (MkNothing _) (Pure _ (SplitAt 2 Id))) Id) [1..5] + , expectPE (PresentT [[1,2],[3,4],[5]]) $ pl @(Unfoldr (MaybeB (Null >> Not) (SplitAt 2 Id)) Id) [1..5] + , expectPE (PresentT [99,1,2,3,4,5]) $ pl @(FlipT (:+) Fst Snd) ([1..5],99) + , expectPE (PresentT [99,1,2,3,4,5]) $ pl @(Fst :+ Snd) (99,[1..5]) + , expectPE (PresentT [[99],[1,99],[2,1,99],[3,2,1,99],[4,3,2,1,99],[5,4,3,2,1,99]]) $ pl @(Scanl (Snd :+ Fst) Fst Snd) ([99],[1..5]) + , expectPE (PresentT [[99]]) $ pl @(Scanl (Snd :+ Fst) Fst Snd) ([99],[]) + , expectPE (FailT "yy") $ pl @(Unfoldr (If Null (MkNothing _) (Guard "yy" (Len < 3) >> Pure _ (SplitAt 2 Id))) Id) [1..5] + , expectPE (FailT "yy") $ pl @(Unfoldr (MaybeB (Null >> Not) (Guard "yy" (Len < 3) >> SplitAt 2 Id)) Id) [1..5] + , expectPE (PresentT [4,1,2,3]) $ pl @(4 :+ '[1,2,3]) () + , expectPE (PresentT [1,2,3,4]) $ pl @('[1,2,3] +: 4) () + , expectPE (PresentT [4,1,2,3]) $ pl @(Fst :+ Snd) (4,[1,2,3]) + , expectPE (PresentT [1,2,3,4]) $ pl @(Snd +: Fst) (4,[1,2,3]) + , expectPE (PresentT "abcx") $ pl @("abc" +: Char1 "x") () + , expectPE (PresentT "abcx") $ pl @(Fst +: Snd) ("abc" :: T.Text,'x') + , expectPE (PresentT [5,1,2,3]) $ pl @(FlipT (:+) '[1,2,3] 5) () + , expectPE (PresentT (map ModifiedJulianDay [0,1,2,3,4,5])) $ pl @(EnumFromTo Fst Snd) (ModifiedJulianDay 0, ModifiedJulianDay 5) + , expectPE (PresentT (map ModifiedJulianDay [0,1,2,3,4,5])) $ pl @((ToEnum Day *** ToEnum Day) >> EnumFromTo Fst Snd) (0,5) + , expectPE (FailT "xx") $ pl @(Unfoldr (Guard "xx" (Len > 4) >> Uncons) Id) [1..10] + , expectPE (PresentT [1,2,3,4,5,6,7,8,9,10]) $ pl @(Unfoldr Uncons Id) [1..10] + , expectPE (PresentT [99,98,97,96]) $ pl @(IterateN 4 Pred) 99 + , expectPE (PresentT (4,'x')) $ pl @('(,) 4 % Char1 "x") () + , expectPE (PresentT (Just False)) $ pl @(FromList (M.Map _ _) >> Lookup Id % Char1 "y") [('x',True),('y',False)] + , expectPE (PresentT (4,"abc")) $ pl @('(,) % 4 % "abc") () + , expectPE (PresentT ("abc",4)) $ pl @(4 %& "abc" %& '(,)) () + , expectPE (PresentT ("abc",4)) $ pl @(FlipT '(,) 4 "abc") () + , expectPE (PresentT (1,[])) $ pl @(Uncons >> MaybeIn '(1,MemptyT _) Id) [] + , expectPE (PresentT []) $ pl @'[] 4 + , expectPE (PresentT (SG.Sum 3)) $ pl @(FromInteger (SG.Sum _) Fst) (3,"A") + , expectPE (PresentT (123 :: DiffTime)) $ pl @(FromInteger DiffTime 123) 'x' + , expectPE (PresentT (0.8 :: Float)) $ pl @(FromRational Float (PosR 4 5)) () + , expectPE (PresentT (14 % 1)) $ pl @(ToRational 14) () + , expectPE (PresentT ('y',3)) $ pl @(Id !! 1) [('x',14),('y',3),('z',5)] + , expectPE (PresentT ('y',3)) $ pl @(Id !!! 1) [('x',14),('y',3),('z',5)] + , expectPE (PresentT (Just ('y',3))) $ pl @(Lookup Id 1) [('x',14),('y',3),('z',5)] + , expectPE (PresentT Nothing) $ pl @(Lookup Id 14) [('x',14),('y',3),('z',5)] + , expectPE (PresentT ('y',3)) $ pl @(Lookup' _ Id 1) [('x',14),('y',3),('z',5)] + , expectPE (FailT "index(14) not found") $ pl @(Lookup' _ Id 14) [('x',14),('y',3),('z',5)] + , expectPE (PresentT 99) $ pl @(FstL _ Id) (99,'a',False,"someval","abc",1.3) + , expectPE (PresentT 'a') $ pl @(SndL _ Id) (99,'a',False,"someval","abc",1.3) + , expectPE (PresentT False) $ pl @(ThdL _ Id) (99,'a',False,"someval","abc",1.3) + , expectPE (PresentT "someval") $ pl @(FthL _ Id) (99,'a',False,"someval","abc",1.3) + , expectPE (PresentT [1,-5,5,-1]) $ pl @('[Pos 1,Negate << PosR 33 7, PosR 21 4,NegR 7 5 >> Signum] >> Map (Floor _ Id) Id) () + , expectPE (PresentT [1,-4,6,-1]) $ pl @('[Pos 1,Negate << PosR 33 7, PosR 21 4,NegR 7 5 >> Signum] >> Map (Ceiling _ Id) Id) () + , expectPE (PresentT [1,-4,5,-1]) $ pl @('[Pos 1,Negate << PosR 33 7, PosR 21 4,NegR 7 5 >> Signum] >> Map (Truncate _ Id) Id) () + , expectPE (PresentT @Integer 2) $ pl @(Truncate' (Fst >> Unproxy ) Snd) (Proxy @Integer,2.3) + , expectPE (PresentT @Int 2) $ pl @(Truncate' Fst Snd) (1::Int,2.3) + , expectPE (PresentT @Float 0.4) $ pl @(FromRational' Fst Snd) (1::Float,2 % 5) + , expectPE (PresentT (5 % 3)) $ pl @((ToRational 5) / (ToRational 3)) 'x' + , expectPE (PresentT (-5 % 3)) $ pl @(Pos 5 / Neg 3) 'x' + , expectPE (PresentT (-5 % 3)) $ pl @(Neg 5 / Fst) (3,'x') + , expectPE (PresentT (-5 % 3)) $ pl @(Snd / Fst) (-3,5) + , expectPE (FailT "DivF zero denominator") $ pl @(Snd / Fst) (0,5) + , expectPE (PresentT 16) $ pl @(Foldl (Guard "someval" (Fst < Snd) >> Snd) (Head' Id) Tail) [1,4,7,9,16] + , expectPE (FailT "7 not less than 6") $ pl @(Foldl (Guard (Printf2 "%d not less than %d") (Fst < Snd) >> Snd) (Head' Id) Tail) [1,4,7,6,16::Int] + , expectPE (PresentT (True,16)) $ pl @(Foldl (If ((Fst >> Fst) && (Snd > (Fst >> Snd))) '( 'True, Snd ) '( 'False, Fst >> Snd )) '( 'True, Head' Id ) Tail) [1,4,7,9,16] + , expectPE (PresentT (False,16)) $ pl @(Foldl (If ((Fst >> Fst) && (Snd > (Fst >> Snd))) '( 'True, Snd ) '( 'False, Fst >> Snd )) '( 'True, Head' Id ) Tail) [1,4,7,9,16,2] + , expectPE (PresentT (False,7)) + $ pl @(Foldl (If (Fst >> Fst) + (If (Snd > (Fst >> Snd)) + '( 'True, Snd ) + '( 'False, Fst >> Snd ) + ) Fst) + '( 'True, Head' Id) Tail) [1,4,7,6,16] + , expectPE (PresentT [1,2,3,4]) $ pl @(Init' Id) [1..5] + , expectPE (FailT "Init(empty)") $ pl @(Init' Id) ([] @()) + , expectPE (PresentT [2,3,4,5]) $ pl @(Tail' Id) [1..5] + , expectPE (FailT "Tail(empty)") $ pl @(Tail' Id) ([] @()) + , expectPE (PresentT [10,12,13]) $ pl @(CatMaybes Id) [Just 10, Just 12, Nothing, Just 13] + , expectPE (PresentT [5,4,3,2,1]) $ pl @(Foldl (Snd :+ Fst) (MemptyT [_]) Id) [1..5] + , expectPE (PresentT (map SG.Min [9,10,11,12,13])) $ pl @(EnumFromTo (Pure SG.Min 9) (Pure _ 13)) () + , expectPE (PresentT (map SG.Min [9,10,11,12,13])) $ pl @(EnumFromTo (Wrap (SG.Min _) 9) (Wrap _ 13)) () +-- , expectPE (PresentT (Just 'x')) $ pl @(Purex Fst Snd) (Just 10,'x') + , expectPE (PresentT (Just 'x')) $ pl @(Snd <$ Fst) (Just 10,'x') + , expectPE (PresentT (Nothing @(SG.Sum _))) $ pl @(MemptyT' Id) (Just (SG.Sum 12)) + , expectPE (PresentT ([4,99],"xy")) $ pl @PartitionEithers [Left 4, Right 'x', Right 'y',Left 99] + , expectPE (PresentT (([4,99],"xy"),[(3,'b'),(5,'x')])) $ pl @PartitionThese [This 4, That 'x', That 'y',These 3 'b', This 99, These 5 'x'] + , expectPE (PresentT [1,2,3]) $ pl @(MapMaybe (MaybeB (Le 3) Id) Id) [1..5] + , expectPE (PresentT [4,5]) $ pl @(MapMaybe (MaybeB (Gt 3) Id) Id) [1..5] + , expectPE (PresentT [94,93,92,91]) $ pl @(IterateWhile (Id > 90) Pred) 94 + , expectPE (PresentT [94,93,92,91,90]) $ pl @(IterateUntil (Id < 90) Pred) 94 + , expectPE (PresentT [95,94,93,92,91]) $ pl @(IterateNWhile 10 (Id > 90) Pred) 95 + , expectPE (PresentT [95,94,93]) $ pl @(IterateNWhile 3 (Id > 90) Pred) 95 + , expectPE (PresentT [95,94,93,92,91]) $ pl @(IterateNUntil 10 (Id <= 90) Pred) 95 + , expectPE (PresentT [95,94,93]) $ pl @(IterateNUntil 3 (Id <= 90) Pred) 95 + -- check for infinite loops + , expectPE (FailT "Unfoldr (9999,1):failed at i=100") $ pl @(IterateNUntil 9999 'False I) 1 + , expectPE (FailT "Scanl:failed at i=100") $ pl @(Foldl Fst '() (EnumFromTo 1 9999)) () + , expectPE (PresentT [1,2,3,4,5,99]) $ pl @(MaybeX Fst ((Fst >> Fst) +: Snd) Snd) ([1..5],Just 99) + , expectPE (PresentT [1,2,3,4,5]) $ pl @(MaybeX Fst ((Fst >> Fst) +: Snd) Snd) ([1..5],Nothing) + , expectPE (PresentT "a=9 b=rhs") $ pl @(TheseX (Printf "a=%d" (Snd >> Succ)) ("b=" <> Snd) (Snd >> Printf2 "a=%d b=%s") Id) (These @Int 9 "rhs") + , expectPE (PresentT "a=10") $ pl @(TheseX (Printf "a=%d" (Snd >> Succ)) ("b=" <> Snd) (Snd >> Printf2 "a=%d b=%s") Id) (This @Int 9) + , expectPE (PresentT "b=rhs") $ pl @(TheseX (Printf "a=%d" (Snd >> Succ)) ("b=" <> Snd) (Snd >> Printf2 "a=%d b=%s") Id) (That @Int "rhs") + , expectPE (PresentT ([] @Int)) $ pl @(HeadP Id) (map (:[]) ([] @Int)) + , expectPE (PresentT ([10] :: [Int])) $ pl @(HeadP Id) (map (:[]) ([10..14] :: [Int])) + , expectPE (PresentT 10) $ pl @(HeadDef Fst Snd) (99,[10..14]) + , expectPE (PresentT 99) $ pl @(HeadDef Fst Snd) (99,[] @Int) + , expectPE (PresentT 43) $ pl @(HeadDef 43 Snd) (99,[] @Int) + , expectPE (PresentT (Just 'd')) $ pl @(Lookup "abcdef" 3) () + , expectPE (PresentT (Just 5)) $ pl @(Lookup '[1,2,3,4,5,6] 4) () + , expectPE (PresentT 5) $ pl @(LookupDef '[1,2,3,4,5,6] 4 Id) 23 + , expectPE (PresentT 5) $ pl @(LookupDef '[1,2,3,4,5,6] 4 Fst) (23,'x') + , expectPE (PresentT 23) $ pl @(LookupDef '[1,2,3,4,5,6] 99 Id) 23 + , expectPE (PresentT 23) $ pl @(LookupDef '[1,2,3,4,5,6] 99 Fst) (23,'x') + , expectPE (PresentT 5) $ pl @(LookupDef '[1,2,3,4,5,6] 4 999) (23,'x') + , expectPE (PresentT 999) $ pl @(LookupDef '[1,2,3,4,5,6] 40 999) (23,'x') + , expectPE (PresentT (SG.Min 5)) $ pl @(LookupP Fst 4) (map SG.Min [1::Int .. 10],'x') + , expectPE (PresentT (mempty @(SG.Min _))) $ pl @(LookupP Fst 999) (map SG.Min [1::Int .. 10],'x') + , expectPE (FailT "someval") $ pl @(LookupFail "someval" Fst 999) (map SG.Min [1::Int .. 10],'x') + , expectPE (FailT "abcsomeval") $ pl @(Fail (Snd >> Unproxy) (Fst <> "someval")) ("abc",Proxy @Int) + , expectPE (FailT "char=x") $ pl @(LookupFail (Printf "char=%c" Snd) Fst 49) (map SG.Min [1::Int ..10],'x') + , expectPE (FailT "someval=13") $ pl @(LeftFail (Printf "someval=%d" Fst) Snd) (13::Int,Right @(SG.Sum Int) "abc") + , expectPE (FailT "someval=Right \"abc\"") $ pl @(LeftFail (Printf "someval=%s" ShowP) Id) (Right @(SG.Sum Int) "abc") + , expectPE (FailT "msg=(\"ASfd\",[]) Asdf") $ pl @(GFail Uncons (Printf "msg=%s Asdf" ShowP) Snd) ("ASfd",[]::[Int]) + , expectPE (PresentT 'c') $ pl @(LookupDef' Fst Snd (Char1 "xx") Id) (['a'..'e'],2) + , expectPE (PresentT 'x') $ pl @(LookupDef' Fst Snd (Char1 "xx") Id) (['a'..'e'],999) + , expectPE (PresentT 'x') $ pl @(LookupDef' Fst Snd (Char1 "xx") Id) ([],2) + , expectPE (PresentT 'x') $ pl @(LookupDef' Fst Snd (Char1 "xx") Snd) ('w',([],2)) + , expectPE (PresentT 'c') $ pl @(LookupDef' Fst Snd Fst Snd) ('x',(['a'..'e'],2)) + , expectPE (PresentT(SG.Min 13)) $ pl @(LookupP' Fst Snd Snd) ('x',(map SG.Min [10..15::Int], 3)) + + , expectPE (PresentT 9) $ pl @(HeadDef 9 Fst) ([],True) + , expectPE (PresentT 1) $ pl @(HeadDef 9 Fst) ([1..5],True) + , expectPE (PresentT 10) $ pl @(HeadDef 3 Fst) ([10..15],True) + + , expectPE (PresentT 9) $ pl @(LastDef 9 Fst) ([],True) + , expectPE (PresentT 5) $ pl @(LastDef 9 Fst) ([1..5],True) + , expectPE (PresentT 15) $ pl @(LastDef 3 Fst) ([10..15],True) + + , expectPE (PresentT [9,7]) $ pl @(InitDef '[9,7] Fst) ([],True) + , expectPE (PresentT [1,2,3,4]) $ pl @(InitDef '[9,7] Fst) ([1..5],True) + , expectPE (PresentT [10,11,12,13,14]) $ pl @(InitDef '[3] Fst) ([10..15],True) + , expectPE (PresentT [10,11,12,13,14]) $ pl @(InitP Fst) ([10..15],True) + , expectPE (PresentT []) $ pl @(InitP Fst) ([] @Int,True) + + , expectPE (PresentT [9,7]) $ pl @(TailDef '[9,7] Fst) ([],True) + , expectPE (PresentT [2,3,4,5]) $ pl @(TailDef '[9,7] Fst) ([1..5],True) + , expectPE (PresentT [11,12,13,14,15]) $ pl @(TailDef '[3] Fst) ([10..15],True) + , expectPE (PresentT [11,12,13,14,15]) $ pl @(TailP Fst) ([10..15],True) + , expectPE (PresentT []) $ pl @(TailP Fst) ([] @Int,True) + + , expectPE (FailT "a=4 b=someval") $ pl @(TailFail (Snd >> Printf2 "a=%d b=%s") Fst) ([]::[()],(4::Int,"someval" :: String)) + , expectPE (PresentT 3) $ pl @(JustDef' 44 (Fst >> Fst >> Fst) Snd) (3,Just 20) + , expectPE (PresentT 999) $ pl @(JustDef' 44 999 Snd) ("xxx",Just 20) + , expectPE (PresentT "xxabcd") $ pl @(JustDef' "dd" ((Fst >> Fst >> Fst) <> Snd) Snd) ("xx",Just "abcd") + , expectPE (PresentT "dd") $ pl @(JustDef' "dd" ((Fst >> Fst >> Fst) <> Snd) Snd) ("xx",Nothing) + , expectPE (PresentT "xx") $ pl @(JustDef' Fst ((Fst >> Fst >> Fst) <> Snd) Snd) ("xx",Nothing) + + , expectPE (PresentT 3) $ pl @(JustDef' 44 (Fst >> Fst >> Snd) Fst) (Just 20,3) + , expectPE (PresentT 999) $ pl @(JustDef' 44 999 Fst) (Just 20,"xxx") + , expectPE (PresentT "xxabcd") $ pl @(JustDef' "dd" ((Fst >> Fst >> Snd) <> Snd) Fst) (Just "abcd","xx") + , expectPE (PresentT "dd") $ pl @(JustDef' "dd" ((Fst >> Fst >> Snd) <> Snd) Fst) (Nothing,"xx") + , expectPE (PresentT "xx") $ pl @(JustDef' Snd ((Fst >> Fst >> Snd) <> Snd) Fst) (Nothing,"xx") + + , expectPE (PresentT "xxya") $ pl @((Id &&& Snd) >> MaybeXP (Fst >> Fst >> Fst) ((Fst >> Fst >> Fst) <> Snd) Snd) ("xx",Just "ya") + , expectPE (PresentT "xxya") $ pl @((Id &&& Fst) >> MaybeXP (Fst >> Fst >> Snd) ((Fst >> Fst >> Snd) <> Snd) Snd) (Just "ya","xx") + + , expectPE (PresentT "xx") $ pl @((Id &&& Snd) >> MaybeXP (Fst >> Fst >> Fst) ((Fst >> Fst >> Fst) <> Snd) Snd) ("xx",Nothing) + + , expectPE (PresentT "aabb") $ pl @(JustDef''' Fst ((Fst >> Fst) <> Snd) Snd) ("aa", Just "bb") + , expectPE (PresentT "aa") $ pl @(JustDef''' Fst ((Fst >> Fst) <> Snd) Snd) ("aa", Nothing) + , expectPE (PresentT "ssbb") $ pl @(JustDef''' Fst ("ss" <> Snd) Snd) ("aa", Just "bb") + + , expectPE (PresentT (Just 1)) $ pl @Fmap_1 (Just (1,'x')) + , expectPE (PresentT (Just 'x')) $ pl @Fmap_2 (Just (1,'x')) + , expectPE (PresentT (Nothing @Int)) $ pl @Fmap_2 (Nothing @(Char,Int)) + , expectPE (PresentT [1,2,3]) $ pl @Fmap_1 [(1,'x'), (2,'y'), (3,'z')] + , expectPE (PresentT (Right 'x')) $ pl @Fmap_2 (Right @() (1,'x')) + , expectPE (PresentT (Left @_ @Double "x")) $ pl @Fmap_2 (Left @_ @(Int,Double) "x") + + , expectPE (PresentT [1,10,99]) $ pl @Thiss [This 1, This 10,That 'x', This 99, That 'y'] + , expectPE (PresentT "xy") $ pl @Thats [This 1, This 10,That 'x', This 99, That 'y'] + , expectPE (PresentT ("xabz",[1,10])) $ pl @PartitionEithers [Left 'x', Right 1,Left 'a', Left 'b',Left 'z', Right 10] + , expectPE (FailT "found rhs=Right 10") $ pl @(LeftFail (Printf "found rhs=%s" ShowP) Id) (Right @String 10) + , expectPE (FailT "found rhs=23") $ pl @(LeftFail (Printf "found rhs=%d" (Snd >> Snd)) (Snd >> Fst)) ('x',(Right @() 10,23::Int)) + , expectPE (PresentT "abc") $ pl @(LeftFail (Printf "found rhs=%d" (Snd >> Snd)) (Snd >> Fst)) ('x',(Left @_ @() "abc",23::Int)) + , expectPE (PresentT (([1,4,10],"xy"),[(9,'z'),(8,'y')])) $ pl @PartitionThese [This 1,That 'x',This 4,That 'y',These 9 'z',This 10,These 8 'y'] + , expectPE (PresentT [('a',1),('a',10),('z',14),('m',22)]) $ pl @(SortOn Snd Snd) ((),[('z',14),('a',10),('m',22),('a',1)]) + , expectPE (PresentT [('z',1),('m',22),('a',10)]) $ pl @(SortOnDesc Fst Snd) ((),[('z',1),('a',10),('m',22)]) + , expectPE (PresentT [('a',10),('m',22),('z',1)]) $ pl @(SortOn Fst Snd) ((),[('z',1),('a',10),('m',22)]) + , expectPE (PresentT [('z',1),('m',22),('a',10)]) $ pl @(SortBy (Swap >> OrdA Fst) Snd) ((),[('z',1),('a',10),('m',22)]) + , expectPE (PresentT ["aa","cx","by","az"]) $ pl @(SortBy (OrdA Reverse) Id) ["az","by","cx","aa"] + , expectPE (PresentT [('a',10),('a',9),('m',22),('m',10),('z',1)]) $ pl @(SortOn Fst Id) [('z',1),('a',10),('m',22),('a',9),('m',10)] + , expectPE (PresentT [('a',9),('a',10),('m',10),('m',22),('z',1)]) $ pl @(SortOn Id Id) [('z',1),('a',10),('m',22),('a',9),('m',10)] + , expectPE (PresentT (False,9)) $ pl @(Just' Uncons >> Foldl (If (Fst>>Fst) (If ((Fst>>Snd) < Snd) '( 'True,Snd ) '( 'False,Snd)) Fst) '( 'True,Fst) Snd) [-10,-2,2,3,4,10,9,11] + , expectPE (PresentT (True,11)) $ pl @(Just' Uncons >> Foldl (If (Fst>>Fst) (If ((Fst>>Snd) < Snd) '( 'True,Snd ) '( 'False,Snd)) Fst) '( 'True,Fst) Snd) [-10,2,3,4,10,11] + , expectPE (FailT "pivot=5 value=3(2)") $ pl @(SortBy (If (Fst==5 && Snd==3) (FailPrt2 _ "pivot=%d value=%d") 'GT) Snd) ((), [5,7,3,1,6,2,1,3]) + , expectPE (PresentT [1,1,2,3,3,5,6,7]) $ pl @(SortBy (If (Fst==50 && Snd==3) (FailPrt2 _ "pivot=%d value=%d") (OrdA Id)) Snd) ((), [5,7,3,1,6,2,1,3]) + , expectPE TrueT $ pl @(Between' (Fst >> Fst) (Fst >> Snd) Snd) ((1,4),3) + , expectPE FalseT $ pl @(Between' (Fst >> Fst) (Fst >> Snd) Snd) ((1,4),10) + , expectPE (FailT "no match on [03/29/0x7]") $ pl @(Map (ParseTimes Day '["%Y-%m-%d", "%m/%d/%y", "%b %d %Y"] Id) Id) ["2001-01-01", "Jan 24 2009", "03/29/0x7"] + , expectPE (PresentT [read @Day "2001-01-01", read @Day "2009-01-24", read @Day "2007-03-29"]) $ pl @(Map (ParseTimes Day '["%Y-%m-%d", "%m/%d/%y", "%b %d %Y"] Id) Id) ["2001-01-01", "Jan 24 2009", "03/29/07"] + + , expectPE (PresentT "gt3") $ pl @(Case (Snd >> Failp "xx") '[Gt 3, Lt 2, Same 3] '["gt3","lt2","eq3"] Id) 15 + , expectPE (PresentT "lt2") $ pl @(Case (Snd >> Failp "xx") '[Gt 3, Lt 2, Same 3] '["gt3","lt2","eq3"] Id) 1 + , expectPE (PresentT "eq3") $ pl @(Case (Snd >> Failp "xx") '[Gt 3, Lt 2, Same 3] '["gt3","lt2","eq3"] Id) 3 + + , expectPE (FailT "no match") $ pl @(Case (Snd >> Failp "no match") '[Same 1, Same 2, Same 3] '["eq1","eq2","eq3"] Id) 15 + , expectPE (FailT "no match for 015") $ pl @(Case (Fail (Snd >> Unproxy) (Printf "no match for %03d" Fst)) '[Same 1, Same 2, Same 3] '["eq1","eq2","eq3"] Id) 15 + , expectPE (FailT "no match for 015") $ pl @(Case (FailCase (Printf "no match for %03d" Id)) '[Same 1, Same 2, Same 3] '["eq1","eq2","eq3"] Id) 15 + , expectPE (FailT "no match for 015") $ pl @(Case'' (Printf "no match for %03d" Id) '[Same 1, Same 2, Same 3] '["eq1","eq2","eq3"] Id) 15 + , expectPE (PresentT "other") $ pl @(Case "other" '[Same 1, Same 2, Same 3] '["eq1","eq2","eq3"] Id) 15 + , expectPE (PresentT "151515") $ pl @(Case (Fst >> ShowP >> Id <> Id <> Id) '[Same 1, Same 2, Same 3] '["eq1","eq2","eq3"] Id) 15 + , expectPE (FailT "Case:no match") $ pl @(Case' '[Same 1, Same 2, Same 3] '["eq1","eq2","eq3"] Id) 15 + , expectPE (FailT "no match for -012") $ pl @(Case (FailCase (Printf "no match for %04d" Id)) '[Between 0 5, Same 6, Between 7 10] '[ 'LT, 'EQ, 'GT] Id) (-12) + , expectPE (PresentT [Left 1,Left 2,Right "fizz",Left 4,Right "buzz",Right "fizz",Left 7,Left 8,Right "fizz",Right "buzz",Left 11,Right "fizz",Left 13,Left 14,Right "fizzbuzz"]) $ pl @(Map Fizzbuzznew Id) [1..15] + , expectPE (PresentT (Left 'x')) $ pl @(EitherB (Fst > 10) (Snd >> Fst) (Snd >> Snd)) (7,('x',99)) + , expectPE (PresentT (Right 99)) $ pl @(EitherB (Fst > 10) (Snd >> Fst) (Snd >> Snd)) (11,('x',99)) + , expectPE (PresentT (Right 99)) $ pl @(EitherB (Gt 10) "found left" 99) 12 + , expectPE (PresentT (Left "found left")) $ pl @(EitherB (Gt 10) "found left" 99) 7 + + , expectPE (FailT "msg=someval caught(044)") $ pl @(Catch' (Failt Int "someval") (Printf2 "msg=%s caught(%03d)")) (44 :: Int) + , expectPE (FailT "msg=expected list of length 1 but found length=3 caught([10,12,13])") $ pl @(Catch' OneP (Second ShowP >> Printf2 "msg=%s caught(%s)")) [10,12,13] + , expectPE (PresentT 10) $ pl @(Catch' OneP (Second ShowP >> Printf2 "msg=%s caught(%s)")) [10] + , expectPE (FailT "msg=expected list of length 1 but found length=2 err s=[10,11]") $ pl @(Catch' OneP (Second ShowP >> Printf2 "msg=%s err s=%s")) [10,11] + , expectPE (PresentT 99) $ pl @(Catch OneP 99) [10,11] + , expectPE (PresentT 10) $ pl @(Catch OneP 99) [10] + , expectPE (PresentT False) $ pl @(Catch OneP 'True) [False] -- cant know that this is FalseT cos is driven by type of the list not the 'True part + , expectPE FalseT $ pl @(Catch OneP 'False) [True,True,False] + , expectPE TrueT $ pl @(Catch OneP 'True) [] + , expectPE (PresentT (-255)) $ pl @(ReadBase Int 16) "-ff" + , expectPE (PresentT 255) $ pl @(ReadBase Int 16) "ff" + , expectPE (PresentT "-7b") $ pl @(ShowBase 16) (-123) + , expectPE (PresentT "7b") $ pl @(ShowBase 16) 123 + , expectPE (PresentT "abc") $ pl @(Fst3 >> Snd >> Thd3) (('x',(13,False,"abc")),True,'y') + , expectPE (PresentT 9.3) $ pl @(Thd3 >> Snd3 >> Fst) ('x',True,(13,(9.3,False),"def")) + ] + +type Fizzbuzz = Id &&& If (Id `Mod` 3==0) "fizz" "" <> If (Id `Mod` 5==0) "buzz" "" +type Fizzbuzz' = Id &&& Case "" '[Id `Mod` 15 == 0, Id `Mod` 3 == 0, Id `Mod` 5 == 0] '["fizzbuzz", "fizz", "buzz"] Id +type Fizzbuzz'' t t1 = Case (MkLeft t Fst) '[Id `Mod` 15 == 0, Id `Mod` 3 == 0, Id `Mod` 5 == 0] '[ MkRight t1 "fizzbuzz", MkRight t1 "fizz", MkRight t1 "buzz"] Id +--type Fizzbuzz''' = Case (MkLeft' (Snd >> Proxyfabb >> Unproxy) Fst) '[Id `Mod` 15 == 0, Id `Mod` 3 == 0, Id `Mod` 5 == 0] '[ MkRight Int "fizzbuzz", MkRight Int "fizz", MkRight Int "buzz"] Id +type Fizzbuzz'''' = Case (MkLeft String Fst) '[Id `Mod` 15 == 0, Id `Mod` 3 == 0, Id `Mod` 5 == 0] '[ MkRight Int "fizzbuzz", MkRight Int "fizz", MkRight Int "buzz"] Id +-- this is also good: makes use of type family MapT which does the apply on ADTs: so type synonyms dont work +type Fizzbuzznew = Case (MkLeft String Fst) '[Id `Mod` 15 == 0, Id `Mod` 3 == 0, Id `Mod` 5 == 0] (MapT (MkRight' (Hole Int)) '[ "fizzbuzz", "fizz", "buzz"]) Id + +type Fizzbuzzalt = '(Id, If (Id `Mod` 3==0) "fizz" "" <> If (Id `Mod` 5==0) "buzz" "") +type Fizzbuzzs = Map Fizzbuzz Id +type Fizzbuzzs1 t t1 = Map (Fizzbuzz >> If (Snd >> Null) (MkLeft t Fst) (MkRight t1 Snd)) Id +type Fizzbuzzs2 = Map (Fizzbuzz >> If (Snd >> Null) (MkLeft String Fst) (MkRight Int Snd)) Id +-- best one cos leverages type info to determine Either a b +type Fizzbuzzs3 = Map (Fizzbuzz >> If (Snd == "") (MkLeft' Snd Fst) (MkRight' Fst Snd)) Id + +
+ test/TestRefined.hs view
@@ -0,0 +1,142 @@+{-# OPTIONS -Wall #-} +{-# OPTIONS -Wcompat #-} +{-# OPTIONS -Wincomplete-record-updates #-} +{-# OPTIONS -Wincomplete-uni-patterns #-} +{-# OPTIONS -Wno-type-defaults #-} +{-# OPTIONS -Wno-redundant-constraints #-} +{-# LANGUAGE TypeOperators #-} +{-# LANGUAGE AllowAmbiguousTypes #-} +{-# LANGUAGE TypeApplications #-} +{-# LANGUAGE DataKinds #-} +{-# LANGUAGE GADTs #-} +{-# LANGUAGE PolyKinds #-} +{-# LANGUAGE ScopedTypeVariables #-} +{-# LANGUAGE OverloadedStrings #-} +{-# LANGUAGE NoOverloadedLists #-} -- overloaded lists breaks some predicates +{-# LANGUAGE ViewPatterns #-} +{-# LANGUAGE FlexibleContexts #-} +{-# LANGUAGE NoStarIsType #-} +{-# LANGUAGE DeriveGeneric #-} +{-# LANGUAGE TemplateHaskell #-} +{-# LANGUAGE LambdaCase #-} +module TestRefined where +import TastyExtras +import Test.Tasty +import Test.Tasty.HUnit +import Test.Tasty.QuickCheck +import Predicate +import Refined +import UtilP +import UtilP_TH + +import Control.Lens +import Data.Aeson +import TH_Orphans () -- need this else refined*TH' fails for dates +import Control.Monad.Cont +import Text.Show.Functions () +import GHC.TypeNats (Nat) + +suite :: IO () +suite = defaultMain $ testGroup "TestRefined" (namedTests <> orderTests unnamedTests <> allProps) + +namedTests :: [TestTree] +namedTests = + [ + testCase "always true" $ (@?=) ($$(refinedTH 7) :: Refined 'True Int) (unsafeRefined 7) + , testCase "between5and9" $ (@?=) ($$(refinedTH 7) :: Refined (Between 5 9) Int) (unsafeRefined 7) + ] + +unnamedTests :: [IO ()] +unnamedTests = [ + (@?=) (unsafeRefined @'True ("1.2.3.4" :: String)) $$(refinedTH "1.2.3.4") + , (@?=) (unsafeRefined @((Len >> Same 4) && Luhn) [1,2,3,0]) $$(refinedTH [1,2,3,0]) + , (@?=) (unsafeRefined @((Len >> Same 4) && Luhn >> Not) [1,2,3,1]) $$(refinedTH [1,2,3,1]) + + , (@?=) [(unsafeRefined 7, "")] (reads @(Refined (Between 2 10) Int) "Refined {unRefined = 7}") + , (@?=) [] (reads @(Refined (Between 2 10) Int) "Refined {unRefined = 0}") + , (@?=) [(unsafeRefined "abcaaaabb", "")] (reads @(Refined (Re "^[abc]+$" Id) String) "Refined {unRefined = \"abcaaaabb\"}") + , (@?=) [] (reads @(Refined (Re "^[abc]+$" Id) String) "Refined {unRefined = \"abcaaaabbx\"}") + + , expectJ (Left ["Error in $: Refined:FalseP"]) (toFrom (unsafeRefined @(Between 4 7 || Gt 14) 12)) + , expectJ (Right (unsafeRefined 22)) (toFrom (unsafeRefined @(Between 4 7 || Gt 14) 22)) + , expectJ (Left ["Error in $: Refined:FailP \"someval\""]) (toFrom (unsafeRefined @(Between 4 7 || Gt 14 || Failt _ "someval") 12)) + + , (fst $ unRavelTI (tst1 ol 10 200)) @?= Right (10,200) + , (fst $ unRavelTI (tst1 ol 11 12)) @?= Left "FalseP" + , (fst <$> unRavelT (tst2 ol 10 200)) >>= (@?= Right (10,200)) + , (fst <$> unRavelT (tst2 ol 11 12)) >>= (@?= Left "FalseP") + ] + +allProps :: [TestTree] +allProps = + [ + testProperty "readshow" $ forAll (arbRefined @(Between 10 45) ol) (\r -> read @(Refined (Between 10 45) Int) (show r) === r) + , testProperty "jsonroundtrip" $ forAll (arbRefined @(Between 10 45) ol) (\r -> testRefinedJ @(Between 10 45) ol (unRefined r) === Right r) + ] + + +type Ip4RE = "^(\\d{1,3})\\.(\\d{1,3})\\.(\\d{1,3})\\.(\\d{1,3})$" + +type Ip4 = Rescan Ip4RE Id >> OneP >> Map (ReadBaseInt 10) Snd >> Ip4guard + +type Ip4guard = Guard "4octets" (Len >> Same 4) >> Guard "0-255" (All (Between 0 255)) + +type Ip6 = Resplit ":" Id + >> Guard "count is bad" (Len >> Between 0 8) + >> Guard "not a hex" (All (All (Elem Id "abcdefABCDEF0123456789"))) + >> Guard "len is bad" (All (Len >> Le 4)) + +type Ip6A = Map (If (Id == "") "0" Id) (Resplit ":" Id) + >> Map (ReadBaseInt 16) Id + +type Ip6B = Guard "count is bad" (Len >> Between 0 8) + >> Guard "out of bounds" (All (Between 0 65535)) + >> 'True + +type Ip6A' = Resplit ":" Id + >> Map (If (Id == "") "0" Id) Id + >> Map (ReadBaseInt 16) Id + >> PadL 8 0 Id + +type Ip6A'' = Map (If (Id == "") 0 (ReadBaseInt 16)) (Resplit ":" Id) >> PadL 8 0 Id + +type Ip6B' = Guard "count is bad" (Len >> Same 8) + >> Guard "out of bounds" (All (Between 0 65535)) + >> 'True + +type Ip4A = Map (ReadBaseInt 10) (Resplit "\\." Id) +type Ip4B = Guard "expected 4 numbers" (Len >> Same 4) + >> Guard "each number must be between 0 and 255" (All (Between 0 255)) + >> 'True + +type Ip4C = Printfnt 4 "%03d.%03d.%03d.%03d" + +-- base n number of length x and then convert to a list of length x of (0 to (n-1)) +-- checks that each digit is between 0 and n-1 +type MM1 (n :: Nat) = Map (ReadBase Int n) Ones +type MM2 (n :: Nat) = ExitWhen "found empty" IsEmpty >> Guard "0<=x<n" (All (Ge 0 && Lt n)) + +-- prtRefinedT tst1 +tst1 :: Monad m => POpts -> Int -> Int -> RefinedT m (Int,Int) +tst1 opts i j = withRefinedT @(Between 2 11) opts i + $ \x -> withRefinedT @(Between 200 211) opts j + $ \y -> return (unRefined x, unRefined y) + +-- prtRefinedTIO tst2 +tst2 :: MonadIO m => POpts -> Int -> Int -> RefinedT m (Int,Int) +tst2 opts i j = withRefinedTIO @(Between 2 11) opts i + $ \x -> withRefinedTIO @(Stderr "start" |> Between 200 211 >| Stderr "end") opts j + $ \y -> return (unRefined x, unRefined y) + +-- roundtrip tojson then fromjson +testRefinedJ :: forall p a + . (ToJSON a, FromJSON a, RefinedC p a) + => POpts + -> a + -> Either String (Refined p a) +testRefinedJ opts a = + let ((bp,e),mr) = runIdentity $ newRefined @p opts a + in case mr of + Nothing -> error $ show bp ++ "\n" ++ e + Just r -> eitherDecode @(Refined p a) $ encode r +
+ test/TestRefined3.hs view
@@ -0,0 +1,387 @@+{-# OPTIONS -Wall #-} +{-# OPTIONS -Wcompat #-} +{-# OPTIONS -Wincomplete-record-updates #-} +{-# OPTIONS -Wincomplete-uni-patterns #-} +{-# OPTIONS -Wno-type-defaults #-} +{-# OPTIONS -Wno-redundant-constraints #-} +{-# LANGUAGE TypeOperators #-} +{-# LANGUAGE AllowAmbiguousTypes #-} +{-# LANGUAGE TypeApplications #-} +{-# LANGUAGE DataKinds #-} +{-# LANGUAGE GADTs #-} +{-# LANGUAGE PolyKinds #-} +{-# LANGUAGE ScopedTypeVariables #-} +{-# LANGUAGE OverloadedStrings #-} +{-# LANGUAGE NoOverloadedLists #-} -- overloaded lists breaks some predicates +{-# LANGUAGE ViewPatterns #-} +{-# LANGUAGE FlexibleContexts #-} +{-# LANGUAGE NoStarIsType #-} +{-# LANGUAGE DeriveGeneric #-} +{-# LANGUAGE TemplateHaskell #-} +{-# LANGUAGE LambdaCase #-} +module TestRefined3 where +import TastyExtras +import Test.Tasty +import Test.Tasty.HUnit +import Test.Tasty.QuickCheck +import Predicate +import TestRefined hiding (namedTests,unnamedTests,allProps) +import Refined +import Refined3 +import Refined3Helper +import UtilP +import UtilP_TH +import Data.Ratio + +import Data.Typeable +import Control.Lens +import Data.Time +import GHC.Generics (Generic) +import Data.Aeson +import TH_Orphans () -- need this else refined*TH' fails for dates +import Control.Monad.Cont +import Text.Show.Functions () +import Data.Tree + +suite :: IO () +suite = defaultMain $ testGroup "TestRefined3" (namedTests <> orderTests unnamedTests <> allProps) + +namedTests :: [TestTree] +namedTests = + [ testCase "ip9" $ (@?=) ($$(refined3TH "121.0.12.13") :: MakeR3 Ip9) (unsafeRefined3 [121,0,12,13] "121.000.012.013") + , testCase "luhn check" $ (@?=) ($$(refined3TH "12345678903") :: MakeR3 CC11) (unsafeRefined3 [1,2,3,4,5,6,7,8,9,0,3] "1234-5678-903") + , testCase "datetime utctime" $ (@?=) ($$(refined3TH "2019-01-04 23:00:59") :: MakeR3 (DateTime1 UTCTime)) (unsafeRefined3 (read "2019-01-04 23:00:59 UTC") "2019-01-04 23:00:59") + , testCase "datetime localtime" $ (@?=) ($$(refined3TH "2019-01-04 09:12:30") :: MakeR3 (DateTime1 LocalTime)) (unsafeRefined3 (read "2019-01-04 09:12:30") "2019-01-04 09:12:30") + , testCase "hms" $ (@?=) ($$(refined3TH "12:0:59") :: MakeR3 Hms) (unsafeRefined3 [12,0,59] "12:00:59") + , testCase "between5and9" $ (@?=) ($$(refined3TH "7") :: Refined3 (ReadP Int) (Between 5 9) (Printf "%03d" Id) String) (unsafeRefined3 7 "007") + , testCase "ssn" $ (@?=) ($$(refined3TH "123-45-6789") :: MakeR3 Ssn) (unsafeRefined3 [123,45,6789] "123-45-6789") + , testCase "base16" $ (@?=) ($$(refined3TH "12f") :: MakeR3 (BaseN 16)) (unsafeRefined3 303 "12f") + , testCase "daten1" $ (@?=) ($$(refined3TH "June 25 1900") :: MakeR3 DateN) (unsafeRefined3 (read "1900-06-25") "1900-06-25") + , testCase "daten2" $ (@?=) ($$(refined3TH "12/02/99") :: MakeR3 DateN) (unsafeRefined3 (read "1999-12-02") "1999-12-02") + , testCase "daten3" $ (@?=) ($$(refined3TH "2011-12-02") :: MakeR3 DateN) (unsafeRefined3 (read "2011-12-02") "2011-12-02") + , testCase "ccn123" $ (@?=) ($$(refined3TH "123455") :: MakeR3 (Ccn '[1,2,3])) (unsafeRefined3 [1,2,3,4,5,5] "1-23-455") + , testCase "readshow" $ (@?=) ($$(refined3TH "12 % 5") :: ReadShowR Rational) (unsafeRefined3 (12 % 5) "12 % 5") + ] + +unnamedTests :: [IO ()] +unnamedTests = [ + (@?=) [(unsafeRefined3 255 "ff", "")] (reads @(Refined3 (ReadBase Int 16) (Between 0 255) (ShowBase 16) String) "Refined3 {r3In = 255, r3Out = \"ff\"}") -- escape quotes cos read instance for String + , (@?=) [] (reads @(Refined3 (ReadBase Int 16) (Between 0 255) (ShowBase 16) String) "Refined3 {r3In = 256, r3Out = \"100\"}") + , (@?=) [(unsafeRefined3 (-1234) "-4d2", "")] (reads @(Refined3 (ReadBase Int 16) (Id < 0) (ShowBase 16) String) "Refined3 {r3In = -1234, r3Out = \"-4d2\"}") + + , (@?=) (unsafeRefined3 [1,2,3,4] "001.002.003.004") ($$(refined3TH "1.2.3.4") :: MakeR3 Ip) + + , expectJ (Right (G4 (unsafeRefined3 12 "12") (unsafeRefined3 [1,2,3,4] "001.002.003.004"))) (toFrom $ G4 (unsafeRefined3 12 "12") (unsafeRefined3 [1,2,3,4] "1.2.3.4")) + , expectJ (Left ["Error in $.g4Ip", "False Boolean Check"]) (toFrom $ G4 (unsafeRefined3 12 "12") (unsafeRefined3 [1,2,3,4] "1.2.3.400")) + , expectJ (Left ["Error in $.g4Ip", "ReadP Int (3x)"]) (toFrom $ G4 (unsafeRefined3 12 "12") (unsafeRefined3 [1,2,3,4] "1.2.3x.4")) + , expectJ (Left ["Error in $.g4Age", "False Boolean Check"]) (toFrom $ G4 (unsafeRefined3 (-2) "-2") (unsafeRefined3 [1,2,3,4] "1.2.3.4")) + , expectRight (testRefined3P (Proxy @(Ccn '[4,4,3])) ol "123-45-6---789-03-") + , expectLeft (testRefined3P (Proxy @(Ccn '[4,4,3])) ol "123-45-6---789-04-") + , expectRight (testRefined3P (Proxy @Hms) ol "1:2:33") + , expectLeft (testRefined3P (Proxy @Hms) ol "1:2:61") + , expectRight (testRefined3P (Proxy @(Ccn '[4,4,3])) ol "6433-1000-006") + , expectRight (testRefined3P (Proxy @(Ccn '[4,4,3])) ol "6433-10000-06") + , expectLeft (testRefined3P (Proxy @(Ccn '[4,4,3])) ol "6433-1000-000") + , expectRight (testRefined3P (Proxy @(Ccn '[1,2,1])) ol "1-23-0") + + , expect3 (Left $ XF "Regex no results") + $ eval3 @(Rescan Ip4RE Id >> HeadFail "failedn" Id >> Map (ReadP Int) Snd) + @((Len >> Same 4) && All (Between 0 255)) + @(Printfnt 4 "%03d.%03d.%03d.%03d") + ol "1.21.x31.4" + + , expect3 (Right $ unsafeRefined3 [1,21,31,4] "001.021.031.004") + $ eval3 @(Rescan Ip4RE Id >> HeadFail "failedn" Id >> Map (ReadP Int) Snd) + @((Len >> Same 4) && All (Between 0 255)) + @(Printfnt 4 "%03d.%03d.%03d.%03d") + ol "1.21.31.4" + + , expect3 (Left $ XTFalse (-6.3)) + $ eval3 @(ReadP Double) + @(Cmp 'Cgt (ToRational Id) (NegR 7 3)) + @(Printf "%5.3f" Id) + ol "-6.3" + + , expect3 (Right $ unsafeRefined3 4.123 "") + $ eval3 @(ReadP Double) @(Cmp 'Cgt (ToRational Id) (NegR 7 3)) @"" + ol "4.123" + + , expect3 (Right $ unsafeRefined3 4.123 (4123 % 1000)) + $ eval3 @Id @(Gt (NegR 7 3)) @(PosR 4123 1000) + ol 4.123 + + , expect3 (Right $ unsafeRefined3 [1,2,3,4] "") + $ eval3 @(Map (ReadP Int) (Resplit "\\." Id)) @(All (Between 0 255) && (Len >> Same 4)) @"" + ol "1.2.3.4" + + , expect3 (Left $ XTF [291,1048319,4387,17,1] "out of bounds") + $ eval3 @Ip6A @Ip6B @"" + ol "123:Ffeff:1123:11:1" + + , expect3 (Right $ unsafeRefined3 [12,2,0,255] "abc") + $ eval3 @Ip4A @Ip4B @"abc" + ol "12.2.0.255" + + , expect3 (Right $ unsafeRefined3 [123,45,6789] "def") + $ eval3 + @(Rescan "^(\\d{3})-(\\d{2})-(\\d{4})$" Id >> OneP >> Map (ReadBaseInt 10) Snd) + @(Guard "expected 3" (Len >> Same 3) + >> Guard "3 digits" (Ix' 0 >> Between 0 999) + >> Guard "2 digits" (Ix' 1 >> Between 0 99) + >> Guard "4 digits" (Ix' 2 >> Between 0 9999) + >> 'True + ) @"def" + ol "123-45-6789" + + , expect3 (Right $ unsafeRefined3 [123,45,6789] "xyz") + $ eval3 + @(Rescan "^(\\d{3})-(\\d{2})-(\\d{4})$" Id >> OneP >> Map (ReadBaseInt 10) Snd) + @(Guards (ToGuardsT (Printf2 "guard(%d) %d is out of range") '[Between 0 999, Between 0 99, Between 0 9999]) >> 'True) + @"xyz" + ol "123-45-6789" + + , expect3 (Left $ XTF [0,0,0,291,1048319,4387,17,1] "out of bounds") + $ eval3 @Ip6A'' @Ip6B' @"xyz" + ol "123:Ffeff:1123:11:1" + + , expect3 (Right $ unsafeRefined3 [0,0,0,291,65535,4387,17,1] "xyz") + $ eval3 @Ip6A'' @Ip6B' @"xyz" + ol "123:Ffff:1123:11:1" + + , expect3 (Right $ unsafeRefined3 [0,0,291,0,65535,0,0,17] "xyz") + $ eval3 @Ip6A'' @Ip6B' @"xyz" + ol "123::Ffff:::11" + + , expect3 (Right $ unsafeRefined3 [31,11,1999] "xyz") + $ eval3 @(Rescan DdmmyyyyRE Id >> OneP >> Map (ReadBaseInt 10) Snd) + @(Ddmmyyyyval >> 'True) + @"xyz" + ol "31-11-1999" + , expect3 (Right $ unsafeRefined3 [123,45,6789] "xyz") $ eval3 + @(Rescan "^(\\d{3})-(\\d{2})-(\\d{4})$" Id >> OneP >> Map (ReadBaseInt 10) Snd) + @(Guards (ToGuardsT (Printf2 "guard(%d) %d is out of range") '[Between 0 999, Between 0 99, Between 0 9999]) >> 'True) + @"xyz" + ol "123-45-6789" + + , expect3 (Right $ unsafeRefined3 [1,2,3,4] "001.002.003.004") $ eval3P ip4 ol "1.2.3.4" + , expect3 (Left $ XF "invalid base 10") $ eval3P ip4 ol "1.2.3x.4" + , expect3 (Left $ XTF [1,2,3,4,5] "expected 4 numbers") $ eval3P ip4 ol "1.2.3.4.5" + , expect3 (Left $ XTF [1,2,300,4] "each number must be between 0 and 255") $ eval3P ip4 ol "1.2.300.4" + , expect3 (Right $ unsafeRefined3 [1,2,3,4,5,6,7,8,9,0,3] "1234-5678-903") $ eval3P cc ol "12345678903" + , expect3 (Left $ XTFalse [1,2,3,4,5,6,7,8,9,0,1]) $ eval3P cc ol "12345678901" +-- , expect3 (Right $ unsafeRefined3 True ["T","r","ue","Tr","ue"]) $ eval3P (Proxy @'(Id, Id, Do '[ShowP, Dup, Sapa, SplitAts '[1,1,2,2]], Bool)) True + , expect3 (Right $ unsafeRefined3 ([12,13,14],TimeOfDay 12 13 14) "12:13:14") $ eval3P hms2E ol "12:13:14" + , expect3 (Left (XTF ([12,13,99], TimeOfDay 12 13 99) "guard(2) 99 secs is out of range")) $ eval3P hms2E ol "12:13:99" + ] + +allProps :: [TestTree] +allProps = + [ + testProperty "base16" $ forAll (arbRefined3 (mkProxy3P @'(ReadBase Int 16, 'True, ShowBase 16, String)) ol) (\r -> evalQuick @(ReadBase Int 16) (r3Out r) === Right (r3In r)) + , testProperty "readshow" $ forAll (arbRefined3 Proxy ol :: Gen HexLtR3) (\r -> read @HexLtR3 (show r) === r) + , testProperty "jsonroundtrip" $ forAll (arbRefined3 Proxy ol :: Gen HexLtR3) (\r -> testRefined3PJ Proxy ol (r3Out r) === Right r) + ] + +type HexLtR3 = Refined3 (ReadBase Int 16) (Id < 500) (ShowBase 16) String +type IntLtR3 = Refined3 (ReadP Int) (Id < 10) ShowP String + +-- printf breaks with negative numbers! +type Tst1 = '(ReadP Int, Between 1 7, Printf "someval val=%03d" Id, String) + +yy1, yy2, yy3, yy4 :: RefinedT Identity (MakeR3 Tst1) + +yy1 = newRefined3TP @Identity (Proxy @Tst1) o2 "4" +yy2 = newRefined3TP @Identity (Proxy @Tst1) o2 "3" + +yy3 = rapply3 o2 (*) yy1 yy2 -- fails +yy4 = rapply3 o2 (+) yy1 yy2 -- pure () + +type Ip4T = '(Ip4A, Ip4B, Ip4C, String) + +ip4 :: Proxy Ip4T -- '(Ip4A, Ip4B, Ip4C, String) +ip4 = mkProxy3 -- safer cos checks that ~ Bool etc + +ip4expands :: Proxy '(Ip4A, Ip4B, Ip4C, String) +ip4expands = mkProxy3 + +-- this works but ParseTimeP is easier +type DdmmyyyyRE = "^(\\d{2})-(\\d{2})-(\\d{4})$" +type Ddmmyyyyval' = Guards (ToGuardsT (Printf2 "guard(%d) %d is out of range") '[Between 1 31, Between 1 12, Between 1990 2050]) +type Ddmmyyyyval = + Guards '[ '(Printf2 "guard(%d) day %d is out of range", Between 1 31) + , '(Printf2 "guard(%d) month %d is out of range", Between 1 12) + , '(Printf2 "guard(%d) year %d is out of range", Between 1990 2050) ] + +cc :: Proxy CC11 +cc = mkProxy3 + +type Ipz1 = '(Id &&& Ip4A + , Snd >> Ip4B + , Snd >> Para (RepeatT 4 (Printf "%03d" Id)) >> Intercalate '["."] Id >> Concat + , String) +type Ipz2 = '(Id, Ip4A, Ip4B, String) -- skips fmt and just uses the original input +type Ipz3 = '(Ip4A, Ip4B, Id, String) + +-- need to add 'True to make it a predicate +-- guards checks also that there are exactly 3 entries! +type Hmsz1 = '(Hmsconv &&& ParseTimeP TimeOfDay "%H:%M:%S" Id + , Fst >> Hmsval >> 'True + , Snd + , String) + +-- better error messages cos doesnt do a strict regex match +type Hmsz2 = '(Hmsip &&& ParseTimeP TimeOfDay "%H:%M:%S" Id + , Fst >> Hmsop >> 'True + , Snd + , String) + +type Hmsip2 = Hmsip &&& ParseTimeP TimeOfDay "%H:%M:%S" Id +type Hmsop2 = Fst >> Hmsop >> 'True + +-- >mkProxy3 @Hmsip2 @Hmsop2 @(Snd >> FormatTimeP "%F %T" Id) @String +hms2E :: Proxy '(Hmsip2, Hmsop2, Snd >> FormatTimeP "%T" Id, String) +hms2E = mkProxy3P + + +-- better to use Guard for op boolean check cos we get better errormessages +-- 1. packaged up as a promoted tuple +type Tst3 = '(Map (ReadP Int) (Resplit "\\." Id), (Len >> Same 4) && All (Between 0 255), ConcatMap (Printf "%03d" Id) Id, String) + +www1, www2 :: String -> Either Msg3 (MakeR3 Tst3) +www1 = prt3 o2 . eval3P (Proxy :: MkProxy3T Tst3) o2 +www2 = prt3 o2 . eval3P tst3 o2 + +-- just pass in an ipaddress as a string: eg 1.2.3.4 or 1.2.3.4.5 (invalid) 1.2.3.400 (invalid) +ww1 = prtEval3P (Proxy :: MkProxy3T Tst3) o2 + +-- 2. packaged as a proxy +tst3 :: Proxy + '(Map (ReadP Int) (Resplit "\\." Id) + ,(Len >> Same 4) && All (Between 0 255) + ,ConcatMap (Printf "%03d" Id) Id + ,String) +tst3 = mkProxy3 + +ww2 = prtEval3P tst3 o2 + +-- 3. direct (has the advantage that we dont need to specify String + +ww3 = prtEval3 + @(Map (ReadP Int) (Resplit "\\." Id)) + @((Len >> Same 4) && All (Between 0 255)) + @(ConcatMap (Printf "%03d" Id) Id) + o2 + +data G4 = G4 { g4Age :: MakeR3 Age + , g4Ip :: MakeR3 Ip9 + } deriving (Show,Generic,Eq) + +type MyAge = Refined3 (ReadP Int) (Gt 4) ShowP String + +type Age = '(ReadP Int, Gt 4, ShowP, String) + +type Ip9 = '( + Map (ReadP Int) (Resplit "\\." Id) -- split String on "." then convert to [Int] + ,(Len >> Same 4) && All (Between 0 255) -- process [Int] and make sure length==4 and each octet is between 0 and 255 + ,Printfnt 4 "%03d.%03d.%03d.%03d" -- printf [Int] + ,String -- input type is string which is also the output type + ) + +instance FromJSON G4 +instance ToJSON G4 +{- ol= summary vs o2 = detail +prtEval3P daten ol "June 25 1900" +prtEval3P daten o2 "12/02/19" +prtEval3P (Proxy @(Ccn '[1,1,1,1])) ol "1230" +prtEval3P (Proxy @(Ccn '[1,2,3])) ol "123455" -- succeeds +-} + +-- prtRefinedT tst1a +tst1a :: Monad m => POpts -> RefinedT m ((Int,String),(Int,String)) +tst1a opts = withRefined3T @(ReadBase Int 16) @(Between 100 200) @(ShowBase 16) @String opts "a3" + $ \r1 -> withRefined3T @(ReadP Int) @'True @ShowP @String opts "12" + $ \r2 -> return ((r3In r1, r3Out r1), (r3In r2, r3Out r2)) + +-- prtRefinedTIO tst2a +tst2a :: MonadIO m => POpts -> RefinedT m ((Int,String),(Int,String)) +tst2a opts = withRefined3TIO @(ReadBase Int 16) @(Stderr "start" |> Between 100 200 >| Stdout "end") @(ShowBase 16) @String opts "a3" + $ \r1 -> withRefined3TIO @(ReadP Int) @'True @ShowP @String opts "12" + $ \r2 -> return ((r3In r1, r3Out r1), (r3In r2, r3Out r2)) + +-- have to use 'i' as we dont hold onto the input +testRefined3PJ :: forall ip op fmt i proxy + . (ToJSON (PP fmt (PP ip i)) + , Show (PP ip i) + , Show (PP fmt (PP ip i)) + , Refined3C ip op fmt i + , FromJSON i) + => proxy '(ip,op,fmt,i) + -> POpts + -> i + -> Either String (Refined3 ip op fmt i) +testRefined3PJ _ opts i = + let (ret,mr) = eval3 @ip @op @fmt opts i + m3 = prt3Impl opts ret + in case mr of + Just r -> eitherDecode @(Refined3 ip op fmt i) $ encode r + Nothing -> Left $ show m3 + +-- test that roundtripping holds ie i ~ PP fmt (PP ip i) +testRefined3P :: forall ip op fmt i proxy + . (Show (PP ip i) + , Show (PP fmt (PP ip i)) + , Refined3C ip op fmt i + , Eq i + , Eq (PP ip i)) + => proxy '(ip,op,fmt,i) + -> POpts + -> i + -> Either (String,String) (Refined3 ip op fmt i, Refined3 ip op fmt i) +testRefined3P _ opts i = + let (ret,mr) = eval3 @ip @op @fmt opts i + m3 = prt3Impl opts ret + in case mr of + Just r -> + let (ret1,mr1) = eval3 @ip @op @fmt opts (r3Out r) + m3a = prt3Impl opts ret1 + in case mr1 of + Nothing -> Left ("testRefined3P(2): round trip failed: old(" ++ show i ++ ") new(" ++ show (r3Out r) ++ ")", show m3a) + Just r1 -> + if r /= r1 then Left ("testRefined3P(3): round trip pure () but values dont match: old(" ++ show i ++ ") new(" ++ show (r3Out r) ++ ")", show (r,r1)) + else Right (r,r1) + Nothing -> Left ("testRefined3P(1): bad initial predicate i=" ++ show i, show m3) + +testRefined3PIO :: forall ip op fmt i proxy + . (Show (PP ip i), Show (PP fmt (PP ip i)), Refined3C ip op fmt i, Eq i, Eq (PP ip i)) + => proxy '(ip,op,fmt,i) + -> POpts + -> i + -> IO (Either String (Refined3 ip op fmt i, Refined3 ip op fmt i)) +testRefined3PIO p opts i = + case testRefined3P p opts i of + Right (r,r1) -> return $ Right (r,r1) + Left (msg, e) -> putStrLn e >> return (Left msg) + +getTTs3 :: RResults a b -> [Tree PE] +getTTs3 = \case + RF _ t1 -> [t1] + RTF _ t1 _ t2 -> [t1,t2] + RTFalse _ t1 t2 -> [t1,t2] + RTTrueF _ t1 t2 _ t3 -> [t1,t2,t3] + RTTrueT _ t1 t2 _ t3 -> [t1,t2,t3] + +toRResults3 :: RResults a b -> Results a b +toRResults3 = \case + RF e _ -> XF e + RTF a _ e _ -> XTF a e + RTFalse a _ _ -> XTFalse a + RTTrueF a _ _ e _ -> XTTrueF a e + RTTrueT a _ _ b _ -> XTTrueT a b + +expect3 :: (HasCallStack, Show i, Show r, Eq i, Eq r, Eq j, Show j) + => Either (Results i j) r + -> (RResults i j, Maybe r) + -> IO () +expect3 lhs (rhs,mr) = do + (@?=) lhs $ maybe (Left $ toRResults3 rhs) Right mr +
+ test/TestSpec.hs view
@@ -0,0 +1,13 @@+module Main where +import qualified TestPredicate +import qualified TestJson +import qualified TestRefined +import qualified TestRefined3 +import Data.Functor + +main :: IO () +main = do + TestPredicate.suite + TestJson.suite + TestRefined.suite + TestRefined3.suite