drone/internal/store/database/wire.go

// Copyright 2022 Harness Inc. All rights reserved.
// Use of this source code is governed by the Polyform Free Trial License
// that can be found in the LICENSE.md file for this repository.

package database

import (
	"context"

	"github.com/harness/gitness/internal/store"
	"github.com/harness/gitness/types"

	"github.com/google/wire"
	"github.com/jmoiron/sqlx"
)

const (
	postgres = "postgres"
)

// WireSet provides a wire set for this package.
var WireSet = wire.NewSet(
	ProvideDatabase,
	ProvideUserStore,
	ProvideServiceAccountStore,
	ProvideServiceStore,
	ProvideSpaceStore,
	ProvideRepoStore,
	ProvidePullReqStore,
	ProvideTokenStore,
)

// ProvideDatabase provides a database connection.
func ProvideDatabase(ctx context.Context, config *types.Config) (*sqlx.DB, error) {
	return Connect(
		ctx,
		config.Database.Driver,
		config.Database.Datasource,
	)
}

// ProvideUserStore provides a user store.
func ProvideUserStore(db *sqlx.DB, uidTransformation store.PrincipalUIDTransformation) store.UserStore {
	switch db.DriverName() {
	case postgres:
		return NewUserStore(db, uidTransformation)
	default:
		return NewUserStoreSync(
			NewUserStore(db, uidTransformation),
		)
	}
}

// ProvideServiceAccountStore provides a service account store.
func ProvideServiceAccountStore(db *sqlx.DB,
	uidTransformation store.PrincipalUIDTransformation) store.ServiceAccountStore {
	switch db.DriverName() {
	case postgres:
		return NewServiceAccountStore(db, uidTransformation)
	default:
		return NewServiceAccountStoreSync(
			NewServiceAccountStore(db, uidTransformation),
		)
	}
}

// ProvideServiceStore provides a service store.
func ProvideServiceStore(db *sqlx.DB, uidTransformation store.PrincipalUIDTransformation) store.ServiceStore {
	switch db.DriverName() {
	case postgres:
		return NewServiceStore(db, uidTransformation)
	default:
		return NewServiceStoreSync(
			NewServiceStore(db, uidTransformation),
		)
	}
}

// ProvideSpaceStore provides a space store.
func ProvideSpaceStore(db *sqlx.DB, pathTransformation store.PathTransformation) store.SpaceStore {
	switch db.DriverName() {
	case postgres:
		return NewSpaceStore(db, pathTransformation)
	default:
		return NewSpaceStoreSync(
			NewSpaceStore(db, pathTransformation),
		)
	}
}

// ProvideRepoStore provides a repo store.
func ProvideRepoStore(db *sqlx.DB, pathTransformation store.PathTransformation) store.RepoStore {
	switch db.DriverName() {
	case postgres:
		return NewRepoStore(db, pathTransformation)
	default:
		return NewRepoStoreSync(
			NewRepoStore(db, pathTransformation),
		)
	}
}

// ProvideTokenStore provides a token store.
func ProvideTokenStore(db *sqlx.DB) store.TokenStore {
	switch db.DriverName() {
	case postgres:
		return NewTokenStore(db)
	default:
		return NewTokenStoreSync(
			NewTokenStore(db),
		)
	}
}

// ProvidePullReqStore provides a pull request store.
func ProvidePullReqStore(db *sqlx.DB) store.PullReqStore {
	switch db.DriverName() {
	case postgres:
		return NewPullReqStore(db)
	default:
		return NewPullReqStoreSync(
			NewPullReqStore(db),
		)
	}
}