mstore/cmd/mstorectl/imagecmd.go

/*
 * Copyright 2026 Oleg Borodin <onborodin@gmail.com>
 *
 * This work is published and licensed under a Creative Commons
 * Attribution-NonCommercial-NoDerivatives 4.0 International License.
 *
 * Distribution of this work is permitted, but commercial use and
 * modifications are strictly prohibited.
 */
package main

import (
	"context"
	"net/url"
	"path"
	"strings"
	"time"

	"mstore/pkg/client"

	"github.com/spf13/cobra"
)

func packUserinfo(resurseuri, username, password string) (string, error) {
	var err error
	var res string
	if !strings.Contains(resurseuri, "://") {
		resurseuri = "https://" + resurseuri
	}
	uri, err := url.Parse(resurseuri)
	if err != nil {
		return res, err
	}
	uri.Path = path.Clean(uri.Path)
	if username != "" && password != "" {
		uri.User = url.UserPassword(username, password)
	}
	res = uri.String()
	return res, err
}

func (util *ImageUtil) CreateImageCmds() *cobra.Command {
	const defaultTimeout uint64 = 30 // Second

	var subCmd = &cobra.Command{
		Use:     "images",
		Short:   "Image operation",
		Aliases: []string{"image"},
	}
	subCmd.PersistentFlags().StringVarP(&util.commonImageParams.Username, "user", "u", "", "Username")
	subCmd.PersistentFlags().StringVarP(&util.commonImageParams.Password, "pass", "p", "", "Password")
	subCmd.PersistentFlags().Uint64VarP(&util.commonImageParams.Timeout, "timeout", "t", defaultTimeout, "Operation timeout")
	subCmd.MarkFlagsRequiredTogether("user", "pass")

	// PushImage
	var pushImageCmd = &cobra.Command{
		Use:   "push",
		Short: "Push container image from local tar file into registry",
		Run:   util.PushImage,
	}
	pushImageCmd.Flags().StringVarP(&util.pushImageParams.Imagepath, "image", "i", "", "Remote image path")
	pushImageCmd.Flags().StringVarP(&util.pushImageParams.Filepath, "file", "f", "", "Local tar file path")
	pushImageCmd.MarkFlagRequired("image")
	pushImageCmd.MarkFlagRequired("file")

	subCmd.AddCommand(pushImageCmd)

	// ImageInfo
	var imageInfoCmd = &cobra.Command{
		Use:   "info",
		Short: "Show container image info",
		Run:   util.ImageInfo,
	}
	imageInfoCmd.Flags().StringVarP(&util.imageInfoParams.Imagepath, "image", "i", "", "Remote image path")
	imageInfoCmd.MarkFlagRequired("image")

	subCmd.AddCommand(imageInfoCmd)

	// PullImage
	var pullImageCmd = &cobra.Command{
		Use:   "pull",
		Short: "Pull container image from registry into local file",
		Run:   util.PullImage,
	}
	pullImageCmd.Flags().StringVarP(&util.pullImageParams.Imagepath, "image", "i", "", "Remote image path")
	pullImageCmd.Flags().StringVarP(&util.pullImageParams.Filepath, "file", "f", "", "Local file path")
	pullImageCmd.MarkFlagRequired("image")
	pullImageCmd.MarkFlagRequired("file")

	subCmd.AddCommand(pullImageCmd)

	// DeleteImage
	var deleteImageCmd = &cobra.Command{
		Use:   "delete",
		Short: "Delete container image from registry",
		Run:   util.DeleteImage,
	}
	deleteImageCmd.Flags().StringVarP(&util.deleteImageParams.Imagepath, "image", "i", "", "Remote image path")
	deleteImageCmd.MarkFlagRequired("image")

	subCmd.AddCommand(deleteImageCmd)

	return subCmd
}

type ImageUtil struct {
	imageInfoParams   ImageInfoParams
	pullImageParams   PullImageParams
	pushImageParams   PushImageParams
	deleteImageParams DeleteImageParams
	commonImageParams CommonImageParams
}

type CommonImageParams struct {
	Timeout  uint64
	Username string
	Password string
}

// PushImage
type PushImageParams struct {
	Imagepath string
	Filepath  string
}

type PushImageResult struct{}

func (util *ImageUtil) PushImage(cmd *cobra.Command, args []string) {
	res, err := util.pushImage(&util.commonImageParams, &util.pushImageParams)
	printResponse(res, err)
}

func (util *ImageUtil) pushImage(common *CommonImageParams, params *PushImageParams) (*PushImageResult, error) {
	var err error
	ctx := context.Background()
	res := &PushImageResult{}

	cli := client.NewClient()
	timeout := time.Duration(common.Timeout) * time.Second
	params.Imagepath, err = packUserinfo(params.Imagepath, common.Username, common.Password)
	if err != nil {
		return res, err
	}
	ctx, _ = context.WithTimeout(ctx, timeout)
	err = cli.PushImage(ctx, params.Filepath, params.Imagepath)
	if err != nil {
		return res, err
	}
	return res, err
}

// ImageInfo
type ImageInfoParams struct {
	Imagepath string
}

type ImageInfoResult struct {
	ImageInfo *client.ImageDescr `yaml:"imageInfo"`
}

func (util *ImageUtil) ImageInfo(cmd *cobra.Command, args []string) {
	res, err := util.imageInfo(&util.commonImageParams, &util.imageInfoParams)
	printResponse(res, err)
}

func (util *ImageUtil) imageInfo(common *CommonImageParams, params *ImageInfoParams) (*ImageInfoResult, error) {
	var err error
	res := &ImageInfoResult{}
	ctx := context.Background()

	cli := client.NewClient()
	timeout := time.Duration(common.Timeout) * time.Second

	params.Imagepath, err = packUserinfo(params.Imagepath, common.Username, common.Password)
	if err != nil {
		return res, err
	}
	ctx, _ = context.WithTimeout(ctx, timeout)
	opres, err := cli.ImageInfo(ctx, params.Imagepath)
	if err != nil {
		return res, err
	}
	res.ImageInfo = opres
	return res, err
}

// PullImage
type PullImageParams struct {
	Imagepath string
	Filepath  string
}

type PullImageResult struct{}

func (util *ImageUtil) PullImage(cmd *cobra.Command, args []string) {
	res, err := util.pullImage(&util.commonImageParams, &util.pullImageParams)
	printResponse(res, err)
}

func (util *ImageUtil) pullImage(common *CommonImageParams, params *PullImageParams) (*PullImageResult, error) {
	var err error

	ctx := context.Background()
	res := &PullImageResult{}

	cli := client.NewClient()
	timeout := time.Duration(common.Timeout) * time.Second
	params.Imagepath, err = packUserinfo(params.Imagepath, common.Username, common.Password)
	if err != nil {
		return res, err
	}
	ctx, _ = context.WithTimeout(ctx, timeout)
	err = cli.PullImage(ctx, params.Imagepath, params.Filepath)
	if err != nil {
		return res, err
	}
	return res, err
}

// DeleteImage
type DeleteImageParams struct {
	Imagepath string
}

type DeleteImageResult struct {
}

func (util *ImageUtil) DeleteImage(cmd *cobra.Command, args []string) {
	res, err := util.deleteImage(&util.commonImageParams, &util.deleteImageParams)
	printResponse(res, err)
}

func (util *ImageUtil) deleteImage(common *CommonImageParams, params *DeleteImageParams) (*DeleteImageResult, error) {
	var err error
	res := &DeleteImageResult{}
	ctx := context.Background()

	cli := client.NewClient()
	timeout := time.Duration(common.Timeout) * time.Second

	params.Imagepath, err = packUserinfo(params.Imagepath, common.Username, common.Password)
	if err != nil {
		return res, err
	}
	ctx, _ = context.WithTimeout(ctx, timeout)
	err = cli.DeleteImage(ctx, params.Imagepath)
	if err != nil {
		return res, err
	}
	return res, err
}