package dockerapi

import (
	"context"
	"errors"
	"net/http"

	"github.com/docker/docker/api/types/container"
	"github.com/docker/docker/pkg/stdcopy"
	"github.com/gin-gonic/gin"
	"github.com/rs/zerolog/log"
	apitypes "github.com/yusing/go-proxy/internal/api/types"
	"github.com/yusing/go-proxy/internal/net/gphttp/websocket"
	"github.com/yusing/go-proxy/internal/task"
)

type LogsPathParams struct {
	Server      string `uri:"server" binding:"required"`
	ContainerID string `uri:"container" binding:"required"`
} //	@name	LogsPathParams

type LogsQueryParams struct {
	Stdout bool   `form:"stdout,default=true"`
	Stderr bool   `form:"stderr,default=true"`
	Since  string `form:"from"`
	Until  string `form:"to"`
	Levels string `form:"levels"`
} //	@name	LogsQueryParams

// @x-id				"logs"
// @BasePath		/api/v1
// @Summary		Get docker container logs
// @Description	Get docker container logs
// @Tags			docker,websocket
// @Accept			json
// @Produce		json
// @Param			server		path	string	true	"server name"
// @Param			container	path	string	true	"container id"
// @Param			stdout		query	bool	false	"show stdout"
// @Param			stderr		query	bool	false	"show stderr"
// @Param			from		query	string	false	"from timestamp"
// @Param			to			query	string	false	"to timestamp"
// @Param			levels		query	string	false	"levels"
// @Success		200
// @Failure		400	{object}	apitypes.ErrorResponse
// @Failure		403	{object}	apitypes.ErrorResponse
// @Failure		404	{object}	apitypes.ErrorResponse
// @Failure		500	{object}	apitypes.ErrorResponse
// @Router			/docker/logs/{server}/{container} [get]
func Logs(c *gin.Context) {
	var pathParams LogsPathParams
	var queryParams LogsQueryParams
	if err := c.ShouldBindQuery(&queryParams); err != nil {
		c.JSON(http.StatusBadRequest, apitypes.Error("invalid query params"))
		return
	}
	if err := c.ShouldBindUri(&pathParams); err != nil {
		c.JSON(http.StatusBadRequest, apitypes.Error("invalid path params"))
		return
	}
	// TODO: implement levels

	dockerClient, found, err := getDockerClient(pathParams.Server)
	if err != nil {
		c.Error(apitypes.InternalServerError(err, "failed to get docker client"))
		return
	}
	if !found {
		c.JSON(http.StatusNotFound, apitypes.Error("server not found"))
		return
	}
	defer dockerClient.Close()

	opts := container.LogsOptions{
		ShowStdout: queryParams.Stdout,
		ShowStderr: queryParams.Stderr,
		Since:      queryParams.Since,
		Until:      queryParams.Until,
		Timestamps: true,
		Follow:     true,
		Tail:       "100",
	}
	if queryParams.Levels != "" {
		opts.Details = true
	}

	logs, err := dockerClient.ContainerLogs(c.Request.Context(), pathParams.ContainerID, opts)
	if err != nil {
		c.Error(apitypes.InternalServerError(err, "failed to get container logs"))
		return
	}
	defer logs.Close()

	manager, err := websocket.NewManagerWithUpgrade(c)
	if err != nil {
		c.Error(apitypes.InternalServerError(err, "failed to create websocket manager"))
		return
	}
	defer manager.Close()

	writer := manager.NewWriter(websocket.TextMessage)

	_, err = stdcopy.StdCopy(writer, writer, logs) // de-multiplex logs
	if err != nil {
		if errors.Is(err, context.Canceled) || errors.Is(err, task.ErrProgramExiting) {
			return
		}
		log.Err(err).
			Str("server", pathParams.Server).
			Str("container", pathParams.ContainerID).
			Msg("failed to de-multiplex logs")
	}
}