summary refs log tree commit diff stats
path: root/tests/objvariant/tyaoption.nim
blob: 7a29b80087e049667c24ab1aaa476c16abdf78e7 (plain) (blame)
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
discard """
  output: '''some(str), some(5), none
some(5!)
some(10)'''
"""

import strutils

type Option[A] = object
  case isDefined*: bool
    of true:
      value*: A
    of false:
      nil

proc some[A](value: A): Option[A] =
  Option[A](isDefined: true, value: value)

proc none[A](): Option[A] =
  Option[A](isDefined: false)

proc `$`[A](o: Option[A]): string =
  if o.isDefined:
    "some($1)" % [$o.value]
  else:
    "none"

let x = some("str")
let y = some(5)
let z = none[int]()

echo x, ", ", y, ", ", z

proc intOrString[A : int | string](o: Option[A]): Option[A] =
  when A is int:
    some(o.value + 5)
  elif A is string:
    some(o.value & "!")
  else:
    o

#let a1 = intOrString(none[String]())
let a2 = intOrString(some("5"))
let a3 = intOrString(some(5))
#echo a1
echo a2
echo a3