adfinis-rclone-mgr/utils.go at v2 · adfinis/adfinis-rclone-mgr · 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
package main

import (
	"fmt"
	"log"
	"os"
	"path"
	"strings"

	"github.com/adrg/xdg"
)

// dont ask...
func sanitizeDriveName(name string) string {
	name = strings.TrimSpace(name)
	name = strings.ReplaceAll(name, " ", "_")
	name = strings.ReplaceAll(name, "/", "_")
	name = strings.ReplaceAll(name, "\\", "_")
	name = strings.ReplaceAll(name, ":", "_")
	name = strings.ReplaceAll(name, "?", "_")
	name = strings.ReplaceAll(name, "*", "_")
	name = strings.ReplaceAll(name, "\"", "_")
	name = strings.ReplaceAll(name, "<", "_")
	name = strings.ReplaceAll(name, ">", "_")
	name = strings.ReplaceAll(name, "|", "_")
	name = strings.ReplaceAll(name, "'", "_")
	name = strings.ReplaceAll(name, "&", "_")
	name = strings.ReplaceAll(name, "%", "_")
	return name
}

func ensureFolderExists(path string) error {
	err := os.MkdirAll(path, os.ModePerm)
	if err != nil && !os.IsExist(err) {
		return fmt.Errorf("failed to create directory %s: %w", path, err)
	}
	return nil
}

func unitNameToDriveName(name string) string {
	name = strings.Split(name, "@")[1]
	name = strings.Split(name, ".service")[0]
	return name
}

func driveNameToUnitName(name string) string {
	return fmt.Sprintf("rclone@%s.service", name)
}

func getGooglePath() string {
	return path.Join(xdg.Home, "google")
}

func getDriveDataPath(name string) string {
	return path.Join(getGooglePath(), name)
}

func getDriveCachePath(name string) string {
	return path.Join(xdg.CacheHome, "google", name)
}

func fileNameToPath(driveName, fileName string) string {
	return path.Join(getDriveDataPath(driveName), fileName)
}

func isDir(p string) bool {
	info, err := os.Stat(p)
	if err != nil {
		log.Println(err)
		return false
	}
	return info.IsDir()
}