mirror of
https://github.com/octoleo/restic.git
synced 2024-12-22 10:58:55 +00:00
141 lines
3.5 KiB
Go
141 lines
3.5 KiB
Go
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"bytes"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"runtime"
|
|
godebug "runtime/debug"
|
|
|
|
"github.com/spf13/cobra"
|
|
"go.uber.org/automaxprocs/maxprocs"
|
|
|
|
"github.com/restic/restic/internal/debug"
|
|
"github.com/restic/restic/internal/errors"
|
|
"github.com/restic/restic/internal/options"
|
|
"github.com/restic/restic/internal/restic"
|
|
)
|
|
|
|
func init() {
|
|
// don't import `go.uber.org/automaxprocs` to disable the log output
|
|
_, _ = maxprocs.Set()
|
|
}
|
|
|
|
// cmdRoot is the base command when no other command has been specified.
|
|
var cmdRoot = &cobra.Command{
|
|
Use: "restic",
|
|
Short: "Backup and restore files",
|
|
Long: `
|
|
restic is a backup program which allows saving multiple revisions of files and
|
|
directories in an encrypted repository stored on different backends.
|
|
|
|
The full documentation can be found at https://restic.readthedocs.io/ .
|
|
`,
|
|
SilenceErrors: true,
|
|
SilenceUsage: true,
|
|
DisableAutoGenTag: true,
|
|
|
|
PersistentPreRunE: func(c *cobra.Command, args []string) error {
|
|
// set verbosity, default is one
|
|
globalOptions.verbosity = 1
|
|
if globalOptions.Quiet && globalOptions.Verbose > 0 {
|
|
return errors.Fatal("--quiet and --verbose cannot be specified at the same time")
|
|
}
|
|
|
|
switch {
|
|
case globalOptions.Verbose >= 2:
|
|
globalOptions.verbosity = 3
|
|
case globalOptions.Verbose > 0:
|
|
globalOptions.verbosity = 2
|
|
case globalOptions.Quiet:
|
|
globalOptions.verbosity = 0
|
|
}
|
|
|
|
// parse extended options
|
|
opts, err := options.Parse(globalOptions.Options)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
globalOptions.extended = opts
|
|
if !needsPassword(c.Name()) {
|
|
return nil
|
|
}
|
|
pwd, err := resolvePassword(globalOptions, "RESTIC_PASSWORD")
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "Resolving password failed: %v\n", err)
|
|
Exit(1)
|
|
}
|
|
globalOptions.password = pwd
|
|
|
|
// run the debug functions for all subcommands (if build tag "debug" is
|
|
// enabled)
|
|
return runDebug()
|
|
},
|
|
}
|
|
|
|
// Distinguish commands that need the password from those that work without,
|
|
// so we don't run $RESTIC_PASSWORD_COMMAND for no reason (it might prompt the
|
|
// user for authentication).
|
|
func needsPassword(cmd string) bool {
|
|
switch cmd {
|
|
case "cache", "generate", "help", "options", "self-update", "version", "__complete":
|
|
return false
|
|
default:
|
|
return true
|
|
}
|
|
}
|
|
|
|
var logBuffer = bytes.NewBuffer(nil)
|
|
|
|
func tweakGoGC() {
|
|
// lower GOGC from 100 to 50, unless it was manually overwritten by the user
|
|
oldValue := godebug.SetGCPercent(50)
|
|
if oldValue != 100 {
|
|
godebug.SetGCPercent(oldValue)
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
tweakGoGC()
|
|
// install custom global logger into a buffer, if an error occurs
|
|
// we can show the logs
|
|
log.SetOutput(logBuffer)
|
|
|
|
debug.Log("main %#v", os.Args)
|
|
debug.Log("restic %s compiled with %v on %v/%v",
|
|
version, runtime.Version(), runtime.GOOS, runtime.GOARCH)
|
|
err := cmdRoot.ExecuteContext(internalGlobalCtx)
|
|
|
|
switch {
|
|
case restic.IsAlreadyLocked(err):
|
|
fmt.Fprintf(os.Stderr, "%v\nthe `unlock` command can be used to remove stale locks\n", err)
|
|
case err == ErrInvalidSourceData:
|
|
fmt.Fprintf(os.Stderr, "Warning: %v\n", err)
|
|
case errors.IsFatal(err):
|
|
fmt.Fprintf(os.Stderr, "%v\n", err)
|
|
case err != nil:
|
|
fmt.Fprintf(os.Stderr, "%+v\n", err)
|
|
|
|
if logBuffer.Len() > 0 {
|
|
fmt.Fprintf(os.Stderr, "also, the following messages were logged by a library:\n")
|
|
sc := bufio.NewScanner(logBuffer)
|
|
for sc.Scan() {
|
|
fmt.Fprintln(os.Stderr, sc.Text())
|
|
}
|
|
}
|
|
}
|
|
|
|
var exitCode int
|
|
switch err {
|
|
case nil:
|
|
exitCode = 0
|
|
case ErrInvalidSourceData:
|
|
exitCode = 3
|
|
default:
|
|
exitCode = 1
|
|
}
|
|
Exit(exitCode)
|
|
}
|