2a150cc610
Introduces a counting bus, and threads it into restore operations so that we can count the number of collision skips that occur. --- #### Does this PR need a docs update or release note? - [x] ⛔ No #### Type of change - [x] 🌻 Feature #### Issue(s) * #3562 #### Test Plan - [x] ⚡ Unit test - [x] 💚 E2E
74 lines
1.5 KiB
Go
74 lines
1.5 KiB
Go
package impl
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/spf13/cobra"
|
|
|
|
. "github.com/alcionai/corso/src/cli/print"
|
|
"github.com/alcionai/corso/src/cli/utils"
|
|
"github.com/alcionai/corso/src/internal/m365/resource"
|
|
"github.com/alcionai/corso/src/pkg/count"
|
|
"github.com/alcionai/corso/src/pkg/fault"
|
|
"github.com/alcionai/corso/src/pkg/logger"
|
|
"github.com/alcionai/corso/src/pkg/path"
|
|
"github.com/alcionai/corso/src/pkg/selectors"
|
|
)
|
|
|
|
var spFilesCmd = &cobra.Command{
|
|
Use: "files",
|
|
Short: "Generate SharePoint files",
|
|
RunE: handleSharePointLibraryFileFactory,
|
|
}
|
|
|
|
func AddSharePointCommands(cmd *cobra.Command) {
|
|
cmd.AddCommand(spFilesCmd)
|
|
}
|
|
|
|
func handleSharePointLibraryFileFactory(cmd *cobra.Command, args []string) error {
|
|
var (
|
|
ctx = cmd.Context()
|
|
service = path.SharePointService
|
|
category = path.LibrariesCategory
|
|
errs = fault.New(false)
|
|
)
|
|
|
|
if utils.HasNoFlagsAndShownHelp(cmd) {
|
|
return nil
|
|
}
|
|
|
|
ctrl, acct, inp, err := getControllerAndVerifyResourceOwner(ctx, resource.Sites, Site, path.SharePointService)
|
|
if err != nil {
|
|
return Only(ctx, err)
|
|
}
|
|
|
|
sel := selectors.NewSharePointBackup([]string{Site}).Selector
|
|
sel.SetDiscreteOwnerIDName(inp.ID(), inp.Name())
|
|
|
|
deets, err := generateAndRestoreDriveItems(
|
|
ctrl,
|
|
inp.ID(),
|
|
SecondaryUser,
|
|
strings.ToLower(SecondaryUser),
|
|
acct,
|
|
service,
|
|
category,
|
|
sel,
|
|
Tenant,
|
|
Destination,
|
|
Count,
|
|
errs,
|
|
count.New())
|
|
if err != nil {
|
|
return Only(ctx, err)
|
|
}
|
|
|
|
for _, e := range errs.Recovered() {
|
|
logger.CtxErr(ctx, err).Error(e.Error())
|
|
}
|
|
|
|
deets.PrintEntries(ctx)
|
|
|
|
return nil
|
|
}
|