git-lfs/commands/command_push.go

package commands

import (
	"fmt"
	"github.com/github/git-media/gitmedia"
	"github.com/github/git-media/gitmediaclient"
	"github.com/spf13/cobra"
	"strings"
)

var (
	pushCmd = &cobra.Command{
		Use:   "push",
		Short: "Push files to the media endpoint.",
		Run:   pushCommand,
	}
)

func pushCommand(cmd *cobra.Command, args []string) {
	q, err := gitmedia.UploadQueue()
	if err != nil {
		Panic(err, "Error setting up the queue")
	}

	q.Walk(func(id string, body []byte) error {
		fileInfo := string(body)
		parts := strings.Split(fileInfo, ":")

		var sha, filename string
		sha = parts[0]
		if len(parts) > 1 {
			filename = parts[1]
		}

		path, err := gitmedia.LocalMediaPath(sha)
		if err == nil {
			err = gitmediaclient.Options(path)
		}
		if err != nil {
			Panic(err, "error uploading file %s/%s", sha, filename)
		}

		err = gitmediaclient.Put(path, filename)
		if err != nil {
			Panic(err, "error uploading file %s/%s", sha, filename)
		}
		fmt.Printf("\n")

		if err := q.Del(id); err != nil {
			Panic(err, "error removing %s from queue", sha)
		}

		return nil
	})
}

func init() {
	RootCmd.AddCommand(pushCmd)
}