aboutsummaryrefslogtreecommitdiff
path: root/src/Result.gleam
blob: 6b9c4135e10cdfe1e9b827e74aa9c42c4cffcc36 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
module Result

doc """
Result represents the result of something that may succeed or fail.
`Ok` means it was successful, `Error` means it failed.
"""
pub type Result(error, value) {
  | Ok(value)
  | Error(error)
;

import Result:*

pub fn is_ok(result) {
  case result {
  | Error(_) => False
  | Ok(_) => True
  }
}

test is_ok {
  is_ok(Ok(1)) |> Assert.true
  is_ok(Error(1)) |> Assert.false
}

pub fn is_error(result) {
  case result {
  | Ok(_) => False
  | Error(_) => True
  }
}

test is_error {
  is_error(Ok(1)) |> Assert.false
  is_error(Error(1)) |> Assert.true
}

pub fn map(result, fun) {
  case result {
  | Ok(x) => fun(x)
  | Error(_) => result
  }
}

test map {
  Ok(1)
    |> map(_, fn(x) { x + 1 })
    |> Assert.equal(_, Ok(2))
  Error(1)
    |> map(_, fn(x) { x + 1 })
    |> Assert.equal(Error(1))
}

pub fn map_error(result, fun) {
  case result {
  | Ok(_) => result
  | Error(error) => Error(fun(error))
  }
}

test map_error {
  Ok(1)
    |> map_error(_, fn(x) { x + 1 })
    |> Assert.equal(_, Ok(1))
  Error(1)
    |> map_error(_, fn(x) { x + 1 })
    |> Assert.equal(_, Error(2))
}

pub fn flatten(result) {
  case result {
  | Ok(x) => x
  | Error(_) => result
  }
}

test flatten {
  flatten(Ok(Ok(1)))
    |> Assert.equal(_, Ok(1))
  flatten(Ok(Error(1)))
    |> Assert.equal(_, Error(1))
  flatten(Error(1))
    |> Assert.equal(_, Error(1))
}

pub fn flat_map(result, fun) {
  result
    |> unwrap(_, fun)
    |> flatten
}

test flat_map {
  Error(1)
    |> flat_map(_, fn(x) { Ok(x + 1) })
    |> Assert.equal(_, Error(1))
  Ok(1)
    |> flat_map(_, fn(x) { Ok(x + 1) })
    |> Assert.equal(_, Ok(2))
  Ok(1)
    |> flat_map(_, fn(_) { Error(1) })
    |> Assert.equal(_, Error(1))
}

pub fn unwrap(result, default) {
  case result {
  | Ok(v) => v
  | Error(_) => default
  }
}

test unwrap {
  unwrap(Ok(1), 50) |> Assert.equal(_, 1)
  unwrap(Error("nope"), 50) |> Assert.equal(_, 50)
}