storj/cmd/piecestore-cli/main.go

// Copyright (C) 2018 Storj Labs, Inc.
// See LICENSE for copying information.

package main

import (
	"context"
	"flag"
	"fmt"
	"io"
	"os"
	"sort"

	"github.com/urfave/cli"
	"github.com/zeebo/errs"
	"storj.io/storj/pkg/piecestore"
	"storj.io/storj/pkg/process"
)

var argError = errs.Class("argError")

func run(ctx context.Context) error {
	app := cli.NewApp()

	app.Name = "Piece Store CLI"
	app.Usage = "Store data in hash folder structure"
	app.Version = "1.0.0"

	app.Flags = []cli.Flag{}

	app.Commands = []cli.Command{
		{
			Name:      "store",
			Aliases:   []string{"s"},
			Usage:     "Store data by hash",
			ArgsUsage: "[hash] [dataPath] [storeDir]",
			Action: func(c *cli.Context) error {
				if c.Args().Get(0) == "" {
					return argError.New("No input file specified")
				}

				if c.Args().Get(1) == "" {
					return argError.New("No output directory specified")
				}

				file, err := os.Open(c.Args().Get(0))
				if err != nil {
					return err
				}

				// Close the file when we are done
				defer file.Close()

				fileInfo, err := os.Stat(c.Args().Get(0))
				if err != nil {
					return err
				}

				if fileInfo.IsDir() {
					return argError.New(fmt.Sprintf("Path (%s) is a directory, not a file", c.Args().Get(0)))
				}

				id := pstore.DetermineID()

				dataFileChunk, err := pstore.StoreWriter(id, c.Args().Get(1))
				if err != nil {
					return err
				}

				// Close when finished
				defer dataFileChunk.Close()

				_, err = io.Copy(dataFileChunk, file)

				return err
			},
		},
		{
			Name:      "retrieve",
			Aliases:   []string{"r"},
			Usage:     "Retrieve data by hash and print to Stdout",
			ArgsUsage: "[hash] [storeDir]",
			Action: func(c *cli.Context) error {
				if c.Args().Get(0) == "" {
					return argError.New("Missing data Hash")
				}
				if c.Args().Get(1) == "" {
					return argError.New("Missing file path")
				}
				fileInfo, err := os.Stat(c.Args().Get(1))
				if err != nil {
					return err
				}

				if fileInfo.IsDir() != true {
					return argError.New(fmt.Sprintf("Path (%s) is a file, not a directory", c.Args().Get(1)))
				}

				dataFileChunk, err := pstore.RetrieveReader(c.Args().Get(0), 0, -1, c.Args().Get(1))
				if err != nil {
					return err
				}

				// Close when finished
				defer dataFileChunk.Close()

				_, err = io.Copy(os.Stdout, dataFileChunk)
				return err
			},
		},
		{
			Name:      "delete",
			Aliases:   []string{"d"},
			Usage:     "Delete data by hash",
			ArgsUsage: "[hash] [storeDir]",
			Action: func(c *cli.Context) error {
				if c.Args().Get(0) == "" {
					return argError.New("Missing data Hash")
				}
				if c.Args().Get(1) == "" {
					return argError.New("No directory specified")
				}
				err := pstore.Delete(c.Args().Get(0), c.Args().Get(1))

				return err
			},
		},
	}

	sort.Sort(cli.FlagsByName(app.Flags))
	sort.Sort(cli.CommandsByName(app.Commands))

	return app.Run(append([]string{os.Args[0]}, flag.Args()...))
}

func main() { process.Must(process.Main(process.ServiceFunc(run))) }