package compat

import (
	"net/http"
	"strings"

	"github.com/containers/podman/v3/libpod"
	"github.com/containers/podman/v3/pkg/api/handlers"
	"github.com/containers/podman/v3/pkg/api/handlers/utils"
	api "github.com/containers/podman/v3/pkg/api/types"
	"github.com/gorilla/schema"
	"github.com/pkg/errors"
)

func TopContainer(w http.ResponseWriter, r *http.Request) {
	runtime := r.Context().Value(api.RuntimeKey).(*libpod.Runtime)
	decoder := r.Context().Value(api.DecoderKey).(*schema.Decoder)

	defaultValue := "-ef"
	if utils.IsLibpodRequest(r) {
		defaultValue = ""
	}
	query := struct {
		PsArgs string `schema:"ps_args"`
	}{
		PsArgs: defaultValue,
	}
	if err := decoder.Decode(&query, r.URL.Query()); err != nil {
		utils.Error(w, http.StatusText(http.StatusBadRequest), http.StatusBadRequest,
			errors.Wrapf(err, "failed to parse parameters for %s", r.URL.String()))
		return
	}

	name := utils.GetName(r)
	c, err := runtime.LookupContainer(name)
	if err != nil {
		utils.ContainerNotFound(w, name, err)
		return
	}

	output, err := c.Top([]string{query.PsArgs})
	if err != nil {
		utils.InternalServerError(w, err)
		return
	}

	var body = handlers.ContainerTopOKBody{}
	if len(output) > 0 {
		body.Titles = strings.Split(output[0], "\t")
		for i := range body.Titles {
			body.Titles[i] = strings.TrimSpace(body.Titles[i])
		}

		for _, line := range output[1:] {
			process := strings.Split(line, "\t")
			for i := range process {
				process[i] = strings.TrimSpace(process[i])
			}
			body.Processes = append(body.Processes, process)
		}
	}
	utils.WriteJSON(w, http.StatusOK, body)
}