zrok/cmd/zrok/copy.go

package main

import (
	"fmt"
	"github.com/openziti/zrok/environment"
	"github.com/openziti/zrok/sdk/golang/sdk"
	"github.com/openziti/zrok/tui"
	"github.com/openziti/zrok/util/sync"
	"github.com/spf13/cobra"
	"net/url"
)

func init() {
	rootCmd.AddCommand(newCopyCommand().cmd)
}

type copyCommand struct {
	cmd *cobra.Command
}

func newCopyCommand() *copyCommand {
	cmd := &cobra.Command{
		Use:     "copy <source> [<target>] (<target> defaults to 'file://.`)",
		Short:   "Copy (unidirectional sync) zrok drive contents from <source> to <target> ('http://', 'file://', and 'zrok://' supported)",
		Aliases: []string{"cp"},
		Args:    cobra.RangeArgs(1, 2),
	}
	command := &copyCommand{cmd: cmd}
	cmd.Run = command.run
	return command
}

func (cmd *copyCommand) run(_ *cobra.Command, args []string) {
	sourceUrl, err := url.Parse(args[0])
	if err != nil {
		tui.Error(fmt.Sprintf("invalid source URL '%v'", args[0]), err)
	}
	if sourceUrl.Scheme == "" {
		sourceUrl.Scheme = "file"
	}

	targetStr := "file://."
	if len(args) == 2 {
		targetStr = args[1]
	}
	targetUrl, err := url.Parse(targetStr)
	if err != nil {
		tui.Error(fmt.Sprintf("invalid target URL '%v'", targetStr), err)
	}
	if targetUrl.Scheme == "" {
		targetUrl.Scheme = "file"
	}

	root, err := environment.LoadRoot()
	if err != nil {
		tui.Error("error loading root", err)
	}

	var allocatedAccesses []*sdk.Access
	if sourceUrl.Scheme == "zrok" {
		access, err := sdk.CreateAccess(root, &sdk.AccessRequest{ShareToken: sourceUrl.Host})
		if err != nil {
			tui.Error("error creating access", err)
		}
		allocatedAccesses = append(allocatedAccesses, access)
	}
	if targetUrl.Scheme == "zrok" {
		access, err := sdk.CreateAccess(root, &sdk.AccessRequest{ShareToken: targetUrl.Host})
		if err != nil {
			tui.Error("error creating access", err)
		}
		allocatedAccesses = append(allocatedAccesses, access)
	}
	defer func() {
		for _, access := range allocatedAccesses {
			err := sdk.DeleteAccess(root, access)
			if err != nil {
				tui.Warning("error deleting target access", err)
			}
		}
	}()

	source, err := sync.TargetForURL(sourceUrl, root)
	if err != nil {
		tui.Error("error creating target", err)
	}
	target, err := sync.TargetForURL(targetUrl, root)
	if err != nil {
		tui.Error("error creating target", err)
	}

	if err := sync.Synchronize(source, target); err != nil {
		tui.Error("error copying", err)
	}

	fmt.Println("copy complete!")
}