f0e1000171
<!-- PR description--> --- #### Does this PR need a docs update or release note? - [x] ✅ Yes, it's included - [ ] 🕐 Yes, but in a later PR - [ ] ⛔ No #### Type of change <!--- Please check the type of change your PR introduces: ---> - [ ] 🌻 Feature - [ ] 🐛 Bugfix - [ ] 🗺️ Documentation - [ ] 🤖 Supportability/Tests - [ ] 💻 CI/Deployment - [ ] 🧹 Tech Debt/Cleanup #### Issue(s) <!-- Can reference multiple issues. Use one of the following "magic words" - "closes, fixes" to auto-close the Github issue. --> * #<issue> #### Test Plan <!-- How will this be tested prior to merging.--> - [ ] 💪 Manual - [ ] ⚡ Unit test - [ ] 💚 E2E
99 lines
2.8 KiB
Go
99 lines
2.8 KiB
Go
package restore
|
|
|
|
import (
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/pflag"
|
|
|
|
"github.com/alcionai/corso/src/cli/flags"
|
|
"github.com/alcionai/corso/src/cli/utils"
|
|
"github.com/alcionai/corso/src/internal/common/dttm"
|
|
)
|
|
|
|
// called by restore.go to map subcommands to provider-specific handling.
|
|
func addOneDriveCommands(cmd *cobra.Command) *cobra.Command {
|
|
var (
|
|
c *cobra.Command
|
|
fs *pflag.FlagSet
|
|
)
|
|
|
|
switch cmd.Use {
|
|
case restoreCommand:
|
|
c, fs = utils.AddCommand(cmd, oneDriveRestoreCmd())
|
|
|
|
c.Use = c.Use + " " + oneDriveServiceCommandUseSuffix
|
|
|
|
// Flags addition ordering should follow the order we want them to appear in help and docs:
|
|
// More generic (ex: --user) and more frequently used flags take precedence.
|
|
fs.SortFlags = false
|
|
|
|
flags.AddBackupIDFlag(c, true)
|
|
flags.AddOneDriveDetailsAndRestoreFlags(c)
|
|
flags.AddNoPermissionsFlag(c)
|
|
flags.AddRestoreConfigFlags(c, true)
|
|
flags.AddFailFastFlag(c)
|
|
}
|
|
|
|
return c
|
|
}
|
|
|
|
const (
|
|
oneDriveServiceCommand = "onedrive"
|
|
oneDriveServiceCommandUseSuffix = "--backup <backupId>"
|
|
|
|
oneDriveServiceCommandRestoreExamples = `# Restore file with ID 98765abcdef in Bob's last backup (1234abcd...)
|
|
corso restore onedrive --backup 1234abcd-12ab-cd34-56de-1234abcd --file 98765abcdef
|
|
|
|
# Restore the file with ID 98765abcdef without its associated permissions
|
|
corso restore onedrive --backup 1234abcd-12ab-cd34-56de-1234abcd --file 98765abcdef --no-permissions
|
|
|
|
# Restore files named "FY2021 Planning.xlsx" in "Documents/Finance Reports"
|
|
corso restore onedrive --backup 1234abcd-12ab-cd34-56de-1234abcd \
|
|
--file "FY2021 Planning.xlsx" --folder "Documents/Finance Reports"
|
|
|
|
# Restore all files and folders in folder "Documents/Finance Reports" that were created before 2020
|
|
corso restore onedrive --backup 1234abcd-12ab-cd34-56de-1234abcd \
|
|
--folder "Documents/Finance Reports" --file-created-before 2020-01-01T00:00:00`
|
|
)
|
|
|
|
// `corso restore onedrive [<flag>...]`
|
|
func oneDriveRestoreCmd() *cobra.Command {
|
|
return &cobra.Command{
|
|
Use: oneDriveServiceCommand,
|
|
Short: "Restore M365 OneDrive service data",
|
|
RunE: restoreOneDriveCmd,
|
|
Args: cobra.NoArgs,
|
|
Example: oneDriveServiceCommandRestoreExamples,
|
|
}
|
|
}
|
|
|
|
// processes an onedrive service restore.
|
|
func restoreOneDriveCmd(cmd *cobra.Command, args []string) error {
|
|
ctx := cmd.Context()
|
|
|
|
if utils.HasNoFlagsAndShownHelp(cmd) {
|
|
return nil
|
|
}
|
|
|
|
opts := utils.MakeOneDriveOpts(cmd)
|
|
opts.RestoreCfg.DTTMFormat = dttm.HumanReadableDriveItem
|
|
|
|
if flags.RunModeFV == flags.RunModeFlagTest {
|
|
return nil
|
|
}
|
|
|
|
if err := utils.ValidateOneDriveRestoreFlags(flags.BackupIDFV, opts); err != nil {
|
|
return err
|
|
}
|
|
|
|
sel := utils.IncludeOneDriveRestoreDataSelectors(opts)
|
|
utils.FilterOneDriveRestoreInfoSelectors(sel, opts)
|
|
|
|
return runRestore(
|
|
ctx,
|
|
cmd,
|
|
opts.RestoreCfg,
|
|
sel.Selector,
|
|
flags.BackupIDFV,
|
|
"OneDrive")
|
|
}
|