wip: migrate to mono-repo. SPN has already been moved to spn/
This commit is contained in:
83
spn/access/token/token.go
Normal file
83
spn/access/token/token.go
Normal file
@@ -0,0 +1,83 @@
|
||||
package token
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"errors"
|
||||
"fmt"
|
||||
"strings"
|
||||
|
||||
"github.com/mr-tron/base58"
|
||||
|
||||
"github.com/safing/portbase/container"
|
||||
)
|
||||
|
||||
// Token represents a token, consisting of a zone (name) and some data.
|
||||
type Token struct {
|
||||
Zone string
|
||||
Data []byte
|
||||
}
|
||||
|
||||
// GetToken returns a token of the given zone.
|
||||
func GetToken(zone string) (*Token, error) {
|
||||
handler, ok := GetHandler(zone)
|
||||
if !ok {
|
||||
return nil, ErrZoneUnknown
|
||||
}
|
||||
|
||||
return handler.GetToken()
|
||||
}
|
||||
|
||||
// VerifyToken verifies the given token.
|
||||
func VerifyToken(token *Token) error {
|
||||
handler, ok := GetHandler(token.Zone)
|
||||
if !ok {
|
||||
return ErrZoneUnknown
|
||||
}
|
||||
|
||||
return handler.Verify(token)
|
||||
}
|
||||
|
||||
// Raw returns the raw format of the token.
|
||||
func (c *Token) Raw() []byte {
|
||||
cont := container.New()
|
||||
cont.Append([]byte(c.Zone))
|
||||
cont.Append([]byte(":"))
|
||||
cont.Append(c.Data)
|
||||
return cont.CompileData()
|
||||
}
|
||||
|
||||
// String returns the stringified format of the token.
|
||||
func (c *Token) String() string {
|
||||
return c.Zone + ":" + base58.Encode(c.Data)
|
||||
}
|
||||
|
||||
// ParseRawToken parses a raw token.
|
||||
func ParseRawToken(code []byte) (*Token, error) {
|
||||
splitted := bytes.SplitN(code, []byte(":"), 2)
|
||||
if len(splitted) < 2 {
|
||||
return nil, errors.New("invalid code format: zone/data separator missing")
|
||||
}
|
||||
|
||||
return &Token{
|
||||
Zone: string(splitted[0]),
|
||||
Data: splitted[1],
|
||||
}, nil
|
||||
}
|
||||
|
||||
// ParseToken parses a stringified token.
|
||||
func ParseToken(code string) (*Token, error) {
|
||||
splitted := strings.SplitN(code, ":", 2)
|
||||
if len(splitted) < 2 {
|
||||
return nil, errors.New("invalid code format: zone/data separator missing")
|
||||
}
|
||||
|
||||
data, err := base58.Decode(splitted[1])
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("invalid code format: %w", err)
|
||||
}
|
||||
|
||||
return &Token{
|
||||
Zone: splitted[0],
|
||||
Data: data,
|
||||
}, nil
|
||||
}
|
||||
Reference in New Issue
Block a user