package k8s

import (
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
)

// PodDisruptionBudget represents a Kubernetes PodDisruptionBudget.
type PodDisruptionBudget struct {
	*base
	Connection
}

// NewPodDisruptionBudget returns a new PodDisruptionBudget.
func NewPodDisruptionBudget(c Connection) *PodDisruptionBudget {
	return &PodDisruptionBudget{&base{}, c}
}

// Get a pdb.
func (p *PodDisruptionBudget) Get(ns, n string) (interface{}, error) {
	return p.DialOrDie().PolicyV1beta1().PodDisruptionBudgets(ns).Get(n, metav1.GetOptions{})
}

// List all pdbs in a given namespace.
func (p *PodDisruptionBudget) List(ns string, opts metav1.ListOptions) (Collection, error) {
	rr, err := p.DialOrDie().PolicyV1beta1().PodDisruptionBudgets(ns).List(opts)
	if err != nil {
		return nil, err
	}
	cc := make(Collection, len(rr.Items))
	for i, r := range rr.Items {
		cc[i] = r
	}

	return cc, nil
}

// Delete a pdb.
func (p *PodDisruptionBudget) Delete(ns, n string, cascade, force bool) error {
	return p.DialOrDie().PolicyV1beta1().PodDisruptionBudgets(ns).Delete(n, nil)
}