mirror of
https://github.com/cupcakearmy/autorestic.git
synced 2025-09-06 02:20:39 +00:00
Compare commits
4 Commits
db6fbd598b
...
master
Author | SHA1 | Date | |
---|---|---|---|
|
9cf919b42b | ||
|
bb29a98614 | ||
|
39f4f87ce3 | ||
|
bd36bbe429 |
@@ -1,4 +1,4 @@
|
|||||||
FROM golang:1.23-alpine as builder
|
FROM golang:1.25-alpine as builder
|
||||||
|
|
||||||
WORKDIR /app
|
WORKDIR /app
|
||||||
COPY go.* .
|
COPY go.* .
|
||||||
|
@@ -6,6 +6,7 @@ import (
|
|||||||
|
|
||||||
"github.com/cupcakearmy/autorestic/internal"
|
"github.com/cupcakearmy/autorestic/internal"
|
||||||
"github.com/cupcakearmy/autorestic/internal/colors"
|
"github.com/cupcakearmy/autorestic/internal/colors"
|
||||||
|
"github.com/cupcakearmy/autorestic/internal/lock"
|
||||||
"github.com/spf13/cobra"
|
"github.com/spf13/cobra"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -14,12 +15,14 @@ var backupCmd = &cobra.Command{
|
|||||||
Short: "Create backups for given locations",
|
Short: "Create backups for given locations",
|
||||||
Run: func(cmd *cobra.Command, args []string) {
|
Run: func(cmd *cobra.Command, args []string) {
|
||||||
internal.GetConfig()
|
internal.GetConfig()
|
||||||
err := internal.Lock()
|
err := lock.Lock()
|
||||||
CheckErr(err)
|
CheckErr(err)
|
||||||
defer internal.Unlock()
|
defer lock.Unlock()
|
||||||
|
dry, _ := cmd.Flags().GetBool("dry-run")
|
||||||
|
|
||||||
selected, err := internal.GetAllOrSelected(cmd, false)
|
selected, err := internal.GetAllOrSelected(cmd, false)
|
||||||
CheckErr(err)
|
CheckErr(err)
|
||||||
|
|
||||||
errors := 0
|
errors := 0
|
||||||
for _, name := range selected {
|
for _, name := range selected {
|
||||||
var splitted = strings.Split(name, "@")
|
var splitted = strings.Split(name, "@")
|
||||||
@@ -28,7 +31,7 @@ var backupCmd = &cobra.Command{
|
|||||||
specificBackend = splitted[1]
|
specificBackend = splitted[1]
|
||||||
}
|
}
|
||||||
location, _ := internal.GetLocation(splitted[0])
|
location, _ := internal.GetLocation(splitted[0])
|
||||||
errs := location.Backup(false, specificBackend)
|
errs := location.Backup(false, dry, specificBackend)
|
||||||
for _, err := range errs {
|
for _, err := range errs {
|
||||||
colors.Error.Printf("%s\n\n", err)
|
colors.Error.Printf("%s\n\n", err)
|
||||||
errors++
|
errors++
|
||||||
@@ -43,4 +46,5 @@ var backupCmd = &cobra.Command{
|
|||||||
func init() {
|
func init() {
|
||||||
rootCmd.AddCommand(backupCmd)
|
rootCmd.AddCommand(backupCmd)
|
||||||
internal.AddFlagsToCommand(backupCmd, false)
|
internal.AddFlagsToCommand(backupCmd, false)
|
||||||
|
backupCmd.Flags().Bool("dry-run", false, "do not write changes, show what would be affected")
|
||||||
}
|
}
|
||||||
|
@@ -3,6 +3,7 @@ package cmd
|
|||||||
import (
|
import (
|
||||||
"github.com/cupcakearmy/autorestic/internal"
|
"github.com/cupcakearmy/autorestic/internal"
|
||||||
"github.com/cupcakearmy/autorestic/internal/colors"
|
"github.com/cupcakearmy/autorestic/internal/colors"
|
||||||
|
"github.com/cupcakearmy/autorestic/internal/lock"
|
||||||
"github.com/spf13/cobra"
|
"github.com/spf13/cobra"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -11,9 +12,9 @@ var checkCmd = &cobra.Command{
|
|||||||
Short: "Check if everything is setup",
|
Short: "Check if everything is setup",
|
||||||
Run: func(cmd *cobra.Command, args []string) {
|
Run: func(cmd *cobra.Command, args []string) {
|
||||||
internal.GetConfig()
|
internal.GetConfig()
|
||||||
err := internal.Lock()
|
err := lock.Lock()
|
||||||
CheckErr(err)
|
CheckErr(err)
|
||||||
defer internal.Unlock()
|
defer lock.Unlock()
|
||||||
|
|
||||||
CheckErr(internal.CheckConfig())
|
CheckErr(internal.CheckConfig())
|
||||||
|
|
||||||
|
@@ -3,6 +3,7 @@ package cmd
|
|||||||
import (
|
import (
|
||||||
"github.com/cupcakearmy/autorestic/internal"
|
"github.com/cupcakearmy/autorestic/internal"
|
||||||
"github.com/cupcakearmy/autorestic/internal/flags"
|
"github.com/cupcakearmy/autorestic/internal/flags"
|
||||||
|
"github.com/cupcakearmy/autorestic/internal/lock"
|
||||||
"github.com/spf13/cobra"
|
"github.com/spf13/cobra"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -12,9 +13,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.`,
|
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) {
|
Run: func(cmd *cobra.Command, args []string) {
|
||||||
internal.GetConfig()
|
internal.GetConfig()
|
||||||
err := internal.Lock()
|
err := lock.Lock()
|
||||||
CheckErr(err)
|
CheckErr(err)
|
||||||
defer internal.Unlock()
|
defer lock.Unlock()
|
||||||
|
|
||||||
err = internal.RunCron()
|
err = internal.RunCron()
|
||||||
CheckErr(err)
|
CheckErr(err)
|
||||||
|
@@ -5,6 +5,7 @@ import (
|
|||||||
|
|
||||||
"github.com/cupcakearmy/autorestic/internal"
|
"github.com/cupcakearmy/autorestic/internal"
|
||||||
"github.com/cupcakearmy/autorestic/internal/colors"
|
"github.com/cupcakearmy/autorestic/internal/colors"
|
||||||
|
"github.com/cupcakearmy/autorestic/internal/lock"
|
||||||
"github.com/spf13/cobra"
|
"github.com/spf13/cobra"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -13,9 +14,9 @@ var execCmd = &cobra.Command{
|
|||||||
Short: "Execute arbitrary native restic commands for given backends",
|
Short: "Execute arbitrary native restic commands for given backends",
|
||||||
Run: func(cmd *cobra.Command, args []string) {
|
Run: func(cmd *cobra.Command, args []string) {
|
||||||
internal.GetConfig()
|
internal.GetConfig()
|
||||||
err := internal.Lock()
|
err := lock.Lock()
|
||||||
CheckErr(err)
|
CheckErr(err)
|
||||||
defer internal.Unlock()
|
defer lock.Unlock()
|
||||||
|
|
||||||
selected, err := internal.GetAllOrSelected(cmd, true)
|
selected, err := internal.GetAllOrSelected(cmd, true)
|
||||||
CheckErr(err)
|
CheckErr(err)
|
||||||
|
@@ -2,6 +2,7 @@ package cmd
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/cupcakearmy/autorestic/internal"
|
"github.com/cupcakearmy/autorestic/internal"
|
||||||
|
"github.com/cupcakearmy/autorestic/internal/lock"
|
||||||
"github.com/spf13/cobra"
|
"github.com/spf13/cobra"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -10,9 +11,9 @@ var forgetCmd = &cobra.Command{
|
|||||||
Short: "Forget and optionally prune snapshots according the specified policies",
|
Short: "Forget and optionally prune snapshots according the specified policies",
|
||||||
Run: func(cmd *cobra.Command, args []string) {
|
Run: func(cmd *cobra.Command, args []string) {
|
||||||
internal.GetConfig()
|
internal.GetConfig()
|
||||||
err := internal.Lock()
|
err := lock.Lock()
|
||||||
CheckErr(err)
|
CheckErr(err)
|
||||||
defer internal.Unlock()
|
defer lock.Unlock()
|
||||||
|
|
||||||
selected, err := internal.GetAllOrSelected(cmd, false)
|
selected, err := internal.GetAllOrSelected(cmd, false)
|
||||||
CheckErr(err)
|
CheckErr(err)
|
||||||
|
@@ -4,6 +4,7 @@ import (
|
|||||||
"fmt"
|
"fmt"
|
||||||
|
|
||||||
"github.com/cupcakearmy/autorestic/internal"
|
"github.com/cupcakearmy/autorestic/internal"
|
||||||
|
"github.com/cupcakearmy/autorestic/internal/lock"
|
||||||
"github.com/spf13/cobra"
|
"github.com/spf13/cobra"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -13,9 +14,9 @@ var restoreCmd = &cobra.Command{
|
|||||||
Args: cobra.MaximumNArgs(1),
|
Args: cobra.MaximumNArgs(1),
|
||||||
Run: func(cmd *cobra.Command, args []string) {
|
Run: func(cmd *cobra.Command, args []string) {
|
||||||
internal.GetConfig()
|
internal.GetConfig()
|
||||||
err := internal.Lock()
|
err := lock.Lock()
|
||||||
CheckErr(err)
|
CheckErr(err)
|
||||||
defer internal.Unlock()
|
defer lock.Unlock()
|
||||||
|
|
||||||
location, _ := cmd.Flags().GetString("location")
|
location, _ := cmd.Flags().GetString("location")
|
||||||
l, ok := internal.GetLocation(location)
|
l, ok := internal.GetLocation(location)
|
||||||
|
@@ -8,6 +8,7 @@ import (
|
|||||||
"github.com/cupcakearmy/autorestic/internal"
|
"github.com/cupcakearmy/autorestic/internal"
|
||||||
"github.com/cupcakearmy/autorestic/internal/colors"
|
"github.com/cupcakearmy/autorestic/internal/colors"
|
||||||
"github.com/cupcakearmy/autorestic/internal/flags"
|
"github.com/cupcakearmy/autorestic/internal/flags"
|
||||||
|
"github.com/cupcakearmy/autorestic/internal/lock"
|
||||||
"github.com/spf13/cobra"
|
"github.com/spf13/cobra"
|
||||||
|
|
||||||
homedir "github.com/mitchellh/go-homedir"
|
homedir "github.com/mitchellh/go-homedir"
|
||||||
@@ -17,7 +18,7 @@ import (
|
|||||||
func CheckErr(err error) {
|
func CheckErr(err error) {
|
||||||
if err != nil {
|
if err != nil {
|
||||||
colors.Error.Fprintln(os.Stderr, "Error:", err)
|
colors.Error.Fprintln(os.Stderr, "Error:", err)
|
||||||
internal.Unlock()
|
lock.Unlock()
|
||||||
os.Exit(1)
|
os.Exit(1)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -41,7 +42,6 @@ func init() {
|
|||||||
rootCmd.PersistentFlags().BoolVarP(&flags.VERBOSE, "verbose", "v", false, "verbose mode")
|
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.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.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)
|
cobra.OnInitialize(initConfig)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@@ -9,6 +9,7 @@ import (
|
|||||||
|
|
||||||
"github.com/cupcakearmy/autorestic/internal"
|
"github.com/cupcakearmy/autorestic/internal"
|
||||||
"github.com/cupcakearmy/autorestic/internal/colors"
|
"github.com/cupcakearmy/autorestic/internal/colors"
|
||||||
|
"github.com/cupcakearmy/autorestic/internal/lock"
|
||||||
"github.com/spf13/cobra"
|
"github.com/spf13/cobra"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -32,7 +33,7 @@ To check you can run "ps aux | grep autorestic".`,
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
err := internal.Unlock()
|
err := lock.Unlock()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
colors.Error.Println("Could not unlock:", err)
|
colors.Error.Println("Could not unlock:", err)
|
||||||
return
|
return
|
||||||
|
@@ -1,11 +1,14 @@
|
|||||||
# Backup
|
# Backup
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
autorestic backup [-l, --location] [-a, --all]
|
autorestic backup [-l, --location] [-a, --all] [--dry-run]
|
||||||
```
|
```
|
||||||
|
|
||||||
Performs a backup of all locations if the `-a` flag is passed. To only backup some locations pass one or more `-l` or `--location` flags.
|
Performs a backup of all locations if the `-a` flag is passed. To only backup some locations pass one or more `-l` or `--location` flags.
|
||||||
|
|
||||||
|
The `--dry-run` flag will do a dry run showing what would have been deleted, but won't touch the actual data.
|
||||||
|
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
# All
|
# All
|
||||||
autorestic backup -a
|
autorestic backup -a
|
||||||
|
@@ -34,11 +34,3 @@ With `--restic-bin` you can specify to run a specific restic binary. This can be
|
|||||||
```bash
|
```bash
|
||||||
autorestic --restic-bin /some/path/to/my/custom/restic/binary
|
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
|
|
||||||
```
|
|
||||||
|
@@ -10,6 +10,7 @@ import (
|
|||||||
|
|
||||||
"github.com/cupcakearmy/autorestic/internal/colors"
|
"github.com/cupcakearmy/autorestic/internal/colors"
|
||||||
"github.com/cupcakearmy/autorestic/internal/flags"
|
"github.com/cupcakearmy/autorestic/internal/flags"
|
||||||
|
"github.com/cupcakearmy/autorestic/internal/lock"
|
||||||
"github.com/joho/godotenv"
|
"github.com/joho/godotenv"
|
||||||
"github.com/mitchellh/go-homedir"
|
"github.com/mitchellh/go-homedir"
|
||||||
"github.com/spf13/cobra"
|
"github.com/spf13/cobra"
|
||||||
@@ -23,7 +24,6 @@ type Options map[string]OptionMap
|
|||||||
|
|
||||||
type Config struct {
|
type Config struct {
|
||||||
Version string `mapstructure:"version" yaml:"version"`
|
Version string `mapstructure:"version" yaml:"version"`
|
||||||
Lockfile string `mapstructure:"lockfile,omitempty" yaml:"lockfile,omitempty"`
|
|
||||||
Extras interface{} `mapstructure:"extras" yaml:"extras"`
|
Extras interface{} `mapstructure:"extras" yaml:"extras"`
|
||||||
Locations map[string]Location `mapstructure:"locations" yaml:"locations"`
|
Locations map[string]Location `mapstructure:"locations" yaml:"locations"`
|
||||||
Backends map[string]Backend `mapstructure:"backends" yaml:"backends"`
|
Backends map[string]Backend `mapstructure:"backends" yaml:"backends"`
|
||||||
@@ -40,7 +40,7 @@ func exitConfig(err error, msg string) {
|
|||||||
if msg != "" {
|
if msg != "" {
|
||||||
colors.Error.Println(msg)
|
colors.Error.Println(msg)
|
||||||
}
|
}
|
||||||
Unlock()
|
lock.Unlock()
|
||||||
os.Exit(1)
|
os.Exit(1)
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -188,15 +188,31 @@ func CheckConfig() error {
|
|||||||
if !CheckIfResticIsCallable() {
|
if !CheckIfResticIsCallable() {
|
||||||
return fmt.Errorf(`%s was not found. Install either with "autorestic install" or manually`, flags.RESTIC_BIN)
|
return fmt.Errorf(`%s was not found. Install either with "autorestic install" or manually`, flags.RESTIC_BIN)
|
||||||
}
|
}
|
||||||
for name, backend := range c.Backends {
|
|
||||||
backend.name = name
|
cwd, _ := GetPathRelativeToConfig(".")
|
||||||
if err := backend.validate(); err != nil {
|
for name, location := range c.Locations {
|
||||||
|
location.name = name
|
||||||
|
|
||||||
|
// Hooks before location validation
|
||||||
|
options := ExecuteOptions{
|
||||||
|
Command: "bash",
|
||||||
|
Dir: cwd,
|
||||||
|
Envs: map[string]string{
|
||||||
|
"AUTORESTIC_LOCATION": location.name,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
if err := location.ExecuteHooks(location.Hooks.PreValidate, options); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := location.validate(); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
for name, location := range c.Locations {
|
|
||||||
location.name = name
|
for name, backend := range c.Backends {
|
||||||
if err := location.validate(); err != nil {
|
backend.name = name
|
||||||
|
if err := backend.validate(); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@@ -1,10 +1,9 @@
|
|||||||
package flags
|
package flags
|
||||||
|
|
||||||
var (
|
var (
|
||||||
CI bool = false
|
CI bool = false
|
||||||
VERBOSE bool = false
|
VERBOSE bool = false
|
||||||
CRON_LEAN bool = false
|
CRON_LEAN bool = false
|
||||||
RESTIC_BIN string
|
RESTIC_BIN string
|
||||||
DOCKER_IMAGE string
|
DOCKER_IMAGE string
|
||||||
LOCKFILE_PATH string
|
|
||||||
)
|
)
|
||||||
|
@@ -11,6 +11,7 @@ import (
|
|||||||
|
|
||||||
"github.com/cupcakearmy/autorestic/internal/colors"
|
"github.com/cupcakearmy/autorestic/internal/colors"
|
||||||
"github.com/cupcakearmy/autorestic/internal/flags"
|
"github.com/cupcakearmy/autorestic/internal/flags"
|
||||||
|
"github.com/cupcakearmy/autorestic/internal/lock"
|
||||||
"github.com/cupcakearmy/autorestic/internal/metadata"
|
"github.com/cupcakearmy/autorestic/internal/metadata"
|
||||||
"github.com/robfig/cron"
|
"github.com/robfig/cron"
|
||||||
)
|
)
|
||||||
@@ -167,7 +168,7 @@ func (l Location) getLocationTags() string {
|
|||||||
return buildTag("location", l.name)
|
return buildTag("location", l.name)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (l Location) Backup(cron bool, specificBackend string) []error {
|
func (l Location) Backup(cron bool, dry bool, specificBackend string) []error {
|
||||||
var errors []error
|
var errors []error
|
||||||
var backends []string
|
var backends []string
|
||||||
colors.PrimaryPrint(" Backing up location \"%s\" ", l.name)
|
colors.PrimaryPrint(" Backing up location \"%s\" ", l.name)
|
||||||
@@ -227,6 +228,9 @@ func (l Location) Backup(cron bool, specificBackend string) []error {
|
|||||||
if cron {
|
if cron {
|
||||||
cmd = append(cmd, "--tag", buildTag("cron"))
|
cmd = append(cmd, "--tag", buildTag("cron"))
|
||||||
}
|
}
|
||||||
|
if dry {
|
||||||
|
cmd = append(cmd, "--dry-run")
|
||||||
|
}
|
||||||
cmd = append(cmd, "--tag", l.getLocationTags())
|
cmd = append(cmd, "--tag", l.getLocationTags())
|
||||||
backupOptions := ExecuteOptions{
|
backupOptions := ExecuteOptions{
|
||||||
Envs: env,
|
Envs: env,
|
||||||
@@ -441,12 +445,12 @@ func (l Location) RunCron() error {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
last := time.Unix(GetCron(l.name), 0)
|
last := time.Unix(lock.GetCron(l.name), 0)
|
||||||
next := schedule.Next(last)
|
next := schedule.Next(last)
|
||||||
now := time.Now()
|
now := time.Now()
|
||||||
if now.After(next) {
|
if now.After(next) {
|
||||||
SetCron(l.name, now.Unix())
|
lock.SetCron(l.name, now.Unix())
|
||||||
errs := l.Backup(true, "")
|
errs := l.Backup(true, false, "")
|
||||||
if len(errs) > 0 {
|
if len(errs) > 0 {
|
||||||
return fmt.Errorf("Failed to backup location \"%s\":\n%w", l.name, errors.Join(errs...))
|
return fmt.Errorf("Failed to backup location \"%s\":\n%w", l.name, errors.Join(errs...))
|
||||||
}
|
}
|
||||||
|
@@ -1,9 +1,8 @@
|
|||||||
package internal
|
package lock
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"os"
|
"os"
|
||||||
"path"
|
"path"
|
||||||
"path/filepath"
|
|
||||||
"sync"
|
"sync"
|
||||||
|
|
||||||
"github.com/cupcakearmy/autorestic/internal/colors"
|
"github.com/cupcakearmy/autorestic/internal/colors"
|
||||||
@@ -13,47 +12,24 @@ import (
|
|||||||
|
|
||||||
var lock *viper.Viper
|
var lock *viper.Viper
|
||||||
var file string
|
var file string
|
||||||
var lockOnce sync.Once
|
var once sync.Once
|
||||||
|
|
||||||
const (
|
const (
|
||||||
RUNNING = "running"
|
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 {
|
func getLock() *viper.Viper {
|
||||||
if lock == nil {
|
if lock == nil {
|
||||||
lockOnce.Do(func() {
|
|
||||||
|
once.Do(func() {
|
||||||
lock = viper.New()
|
lock = viper.New()
|
||||||
lock.SetDefault("running", false)
|
lock.SetDefault("running", false)
|
||||||
file = getLockfilePath()
|
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")
|
||||||
if !flags.CRON_LEAN {
|
if !flags.CRON_LEAN {
|
||||||
colors.Faint.Println("Using lock:\t", file)
|
colors.Faint.Println("Using lock:\t", file)
|
||||||
}
|
}
|
@@ -1,4 +1,4 @@
|
|||||||
package internal
|
package lock
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"log"
|
"log"
|
||||||
@@ -7,7 +7,6 @@ import (
|
|||||||
"strconv"
|
"strconv"
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
"github.com/cupcakearmy/autorestic/internal/flags"
|
|
||||||
"github.com/spf13/viper"
|
"github.com/spf13/viper"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -29,37 +28,6 @@ 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) {
|
func TestLock(t *testing.T) {
|
||||||
setup(t)
|
setup(t)
|
||||||
|
|
4
main.go
4
main.go
@@ -22,7 +22,7 @@ import (
|
|||||||
"syscall"
|
"syscall"
|
||||||
|
|
||||||
"github.com/cupcakearmy/autorestic/cmd"
|
"github.com/cupcakearmy/autorestic/cmd"
|
||||||
"github.com/cupcakearmy/autorestic/internal"
|
"github.com/cupcakearmy/autorestic/internal/lock"
|
||||||
)
|
)
|
||||||
|
|
||||||
func handleCtrlC() {
|
func handleCtrlC() {
|
||||||
@@ -31,7 +31,7 @@ func handleCtrlC() {
|
|||||||
go func() {
|
go func() {
|
||||||
sig := <-c
|
sig := <-c
|
||||||
fmt.Println("Signal:", sig)
|
fmt.Println("Signal:", sig)
|
||||||
internal.Unlock()
|
lock.Unlock()
|
||||||
os.Exit(0)
|
os.Exit(0)
|
||||||
}()
|
}()
|
||||||
}
|
}
|
||||||
|
Reference in New Issue
Block a user