horizoncd/horizon

// 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"
 
    herrors "github.com/horizoncd/horizon/core/errors"
    perror "github.com/horizoncd/horizon/pkg/errors"
    "github.com/horizoncd/horizon/pkg/idp/utils"
    "github.com/horizoncd/horizon/pkg/userlink/dao"
    "github.com/horizoncd/horizon/pkg/userlink/models"
    "gorm.io/gorm"
)
 
// nolint
//
//go:generate mockgen -source=$GOFILE -destination=../../../mock/pkg/userlink/manager/manager_mock.go -package=mock_manager
type Manager interface {
    CreateLink(ctx context.Context, uid uint, idpID uint, claims *utils.Claims, deletable bool) (*models.UserLink, error)
    ListByUserID(ctx context.Context, uid uint) ([]*models.UserLink, error)
    GetByID(ctx context.Context, id uint) (*models.UserLink, error)
    GetByIDPAndSub(ctx context.Context, idpID uint, sub string) (*models.UserLink, error)
    DeleteByID(ctx context.Context, id uint) error
}
 
type manager struct {
    dao dao.DAO
}
 
func New(db *gorm.DB) Manager {
    return &manager{dao: dao.NewDAO(db)}
}
 
Method `manager.CreateLink` has 5 arguments (exceeds 4 allowed). Consider refactoring.
func (m *manager) CreateLink(ctx context.Context, uid uint,
    idpID uint, claims *utils.Claims, deletable bool) (*models.UserLink, error) {
    if claims == nil {
        return nil, perror.Wrapf(herrors.ErrParamInvalid, "claims is required")
    }
    link := models.UserLink{
        Sub:       claims.Sub,
        IdpID:     idpID,
        UserID:    uid,
        Name:      claims.Name,
        Email:     claims.Email,
        Deletable: deletable,
    }
    return m.dao.CreateLink(ctx, &link)
}
 
func (m *manager) ListByUserID(ctx context.Context, uid uint) ([]*models.UserLink, error) {
    return m.dao.ListByUserID(ctx, uid)
}
 
func (m *manager) GetByID(ctx context.Context, id uint) (*models.UserLink, error) {
    return m.dao.GetByID(ctx, id)
}
 
func (m *manager) GetByIDPAndSub(ctx context.Context,
    idpID uint, sub string) (*models.UserLink, error) {
    return m.dao.GetByIDPAndSub(ctx, idpID, sub)
}
 
func (m *manager) DeleteByID(ctx context.Context, id uint) error {
    return m.dao.DeleteByID(ctx, id)
}