~wallyworld/juju-core/fast-lxc-everywhere

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
// Copyright 2013 Canonical Ltd.
// Licensed under the AGPLv3, see LICENCE file for details.

package names

import (
	"fmt"
	"strings"
)

const (
	UnitTagKind     = "unit"
	MachineTagKind  = "machine"
	ServiceTagKind  = "service"
	EnvironTagKind  = "environment"
	UserTagKind     = "user"
	RelationTagKind = "relation"
	NetworkTagKind  = "network"
)

var validKinds = map[string]bool{
	UnitTagKind:     true,
	MachineTagKind:  true,
	ServiceTagKind:  true,
	EnvironTagKind:  true,
	UserTagKind:     true,
	RelationTagKind: true,
	NetworkTagKind:  true,
}

var tagSuffixToId = map[string]func(string) string{
	UnitTagKind:     unitTagSuffixToId,
	MachineTagKind:  machineTagSuffixToId,
	RelationTagKind: relationTagSuffixToKey,
}

var verifyId = map[string]func(string) bool{
	UnitTagKind:     IsUnit,
	MachineTagKind:  IsMachine,
	ServiceTagKind:  IsService,
	UserTagKind:     IsUser,
	EnvironTagKind:  IsEnvironment,
	RelationTagKind: IsRelation,
	NetworkTagKind:  IsNetwork,
}

// TagKind returns one of the *TagKind constants for the given tag, or
// an error if none matches.
func TagKind(tag string) (string, error) {
	i := strings.Index(tag, "-")
	if i <= 0 || !validKinds[tag[:i]] {
		return "", fmt.Errorf("%q is not a valid tag", tag)
	}
	return tag[:i], nil
}

func splitTag(tag string) (kind, rest string, err error) {
	kind, err = TagKind(tag)
	if err != nil {
		return "", "", err
	}
	return kind, tag[len(kind)+1:], nil
}

func makeTag(kind, rest string) string {
	return kind + "-" + rest
}

// ParseTag parses a tag into its kind and identifier
// components. It returns an error if the tag is malformed,
// or if expectKind is not empty and the kind is
// not as expected.
func ParseTag(tag, expectKind string) (kind, id string, err error) {
	kind, id, err = splitTag(tag)
	if err != nil {
		return "", "", invalidTagError(tag, expectKind)
	}
	if expectKind != "" && kind != expectKind {
		return "", "", invalidTagError(tag, expectKind)
	}
	if toId := tagSuffixToId[kind]; toId != nil {
		id = toId(id)
	}
	if verify := verifyId[kind]; verify != nil && !verify(id) {
		return "", "", invalidTagError(tag, kind)
	}
	return kind, id, nil
}

func invalidTagError(tag, kind string) error {
	if kind != "" {
		return fmt.Errorf("%q is not a valid %s tag", tag, kind)
	}
	return fmt.Errorf("%q is not a valid tag", tag)
}