auth.go

// Copyright 2023 sigma
//
// 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 auth

import (
	"github.com/go-sigma/sigma/pkg/dal/dao"
	"github.com/go-sigma/sigma/pkg/dal/models"
	"github.com/go-sigma/sigma/pkg/types/enums"
)

//go:generate mockgen -destination=mocks/service.go -package=mocks github.com/go-sigma/sigma/pkg/auth AuthService
//go:generate mockgen -destination=mocks/service_factory.go -package=mocks github.com/go-sigma/sigma/pkg/auth AuthServiceFactory

// AuthService is the interface for the auth service
type AuthService interface {
	// Namespace ...
	Namespace(user models.User, namespaceID int64, auth enums.Auth) (bool, error)
	// NamespaceRole get user role in namespace
	NamespaceRole(user models.User, namespaceID int64) (*enums.NamespaceRole, error)
	// NamespacesRole ...
	NamespacesRole(user models.User, namespaceIDs []int64) (map[int64]*enums.NamespaceRole, error)
	// Repository ...
	Repository(user models.User, repositoryID int64, auth enums.Auth) (bool, error)
	// Tag ...
	Tag(user models.User, tagID int64, auth enums.Auth) (bool, error)
	// Artifact ...
	Artifact(user models.User, artifactID int64, auth enums.Auth) (bool, error)
}

// AuthServiceFactory is the interface that provides the artifact service factory methods.
type AuthServiceFactory interface {
	New() AuthService
}

type authService struct {
	namespaceMemberServiceFactory dao.NamespaceMemberServiceFactory
	namespaceServiceFactory       dao.NamespaceServiceFactory
	repositoryServiceFactory      dao.RepositoryServiceFactory
	tagServiceFactory             dao.TagServiceFactory
	artifactServiceFactory        dao.ArtifactServiceFactory
}

type inject struct {
	namespaceMemberServiceFactory dao.NamespaceMemberServiceFactory
	namespaceServiceFactory       dao.NamespaceServiceFactory
	repositoryServiceFactory      dao.RepositoryServiceFactory
	tagServiceFactory             dao.TagServiceFactory
	artifactServiceFactory        dao.ArtifactServiceFactory
}

type authServiceFactory struct {
	namespaceMemberServiceFactory dao.NamespaceMemberServiceFactory
	namespaceServiceFactory       dao.NamespaceServiceFactory
	repositoryServiceFactory      dao.RepositoryServiceFactory
	tagServiceFactory             dao.TagServiceFactory
	artifactServiceFactory        dao.ArtifactServiceFactory
}

// NewAuthServiceFactory creates a new auth service factory.
func NewAuthServiceFactory(injects ...inject) AuthServiceFactory {
	namespaceMemberServiceFactory := dao.NewNamespaceMemberServiceFactory()
	namespaceServiceFactory := dao.NewNamespaceServiceFactory()
	repositoryServiceFactory := dao.NewRepositoryServiceFactory()
	tagServiceFactory := dao.NewTagServiceFactory()
	artifactServiceFactory := dao.NewArtifactServiceFactory()
	if len(injects) > 0 {
		ij := injects[0]
		if ij.namespaceMemberServiceFactory != nil {
			namespaceMemberServiceFactory = ij.namespaceMemberServiceFactory
		}
		if ij.namespaceServiceFactory != nil {
			namespaceServiceFactory = ij.namespaceServiceFactory
		}
		if ij.repositoryServiceFactory != nil {
			repositoryServiceFactory = ij.repositoryServiceFactory
		}
		if ij.tagServiceFactory != nil {
			tagServiceFactory = ij.tagServiceFactory
		}
		if ij.artifactServiceFactory != nil {
			artifactServiceFactory = ij.artifactServiceFactory
		}
	}
	return &authServiceFactory{
		namespaceMemberServiceFactory: namespaceMemberServiceFactory,
		namespaceServiceFactory:       namespaceServiceFactory,
		repositoryServiceFactory:      repositoryServiceFactory,
		tagServiceFactory:             tagServiceFactory,
		artifactServiceFactory:        artifactServiceFactory,
	}
}

// New ...
func (f *authServiceFactory) New() AuthService {
	s := &authService{
		namespaceMemberServiceFactory: f.namespaceMemberServiceFactory,
		namespaceServiceFactory:       f.namespaceServiceFactory,
		repositoryServiceFactory:      f.repositoryServiceFactory,
		tagServiceFactory:             f.tagServiceFactory,
		artifactServiceFactory:        f.artifactServiceFactory,
	}
	return s
}

var _ AuthService = &authService{}