horizoncd/horizon

View on GitHub
pkg/token/manager/manager.go

Summary

Maintainability
A
0 mins
Test Coverage
A
100%
// Copyright © 2023 Horizoncd.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
 
package manager
 
import (
"context"
 
"github.com/horizoncd/horizon/pkg/token/models"
"github.com/horizoncd/horizon/pkg/token/store"
"gorm.io/gorm"
)
 
type Manager interface {
CreateToken(context.Context, *models.Token) (*models.Token, error)
LoadTokenByID(context.Context, uint) (*models.Token, error)
LoadTokenByCode(ctx context.Context, code string) (*models.Token, error)
RevokeTokenByID(context.Context, uint) error
RevokeTokenByClientID(ctx context.Context, clientID string) error
}
 
func New(db *gorm.DB) Manager {
return &manager{store: store.NewStore(db)}
}
 
type manager struct {
store store.Store
}
 
func (m *manager) CreateToken(ctx context.Context, token *models.Token) (*models.Token, error) {
return m.store.Create(ctx, token)
}
 
func (m *manager) LoadTokenByID(ctx context.Context, id uint) (*models.Token, error) {
return m.store.GetByID(ctx, id)
}
 
func (m *manager) LoadTokenByCode(ctx context.Context, code string) (*models.Token, error) {
return m.store.GetByCode(ctx, code)
}
 
func (m *manager) RevokeTokenByID(ctx context.Context, id uint) error {
return m.store.DeleteByID(ctx, id)
}
 
func (m *manager) RevokeTokenByClientID(ctx context.Context, clientID string) error {
return m.store.DeleteByClientID(ctx, clientID)
}