84 lines
1.7 KiB
Go
84 lines
1.7 KiB
Go
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
|
|
}
|