summaryrefslogtreecommitdiff
path: root/internal/userinput/userinput.go
blob: 8f1ecdc675ed7baeb81659e6cfaf4ecce9427533 (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
package userinput

import (
	"errors"
	"fmt"
	"strings"

	"github.com/ChausseBenjamin/termpicker/internal/colors"
)

var (
	errUnknownColorFormat = errors.New("Unrecognized color format")
	errHexParsing         = errors.New("Failed to parse hex color")
	errRGBParsing         = errors.New("Failed to parse RGB color")
	errHSLParsing         = errors.New("Failed to parse HSL color")
	errCMYKParsing        = errors.New("Failed to parse CMYK color")
)

func sanitize(s string) string {
	s = strings.ReplaceAll(s, "\"", "")
	s = strings.ReplaceAll(s, "%", "")
	s = strings.ReplaceAll(s, "°", "")
	s = strings.TrimSpace(s)
	s = strings.ToLower(s)
	return s
}

func ParseColor(s string) (colors.ColorSpace, error) {
	s = sanitize(s)
	switch {
	case strings.Contains(s, "#"):
		return parseHex(s)
	case strings.Contains(s, "rgb"):
		return parseRGB(s)
	case strings.Contains(s, "hsl"):
		return parseHSL(s)
	case strings.Contains(s, "cmyk"):
		return parseCMYK(s)
	default:
		return nil, errUnknownColorFormat
	}
}

func parseRGB(s string) (colors.ColorSpace, error) {
	var r, g, b int
	_, err := fmt.Sscanf(s, "rgb(%d,%d,%d)", &r, &g, &b)
	if err != nil {
		return nil, errors.Join(errRGBParsing, err)
	}
	return colors.RGB{R: r, G: g, B: b}, nil
}

func parseHex(s string) (colors.ColorSpace, error) {
	var r, g, b int
	_, err := fmt.Sscanf(s, "#%02x%02x%02x", &r, &g, &b)
	if err != nil {
		return nil, errors.Join(errHexParsing, err)
	}
	return colors.RGB{R: r, G: g, B: b}, nil
}

func parseCMYK(s string) (colors.ColorSpace, error) {
	var c, m, y, k int
	_, err := fmt.Sscanf(s, "cmyk(%d,%d,%d,%d)", &c, &m, &y, &k)
	if err != nil {
		return nil, errors.Join(errCMYKParsing, err)
	}
	return colors.CMYK{C: c, M: m, Y: y, K: k}, nil
}

func parseHSL(str string) (colors.ColorSpace, error) {
	var h, s, l int
	_, err := fmt.Sscanf(str, "hsl(%d,%d,%d)", &h, &s, &l)
	if err != nil {
		return nil, errors.Join(errHSLParsing, err)
	}
	return colors.HSL{H: h, S: s, L: l}, nil
}