schollz_croc/cmd/receive/cmd.go

package receive

import (
	"fmt"
	"os"

	log "github.com/sirupsen/logrus"

	"github.com/schollz/croc/v5/src/webrtc/pkg/session/receiver"
	"gopkg.in/urfave/cli.v1"
)

func handler(c *cli.Context) error {
	output := c.String("output")
	if output == "" {
		return fmt.Errorf("output parameter missing")
	}
	f, err := os.OpenFile(output, os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0644)
	if err != nil {
		return err
	}
	defer f.Close()

	sess := receiver.NewWith(receiver.Config{
		Stream: f,
	})
	return sess.Start()
}

// New creates the command
func New() cli.Command {
	log.Traceln("Installing 'receive' command")
	return cli.Command{
		Name:    "receive",
		Aliases: []string{"r"},
		Usage:   "Receive a file",
		Action:  handler,
		Flags: []cli.Flag{
			cli.StringFlag{
				Name:  "output, o",
				Usage: "Output",
			},
		},
	}
}