about summary refs log tree commit diff
path: root/vendor/github.com/ncruces/go-strftime/parser.go
blob: b006de38a85cde9d5fa153887daecd21d6416cd4 (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
package strftime

import "unicode/utf8"

type parser struct {
	format  func(spec, flag byte) error
	literal func(byte) error
}

func (p *parser) parse(fmt string) error {
	const (
		initial = iota
		percent
		flagged
		modified
	)

	var flag, modifier byte
	var err error
	state := initial
	start := 0
	for i, b := range []byte(fmt) {
		switch state {
		default:
			if b == '%' {
				state = percent
				start = i
				continue
			}
			err = p.literal(b)

		case percent:
			if b == '-' || b == ':' {
				state = flagged
				flag = b
				continue
			}
			if b == 'E' || b == 'O' {
				state = modified
				modifier = b
				flag = 0
				continue
			}
			err = p.format(b, 0)
			state = initial

		case flagged:
			if b == 'E' || b == 'O' {
				state = modified
				modifier = b
				continue
			}
			err = p.format(b, flag)
			state = initial

		case modified:
			if okModifier(modifier, b) {
				err = p.format(b, flag)
			} else {
				err = p.literals(fmt[start : i+1])
			}
			state = initial
		}

		if err != nil {
			if err, ok := err.(formatError); ok {
				err.setDirective(fmt, start, i)
				return err
			}
			return err
		}
	}

	if state != initial {
		return p.literals(fmt[start:])
	}
	return nil
}

func (p *parser) literals(literal string) error {
	for _, b := range []byte(literal) {
		if err := p.literal(b); err != nil {
			return err
		}
	}
	return nil
}

type literalErr string

func (e literalErr) Error() string {
	return "strftime: unsupported literal: " + string(e)
}

type formatError struct {
	message   string
	directive string
}

func (e formatError) Error() string {
	return "strftime: unsupported directive: " + e.directive + " " + e.message
}

func (e *formatError) setDirective(str string, i, j int) {
	_, n := utf8.DecodeRuneInString(str[j:])
	e.directive = str[i : j+n]
}