This is an alternative site for discovering Elm packages. You may be looking for the official Elm package site instead.

# List.Extra

Convenience functions for working with List

# Basics

last : List a -> Maybe a

Extract the last element of a list.

last [1,2,3] == Just 3
last [] == Nothing

init : List a -> Maybe (List a)

Return all elements of the list except the last one.

init [1,2,3] == Just [1,2]
init [] == Nothing

getAt : Int -> List a -> Maybe a

Returns Just the element at the given index in the list, or Nothing if the index is out of range.

uncons : List a -> Maybe (a, List a)

Decompose a list into its head and tail. If the list is empty, return Nothing. Otherwise, return Just (x, xs), where x is head and xs is tail.

uncons [1,2,3] == Just (1, [2,3])
uncons [] = Nothing

maximumBy : (a -> comparable) -> List a -> Maybe a

Find the first maximum element in a list using a comparable transformation

minimumBy : (a -> comparable) -> List a -> Maybe a

Find the first minimum element in a list using a comparable transformation

andMap : List (a -> b) -> List a -> List b

Map functions taking multiple arguments over multiple lists. Each list should be of the same length.

( (\a b c -> a + b * c)
map [1,2,3]
andMap [4,5,6]
andMap [2,1,1]
) == [9,7,9]

andThen : List a -> (a -> List b) -> List b

Equivalent to concatMap with arguments reversed. Ideal to use as an infix function, chaining together functions that return List. For example, suppose you want to have a cartesian product of [1,2] and [3,4]:

[1,2] andThen \x ->
[3,4] andThen \y ->
[(x,y)]


will give back the list:

[(1,3),(1,4),(2,3),(2,4)]


Now suppose we want to have a cartesian product between the first list and the second list and its doubles:

[1,2] andThen \x ->
[3,4] andThen \y ->
[y,y*2] andThen \z ->
[(x,z)]


will give back the list:

[(1,3),(1,6),(1,4),(1,8),(2,3),(2,6),(2,4),(2,8)]


Advanced functional programmers will recognize this as the implementation of bind operator (>>=) for lists from the Monad typeclass.

takeWhile : (a -> Bool) -> List a -> List a

Take elements in order as long as the predicate evaluates to True

dropWhile : (a -> Bool) -> List a -> List a

Drop elements in order as long as the predicate evaluates to True

dropDuplicates : List comparable -> List comparable

Drop all duplicates

replaceIf : (a -> Bool) -> a -> List a -> List a

Replace all values that satisfy a predicate with a replacement value.

setAt : Int -> a -> List a -> Maybe (List a)

Set a value in a list by index. Returns the updated list if the index is in range, or Nothing if it is out of range.

deleteIf : (a -> Bool) -> List a -> List a

Remove all values that satisfy a predicate

updateIf : (a -> Bool) -> (a -> a) -> List a -> List a

Replace all values that satisfy a predicate by calling an update function.

updateAt : Int -> (a -> a) -> List a -> Maybe (List a)

Replace a value at a specific index by calling an update function.

updateIfIndex : (Int -> Bool) -> (a -> a) -> List a -> List a

Replace a value at an index that satisfies a predicate.

singleton : a -> List a

Convert a value to a list containing one value.

singleton 3 == 

removeAt : Int -> List a -> List a

Remove the element at an index from a list. If the index is out of range, this returns the original list unchanged. Otherwise, it returns the updated list.

removeWhen : (a -> Bool) -> List a -> List a

Take a predicate and a list, and return a list that contains elements which fails to satisfy the predicate. This is equivalent to List.filter (not << predicate) list.

removeWhen isEven [1,2,3,4] == [1,3]


# List transformations

intercalate : List a -> List (List a) -> List a

Take a list and a list of lists, insert that list between every list in the list of lists, concatenate the result. intercalate xs xss is equivalent to concat (intersperse xs xss).

intercalate [0,0] [[1,2],[3,4],[5,6]] == [1,2,0,0,3,4,0,0,5,6]

transpose : List (List a) -> List (List a)

Transpose rows and columns of the list of lists.

transpose [[1,2,3],[4,5,6]] == [[1,4],[2,5],[3,6]]


If some rows are shorter than the following rows, their elements are skipped:

transpose [[10,11],,[],[30,31,32]] == [[10,20,30],[11,31],]

subsequences : List a -> List (List a)

Return the list of all subsequences of a list.

subsequences [1,2,3] == [[],,,[1,2],,[1,3],[2,3],[1,2,3]]

permutations : List a -> List (List a)

Return the list of of all permutations of a list. The result is in lexicographic order.

permutations [1,2,3] == [[1,2,3],[1,3,2],[2,1,3],[2,3,1],[3,1,2],[3,2,1]]

interweave : List a -> List a -> List a

Return a list that contains elements from the two provided, in alternate order. If one list runs out of items, append the items from the remaining list.

interweave [1,3] [2,4] == [1,2,3,4]
interweave [1,3,5,7] [2,4] == [1,2,3,4,5,7]


# Folds

foldl1 : (a -> a -> a) -> List a -> Maybe a

Variant of foldl that has no starting value argument and treats the head of the list as its starting value. If the list is empty, return Nothing.

foldl1 max [1,2,3,2,1] == Just 3
foldl1 max [] == Nothing
foldl1 (-) [1,2,3] == Just -4

foldr1 : (a -> a -> a) -> List a -> Maybe a

Variant of foldr that has no starting value argument and treats the last element of the list as its starting value. If the list is empty, return Nothing.

foldr1 min [1,2,3,2,1] == Just 1
foldr1 min [] == Nothing
foldr1 (-) [1,2,3] == Just 2


# Building lists

scanl1 : (a -> a -> a) -> List a -> List a

scanl1 is a variant of scanl that has no starting value argument.

Compare:

List.scanl (+) 0 [1,2,3] == [0,1,3,6]
scanl1 (+) [1,2,3] == [1,3,6]

List.scanl (-) 0 [1,2,3] == [0,1,1,2]
scanl1 (-) [1,2,3] == [1,1,2]

List.scanl (flip (-)) 0 [1,2,3] == [0,-1,-3,-6]
scanl1 (flip (-)) [1,2,3] == [1,-1,4]

scanr : (a -> b -> b) -> b -> List a -> List b

scanr is a right-to-left dual of scanl. Note that:

head (scanr f z xs) == foldr f z xs


Examples:

scanr (+) 0 [1,2,3] == [6,5,3,0]
scanr (-) 0 [1,2,3] == [2,-1,3,0]

scanr1 : (a -> a -> a) -> List a -> List a

scanr1 is a variant of scanr that has no starting value argument.

scanr1 (+) [1,2,3] == [6,5,3]
scanr1 (-) [1,2,3] == [2,-1,3]
scanr1 (flip (-)) [1,2,3] == [0,1,3]

unfoldr : (b -> Maybe (a, b)) -> b -> List a

The unfoldr function is "dual" to foldr. foldr reduces a list to a summary value, unfoldr builds a list from a seed. The function takes a function and a starting element. It applies the function to the element. If the result is Just (a, b), a is accumulated and the function is applied to b. If the result is Nothing, the list accumulated so far is returned.

unfoldr (\b -> if b == 0 then Nothing else Just (b, b-1)) 5 == [5,4,3,2,1]

iterate : (a -> Maybe a) -> a -> List a

Returns a list of repeated applications of f.

If f returns Nothing the iteration will stop. If it returns Just y then y will be added to the list and the iteration will continue with f y. nextYear : Int -> Maybe Int nextYear year = if year >= 2030 then Nothing else Just (year + 1) -- Will evaluate to [2010, 2011, ..., 2030] iterate nextYear 2010

# Sublists

splitAt : Int -> List a -> (List a, List a)

Take a number and a list, return a tuple of lists, where first part is prefix of the list of length equal the number, and second part is the remainder of the list. splitAt n xs is equivalent to (take n xs, drop n xs).

splitAt 3 [1,2,3,4,5] == ([1,2,3],[4,5])
splitAt 1 [1,2,3] == (,[2,3])
splitAt 3 [1,2,3] == ([1,2,3],[])
splitAt 4 [1,2,3] == ([1,2,3],[])
splitAt 0 [1,2,3] == ([],[1,2,3])
splitAt (-1) [1,2,3] == ([],[1,2,3])

takeWhileEnd : (a -> Bool) -> List a -> List a

Take elements from the end, while predicate still holds.

takeWhileEnd ((<)5) [1..10] == [6,7,8,9,10]

dropWhileEnd : (a -> Bool) -> List a -> List a

Drop elements from the end, while predicate still holds.

dropWhileEnd ((<)5) [1..10] == [1,2,3,4,5]

span : (a -> Bool) -> List a -> (List a, List a)

Take a predicate and a list, return a tuple. The first part of the tuple is longest prefix of that list, for each element of which the predicate holds. The second part of the tuple is the remainder of the list. span p xs is equivalent to (takeWhile p xs, dropWhile p xs).

span (< 3) [1,2,3,4,1,2,3,4] == ([1,2],[3,4,1,2,3,4])
span (< 5) [1,2,3] == ([1,2,3],[])
span (< 0) [1,2,3] == ([],[1,2,3])

break : (a -> Bool) -> List a -> (List a, List a)

Take a predicate and a list, return a tuple. The first part of the tuple is longest prefix of that list, for each element of which the predicate does not hold. The second part of the tuple is the remainder of the list. span p xs is equivalent to (dropWhile p xs, takeWhile p xs).

break (> 3) [1,2,3,4,1,2,3,4] == ([1,2,3],[4,1,2,3,4])
break (< 5) [1,2,3] == ([],[1,2,3])
break (> 5) [1,2,3] == ([1,2,3],[])

stripPrefix : List a -> List a -> Maybe (List a)

Drop the given prefix from the list. If the list doesn't start with that prefix, return Nothing.

stripPrefix [1,2] [1,2,3,4] == Just [3,4]
stripPrefix [1,2,3] [1,2,3,4,5] == Just [4,5]
stripPrefix [1,2,3] [1,2,3] == Just []
stripPrefix [1,2,3] [1,2] == Nothing
stripPrefix [3,2,1] [1,2,3,4,5] == Nothing

group : List a -> List (List a)

Group similar elements together. group is equivalent to groupWhile (==).

group [1,2,2,3,3,3,2,2,1] == [,[2,2],[3,3,3],[2,2],]

groupWhile : (a -> a -> Bool) -> List a -> List (List a)

Group elements together, using a custom equality test.

groupWhile (\x y -> fst x == fst y) [(0,'a'),(0,'b'),(1,'c'),(1,'d')] == [[(0,'a'),(0,'b')],[(1,'c'),(1,'d')]]


The equality test should be an equivalent relationship, i.e. it should have the properties of reflexivity, symmetry, and transitivity. For non-equivalent relations it gives non-intuitive behavior:

groupWhile (<) [1,2,3,2,4,1,3,2,1] == [[1,2,3,2,4],[1,3,2],]


For grouping elements with a comparison test, which must only hold the property of transitivity, see groupWhileTransitively.

groupWhileTransitively : (a -> a -> Bool) -> List a -> List (List a)

Group elements together, using a custom comparison test. Start a new group each time the comparison test doesn't hold for two adjacent elements.

groupWhileTransitively (<) [1,2,3,2,4,1,3,2,1] == [[1,2,3],[2,4],[1,3],,]

inits : List a -> List (List a)

Return all initial segments of a list, from shortest to longest, empty list first, the list itself last.

inits [1,2,3] == [[],,[1,2],[1,2,3]]

tails : List a -> List (List a)

Return all final segments of a list, from longest to shortest, the list itself first, empty list last.

tails [1,2,3] == [[1,2,3],[2,3],,[]]

select : List a -> List (a, List a)

Return all combinations in the form of (element, rest of the list). Read Haskell Libraries proposal for further ideas on how to use this function.

select [1,2,3,4] == [(1,[2,3,4]),(2,[1,3,4]),(3,[1,2,4]),(4,[1,2,3])]

selectSplit : List a -> List (List a, a, List a)

Return all combinations in the form of (elements before, element, elements after).

selectSplit [1,2,3] == [([],1,[2,3]),(,2,),([1,2],3,[])]


# Predicates

isPrefixOf : List a -> List a -> Bool

Take 2 lists and return True, if the first list is the prefix of the second list.

isSuffixOf : List a -> List a -> Bool

Take 2 lists and return True, if the first list is the suffix of the second list.

isInfixOf : List a -> List a -> Bool

Take 2 lists and return True, if the first list is an infix of the second list.

isSubsequenceOf : List a -> List a -> Bool

Take 2 lists and return True, if the first list is a subsequence of the second list.

isPermutationOf : List a -> List a -> Bool

Take 2 lists and return True, if the first list is a permutation of the second list.

# Searching

notMember : a -> List a -> Bool

Negation of member.

1 notMember [1,2,3] == False
4 notMember [1,2,3] == True

find : (a -> Bool) -> List a -> Maybe a

Find the first element that satisfies a predicate and return Just that element. If none match, return Nothing.

find (\num -> num > 5) [2, 4, 6, 8] == Just 6

elemIndex : a -> List a -> Maybe Int

Return the index of the first occurrence of the element. Otherwise, return Nothing. Indexing starts from 0.

elemIndex 1 [1,2,3] == Just 0
elemIndex 4 [1,2,3] == Nothing
elemIndex 1 [1,2,1] == Just 0

elemIndices : a -> List a -> List Int

Return all indices of occurrences of the element. If element is not found, return empty list. Indexing starts from 0.

elemIndices 1 [1,2,3] == 
elemIndices 4 [1,2,3] == []
elemIndices 1 [1,2,1] == [0,2]

findIndex : (a -> Bool) -> List a -> Maybe Int

Take a predicate and a list, return the index of the first element that satisfies the predicate. Otherwise, return Nothing. Indexing starts from 0.

findIndex isEven [1,2,3] == Just 1
findIndex isEven [1,3,5] == Nothing
findIndex isEven [1,2,4] == Just 1

findIndices : (a -> Bool) -> List a -> List Int

Take a predicate and a list, return indices of all elements satisfying the predicate. Otherwise, return empty list. Indexing starts from 0.

findIndices isEven [1,2,3] == 
findIndices isEven [1,3,5] == []
findIndices isEven [1,2,4] == [1,2]


# Zipping

zip : List a -> List b -> List (a,b)

Take two lists and returns a list of corresponding pairs

zip3 : List a -> List b -> List c -> List (a,b,c)

Take three lists and returns a list of triples

zip4 : List a -> List b -> List c -> List d -> List (a,b,c,d)

Take four lists and returns a list of quadruples

zip5 : List a -> List b -> List c -> List d -> List e -> List (a,b,c,d,e)

Take five lists and returns a list of quintuples

# Lift functions onto multiple lists of arguments

lift2 : (a -> b -> c) -> List a -> List b -> List c

Map functions taking multiple arguments over multiple lists, regardless of list length. All possible combinations will be explored.

lift2 (+) [1,2,3] [4,5] == [5,6,6,7,7,8]

lift3 : (a -> b -> c -> d) -> List a -> List b -> List c -> List d
lift4 : (a -> b -> c -> d -> e) -> List a -> List b -> List c -> List d -> List e

# Split to groups of given size

groupsOf : Int -> List a -> List (List a)

Split list into groups of size given by the first argument.

groupsOf 3 [1..10]
== [[1,2,3],[4,5,6],[7,8,9]]

groupsOfWithStep : Int -> Int -> List a -> List (List a)

Split list into groups of size given by the first argument. After each group, drop a number of elements given by the second argumet before starting the next group.

groupsOfWithStep 2 1 [1..4]
== [[1,2],[2,3],[3,4]]

greedyGroupsOf : Int -> List a -> List (List a)

Split list into groups of size given by the first argument "greedily" (don't throw the group away if not long enough).

greedyGroupsOf 3 [1..10]
== [[1,2,3],[4,5,6],[7,8,9],]

greedyGroupsOfWithStep : Int -> Int -> List a -> List (List a)

Split list into groups of size given by the first argument "greedily" (don't throw the group away if not long enough). After each group, drop a number of elements given by the second argumet before starting the next group.

greedyGroupsOfWithStep 3 2 [1..6]
== [[1,2,3],[3,4,5],[5,6]]

module List.Extra exposing ( last
, init
, getAt, (!!)
, uncons
, minimumBy
, maximumBy
, andMap, andThen
, takeWhile
, dropWhile
, dropDuplicates
, replaceIf
, setAt
, deleteIf
, updateIf
, updateAt
, updateIfIndex
, singleton
, removeAt
, removeWhen
, iterate
, intercalate, transpose, subsequences, permutations, interweave
, foldl1, foldr1
, scanl1, scanr, scanr1, unfoldr
, splitAt, takeWhileEnd, dropWhileEnd, span, break, stripPrefix
, group, groupWhile, groupWhileTransitively, inits, tails, select, selectSplit
, isPrefixOf, isSuffixOf, isInfixOf, isSubsequenceOf, isPermutationOf
, notMember, find
, elemIndex, elemIndices, findIndex, findIndices
, zip
, zip3
, zip4
, zip5
, lift2
, lift3
, lift4
, groupsOf, groupsOfWithStep, greedyGroupsOf, greedyGroupsOfWithStep
)
{-| Convenience functions for working with List

# Basics
@docs last, init, getAt, (!!), uncons, maximumBy, minimumBy, andMap, andThen, takeWhile, dropWhile, dropDuplicates, replaceIf, setAt, deleteIf, updateIf, updateAt, updateIfIndex, singleton, removeAt, removeWhen

# List transformations
@docs intercalate, transpose, subsequences, permutations, interweave

# Folds
@docs foldl1, foldr1

# Building lists
@docs scanl1, scanr, scanr1, unfoldr, iterate

# Sublists
@docs splitAt, takeWhileEnd, dropWhileEnd, span, break, stripPrefix, group, groupWhile, groupWhileTransitively, inits, tails, select, selectSplit

# Predicates
@docs isPrefixOf, isSuffixOf, isInfixOf, isSubsequenceOf, isPermutationOf

# Searching
@docs notMember, find, elemIndex, elemIndices, findIndex, findIndices

# Zipping
@docs zip, zip3, zip4, zip5

# Lift functions onto multiple lists of arguments
@docs lift2, lift3, lift4

# Split to groups of given size
@docs groupsOf, groupsOfWithStep, greedyGroupsOf, greedyGroupsOfWithStep
-}

import List exposing (..)
import Set exposing (Set)

{-| Extract the last element of a list.

last [1,2,3] == Just 3
last [] == Nothing
-}
last : List a -> Maybe a
last = foldl1 (flip always)

{-| Return all elements of the list except the last one.

init [1,2,3] == Just [1,2]
init [] == Nothing
-}
init : List a -> Maybe (List a)
init =
let
maybe : b -> (a -> b) -> Maybe a -> b
maybe d f = Maybe.withDefault d << Maybe.map f
in
foldr ((<<) Just << maybe [] << (::)) Nothing

{-| Returns Just the element at the given index in the list,
or Nothing if the index is out of range.
-}
getAt : Int -> List a -> Maybe a
getAt idx xs =
if idx < 0 then
Nothing
else

{-| Alias for getAt, but with the parameters flipped.
-}
(!!) : List a -> Int -> Maybe a
(!!) = flip getAt

{-| Returns a list of repeated applications of f.

If f returns Nothing the iteration will stop. If it returns Just y then y will be added to the list and the iteration will continue with f y.
nextYear : Int -> Maybe Int
nextYear year =
if year >= 2030 then
Nothing
else
Just (year + 1)
-- Will evaluate to [2010, 2011, ..., 2030]
iterate nextYear 2010
-}
iterate : (a -> Maybe a) -> a -> List a
iterate f x =
case f x of
Just x' -> x :: iterate f x'
Nothing -> [x]

{-| Decompose a list into its head and tail. If the list is empty, return Nothing. Otherwise, return Just (x, xs), where x is head and xs is tail.

uncons [1,2,3] == Just (1, [2,3])
uncons [] = Nothing
-}
uncons : List a -> Maybe (a, List a)
uncons xs =
case xs of
[] -> Nothing
(x::xs) -> Just (x,xs)

{-| Find the first maximum element in a list using a comparable transformation
-}
maximumBy : (a -> comparable) -> List a -> Maybe a
maximumBy f ls =
let maxBy x (y, fy) = let fx = f x in if fx > fy then (x, fx) else (y, fy)
in case ls of
[l']    -> Just l'
l'::ls' -> Just <| fst <| foldl maxBy (l', f l') ls'
_       -> Nothing

{-| Find the first minimum element in a list using a comparable transformation
-}
minimumBy : (a -> comparable) -> List a -> Maybe a
minimumBy f ls =
let minBy x (y, fy) = let fx = f x in if fx < fy then (x, fx) else (y, fy)
in case ls of
[l']    -> Just l'
l'::ls' -> Just <| fst <| foldl minBy (l', f l') ls'
_       -> Nothing

{-| Take elements in order as long as the predicate evaluates to True
-}
takeWhile : (a -> Bool) -> List a -> List a
takeWhile predicate list =
case list of
[]      -> []
x::xs   -> if (predicate x) then x :: takeWhile predicate xs
else []

{-| Drop elements in order as long as the predicate evaluates to True
-}
dropWhile : (a -> Bool) -> List a -> List a
dropWhile predicate list =
case list of
[]      -> []
x::xs   -> if (predicate x) then dropWhile predicate xs
else list

{-| Drop all duplicates
-}
dropDuplicates : List comparable -> List comparable
dropDuplicates list =
dropDuplicatesHelp Set.empty list

dropDuplicatesHelp : Set comparable -> List comparable -> List comparable
dropDuplicatesHelp existing remaining =
case remaining of
[] ->
[]

first :: rest ->
if Set.member first existing then
dropDuplicatesHelp existing rest
else
first :: dropDuplicatesHelp (Set.insert first existing) rest

{-| Map functions taking multiple arguments over multiple lists. Each list should be of the same length.

( (\a b c -> a + b * c)
map [1,2,3]
andMap [4,5,6]
andMap [2,1,1]
) == [9,7,9]
-}
andMap : List (a -> b) -> List a -> List b
andMap fl l = map2 (<|) fl l

{-| Equivalent to concatMap with arguments reversed. Ideal to use as an infix function, chaining together functions that return List. For example, suppose you want to have a cartesian product of [1,2] and [3,4]:

[1,2] andThen \x ->
[3,4] andThen \y ->
[(x,y)]

will give back the list:

[(1,3),(1,4),(2,3),(2,4)]

Now suppose we want to have a cartesian product between the first list and the second list and its doubles:

[1,2] andThen \x ->
[3,4] andThen \y ->
[y,y*2] andThen \z ->
[(x,z)]

will give back the list:

[(1,3),(1,6),(1,4),(1,8),(2,3),(2,6),(2,4),(2,8)]

Advanced functional programmers will recognize this as the implementation of bind operator (>>=) for lists from the Monad typeclass.
-}
andThen : List a -> (a -> List b) -> List b
andThen = flip concatMap

{-| Negation of member.

1 notMember [1,2,3] == False
4 notMember [1,2,3] == True
-}
notMember : a -> List a -> Bool
notMember x = not << member x

{-| Find the first element that satisfies a predicate and return
Just that element. If none match, return Nothing.

find (\num -> num > 5) [2, 4, 6, 8] == Just 6
-}
find : (a -> Bool) -> List a -> Maybe a
find predicate list =
case list of
[] ->
Nothing

first::rest ->
if predicate first then
Just first
else
find predicate rest

{-| Return the index of the first occurrence of the element. Otherwise, return Nothing. Indexing starts from 0.

elemIndex 1 [1,2,3] == Just 0
elemIndex 4 [1,2,3] == Nothing
elemIndex 1 [1,2,1] == Just 0
-}
elemIndex : a -> List a -> Maybe Int
elemIndex x = findIndex ((==)x)

{-| Return all indices of occurrences of the element. If element is not found, return empty list. Indexing starts from 0.

elemIndices 1 [1,2,3] == 
elemIndices 4 [1,2,3] == []
elemIndices 1 [1,2,1] == [0,2]
-}
elemIndices : a -> List a -> List Int
elemIndices x = findIndices ((==)x)

{-| Take a predicate and a list, return the index of the first element that satisfies the predicate. Otherwise, return Nothing. Indexing starts from 0.

findIndex isEven [1,2,3] == Just 1
findIndex isEven [1,3,5] == Nothing
findIndex isEven [1,2,4] == Just 1
-}
findIndex : (a -> Bool) -> List a -> Maybe Int
findIndex p = head << findIndices p

{-| Take a predicate and a list, return indices of all elements satisfying the predicate. Otherwise, return empty list. Indexing starts from 0.

findIndices isEven [1,2,3] == 
findIndices isEven [1,3,5] == []
findIndices isEven [1,2,4] == [1,2]
-}
findIndices : (a -> Bool) -> List a -> List Int
findIndices p = map fst << filter (\(i,x) -> p x) << indexedMap (,)

{-| Replace all values that satisfy a predicate with a replacement value.
-}
replaceIf : (a -> Bool) -> a -> List a -> List a
replaceIf predicate replacement list =
updateIf predicate (always replacement) list

{-| Replace all values that satisfy a predicate by calling an update function.
-}
updateIf : (a -> Bool) -> (a -> a) -> List a -> List a
updateIf predicate update list =
List.map (\item -> if predicate item then update item else item) list

{-| Replace a value at a specific index by calling an update function.
-}
updateAt : Int -> (a -> a) -> List a -> Maybe (List a)
updateAt index update list =
if index < 0 || index >= List.length list then
Nothing
else
Just <| updateIfIndex ((==) index) update list

{-| Replace a value at an index that satisfies a predicate.
-}
updateIfIndex : (Int -> Bool) -> (a -> a) -> List a -> List a
updateIfIndex predicate update list =
List.indexedMap (\i x -> if predicate i then update x else x) list

{-| Remove all values that satisfy a predicate
-}
deleteIf : (a -> Bool) -> List a -> List a
deleteIf predicate items =
List.filter (not << predicate) items

{-| Set a value in a list by index. Returns the updated list if the index is in range, or Nothing if it is out of range.
-}
setAt : Int -> a -> List a -> Maybe (List a)
setAt index value l =
if index < 0 then
Nothing
else
let
tail = List.drop index l |> List.tail
in
case tail of
Nothing ->
Nothing

Just t ->
Just (value :: t |> List.append head)

{-| Convert a value to a list containing one value.

singleton 3 == 
-}
singleton : a -> List a
singleton x = [x]

{-| Remove the element at an index from a list. If the index is out of range, this returns the original list unchanged. Otherwise, it returns the updated list.
-}
removeAt : Int -> List a -> List a
removeAt index l =
if index < 0 then
l
else
let
tail = List.drop index l |> List.tail
in
case tail of
Nothing ->
l

Just t ->

{-| Take a predicate and a list, and return a list that contains elements which fails to satisfy the predicate.
This is equivalent to List.filter (not << predicate) list.

removeWhen isEven [1,2,3,4] == [1,3]
-}
removeWhen : (a -> Bool) -> List a -> List a
removeWhen pred list =
List.filter (not << pred) list

{-| Take a list and a list of lists, insert that list between every list in the list of lists, concatenate the result. intercalate xs xss is equivalent to concat (intersperse xs xss).

intercalate [0,0] [[1,2],[3,4],[5,6]] == [1,2,0,0,3,4,0,0,5,6]
-}
intercalate : List a -> List (List a) -> List a
intercalate xs = concat << intersperse xs

{-| Transpose rows and columns of the list of lists.

transpose [[1,2,3],[4,5,6]] == [[1,4],[2,5],[3,6]]

If some rows are shorter than the following rows, their elements are skipped:

transpose [[10,11],,[],[30,31,32]] == [[10,20,30],[11,31],]
-}
transpose : List (List a) -> List (List a)
transpose ll =
case ll of
[] -> []
([]::xss) -> transpose xss
((x::xs)::xss) ->
let
tails = filterMap tail xss
in

{-| Return the list of all subsequences of a list.

subsequences [1,2,3] == [[],,,[1,2],,[1,3],[2,3],[1,2,3]]
-}
subsequences : List a -> List (List a)
subsequences xs = []::subsequencesNonEmpty xs

{-| Return the list of all subsequences of the argument, except for the empty list.

subsequencesNonEmpty [1,2,3] == [,,[1,2],,[1,3],[2,3],[1,2,3]]
-}
subsequencesNonEmpty : List a -> List (List a)
subsequencesNonEmpty xs =
case xs of
[] -> []
(x::xs) ->
let f ys r = ys::(x::ys)::r
in [x]::foldr f [] (subsequencesNonEmpty xs)

{-| Return the list of of all permutations of a list. The result is in lexicographic order.

permutations [1,2,3] == [[1,2,3],[1,3,2],[2,1,3],[2,3,1],[3,1,2],[3,2,1]]
-}
permutations : List a -> List (List a)
permutations xs' =
case xs' of
[] -> [[]]
xs -> let f (y,ys) = map ((::)y) (permutations ys)
in concatMap f (select xs)

{-| Return a list that contains elements from the two provided, in alternate order.
If one list runs out of items, append the items from the remaining list.

interweave [1,3] [2,4] == [1,2,3,4]
interweave [1,3,5,7] [2,4] == [1,2,3,4,5,7]
-}
interweave : List a -> List a -> List a
interweave l1 l2 =
interweaveHelp l1 l2 []

interweaveHelp : List a -> List a -> List a -> List a
interweaveHelp l1 l2 acc =
case (l1, l2) of
(x::xs, y::ys) ->
interweaveHelp xs ys <| acc ++ [x, y]

(x, []) ->
acc ++ x

([], y) ->
acc ++ y

{-| Variant of foldl that has no starting value argument and treats the head of the list as its starting value. If the list is empty, return Nothing.

foldl1 max [1,2,3,2,1] == Just 3
foldl1 max [] == Nothing
foldl1 (-) [1,2,3] == Just -4
-}
foldl1 : (a -> a -> a) -> List a -> Maybe a
foldl1 f xs =
let
mf x m = Just (case m of
Nothing -> x
Just y -> f y x)
in
List.foldl mf Nothing xs

{-| Variant of foldr that has no starting value argument and treats the last element of the list as its starting value. If the list is empty, return Nothing.

foldr1 min [1,2,3,2,1] == Just 1
foldr1 min [] == Nothing
foldr1 (-) [1,2,3] == Just 2
-}
foldr1 : (a -> a -> a) -> List a -> Maybe a
foldr1 f xs =
let
mf x m = Just (case m of
Nothing -> x
Just y -> f x y)
in
List.foldr mf Nothing xs

{-| scanl1 is a variant of scanl that has no starting value argument.

Compare:

List.scanl (+) 0 [1,2,3] == [0,1,3,6]
scanl1 (+) [1,2,3] == [1,3,6]

List.scanl (-) 0 [1,2,3] == [0,1,1,2]
scanl1 (-) [1,2,3] == [1,1,2]

List.scanl (flip (-)) 0 [1,2,3] == [0,-1,-3,-6]
scanl1 (flip (-)) [1,2,3] == [1,-1,4]
-}
scanl1 : (a -> a -> a) -> List a -> List a
scanl1 f xs' =
case xs' of
[] -> []
(x::xs) -> scanl f x xs

{-| scanr is a right-to-left dual of scanl. Note that:

head (scanr f z xs) == foldr f z xs

Examples:

scanr (+) 0 [1,2,3] == [6,5,3,0]
scanr (-) 0 [1,2,3] == [2,-1,3,0]
-}
scanr : (a -> b -> b) -> b -> List a -> List b
scanr f acc xs' =
case xs' of
[] -> [acc]
(x::xs) ->
case scanr f acc xs of
(q::_) as qs ->
f x q :: qs

[] ->
[]

{-| scanr1 is a variant of scanr that has no starting value argument.

scanr1 (+) [1,2,3] == [6,5,3]
scanr1 (-) [1,2,3] == [2,-1,3]
scanr1 (flip (-)) [1,2,3] == [0,1,3]
-}
scanr1 : (a -> a -> a) -> List a -> List a
scanr1 f xs' =
case xs' of
[] -> []
[x] -> [x]
(x::xs) ->
case scanr1 f xs of
(q::_) as qs ->
f x q :: qs

[] ->
[]

{-| The unfoldr function is "dual" to foldr. foldr reduces a list to a summary value, unfoldr builds a list from a seed. The function takes a function and a starting element. It applies the function to the element. If the result is Just (a, b), a is accumulated and the function is applied to b. If the result is Nothing, the list accumulated so far is returned.

unfoldr (\b -> if b == 0 then Nothing else Just (b, b-1)) 5 == [5,4,3,2,1]
-}
unfoldr : (b -> Maybe (a, b)) -> b -> List a
unfoldr f seed =
case f seed of
Nothing -> []
Just (a, b) -> a :: unfoldr f b

{-| Take a number and a list, return a tuple of lists, where first part is prefix of the list of length equal the number, and second part is the remainder of the list. splitAt n xs is equivalent to (take n xs, drop n xs).

splitAt 3 [1,2,3,4,5] == ([1,2,3],[4,5])
splitAt 1 [1,2,3] == (,[2,3])
splitAt 3 [1,2,3] == ([1,2,3],[])
splitAt 4 [1,2,3] == ([1,2,3],[])
splitAt 0 [1,2,3] == ([],[1,2,3])
splitAt (-1) [1,2,3] == ([],[1,2,3])
-}
splitAt : Int -> List a -> (List a, List a)
splitAt n xs = (take n xs, drop n xs)

{-| Take elements from the end, while predicate still holds.

takeWhileEnd ((<)5) [1..10] == [6,7,8,9,10]
-}
takeWhileEnd : (a -> Bool) -> List a -> List a
takeWhileEnd p =
let
step x (xs,free) = if p x && free then (x::xs,True) else (xs, False)
in
fst << foldr step ([], True)

{-| Drop elements from the end, while predicate still holds.

dropWhileEnd ((<)5) [1..10] == [1,2,3,4,5]
-}
dropWhileEnd : (a -> Bool) -> List a -> List a
dropWhileEnd p = foldr (\x xs -> if p x && isEmpty xs then [] else x::xs) []

{-| Take a predicate and a list, return a tuple. The first part of the tuple is longest prefix of that list, for each element of which the predicate holds. The second part of the tuple is the remainder of the list. span p xs is equivalent to (takeWhile p xs, dropWhile p xs).

span (< 3) [1,2,3,4,1,2,3,4] == ([1,2],[3,4,1,2,3,4])
span (< 5) [1,2,3] == ([1,2,3],[])
span (< 0) [1,2,3] == ([],[1,2,3])
-}
span : (a -> Bool) -> List a -> (List a, List a)
span p xs = (takeWhile p xs, dropWhile p xs)

{-| Take a predicate and a list, return a tuple. The first part of the tuple is longest prefix of that list, for each element of which the predicate *does not* hold. The second part of the tuple is the remainder of the list. span p xs is equivalent to (dropWhile p xs, takeWhile p xs).

break (> 3) [1,2,3,4,1,2,3,4] == ([1,2,3],[4,1,2,3,4])
break (< 5) [1,2,3] == ([],[1,2,3])
break (> 5) [1,2,3] == ([1,2,3],[])
-}
break : (a -> Bool) -> List a -> (List a, List a)
break p = span (not << p)

{-| Drop the given prefix from the list. If the list doesn't start with that prefix, return Nothing.

stripPrefix [1,2] [1,2,3,4] == Just [3,4]
stripPrefix [1,2,3] [1,2,3,4,5] == Just [4,5]
stripPrefix [1,2,3] [1,2,3] == Just []
stripPrefix [1,2,3] [1,2] == Nothing
stripPrefix [3,2,1] [1,2,3,4,5] == Nothing
-}
stripPrefix : List a -> List a -> Maybe (List a)
stripPrefix prefix xs =
let
step e m =
case m of
Nothing -> Nothing
Just [] -> Nothing
Just (x::xs') -> if e == x
then Just xs'
else Nothing
in
foldl step (Just xs) prefix

{-| Group similar elements together. group is equivalent to groupWhile (==).

group [1,2,2,3,3,3,2,2,1] == [,[2,2],[3,3,3],[2,2],]
-}
group : List a -> List (List a)
group = groupWhile (==)

{-| Group elements together, using a custom equality test.

groupWhile (\x y -> fst x == fst y) [(0,'a'),(0,'b'),(1,'c'),(1,'d')] == [[(0,'a'),(0,'b')],[(1,'c'),(1,'d')]]

The equality test should be an equivalent relationship, i.e. it should have the properties of reflexivity, symmetry, and transitivity. For non-equivalent relations it gives non-intuitive behavior:

groupWhile (<) [1,2,3,2,4,1,3,2,1] == [[1,2,3,2,4],[1,3,2],]

For grouping elements with a comparison test, which must only hold the property of transitivity, see groupWhileTransitively.
-}
groupWhile : (a -> a -> Bool) -> List a -> List (List a)
groupWhile eq xs' =
case xs' of
[] -> []
(x::xs) -> let (ys,zs) = span (eq x) xs
in (x::ys)::groupWhile eq zs

{-| Group elements together, using a custom comparison test. Start a new group each time the comparison test doesn't hold for two adjacent elements.

groupWhileTransitively (<) [1,2,3,2,4,1,3,2,1] == [[1,2,3],[2,4],[1,3],,]
-}
groupWhileTransitively : (a -> a -> Bool) -> List a -> List (List a)
groupWhileTransitively cmp xs' =
case xs' of
[] -> []
[x] -> [[x]]
(x::((x'::_) as xs)) ->
case groupWhileTransitively cmp xs of
(y::ys) as r ->
if cmp x x'
then (x::y)::ys
else [x]::r

[] ->
[]

{-| Return all initial segments of a list, from shortest to longest, empty list first, the list itself last.

inits [1,2,3] == [[],,[1,2],[1,2,3]]
-}
inits : List a -> List (List a)
inits = foldr (\e acc -> []::map ((::)e) acc) [[]]

{-| Return all final segments of a list, from longest to shortest, the list itself first, empty list last.

tails [1,2,3] == [[1,2,3],[2,3],,[]]
-}
tails : List a -> List (List a)
tails = foldr tailsHelp [[]]

tailsHelp : a -> List (List a) -> List (List a)
tailsHelp e list =
case list of
(x::xs) ->
(e::x)::x::xs

[] ->
[]

{-| Return all combinations in the form of (element, rest of the list). Read [Haskell Libraries proposal](https://mail.haskell.org/pipermail/libraries/2008-February/009270.html) for further ideas on how to use this function.

select [1,2,3,4] == [(1,[2,3,4]),(2,[1,3,4]),(3,[1,2,4]),(4,[1,2,3])]
-}
select : List a -> List (a, List a)
select xs =
case xs of
[] -> []
(x::xs) -> (x,xs)::map (\(y,ys) -> (y,x::ys)) (select xs)

{-| Return all combinations in the form of (elements before, element, elements after).

selectSplit [1,2,3] == [([],1,[2,3]),(,2,),([1,2],3,[])]
-}
selectSplit : List a -> List (List a, a, List a)
selectSplit xs =
case xs of
[] -> []
(x::xs) -> ([],x,xs)::map (\(lys,y,rys) -> (x::lys,y,rys)) (selectSplit xs)

{-| Take 2 lists and return True, if the first list is the prefix of the second list.
-}
isPrefixOf : List a -> List a -> Bool
isPrefixOf prefix = all identity << map2 (==) prefix

{-| Take 2 lists and return True, if the first list is the suffix of the second list.
-}
isSuffixOf : List a -> List a -> Bool
isSuffixOf suffix xs = isPrefixOf (reverse suffix) (reverse xs)

{-| Take 2 lists and return True, if the first list is an infix of the second list.
-}
isInfixOf : List a -> List a -> Bool
isInfixOf infix xs = any (isPrefixOf infix) (tails xs)

{-| Take 2 lists and return True, if the first list is a subsequence of the second list.
-}
isSubsequenceOf : List a -> List a -> Bool
isSubsequenceOf subseq xs = subseq member subsequences xs

{-| Take 2 lists and return True, if the first list is a permutation of the second list.
-}
isPermutationOf : List a -> List a -> Bool
isPermutationOf permut xs = permut member permutations xs

{-| Take two lists and returns a list of corresponding pairs
-}
zip : List a -> List b -> List (a,b)
zip = map2 (,)

{-| Take three lists and returns a list of triples
-}
zip3 : List a -> List b -> List c -> List (a,b,c)
zip3 = map3 (,,)

{-| Take four lists and returns a list of quadruples
-}
zip4 : List a -> List b -> List c -> List d -> List (a,b,c,d)
zip4 = map4 (,,,)

{-| Take five lists and returns a list of quintuples
-}
zip5 : List a -> List b -> List c -> List d -> List e -> List (a,b,c,d,e)
zip5 = map5 (,,,,)

{-| Map functions taking multiple arguments over multiple lists, regardless of list length.
All possible combinations will be explored.

lift2 (+) [1,2,3] [4,5] == [5,6,6,7,7,8]
-}
lift2 : (a -> b -> c) -> List a -> List b -> List c
lift2 f la lb =
la andThen (\a -> lb andThen (\b -> [f a b]))

{-|
-}
lift3 : (a -> b -> c -> d) -> List a -> List b -> List c -> List d
lift3 f la lb lc =
la andThen (\a -> lb andThen (\b -> lc andThen (\c -> [f a b c])))

{-|
-}
lift4 : (a -> b -> c -> d -> e) -> List a -> List b -> List c -> List d -> List e
lift4 f la lb lc ld =
la andThen (\a -> lb andThen (\b -> lc andThen (\c -> ld andThen (\d -> [f a b c d]))))

{-| Split list into groups of size given by the first argument.

groupsOf 3 [1..10]
== [[1,2,3],[4,5,6],[7,8,9]]
-}
groupsOf : Int -> List a -> List (List a)
groupsOf size xs =
groupsOfWithStep size size xs

{-| Split list into groups of size given by the first argument.  After each group, drop a number of elements given by the second argumet before starting the next group.

groupsOfWithStep 2 1 [1..4]
== [[1,2],[2,3],[3,4]]
-}
groupsOfWithStep : Int -> Int -> List a -> List (List a)
groupsOfWithStep size step xs =
let
group =
List.take size xs

xs' =
List.drop step xs

okayArgs =
size > 0 && step > 0

okayLength =
size == List.length group
in
if okayArgs && okayLength then
group :: groupsOfWithStep size step xs'
else
[]

{-| Split list into groups of size given by the first argument "greedily" (don't throw the group away if not long enough).

greedyGroupsOf 3 [1..10]
== [[1,2,3],[4,5,6],[7,8,9],]
-}
greedyGroupsOf : Int -> List a -> List (List a)
greedyGroupsOf size xs =
greedyGroupsOfWithStep size size xs

{-| Split list into groups of size given by the first argument "greedily" (don't throw the group away if not long enough). After each group, drop a number of elements given by the second argumet before starting the next group.

greedyGroupsOfWithStep 3 2 [1..6]
== [[1,2,3],[3,4,5],[5,6]]
-}
greedyGroupsOfWithStep : Int -> Int -> List a -> List (List a)
greedyGroupsOfWithStep size step xs =
let
group =
List.take size xs

xs' =
List.drop step xs

okayArgs =
size > 0 && step > 0

okayXs =
List.length xs > 0
in
if okayArgs && okayXs then
group :: greedyGroupsOfWithStep size step xs'
else
[]