aboutsummaryrefslogtreecommitdiff
path: root/src/result.gleam
blob: 18503ec8f292017b5da66f72f48b2f8280c3df64 (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
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
import expect

// Result represents the result of something that may succeed or fail.
// `Ok` means it was successful, `Error` means it failed.

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

test is_ok {
  is_ok(Ok(1)) |> expect:true
  is_ok(Error(1)) |> expect:false
}

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

test is_error {
  is_error(Ok(1))
    |> expect:false

  is_error(Error(1))
    |> expect:true
}

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

test map {
  Ok(1)
    |> map(_, fn(x) { x + 1 })
    |> expect:equal(_, Ok(2))

  Ok(1)
    |> map(_, fn(_) { "2" })
    |> expect:equal(_, Ok("2"))

  Error(1)
    |> map(_, fn(x) { x + 1 })
    |> expect: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 })
    |> expect:equal(_, Ok(1))

  Error(1)
    |> map_error(_, fn(x) { x + 1 })
    |> expect:equal(_, Error(2))
}

pub fn flatten(result) {
  case result {
  | Ok(x) -> x
  | Error(error) -> Error(error)
  }
}

test flatten {
  flatten(Ok(Ok(1)))
    |> expect:equal(_, Ok(1))

  flatten(Ok(Error(1)))
    |> expect:equal(_, Error(1))

  flatten(Error(1))
    |> expect:equal(_, Error(1))

  flatten(Error(Error(1)))
    |> expect:equal(_, Error(Error(1)))
}

pub fn then(result, fun) {
  case result {
  | Ok(x) ->
      case fun(x) {
      | Ok(y) -> Ok(y)
      | Error(y) -> Error(y)
      }
  | Error(_) -> result
  }
}

test then {
  Error(1)
    |> then(_, fn(x) { Ok(x + 1) })
    |> expect:equal(_, Error(1))

  Ok(1)
    |> then(_, fn(x) { Ok(x + 1) })
    |> expect:equal(_, Ok(2))

  Ok(1)
    |> then(_, fn(_) { Error(1) })
    |> expect:equal(_, Error(1))
}

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

test unwrap {
  unwrap(Ok(1), 50)
    |> expect:equal(_, 1)

  unwrap(Error("nope"), 50)
    |> expect:equal(_, 50)
}