package k8s

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

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

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

// Get a deployment.
func (d *Deployment) Get(ns, n string) (interface{}, error) {
	return d.DialOrDie().AppsV1().Deployments(ns).Get(n, metav1.GetOptions{})
}

// List all Deployments in a given namespace.
func (d *Deployment) List(ns string, opts metav1.ListOptions) (Collection, error) {
	rr, err := d.DialOrDie().AppsV1().Deployments(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 Deployment.
func (d *Deployment) Delete(ns, n string, cascade, force bool) error {
	return d.DialOrDie().AppsV1().Deployments(ns).Delete(n, nil)
}

// Scale a Deployment.
func (d *Deployment) Scale(ns, n string, replicas int32) error {
	scale, err := d.DialOrDie().AppsV1().Deployments(ns).GetScale(n, metav1.GetOptions{})
	if err != nil {
		return err
	}

	scale.Spec.Replicas = replicas
	_, err = d.DialOrDie().AppsV1().Deployments(ns).UpdateScale(n, scale)
	return err
}

// Restart a Deployment rollout.
func (d *Deployment) Restart(ns, n string) error {

	dp, err := d.DialOrDie().AppsV1().Deployments(ns).Get(n, metav1.GetOptions{})
	if err != nil {
		return err
	}
	update, err := polymorphichelpers.ObjectRestarterFn(dp)
	if err != nil {
		return err
	}

	_, err = d.DialOrDie().AppsV1().Deployments(ns).Patch(dp.Name, types.StrategicMergePatchType, update)
	return err
}