package supervisor

import "github.com/docker/containerd/runtime"

type UpdateEvent struct {
	s *Supervisor
}

func (h *UpdateEvent) Handle(e *Event) error {
	i, ok := h.s.containers[e.ID]
	if !ok {
		return ErrContainerNotFound
	}
	container := i.container
	if e.State != "" {
		switch e.State {
		case runtime.Running:
			if err := container.Resume(); err != nil {
				return ErrUnknownContainerStatus
			}
		case runtime.Paused:
			if err := container.Pause(); err != nil {
				return ErrUnknownContainerStatus
			}
		default:
			return ErrUnknownContainerStatus
		}
	}
	return nil
}

type UpdateProcessEvent struct {
	s *Supervisor
}

func (h *UpdateProcessEvent) Handle(e *Event) error {
	i, ok := h.s.containers[e.ID]
	if !ok {
		return ErrContainerNotFound
	}
	processes, err := i.container.Processes()
	if err != nil {
		return err
	}
	var process runtime.Process
	for _, p := range processes {
		if p.ID() == e.Pid {
			process = p
			break
		}
	}
	if process == nil {
		return ErrProcessNotFound
	}
	if e.CloseStdin {
		if err := process.CloseStdin(); err != nil {
			return err
		}
	}
	if e.Width > 0 || e.Height > 0 {
		if err := process.Resize(e.Width, e.Height); err != nil {
			return err
		}
	}
	return nil
}