91 lines
3.1 KiB
Go
91 lines
3.1 KiB
Go
// Copyright (C) 2021 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/spacemonkeygo/monkit/v3"
|
|
"github.com/zeebo/clingy"
|
|
|
|
"storj.io/common/pb"
|
|
_ "storj.io/common/rpc/quic" // include quic connector
|
|
"storj.io/storj/cmd/uplink/ulext"
|
|
"storj.io/uplink/private/piecestore"
|
|
)
|
|
|
|
var mon = monkit.Package()
|
|
|
|
func main() {
|
|
ex := newExternal()
|
|
raiseUlimits()
|
|
ctx := context.Background()
|
|
ctx = withPieceHashAlgorithm(ctx)
|
|
|
|
ok, err := clingy.Environment{
|
|
Name: "uplink",
|
|
Args: os.Args[1:],
|
|
Dynamic: ex.Dynamic,
|
|
Wrap: ex.Wrap,
|
|
}.Run(ctx, func(cmds clingy.Commands) {
|
|
ex.Setup(cmds) // setup ex first so that stdlib flags can consult config
|
|
newStdlibFlags(flag.CommandLine).Setup(cmds)
|
|
commands(cmds, ex)
|
|
})
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "%+v\n", err)
|
|
}
|
|
if !ok || err != nil {
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
func withPieceHashAlgorithm(ctx context.Context) context.Context {
|
|
customHashAlgo := os.Getenv("STORJ_PIECE_HASH_ALGORITHM_EXPERIMENTAL")
|
|
if customHashAlgo != "" {
|
|
var available []string
|
|
for value, name := range pb.PieceHashAlgorithm_name {
|
|
available = append(available, name)
|
|
if name == customHashAlgo {
|
|
return piecestore.WithPieceHashAlgo(ctx, pb.PieceHashAlgorithm(value))
|
|
|
|
}
|
|
}
|
|
panic("Piece hash algorithm is invalid. Available options: " + strings.Join(available, ","))
|
|
}
|
|
return ctx
|
|
}
|
|
|
|
func commands(cmds clingy.Commands, ex ulext.External) {
|
|
cmds.Group("access", "Access related commands", func() {
|
|
cmds.New("create", "Create an access from the satellite UI", newCmdAccessCreate(ex))
|
|
cmds.New("export", "Export an access to a file", newCmdAccessExport(ex))
|
|
cmds.New("import", "Import an existing access", newCmdAccessImport(ex))
|
|
cmds.New("inspect", "Inspect shows verbose details about an access", newCmdAccessInspect(ex))
|
|
cmds.New("list", "List saved accesses", newCmdAccessList(ex))
|
|
cmds.New("register", "Register an access grant for use with a hosted S3 compatible gateway and linksharing", newCmdAccessRegister(ex))
|
|
cmds.New("remove", "Removes an access from local store", newCmdAccessRemove(ex))
|
|
cmds.New("restrict", "Restrict an access", newCmdAccessRestrict(ex))
|
|
cmds.New("revoke", "Revoke an access", newCmdAccessRevoke(ex))
|
|
cmds.New("setup", "Wizard for setting up uplink from satellite UI", newCmdAccessSetup(ex))
|
|
cmds.New("use", "Set default access to use", newCmdAccessUse(ex))
|
|
})
|
|
cmds.New("setup", "Wizard for setting up uplink from satellite UI", newCmdAccessSetup(ex))
|
|
cmds.New("mb", "Create a new bucket", newCmdMb(ex))
|
|
cmds.New("rb", "Remove a bucket bucket", newCmdRb(ex))
|
|
cmds.New("cp", "Copies files or objects into or out of storj", newCmdCp(ex))
|
|
cmds.New("mv", "Moves files or objects", newCmdMv(ex))
|
|
cmds.New("ls", "Lists buckets, prefixes, or objects", newCmdLs(ex))
|
|
cmds.New("rm", "Remove an object", newCmdRm(ex))
|
|
cmds.Group("meta", "Object metadata related commands", func() {
|
|
cmds.New("get", "Get an object's metadata", newCmdMetaGet(ex))
|
|
})
|
|
cmds.New("share", "Shares restricted accesses to objects", newCmdShare(ex))
|
|
cmds.New("version", "Prints version information", newCmdVersion())
|
|
}
|