g0v0-database-backup/utils.go at main · GooGuTeam/g0v0-database-backup · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
// Handle time formatting and parsing for backup timestamps.
package main

import (
	"os/exec"
	"path/filepath"
	"strings"
	"time"
)

const backupDateLayout = "0601/02" // YYMM/DD

func FormatBackupTime(t time.Time) string {
	return t.Format("20060102_1504")
}

func FormatBackupDatePath(t time.Time) string {
	return t.Format(backupDateLayout)
}

func FormatFullBackupName(t time.Time) string {
	return "db_" + FormatBackupTime(t)
}

func FormatIncrementalBackupName(t time.Time) string {
	return FormatFullBackupName(t) + "_inc"
}

func FormatBackupRelativePath(t time.Time, isIncremental bool) string {
	backupName := FormatFullBackupName(t)
	if isIncremental {
		backupName = FormatIncrementalBackupName(t)
	}
	return filepath.ToSlash(filepath.Join(FormatBackupDatePath(t), backupName))
}

func FormatLegacyBackupRelativePath(t time.Time, isIncremental bool) string {
	if isIncremental {
		return FormatIncrementalBackupName(t)
	}
	return FormatFullBackupName(t)
}

func NormalizeBackupRelativePath(path string) string {
	normalized := filepath.ToSlash(strings.TrimSpace(path))
	normalized = strings.TrimPrefix(normalized, backupPath)
	normalized = strings.TrimPrefix(normalized, strings.TrimPrefix(backupPath, "/"))
	normalized = strings.TrimPrefix(normalized, remoteBackupBasePath)
	normalized = strings.TrimPrefix(normalized, "/")
	return normalized
}

func FormatBackupLocalPath(relativePath string) string {
	return filepath.Join(backupPath, filepath.FromSlash(NormalizeBackupRelativePath(relativePath)))
}

func FormatRemoteBackupPath(relativePath string) string {
	return remoteBackupBasePath + NormalizeBackupRelativePath(relativePath)
}

func FormatFullBackupDir(t time.Time) string {
	return FormatBackupLocalPath(FormatBackupRelativePath(t, false))
}

func FormatIncrementalBackupDir(t time.Time) string {
	return FormatBackupLocalPath(FormatBackupRelativePath(t, true))
}

func BuildRCloneRemotePath(remote string, relativePath string) string {
	return remote + FormatRemoteBackupPath(relativePath)
}

func ResolveBackupRelativePaths(backupName string) []string {
	backupName = strings.TrimSpace(backupName)
	if backupName == "" {
		return nil
	}

	normalized := NormalizeBackupRelativePath(backupName)
	if strings.Contains(normalized, "/") {
		return []string{normalized}
	}

	layouts := []string{
		"db_20060102_1504",
		"db_20060102_1504_inc",
	}
	for _, layout := range layouts {
		backupTime, err := time.Parse(layout, backupName)
		if err == nil {
			isIncremental := strings.HasSuffix(backupName, "_inc")
			return []string{
				FormatBackupRelativePath(backupTime, isIncremental),
				FormatLegacyBackupRelativePath(backupTime, isIncremental),
			}
		}
	}

	return []string{normalized}
}

func RunSubprocess(name string, args ...string) (string, error) {
	cmd := exec.Command(name, args...)
	output, err := cmd.CombinedOutput()
	return string(output), err
}