porter/api/server/handlers/project/images.go

package project

import (
	"net/http"
	"time"

	"connectrpc.com/connect"
	porterv1 "github.com/porter-dev/api-contracts/generated/go/porter/v1"
	"github.com/porter-dev/porter/api/server/handlers"
	"github.com/porter-dev/porter/api/server/shared"
	"github.com/porter-dev/porter/api/server/shared/apierrors"
	"github.com/porter-dev/porter/api/server/shared/config"
	"github.com/porter-dev/porter/api/types"
	"github.com/porter-dev/porter/internal/models"
	"github.com/porter-dev/porter/internal/telemetry"
)

// ImagesHandler serves the /images endpoint
type ImagesHandler struct {
	handlers.PorterHandlerWriter
}

// NewImagesHandler returns a new ImagesHandler
func NewImagesHandler(
	config *config.Config,
	writer shared.ResultWriter,
) *ImagesHandler {
	return &ImagesHandler{
		PorterHandlerWriter: handlers.NewDefaultPorterHandler(config, nil, writer),
	}
}

// ImageArtifact is an instance of an image in an image repository
type ImageArtifact struct {
	Tag       string    `json:"tag"`
	UpdatedAt time.Time `json:"updated_at"`
}

// Image is a representation of a docker image
// To pull one of the image instances, you must use a string of format <image.uri>:<image.artifact.tag>
type Image struct {
	Uri       string          `json:"uri"`
	Artifacts []ImageArtifact `json:"artifacts"`
}

// ImagesReponse is the response payload for the /images endpoint
type ImagesReponse struct {
	Images []Image `json:"images"`
}

// ServeHTTP handles the GET request to retrieve a list of images for a given project
func (p *ImagesHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
	ctx, span := telemetry.NewSpan(r.Context(), "serve-get-images")
	defer span.End()

	project, _ := r.Context().Value(types.ProjectScope).(*models.Project)

	if project.ID == 0 {
		err := telemetry.Error(ctx, span, nil, "project id is 0")
		p.HandleAPIError(w, r, apierrors.NewErrPassThroughToClient(err, http.StatusBadRequest))
		return
	}

	var resp ImagesReponse
	resp.Images = make([]Image, 0)

	imagesReq := connect.NewRequest(&porterv1.ImagesRequest{
		ProjectId: int64(project.ID),
	})
	ccpResp, err := p.Config().ClusterControlPlaneClient.Images(ctx, imagesReq)
	if err != nil {
		err := telemetry.Error(ctx, span, err, "error calling ccp rollback porter app")
		p.HandleAPIError(w, r, apierrors.NewErrPassThroughToClient(err, http.StatusInternalServerError))
		return
	}

	if ccpResp == nil {
		err := telemetry.Error(ctx, span, err, "ccp resp is nil")
		p.HandleAPIError(w, r, apierrors.NewErrPassThroughToClient(err, http.StatusInternalServerError))
		return
	}
	if ccpResp.Msg == nil {
		err := telemetry.Error(ctx, span, err, "ccp resp msg is nil")
		p.HandleAPIError(w, r, apierrors.NewErrPassThroughToClient(err, http.StatusInternalServerError))
		return
	}
	if ccpResp.Msg.Images == nil { // this means that no images were returned
		p.WriteResult(w, r, resp)
		return
	}

	for _, image := range ccpResp.Msg.Images {
		artifacts := make([]ImageArtifact, 0)
		for _, artifact := range image.Artifacts {
			artifacts = append(artifacts, ImageArtifact{
				Tag:       artifact.Tag,
				UpdatedAt: artifact.UpdatedAt.AsTime().UTC(),
			})
		}
		resp.Images = append(resp.Images, Image{
			Uri:       image.Uri,
			Artifacts: artifacts,
		})
	}

	p.WriteResult(w, r, resp)
}