validator-go/string.go at main · Jamess-Lucass/validator-go · GitHub

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
package schema

import (
	"fmt"
	"net/url"
	"strings"
)

type StringSchema struct {
	Schema[string]
}

var _ ISchema = (*StringSchema)(nil)

func String() *StringSchema {
	return &StringSchema{}
}

func (s *StringSchema) Max(maxLength int) *StringSchema {
	validator := Validator[string]{
		MessageFunc: func(value string) string {
			return fmt.Sprintf("String must contain at most %d character(s)", maxLength)
		},
		ValidateFunc: func(value string) bool {
			return len(value) <= maxLength
		},
	}

	s.validators = append(s.validators, validator)

	return s
}

func (s *StringSchema) Min(minLength int) *StringSchema {
	validator := Validator[string]{
		MessageFunc: func(value string) string {
			return fmt.Sprintf("String must contain at least %d character(s)", minLength)
		},
		ValidateFunc: func(value string) bool {
			return len(value) >= minLength
		},
	}

	s.validators = append(s.validators, validator)

	return s
}

func (s *StringSchema) Length(length int) *StringSchema {
	validator := Validator[string]{
		MessageFunc: func(value string) string {
			return fmt.Sprintf("String must contain exactly %d character(s)", length)
		},
		ValidateFunc: func(value string) bool {
			return len(value) == length
		},
	}

	s.validators = append(s.validators, validator)

	return s
}

func (s *StringSchema) Url() *StringSchema {
	validator := Validator[string]{
		MessageFunc: func(value string) string {
			return "Invalid url"
		},
		ValidateFunc: func(value string) bool {
			uri, err := url.ParseRequestURI(value)
			return err == nil && uri.Host != ""
		},
	}

	s.validators = append(s.validators, validator)

	return s
}

func (s *StringSchema) Includes(str string) *StringSchema {
	validator := Validator[string]{
		MessageFunc: func(value string) string {
			return fmt.Sprintf("Invalid input: must include \"%s\"", str)
		},
		ValidateFunc: func(value string) bool {
			return strings.Contains(value, str)
		},
	}

	s.validators = append(s.validators, validator)

	return s
}

func (s *StringSchema) StartsWith(str string) *StringSchema {
	validator := Validator[string]{
		MessageFunc: func(value string) string {
			return fmt.Sprintf("Invalid input: must start with \"%s\"", str)
		},
		ValidateFunc: func(value string) bool {
			return strings.HasPrefix(value, str)
		},
	}

	s.validators = append(s.validators, validator)

	return s
}

func (s *StringSchema) EndsWith(str string) *StringSchema {
	validator := Validator[string]{
		MessageFunc: func(value string) string {
			return fmt.Sprintf("Invalid input: must end with \"%s\"", str)
		},
		ValidateFunc: func(value string) bool {
			return strings.HasSuffix(value, str)
		},
	}

	s.validators = append(s.validators, validator)

	return s
}