db.go

package db

import (
	appv1 "github.com/argoproj/argo-cd/pkg/apis/application/v1alpha1"
	"golang.org/x/net/context"
	"k8s.io/client-go/kubernetes"
)

type ArgoDB interface {
	// ListClusters lists configured clusters
	ListClusters(ctx context.Context) (*appv1.ClusterList, error)
	// CreateCluster creates a cluster
	CreateCluster(ctx context.Context, c *appv1.Cluster) (*appv1.Cluster, error)
	// WatchClusters allow watching for cluster events
	WatchClusters(ctx context.Context, callback func(*ClusterEvent)) error
	// Get returns a cluster from a query
	GetCluster(ctx context.Context, name string) (*appv1.Cluster, error)
	// UpdateCluster updates a cluster
	UpdateCluster(ctx context.Context, c *appv1.Cluster) (*appv1.Cluster, error)
	// DeleteCluster deletes a cluster by name
	DeleteCluster(ctx context.Context, name string) error

	// ListRepositories lists repositories
	ListRepositories(ctx context.Context) (*appv1.RepositoryList, error)
	// CreateRepository creates a repository
	CreateRepository(ctx context.Context, r *appv1.Repository) (*appv1.Repository, error)
	// GetRepository returns a repository by URL
	GetRepository(ctx context.Context, name string) (*appv1.Repository, error)
	// UpdateRepository updates a repository
	UpdateRepository(ctx context.Context, r *appv1.Repository) (*appv1.Repository, error)
	// DeleteRepository updates a repository
	DeleteRepository(ctx context.Context, name string) error
}

type db struct {
	ns            string
	kubeclientset kubernetes.Interface
}

// NewDB returns a new instance of the argo database
func NewDB(namespace string, kubeclientset kubernetes.Interface) ArgoDB {
	return &db{
		ns:            namespace,
		kubeclientset: kubeclientset,
	}
}