97ed97c1c3
adds logical changes required for sharepoint permission handling, both backup and restore. --- #### Does this PR need a docs update or release note? - [x] ✅ Yes, it's included #### Type of change - [x] 🌻 Feature #### Issue(s) * #3135 #### Test Plan - [x] 💪 Manual - [x] ⚡ Unit test - [x] 💚 E2E
98 lines
2.8 KiB
Go
98 lines
2.8 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
|
|
"github.com/spf13/cobra"
|
|
|
|
. "github.com/alcionai/corso/src/cli/print"
|
|
"github.com/alcionai/corso/src/cmd/factory/impl"
|
|
"github.com/alcionai/corso/src/internal/common/crash"
|
|
"github.com/alcionai/corso/src/pkg/logger"
|
|
)
|
|
|
|
var factoryCmd = &cobra.Command{
|
|
Use: "factory",
|
|
Short: "Generate all types of m365 folders",
|
|
RunE: handleFactoryRoot,
|
|
}
|
|
|
|
var exchangeCmd = &cobra.Command{
|
|
Use: "exchange",
|
|
Short: "Generate exchange data",
|
|
RunE: handleExchangeFactory,
|
|
}
|
|
|
|
var oneDriveCmd = &cobra.Command{
|
|
Use: "onedrive",
|
|
Short: "Generate onedrive data",
|
|
RunE: handleOneDriveFactory,
|
|
}
|
|
|
|
var sharePointCmd = &cobra.Command{
|
|
Use: "sharepoint",
|
|
Short: "Generate shareopint data",
|
|
RunE: handleSharePointFactory,
|
|
}
|
|
|
|
// ------------------------------------------------------------------------------------------
|
|
// CLI command handlers
|
|
// ------------------------------------------------------------------------------------------
|
|
|
|
func main() {
|
|
ctx, _ := logger.SeedLevel(context.Background(), logger.Development)
|
|
ctx = SetRootCmd(ctx, factoryCmd)
|
|
|
|
defer func() {
|
|
if err := crash.Recovery(ctx, recover(), "backup"); err != nil {
|
|
logger.CtxErr(ctx, err).Error("panic in factory")
|
|
}
|
|
|
|
logger.Flush(ctx)
|
|
}()
|
|
|
|
// persistent flags that are common to all use cases
|
|
fs := factoryCmd.PersistentFlags()
|
|
fs.StringVar(&impl.Tenant, "tenant", "", "m365 tenant containing the user")
|
|
fs.StringVar(&impl.Site, "site", "", "sharepoint site owning the new data")
|
|
fs.StringVar(&impl.User, "user", "", "m365 user owning the new data")
|
|
fs.StringVar(&impl.SecondaryUser, "secondaryuser", "", "m365 secondary user owning the new data")
|
|
fs.IntVar(&impl.Count, "count", 0, "count of items to produce")
|
|
cobra.CheckErr(factoryCmd.MarkPersistentFlagRequired("count"))
|
|
fs.StringVar(&impl.Destination, "destination", "", "destination of the new data (will create as needed)")
|
|
cobra.CheckErr(factoryCmd.MarkPersistentFlagRequired("destination"))
|
|
|
|
factoryCmd.AddCommand(exchangeCmd)
|
|
impl.AddExchangeCommands(exchangeCmd)
|
|
factoryCmd.AddCommand(oneDriveCmd)
|
|
impl.AddOneDriveCommands(oneDriveCmd)
|
|
factoryCmd.AddCommand(sharePointCmd)
|
|
impl.AddSharePointCommands(sharePointCmd)
|
|
|
|
if err := factoryCmd.ExecuteContext(ctx); err != nil {
|
|
logger.Flush(ctx)
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
func handleFactoryRoot(cmd *cobra.Command, args []string) error {
|
|
Err(cmd.Context(), impl.ErrNotYetImplemented)
|
|
return cmd.Help()
|
|
}
|
|
|
|
func handleExchangeFactory(cmd *cobra.Command, args []string) error {
|
|
Err(cmd.Context(), impl.ErrNotYetImplemented)
|
|
return cmd.Help()
|
|
}
|
|
|
|
func handleOneDriveFactory(cmd *cobra.Command, args []string) error {
|
|
Err(cmd.Context(), impl.ErrNotYetImplemented)
|
|
return cmd.Help()
|
|
}
|
|
|
|
func handleSharePointFactory(cmd *cobra.Command, args []string) error {
|
|
Err(cmd.Context(), impl.ErrNotYetImplemented)
|
|
return cmd.Help()
|
|
}
|