docker-db-backup/config.go

package main

import (
	"log"
	"os"
	"strconv"
	"strings"
	"time"
)

// Backup folder
// Backup interval
// Max backups to keep

type conf struct {
	Folder                string // Backup folder _with_ trailing slash
	fullCurrentBackupPath string
	Schedule              string
	MaxBackups            int
	CompletionWebhookURL  string
	CompressBackups       bool
}

var (
	config   *conf
	dumpTime time.Time
)

const (
	envBackupFolder         = `BACKUP_FOLDER`
	envSchedule             = `BACKUP_SCHEDULE`
	envMax                  = `BACKUP_MAX`
	envCompletionWebhookURL = `BACKUP_COMPLETION_WEBHOOK_URL`
	envCompressBackups      = `BACKUP_COMPRESS`
)

func init() {
	config = &conf{
		Folder:     "/backups/",
		Schedule:   "0 */6 * * *",
		MaxBackups: 12,
	}

	folder, has := os.LookupEnv(envBackupFolder)
	if has {
		if !strings.HasSuffix(folder, "/") {
			folder = folder + "/"
		}

		config.Folder = folder
	}

	schedule, has := os.LookupEnv(envSchedule)
	if has {
		config.Schedule = schedule
	}

	max, has := os.LookupEnv(envMax)
	if has {
		maxBackups, err := strconv.ParseInt(max, 10, 32)
		if err != nil {
			log.Fatalf("Invalid max: %s\n", err)
		}
		config.MaxBackups = int(maxBackups)
	}

	webhookURL, has := os.LookupEnv(envCompletionWebhookURL)
	if has {
		config.CompletionWebhookURL = webhookURL
	}

	compress, has := os.LookupEnv(envCompressBackups)
	if has {
		config.CompressBackups = compress == "1" || compress == "true"
	}
}

func updateFullBackupPath() {
	dumpTime = time.Now()
	config.fullCurrentBackupPath = config.Folder + dumpTime.Format("2006-01-02_15-04-05") + "/"
	err := os.MkdirAll(config.fullCurrentBackupPath, 0744)
	if err != nil {
		log.Fatalf("Could not create backup folder: %s\n", err)
	}
}