chore(cmd/backup): split up backup function to helper package

This commit is contained in:
Tom Neuber 2024-07-30 20:43:58 +02:00
parent 1d949f1fb8
commit 0c05b9ee2c
Signed by: tom
GPG key ID: F17EFE4272D89FF6
2 changed files with 112 additions and 61 deletions

View file

@ -2,11 +2,12 @@ package cmd
import (
"context"
"encoding/json"
"errors"
"fmt"
"slices"
"git.ar21.de/yolokube/grafana-backuper/internal/config"
"git.ar21.de/yolokube/grafana-backuper/internal/helper"
"git.ar21.de/yolokube/grafana-backuper/pkg/git"
"git.ar21.de/yolokube/grafana-backuper/pkg/grafana"
"github.com/spf13/cobra"
@ -38,29 +39,15 @@ func backup(ctx context.Context, c *config.Config) error {
grafana.WithToken(c.GrafanaToken),
)
project := git.NewProject(
c.GitRepo,
git.WithBasicAuth(c.GitUser, c.GitPass),
git.WithBranch(c.GitBranch),
git.WithOutputWriter(c.Output),
)
if err := project.Clone(ctx); err != nil {
return err
}
if err := project.Checkout(); err != nil {
return err
}
if err := project.Pull(ctx); err != nil {
project, err := helper.PrepareProject(ctx, c)
if err != nil {
return err
}
var signer git.SignKey
if c.GPGKey != "" {
signer = git.SignKey{KeyFile: c.GPGKey}
if err := signer.ReadKeyFile(); err != nil {
if err = signer.ReadKeyFile(); err != nil {
return err
}
}
@ -85,7 +72,7 @@ func backup(ctx context.Context, c *config.Config) error {
slices.Reverse(versions)
var uncommitedVersion bool
uncommitedVersion := c.ForceCommits
for _, version := range versions {
var dashboardVersion *grafana.DashboardVersion
dashboardVersion, _, err = client.DashboardVersion.Get(ctx, dashboardInfo.UID, version.Version)
@ -93,57 +80,31 @@ func backup(ctx context.Context, c *config.Config) error {
return err
}
dashboard.Dashboard = dashboardVersion.Data
dashboard.UpdatedBy = dashboardVersion.CreatedBy
dashboard.Updated = dashboardVersion.Created
dashboard.Version = dashboardVersion.Version
var data []byte
data, err = json.MarshalIndent(grafana.SchemaFromDashboardMeta(dashboard), "", " ")
if err != nil {
return err
}
commitMsg := fmt.Sprintf(
"%s: Update %s to version %d",
dashboardInfo.Title,
dashboardInfo.UID,
dashboardVersion.ID,
var commitMsg string
commitMsg, err = helper.CommitVersion(
dashboard,
dashboardVersion,
dashboardInfo,
project,
c,
signer,
uncommitedVersion,
)
if dashboardVersion.Message != "" {
commitMsg += fmt.Sprintf(" => %s", dashboardVersion.Message)
}
commitOpts := []git.CommitOption{
git.WithAuthor(dashboardVersion.CreatedBy, ""),
git.WithFileContent(data, dashboardInfo.Title, dashboard.FolderTitle),
git.WithSigner(signer),
}
if c.GitUser != "" {
commitOpts = append(commitOpts, git.WithCommitter(c.GitUser, c.GitEmail))
}
commit := project.NewCommit(commitOpts...)
if commit.Exists(dashboardVersion.DashboardUID, dashboardVersion.ID) && !c.ForceCommits && !uncommitedVersion {
fmt.Fprintf(c.Output, "%s -> already committed\n", commitMsg)
if errors.Is(err, helper.ErrAlreadyCommited) {
fmt.Fprintf(c.Output, "%s -> %v\n", commitMsg, err)
continue
} else if err != nil {
return err
}
uncommitedVersion = true
if err = commit.Create(commitMsg); err != nil {
return err
}
fmt.Fprintln(c.Output, commitMsg)
}
}
if project.HasChanges() {
if err = project.Push(ctx); err != nil {
return err
}
}
if !project.HasChanges() {
return nil
}
return project.Push(ctx)
}

90
internal/helper/helper.go Normal file
View file

@ -0,0 +1,90 @@
package helper
import (
"context"
"encoding/json"
"errors"
"fmt"
"git.ar21.de/yolokube/grafana-backuper/internal/config"
"git.ar21.de/yolokube/grafana-backuper/pkg/git"
"git.ar21.de/yolokube/grafana-backuper/pkg/grafana"
)
var ErrAlreadyCommited = errors.New("already committed")
func CreateCommitMsg(d *grafana.DashboardVersion, title, uid string) string {
commitMsg := fmt.Sprintf("%s: Update %s to version %d", title, uid, d.ID)
if d.Message != "" {
commitMsg += fmt.Sprintf(" => %s", d.Message)
}
return commitMsg
}
func CommitVersion(
dashboard *grafana.Dashboard,
dashboardVersion *grafana.DashboardVersion,
dashboardInfo *grafana.SearchResult,
project *git.Project,
cfg *config.Config,
signer git.SignKey,
force bool,
) (string, error) {
commitMsg := CreateCommitMsg(dashboardVersion, dashboardInfo.Title, dashboardInfo.UID)
if !force && project.CommitExists(dashboardVersion.DashboardUID, dashboardVersion.ID) {
return commitMsg, ErrAlreadyCommited
}
UpdateDashboardInfo(dashboard, dashboardVersion)
data, err := json.MarshalIndent(grafana.SchemaFromDashboardMeta(dashboard), "", " ")
if err != nil {
return commitMsg, err
}
commitOpts := []git.CommitOption{
git.WithAuthor(dashboardVersion.CreatedBy, ""),
git.WithFileContent(data, dashboardInfo.Title, dashboard.FolderTitle),
git.WithSigner(signer),
}
if cfg.GitUser != "" {
commitOpts = append(commitOpts, git.WithCommitter(cfg.GitUser, cfg.GitEmail))
}
commit := project.NewCommit(commitOpts...)
return commitMsg, commit.Create(commitMsg)
}
func PrepareProject(ctx context.Context, c *config.Config) (*git.Project, error) {
project := git.NewProject(
c.GitRepo,
git.WithBasicAuth(c.GitUser, c.GitPass),
git.WithBranch(c.GitBranch),
git.WithOutputWriter(c.Output),
)
if err := project.Clone(ctx); err != nil {
return nil, err
}
if err := project.Checkout(); err != nil {
return nil, err
}
if err := project.Pull(ctx); err != nil {
return nil, err
}
return project, nil
}
func UpdateDashboardInfo(d *grafana.Dashboard, dv *grafana.DashboardVersion) {
d.Dashboard = dv.Data
d.UpdatedBy = dv.CreatedBy
d.Updated = dv.Created
d.Version = dv.Version
}