-
Notifications
You must be signed in to change notification settings - Fork 3
/
variant.go
63 lines (54 loc) · 1.49 KB
/
variant.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
package plaintext
import (
"github.com/go-crypt/crypt/internal/encoding"
)
// NewVariant converts an identifier string to a plaintext.Variant.
func NewVariant(identifier string) (variant Variant) {
switch identifier {
case AlgIdentifierPlainText:
return VariantPlainText
case AlgIdentifierBase64:
return VariantBase64
default:
return VariantNone
}
}
// Variant is a variant of the plaintext.Digest.
type Variant int
const (
// VariantNone is a variant of the plaintext.Digest which is unknown.
VariantNone Variant = iota
// VariantPlainText is a variant of the plaintext.Digest which stores the key as plain text.
VariantPlainText
// VariantBase64 is a variant of the plaintext.Digest which stores the key as a base64 string.
VariantBase64
)
// Prefix returns the plaintext.Variant prefix identifier.
func (v Variant) Prefix() (prefix string) {
switch v {
case VariantPlainText:
return AlgIdentifierPlainText
case VariantBase64:
return AlgIdentifierBase64
default:
return
}
}
// Decode performs the decode operation for this plaintext.Variant.
func (v Variant) Decode(src string) (dst []byte, err error) {
switch v {
case VariantBase64:
return encoding.Base64RawAdaptedEncoding.DecodeString(src)
default:
return []byte(src), nil
}
}
// Encode performs the encode operation for this plaintext.Variant.
func (v Variant) Encode(src []byte) (dst string) {
switch v {
case VariantBase64:
return encoding.Base64RawAdaptedEncoding.EncodeToString(src)
default:
return string(src)
}
}