Initial QSfera import

Server/qsfera/pkg/command/backup.go

@@ -0,0 +1,82 @@
+package command
+
+import (
+	"errors"
+	"fmt"
+
+	"github.com/spf13/cobra"
+
+	"github.com/qsfera/server/qsfera/pkg/backup"
+	"github.com/qsfera/server/qsfera/pkg/register"
+	"github.com/qsfera/server/pkg/config"
+	"github.com/qsfera/server/pkg/config/configlog"
+	"github.com/qsfera/server/pkg/config/parser"
+	decomposedbs "github.com/opencloud-eu/reva/v2/pkg/storage/fs/decomposed/blobstore"
+	decomposeds3bs "github.com/opencloud-eu/reva/v2/pkg/storage/fs/decomposeds3/blobstore"
+)
+
+// BackupCommand is the entrypoint for the backup command
+func BackupCommand(cfg *config.Config) *cobra.Command {
+	bckCmd := &cobra.Command{
+		Use:   "backup",
+		Short: "КуСфера backup functionality",
+		PersistentPreRunE: func(cmd *cobra.Command, args []string) error {
+			return configlog.ReturnError(parser.ParseConfig(cfg, true))
+		},
+	}
+	bckCmd.AddCommand(ConsistencyCommand(cfg))
+	return bckCmd
+}
+
+// ConsistencyCommand is the entrypoint for the consistency Command
+func ConsistencyCommand(cfg *config.Config) *cobra.Command {
+	consCmd := &cobra.Command{
+		Use:   "consistency",
+		Short: "check backup consistency",
+		RunE: func(cmd *cobra.Command, args []string) error {
+			var (
+				bs  backup.ListBlobstore
+				err error
+			)
+			basePath, _ := cmd.Flags().GetString("basepath")
+			blobstoreFlag, _ := cmd.Flags().GetString("blobstore")
+			switch blobstoreFlag {
+			case "decomposeds3":
+				bs, err = decomposeds3bs.New(
+					cfg.StorageUsers.Drivers.DecomposedS3.Endpoint,
+					cfg.StorageUsers.Drivers.DecomposedS3.Region,
+					cfg.StorageUsers.Drivers.DecomposedS3.Bucket,
+					cfg.StorageUsers.Drivers.DecomposedS3.AccessKey,
+					cfg.StorageUsers.Drivers.DecomposedS3.SecretKey,
+					decomposeds3bs.Options{},
+				)
+			case "decomposed":
+				bs, err = decomposedbs.New(basePath)
+			case "none":
+				bs = nil
+			default:
+				err = errors.New("blobstore type not supported")
+			}
+			if err != nil {
+				fmt.Println(err)
+				return err
+			}
+			fail, _ := cmd.Flags().GetBool("fail")
+			if err := backup.CheckProviderConsistency(basePath, bs, fail); err != nil {
+				fmt.Println(err)
+				return err
+			}
+
+			return nil
+		},
+	}
+	consCmd.Flags().StringP("basepath", "p", "", "the basepath of the decomposedfs (e.g. /var/tmp/qsfera/storage/users)")
+	_ = consCmd.MarkFlagRequired("basepath")
+	consCmd.Flags().StringP("blobstore", "b", "decomposed", "the blobstore type. Can be (none, decomposed, decomposeds3). Default decomposed")
+	consCmd.Flags().Bool("fail", false, "exit with non-zero status if consistency check fails")
+	return consCmd
+}
+
+func init() {
+	register.AddCommand(BackupCommand)
+}