// Copyright Project Harbor Authors
//
// 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 pkg

import (
	"github.com/goharbor/harbor/src/lib/config"
	"github.com/goharbor/harbor/src/pkg/artifact"
	cachedArtifact "github.com/goharbor/harbor/src/pkg/cached/artifact/redis"
	cachedManifest "github.com/goharbor/harbor/src/pkg/cached/manifest/redis"
	cachedProject "github.com/goharbor/harbor/src/pkg/cached/project/redis"
	cachedProjectMeta "github.com/goharbor/harbor/src/pkg/cached/project_metadata/redis"
	cachedRepo "github.com/goharbor/harbor/src/pkg/cached/repository/redis"
	"github.com/goharbor/harbor/src/pkg/project"
	"github.com/goharbor/harbor/src/pkg/project/metadata"
	"github.com/goharbor/harbor/src/pkg/repository"

	// init the db config
	_ "github.com/goharbor/harbor/src/pkg/config/db"
	// init the rest config
	_ "github.com/goharbor/harbor/src/pkg/config/rest"
)

// Define global resource manager.
var (
	// ArtifactMgr is the manager for artifact.
	ArtifactMgr artifact.Manager
	// ProjectMgr is the manager for project.
	ProjectMgr project.Manager
	// ProjectMetaMgr is the manager for project metadata.
	ProjectMetaMgr metadata.Manager
	// RepositoryMgr is the manager for repository.
	RepositoryMgr repository.Manager
	// ManifestMgr is the manager for manifest.
	ManifestMgr cachedManifest.CachedManager
)

// init initialize mananger for resources
func init() {
	cacheEnabled := config.CacheEnabled()
	initArtifactMgr(cacheEnabled)
	initProjectMgr(cacheEnabled)
	initProjectMetaMgr(cacheEnabled)
	initRepositoryMgr(cacheEnabled)
	initManifestManager(cacheEnabled)
}

func initArtifactMgr(cacheEnabled bool) {
	artMgr := artifact.NewManager()
	// check cache enable
	if cacheEnabled {
		ArtifactMgr = cachedArtifact.NewManager(artMgr)
	} else {
		ArtifactMgr = artMgr
	}
}

func initProjectMgr(cacheEnabled bool) {
	projectMgr := project.New()
	// check cache enable
	if cacheEnabled {
		ProjectMgr = cachedProject.NewManager(projectMgr)
	} else {
		ProjectMgr = projectMgr
	}
}

func initProjectMetaMgr(cacheEnabled bool) {
	projectMetaMgr := metadata.New()
	if cacheEnabled {
		ProjectMetaMgr = cachedProjectMeta.NewManager(projectMetaMgr)
	} else {
		ProjectMetaMgr = projectMetaMgr
	}
}

func initRepositoryMgr(cacheEnabled bool) {
	repoMgr := repository.New()
	if cacheEnabled {
		RepositoryMgr = cachedRepo.NewManager(repoMgr)
	} else {
		RepositoryMgr = repoMgr
	}
}

func initManifestManager(_ bool) {
	ManifestMgr = cachedManifest.NewManager()
}