k9s/internal/views/restartable_resource.go

61 lines
1.4 KiB
Go

package views
import (
"errors"
"github.com/derailed/k9s/internal/resource"
"github.com/derailed/k9s/internal/ui"
"github.com/derailed/k9s/internal/ui/dialog"
"github.com/gdamore/tcell"
)
type (
restartableResourceView struct {
*resourceView
}
)
func newRestartableResourceViewForParent(parent *resourceView) *restartableResourceView {
v := restartableResourceView{
parent,
}
parent.extraActionsFn = v.extraActions
return &v
}
func (v *restartableResourceView) extraActions(aa ui.KeyActions) {
aa[tcell.KeyCtrlT] = ui.NewKeyAction("Restart Rollout", v.restartCmd, true)
}
func (v *restartableResourceView) restartCmd(evt *tcell.EventKey) *tcell.EventKey {
if !v.masterPage().RowSelected() {
return evt
}
sel := v.masterPage().GetSelectedItem()
v.stopUpdates()
defer v.restartUpdates()
msg := "Please confirm rollout restart for " + sel
dialog.ShowConfirm(v.Pages, "<Confirm Restart>", msg, func() {
if err := v.restartRollout(sel); err != nil {
v.app.Flash().Err(err)
} else {
v.app.Flash().Infof("Rollout restart in progress for `%s...", sel)
}
}, func() {
v.showMaster()
})
return nil
}
func (v *restartableResourceView) restartRollout(selection string) error {
r, ok := v.list.Resource().(resource.Restartable)
if !ok {
return errors.New("resource is not of type resource.Restartable")
}
ns, n := namespaced(selection)
return r.Restart(ns, n)
}