/
switch_case.go
87 lines (71 loc) · 1.67 KB
/
switch_case.go
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
// Copyright © 2023 Ory Corp
// SPDX-License-Identifier: Apache-2.0
package stringsx
import (
"fmt"
"strings"
)
type (
RegisteredCases struct {
cases []string
actual string
}
errUnknownCase struct {
*RegisteredCases
}
RegisteredPrefixes struct {
prefixes []string
actual string
}
errUnknownPrefix struct {
*RegisteredPrefixes
}
)
var (
ErrUnknownCase = errUnknownCase{}
ErrUnknownPrefix = errUnknownPrefix{}
)
func SwitchExact(actual string) *RegisteredCases {
return &RegisteredCases{
actual: actual,
}
}
func SwitchPrefix(actual string) *RegisteredPrefixes {
return &RegisteredPrefixes{
actual: actual,
}
}
func (r *RegisteredCases) AddCase(c string) bool {
r.cases = append(r.cases, c)
return r.actual == c
}
func (r *RegisteredPrefixes) HasPrefix(prefix string) bool {
r.prefixes = append(r.prefixes, prefix)
return strings.HasPrefix(r.actual, prefix)
}
func (r *RegisteredCases) String() string {
return "[" + strings.Join(r.cases, ", ") + "]"
}
func (r *RegisteredPrefixes) String() string {
return "[" + strings.Join(r.prefixes, ", ") + "]"
}
func (r *RegisteredCases) ToUnknownCaseErr() error {
return errUnknownCase{r}
}
func (r *RegisteredPrefixes) ToUnknownPrefixErr() error {
return errUnknownPrefix{r}
}
func (e errUnknownCase) Error() string {
return fmt.Sprintf("expected one of %s but got %s", e.String(), e.actual)
}
func (e errUnknownCase) Is(err error) bool {
_, ok := err.(errUnknownCase)
return ok
}
func (e errUnknownPrefix) Error() string {
return fmt.Sprintf("expected %s to have one of the prefixes %s", e.actual, e.String())
}
func (e errUnknownPrefix) Is(err error) bool {
_, ok := err.(errUnknownPrefix)
return ok
}