korm/dash.go at master · kamalshkeir/korm · 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
package korm

import (
	"embed"
	"os"
	"os/exec"

	"github.com/kamalshkeir/lg"
)

var (
	staticAndTemplatesFS []embed.FS
	statsFuncs           []StatsFunc
)

type StatsFunc struct {
	Name string
	Func func() string
}

func cloneAndMigrateDashboard(staticAndTemplatesEmbeded ...embed.FS) {
	AddDashStats(StatsFunc{
		Name: "Total records",
		Func: statsNbRecords,
	})
	AddDashStats(StatsFunc{
		Name: "Database <span style='color:var(--theme-color)'>" + defaultDB + "</span> size",
		Func: statsDbSize,
	})
	if _, err := os.Stat(assetsDir); err != nil && !embededDashboard {
		// if not generated
		cmd := exec.Command("git", "clone", "https://github.com/"+repoUser+"/"+repoName)
		err := cmd.Run()
		if lg.CheckError(err) {
			return
		}
		err = os.RemoveAll(repoName + "/.git")
		if err != nil {
			lg.ErrorC("unable to delete .git", "repo", repoName, "err", err)
		}
		err = os.Rename(repoName, assetsDir)
		if err != nil {
			lg.ErrorC("unable to rename", "repo", repoName, "err", err)
		}
		lg.Printfs("grdashboard assets cloned\n")
	}

	if len(staticAndTemplatesEmbeded) > 0 {
		staticAndTemplatesFS = staticAndTemplatesEmbeded
		lg.CheckError(serverBus.App().EmbededStatics(staticAndTemplatesEmbeded[0], staticDir, staticUrl))
		err := serverBus.App().EmbededTemplates(staticAndTemplatesEmbeded[1], templatesDir)
		lg.CheckError(err)
	} else {
		lg.CheckError(serverBus.App().LocalStatics(staticDir, staticUrl))
		err := serverBus.App().LocalTemplates(templatesDir)
		lg.CheckError(err)
	}
	err := AutoMigrate[User]("users", defaultDB)
	if lg.CheckError(err) {
		return
	}
	if kanbanUIEnabled {
		err = AutoMigrate[Board]("_boards", defaultDB)
		if lg.CheckError(err) {
			return
		}
		err = AutoMigrate[Task]("_tasks", defaultDB)
		if lg.CheckError(err) {
			return
		}
	}
}

func AddDashStats(fn ...StatsFunc) {
	statsFuncs = append(statsFuncs, fn...)
}

func GetStats() []StatsFunc {
	return statsFuncs
}