service_account_action.go

package actions

import (
	"context"

	"github.com/andytechcastro/swiss-knife/errors"
	apiv1 "k8s.io/api/core/v1"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	coreInterface "k8s.io/client-go/kubernetes/typed/core/v1"
)

// ServiceAccount for manager namespace actions
type ServiceAccount struct {
	client           coreInterface.CoreV1Interface
	CurrentNamespace string
}

// NewServiceAccountAction get ServiceAccount action
func NewServiceAccountAction(client coreInterface.CoreV1Interface) *ServiceAccount {
	return &ServiceAccount{
		client:           client,
		CurrentNamespace: "",
	}
}

// Namespace set namespace
func (sa *ServiceAccount) Namespace(namespace string) *ServiceAccount {
	sa.CurrentNamespace = namespace
	return sa
}

// Get get namespace
func (sa *ServiceAccount) Get(saName string) (*apiv1.ServiceAccount, error) {
	if saName == "" {
		return nil, errors.GetEmptyError("Name")
	}
	serviceAccount, err := sa.client.ServiceAccounts(sa.CurrentNamespace).Get(
		context.TODO(),
		saName,
		metav1.GetOptions{})
	if err != nil {
		return nil, err
	}
	return serviceAccount, nil
}

// Create Create an ServiceAccount
func (sa *ServiceAccount) Create(serviceAccount *apiv1.ServiceAccount) error {
	if serviceAccount == nil {
		return errors.GetEmptyError("ServiceAccount")
	}
	_, err := sa.client.ServiceAccounts(sa.CurrentNamespace).Create(
		context.TODO(),
		serviceAccount,
		metav1.CreateOptions{},
	)
	if err != nil {
		return err
	}
	return nil
}

// Update Update a pods in the client
func (sa *ServiceAccount) Update(serviceAccount *apiv1.ServiceAccount) error {
	if serviceAccount == nil {
		return errors.GetEmptyError("ServiceAccount")
	}
	_, err := sa.client.ServiceAccounts(sa.CurrentNamespace).Update(
		context.TODO(),
		serviceAccount,
		metav1.UpdateOptions{},
	)
	if err != nil {
		return err
	}
	return nil
}

// Delete Delete a pod in the client
func (sa *ServiceAccount) Delete(saName string) error {
	if saName == "" {
		return errors.GetEmptyError("Name")
	}
	deletePolicy := metav1.DeletePropagationForeground
	err := sa.client.ServiceAccounts(sa.CurrentNamespace).Delete(
		context.TODO(),
		saName,
		metav1.DeleteOptions{
			PropagationPolicy: &deletePolicy,
		},
	)
	if err != nil {
		return err
	}
	return nil
}

// List List all pods in a namespace
func (sa *ServiceAccount) List() (*apiv1.ServiceAccountList, error) {
	serviceAccountList, err := sa.client.ServiceAccounts(sa.CurrentNamespace).List(
		context.TODO(),
		metav1.ListOptions{},
	)
	if err != nil {
		return nil, err
	}
	return serviceAccountList, nil
}