diff --git a/cmd/backup.go b/cmd/backup.go index a575220..da999a7 100644 --- a/cmd/backup.go +++ b/cmd/backup.go @@ -6,7 +6,6 @@ import ( "github.com/cupcakearmy/autorestic/internal" "github.com/cupcakearmy/autorestic/internal/colors" - "github.com/cupcakearmy/autorestic/internal/lock" "github.com/spf13/cobra" ) @@ -15,9 +14,9 @@ var backupCmd = &cobra.Command{ Short: "Create backups for given locations", Run: func(cmd *cobra.Command, args []string) { internal.GetConfig() - err := lock.Lock() + err := internal.Lock() CheckErr(err) - defer lock.Unlock() + defer internal.Unlock() selected, err := internal.GetAllOrSelected(cmd, false) CheckErr(err) diff --git a/cmd/check.go b/cmd/check.go index 159f341..6d458e4 100644 --- a/cmd/check.go +++ b/cmd/check.go @@ -3,7 +3,6 @@ package cmd import ( "github.com/cupcakearmy/autorestic/internal" "github.com/cupcakearmy/autorestic/internal/colors" - "github.com/cupcakearmy/autorestic/internal/lock" "github.com/spf13/cobra" ) @@ -12,9 +11,9 @@ var checkCmd = &cobra.Command{ Short: "Check if everything is setup", Run: func(cmd *cobra.Command, args []string) { internal.GetConfig() - err := lock.Lock() + err := internal.Lock() CheckErr(err) - defer lock.Unlock() + defer internal.Unlock() CheckErr(internal.CheckConfig()) diff --git a/cmd/cron.go b/cmd/cron.go index 175ae13..be5a4bd 100644 --- a/cmd/cron.go +++ b/cmd/cron.go @@ -3,7 +3,6 @@ package cmd import ( "github.com/cupcakearmy/autorestic/internal" "github.com/cupcakearmy/autorestic/internal/flags" - "github.com/cupcakearmy/autorestic/internal/lock" "github.com/spf13/cobra" ) @@ -13,9 +12,9 @@ var cronCmd = &cobra.Command{ Long: `Intended to be mainly triggered by an automated system like systemd or crontab. For each location checks if a cron backup is due and runs it.`, Run: func(cmd *cobra.Command, args []string) { internal.GetConfig() - err := lock.Lock() + err := internal.Lock() CheckErr(err) - defer lock.Unlock() + defer internal.Unlock() err = internal.RunCron() CheckErr(err) diff --git a/cmd/exec.go b/cmd/exec.go index 107b0e1..b3f5db1 100644 --- a/cmd/exec.go +++ b/cmd/exec.go @@ -5,7 +5,6 @@ import ( "github.com/cupcakearmy/autorestic/internal" "github.com/cupcakearmy/autorestic/internal/colors" - "github.com/cupcakearmy/autorestic/internal/lock" "github.com/spf13/cobra" ) @@ -14,9 +13,9 @@ var execCmd = &cobra.Command{ Short: "Execute arbitrary native restic commands for given backends", Run: func(cmd *cobra.Command, args []string) { internal.GetConfig() - err := lock.Lock() + err := internal.Lock() CheckErr(err) - defer lock.Unlock() + defer internal.Unlock() selected, err := internal.GetAllOrSelected(cmd, true) CheckErr(err) diff --git a/cmd/forget.go b/cmd/forget.go index 93d7137..ba3b594 100644 --- a/cmd/forget.go +++ b/cmd/forget.go @@ -2,7 +2,6 @@ package cmd import ( "github.com/cupcakearmy/autorestic/internal" - "github.com/cupcakearmy/autorestic/internal/lock" "github.com/spf13/cobra" ) @@ -11,9 +10,9 @@ var forgetCmd = &cobra.Command{ Short: "Forget and optionally prune snapshots according the specified policies", Run: func(cmd *cobra.Command, args []string) { internal.GetConfig() - err := lock.Lock() + err := internal.Lock() CheckErr(err) - defer lock.Unlock() + defer internal.Unlock() selected, err := internal.GetAllOrSelected(cmd, false) CheckErr(err) diff --git a/cmd/restore.go b/cmd/restore.go index 8faf4a7..f84b240 100644 --- a/cmd/restore.go +++ b/cmd/restore.go @@ -4,7 +4,6 @@ import ( "fmt" "github.com/cupcakearmy/autorestic/internal" - "github.com/cupcakearmy/autorestic/internal/lock" "github.com/spf13/cobra" ) @@ -14,9 +13,9 @@ var restoreCmd = &cobra.Command{ Args: cobra.MaximumNArgs(1), Run: func(cmd *cobra.Command, args []string) { internal.GetConfig() - err := lock.Lock() + err := internal.Lock() CheckErr(err) - defer lock.Unlock() + defer internal.Unlock() location, _ := cmd.Flags().GetString("location") l, ok := internal.GetLocation(location) diff --git a/cmd/root.go b/cmd/root.go index 3da9e25..4db353e 100644 --- a/cmd/root.go +++ b/cmd/root.go @@ -8,7 +8,6 @@ import ( "github.com/cupcakearmy/autorestic/internal" "github.com/cupcakearmy/autorestic/internal/colors" "github.com/cupcakearmy/autorestic/internal/flags" - "github.com/cupcakearmy/autorestic/internal/lock" "github.com/spf13/cobra" homedir "github.com/mitchellh/go-homedir" @@ -18,7 +17,7 @@ import ( func CheckErr(err error) { if err != nil { colors.Error.Fprintln(os.Stderr, "Error:", err) - lock.Unlock() + internal.Unlock() os.Exit(1) } } @@ -42,6 +41,7 @@ func init() { rootCmd.PersistentFlags().BoolVarP(&flags.VERBOSE, "verbose", "v", false, "verbose mode") rootCmd.PersistentFlags().StringVar(&flags.RESTIC_BIN, "restic-bin", "restic", "specify custom restic binary") rootCmd.PersistentFlags().StringVar(&flags.DOCKER_IMAGE, "docker-image", "cupcakearmy/autorestic:"+internal.VERSION, "specify a custom docker image") + rootCmd.PersistentFlags().StringVar(&flags.LOCKFILE_PATH, "lockfile-path", "", "specify a custom path for the lockfile (defaults to .autorestic.lock.yml next to the loaded autorestic config file)") cobra.OnInitialize(initConfig) } diff --git a/cmd/unlock.go b/cmd/unlock.go index 75aabf5..0b79c8d 100644 --- a/cmd/unlock.go +++ b/cmd/unlock.go @@ -9,7 +9,6 @@ import ( "github.com/cupcakearmy/autorestic/internal" "github.com/cupcakearmy/autorestic/internal/colors" - "github.com/cupcakearmy/autorestic/internal/lock" "github.com/spf13/cobra" ) @@ -33,7 +32,7 @@ To check you can run "ps aux | grep autorestic".`, } } - err := lock.Unlock() + err := internal.Unlock() if err != nil { colors.Error.Println("Could not unlock:", err) return diff --git a/docs/pages/cli/general.md b/docs/pages/cli/general.md index 9fa16eb..e9fad29 100644 --- a/docs/pages/cli/general.md +++ b/docs/pages/cli/general.md @@ -34,3 +34,11 @@ With `--restic-bin` you can specify to run a specific restic binary. This can be ```bash autorestic --restic-bin /some/path/to/my/custom/restic/binary ``` + +## `--lockfile-path` + +Specify the path for the lockfile used by `autorestic`. If omitted, this will default to `.autorestic.lock.yml` next to the loaded config file. + +```bash +autorestic --lockfile-path /path/to/my/.autorestic.lock.yml +``` diff --git a/internal/config.go b/internal/config.go index c5430d3..f3c63b0 100644 --- a/internal/config.go +++ b/internal/config.go @@ -10,7 +10,6 @@ import ( "github.com/cupcakearmy/autorestic/internal/colors" "github.com/cupcakearmy/autorestic/internal/flags" - "github.com/cupcakearmy/autorestic/internal/lock" "github.com/joho/godotenv" "github.com/mitchellh/go-homedir" "github.com/spf13/cobra" @@ -24,6 +23,7 @@ type Options map[string]OptionMap type Config struct { Version string `mapstructure:"version" yaml:"version"` + Lockfile string `mapstructure:"lockfile,omitempty" yaml:"lockfile,omitempty"` Extras interface{} `mapstructure:"extras" yaml:"extras"` Locations map[string]Location `mapstructure:"locations" yaml:"locations"` Backends map[string]Backend `mapstructure:"backends" yaml:"backends"` @@ -40,7 +40,7 @@ func exitConfig(err error, msg string) { if msg != "" { colors.Error.Println(msg) } - lock.Unlock() + Unlock() os.Exit(1) } diff --git a/internal/flags/flags.go b/internal/flags/flags.go index e03cc25..801aed7 100644 --- a/internal/flags/flags.go +++ b/internal/flags/flags.go @@ -1,9 +1,10 @@ package flags var ( - CI bool = false - VERBOSE bool = false - CRON_LEAN bool = false - RESTIC_BIN string - DOCKER_IMAGE string + CI bool = false + VERBOSE bool = false + CRON_LEAN bool = false + RESTIC_BIN string + DOCKER_IMAGE string + LOCKFILE_PATH string ) diff --git a/internal/location.go b/internal/location.go index 221c5f3..5ba7b56 100644 --- a/internal/location.go +++ b/internal/location.go @@ -11,7 +11,6 @@ import ( "github.com/cupcakearmy/autorestic/internal/colors" "github.com/cupcakearmy/autorestic/internal/flags" - "github.com/cupcakearmy/autorestic/internal/lock" "github.com/cupcakearmy/autorestic/internal/metadata" "github.com/robfig/cron" ) @@ -442,11 +441,11 @@ func (l Location) RunCron() error { if err != nil { return err } - last := time.Unix(lock.GetCron(l.name), 0) + last := time.Unix(GetCron(l.name), 0) next := schedule.Next(last) now := time.Now() if now.After(next) { - lock.SetCron(l.name, now.Unix()) + SetCron(l.name, now.Unix()) errs := l.Backup(true, "") if len(errs) > 0 { return fmt.Errorf("Failed to backup location \"%s\":\n%w", l.name, errors.Join(errs...)) diff --git a/internal/lock/lock.go b/internal/lock.go similarity index 57% rename from internal/lock/lock.go rename to internal/lock.go index 4e9fd52..a128bdf 100644 --- a/internal/lock/lock.go +++ b/internal/lock.go @@ -1,8 +1,9 @@ -package lock +package internal import ( "os" "path" + "path/filepath" "sync" "github.com/cupcakearmy/autorestic/internal/colors" @@ -12,24 +13,47 @@ import ( var lock *viper.Viper var file string -var once sync.Once +var lockOnce sync.Once const ( RUNNING = "running" ) +// getLockfilePath returns the path to the lockfile. The path for the lockfile +// can be sources from multiple places If flags.LOCKFILE_PATH is set, its value +// is used; if the config has the `lockfile` option set, its value is used; +// otherwise the path is generated relative to the config file. +func getLockfilePath() string { + if flags.LOCKFILE_PATH != "" { + abs, err := filepath.Abs(flags.LOCKFILE_PATH) + if err != nil { + return flags.LOCKFILE_PATH + } + return abs + } + + if lockfile := GetConfig().Lockfile; lockfile != "" { + abs, err := filepath.Abs(lockfile) + if err != nil { + return lockfile + } + return abs + } + + p := viper.ConfigFileUsed() + if p == "" { + colors.Error.Println("cannot lock before reading config location") + os.Exit(1) + } + return path.Join(path.Dir(p), ".autorestic.lock.yml") +} + func getLock() *viper.Viper { if lock == nil { - - once.Do(func() { + lockOnce.Do(func() { lock = viper.New() lock.SetDefault("running", false) - p := viper.ConfigFileUsed() - if p == "" { - colors.Error.Println("cannot lock before reading config location") - os.Exit(1) - } - file = path.Join(path.Dir(p), ".autorestic.lock.yml") + file = getLockfilePath() if !flags.CRON_LEAN { colors.Faint.Println("Using lock:\t", file) } diff --git a/internal/lock/lock_test.go b/internal/lock_test.go similarity index 73% rename from internal/lock/lock_test.go rename to internal/lock_test.go index 2b9d9f7..83ea7cc 100644 --- a/internal/lock/lock_test.go +++ b/internal/lock_test.go @@ -1,4 +1,4 @@ -package lock +package internal import ( "log" @@ -7,6 +7,7 @@ import ( "strconv" "testing" + "github.com/cupcakearmy/autorestic/internal/flags" "github.com/spf13/viper" ) @@ -28,6 +29,37 @@ func setup(t *testing.T) { }) } +func TestGetLockfilePath(t *testing.T) { + t.Run("when flags.LOCKFILE_PATH is set", func(t *testing.T) { + flags.LOCKFILE_PATH = "/path/to/my/autorestic.lock.yml" + defer func() { flags.LOCKFILE_PATH = "" }() + + p := getLockfilePath() + + if p != "/path/to/my/autorestic.lock.yml" { + t.Errorf("got %v, want %v", p, "/path/to/my/autorestic.lock.yml") + } + }) + + t.Run("when flags.LOCKFILE_PATH is set", func(t *testing.T) { + d, err := os.MkdirTemp("", testDirectory) + if err != nil { + log.Fatalf("error creating temp dir: %v", err) + return + } + viper.SetConfigFile(d + "/.autorestic.yml") + defer viper.Reset() + + flags.LOCKFILE_PATH = "" + + p := getLockfilePath() + + if p != d+"/.autorestic.lock.yml" { + t.Errorf("got %v, want %v", p, d+"/.autorestic.lock.yml") + } + }) +} + func TestLock(t *testing.T) { setup(t) diff --git a/main.go b/main.go index ae644c3..e0d9ae5 100644 --- a/main.go +++ b/main.go @@ -22,7 +22,7 @@ import ( "syscall" "github.com/cupcakearmy/autorestic/cmd" - "github.com/cupcakearmy/autorestic/internal/lock" + "github.com/cupcakearmy/autorestic/internal" ) func handleCtrlC() { @@ -31,7 +31,7 @@ func handleCtrlC() { go func() { sig := <-c fmt.Println("Signal:", sig) - lock.Unlock() + internal.Unlock() os.Exit(0) }() }