route/cmd/routed/storage.go

35 lines
635 B
Go
Raw Normal View History

2017-12-15 18:18:13 +00:00
package main
import (
"context"
2017-09-30 13:41:35 +00:00
"git.xeserv.us/xena/route/internal/database"
2017-10-01 15:23:08 +00:00
"git.xeserv.us/xena/route/internal/tun2"
)
type storageWrapper struct {
database.Storage
}
var (
_ tun2.Storage = &storageWrapper{}
)
func (s *storageWrapper) HasToken(token string) (string, []string, error) {
t, err := s.Storage.Tokens().Get(context.Background(), token)
if err != nil {
return "", nil, err
}
return t.Owner, t.Scopes, nil
}
func (s *storageWrapper) HasRoute(domain string) (string, error) {
r, err := s.Storage.Routes().GetHost(context.Background(), domain)
if err != nil {
return "", err
}
return r.Creator, nil
}