storj/cmd/piecestore/main.go
Alexander Leitner 4b03c08cf6 Added piecestore (#36)
* Added piecestore

* gofmt

* Added requested changes

* Added cli

* Removed ranger because I wanted something that can stand alone

* Changed piecestore code to make it more optial for error handelling

* Added missing package

* Forgot io import

* gofmt

* Make path by hash exported

* updated to simplify again whoops

* Forgot ampersand

* Updated to match FilePiece

* Change store to use io.CopyN

* Clear golinter errors

* Updated go.mod

* Updated go.mod
2018-05-16 06:40:44 -06:00

126 lines
2.6 KiB
Go

// Copyright (C) 2018 Storj Labs, Inc.
// See LICENSE for copying information.
package main
import (
"fmt"
"log"
"os"
"sort"
"github.com/urfave/cli"
"github.com/zeebo/errs"
"storj.io/storj/pkg/piecestore"
)
var argError = errs.Class("argError")
func main() {
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("Missing data Hash")
}
if c.Args().Get(1) == "" {
return argError.New("No input file specified")
}
if c.Args().Get(2) == "" {
return argError.New("No output directory specified")
}
file, err := os.Open(c.Args().Get(1))
if err != nil {
return err
}
// Close the file when we are done
defer file.Close()
fileInfo, err := os.Stat(c.Args().Get(1))
if err != nil {
return err
}
if fileInfo.IsDir() {
return argError.New(fmt.Sprintf("Path (%s) is a directory, not a file", c.Args().Get(1)))
}
_, err = pstore.Store(c.Args().Get(0), file, int64(fileInfo.Size()), 0, c.Args().Get(2))
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)))
}
_, err = pstore.Retrieve(c.Args().Get(0), os.Stdout, -1, 0, c.Args().Get(1))
if err != nil {
return err
}
return nil
},
},
{
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))
err := app.Run(os.Args)
if err != nil {
log.Fatal(err)
}
}