mirror of
https://github.com/kevin-DL/services.git
synced 2026-01-12 03:05:14 +00:00
b8877f8312
* fixup streams stuff * use cache instead of pg in streams * hash out issuer streams test * fix compile error
62 lines
1.6 KiB
Go
62 lines
1.6 KiB
Go
package handler
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/micro/micro/v3/service/auth"
|
|
"github.com/micro/micro/v3/service/errors"
|
|
"github.com/micro/micro/v3/service/events"
|
|
"github.com/micro/services/pkg/cache"
|
|
"github.com/nats-io/nats-streaming-server/util"
|
|
)
|
|
|
|
var (
|
|
TokenTTL = time.Minute
|
|
ErrMissingTopic = errors.BadRequest("MISSING_TOPIC", "Missing topic")
|
|
ErrInvalidTopic = errors.BadRequest("MISSING_TOPIC", "Invalid topic")
|
|
ErrMissingToken = errors.BadRequest("MISSING_TOKEN", "Missing token")
|
|
ErrMissingMessage = errors.BadRequest("MISSING_MESSAGE", "Missing message")
|
|
ErrInvalidToken = errors.Forbidden("INVALID_TOKEN", "Invalid token")
|
|
ErrExpiredToken = errors.Forbidden("EXPIRED_TOKEN", "Token expired")
|
|
ErrForbiddenTopic = errors.Forbidden("FORBIDDEN_TOPIC", "Token has not have permission to subscribe to this topic")
|
|
)
|
|
|
|
type Token struct {
|
|
Token string
|
|
Topic string
|
|
Account string
|
|
ExpiresAt time.Time
|
|
}
|
|
|
|
type Streams struct {
|
|
Cache cache.Cache
|
|
Events events.Stream
|
|
Time func() time.Time
|
|
}
|
|
|
|
func (t *Token) Key() string {
|
|
return fmt.Sprintf("%s:%s", t.Account, t.Token)
|
|
}
|
|
|
|
func getAccount(acc *auth.Account) string {
|
|
owner := acc.Metadata["apikey_owner"]
|
|
if len(owner) == 0 {
|
|
owner = acc.ID
|
|
}
|
|
return fmt.Sprintf("%s.%s", acc.Issuer, owner)
|
|
}
|
|
|
|
// fmtTopic returns a topic string with namespace prefix
|
|
func fmtTopic(acc *auth.Account, topic string) string {
|
|
return fmt.Sprintf("%s.%s", getAccount(acc), topic)
|
|
}
|
|
|
|
// validateTopicInput validates that topic is alphanumeric
|
|
func validateTopicInput(topic string) error {
|
|
if !util.IsChannelNameValid(topic, false) {
|
|
return ErrInvalidTopic
|
|
}
|
|
return nil
|
|
}
|