121 lines
2.8 KiB
Go
121 lines
2.8 KiB
Go
package backup
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/pflag"
|
|
|
|
"github.com/alcionai/corso/cli/config"
|
|
"github.com/alcionai/corso/cli/utils"
|
|
"github.com/alcionai/corso/pkg/logger"
|
|
"github.com/alcionai/corso/pkg/repository"
|
|
)
|
|
|
|
// exchange bucket info from flags
|
|
var (
|
|
user string
|
|
)
|
|
|
|
// called by backup.go to map parent subcommands to provider-specific handling.
|
|
func addExchangeCommands(parent *cobra.Command) *cobra.Command {
|
|
var (
|
|
c *cobra.Command
|
|
fs *pflag.FlagSet
|
|
)
|
|
switch parent.Use {
|
|
case createCommand:
|
|
c, fs = utils.AddCommand(parent, exchangeCreateCmd)
|
|
fs.StringVar(&user, "user", "", "ID of the user whose Exchange data is to be backed up.")
|
|
case listCommand:
|
|
c, _ = utils.AddCommand(parent, exchangeListCmd)
|
|
}
|
|
return c
|
|
}
|
|
|
|
const exchangeServiceCommand = "exchange"
|
|
|
|
// `corso backup create exchange [<flag>...]`
|
|
var exchangeCreateCmd = &cobra.Command{
|
|
Use: exchangeServiceCommand,
|
|
Short: "Backup M365 Exchange service data",
|
|
RunE: createExchangeCmd,
|
|
Args: cobra.NoArgs,
|
|
}
|
|
|
|
// processes an exchange service backup.
|
|
func createExchangeCmd(cmd *cobra.Command, args []string) error {
|
|
ctx := cmd.Context()
|
|
|
|
if utils.HasNoFlagsAndShownHelp(cmd) {
|
|
return nil
|
|
}
|
|
|
|
s, acct, err := config.GetStorageAndAccount(true, nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
m365, err := acct.M365Config()
|
|
if err != nil {
|
|
return errors.Wrap(err, "Failed to parse m365 account config")
|
|
}
|
|
|
|
logger.Ctx(ctx).Debugw(
|
|
"Called - "+cmd.CommandPath(),
|
|
"tenantID", m365.TenantID,
|
|
"clientID", m365.ClientID,
|
|
"hasClientSecret", len(m365.ClientSecret) > 0)
|
|
|
|
r, err := repository.Connect(ctx, acct, s)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "Failed to connect to the %s repository", s.Provider)
|
|
}
|
|
defer utils.CloseRepo(ctx, r)
|
|
|
|
bo, err := r.NewBackup(ctx, []string{user})
|
|
if err != nil {
|
|
return errors.Wrap(err, "Failed to initialize Exchange backup")
|
|
}
|
|
|
|
result, err := bo.Run(ctx)
|
|
if err != nil {
|
|
return errors.Wrap(err, "Failed to run Exchange backup")
|
|
}
|
|
|
|
fmt.Printf("Backed up restore point %s in %s for Exchange user %s.\n", result.SnapshotID, s.Provider, user)
|
|
return nil
|
|
}
|
|
|
|
// `corso backup list exchange [<flag>...]`
|
|
var exchangeListCmd = &cobra.Command{
|
|
Use: exchangeServiceCommand,
|
|
Short: "List the history of M365 Exchange service backups",
|
|
RunE: listExchangeCmd,
|
|
Args: cobra.NoArgs,
|
|
}
|
|
|
|
// lists the history of backup operations
|
|
func listExchangeCmd(cmd *cobra.Command, args []string) error {
|
|
ctx := cmd.Context()
|
|
|
|
_, acct, err := config.GetStorageAndAccount(true, nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
m365, err := acct.M365Config()
|
|
if err != nil {
|
|
return errors.Wrap(err, "Failed to parse m365 account config")
|
|
}
|
|
|
|
logger.Ctx(ctx).Debugw(
|
|
"Called - "+cmd.CommandPath(),
|
|
"tenantID", m365.TenantID)
|
|
|
|
// todo (keepers issue #251): e2e hookup
|
|
|
|
return nil
|
|
}
|