2b65ff80f2
* hook up restore end-to-end Now that GC and KW both provide restore operations for a single message, we can hook up the end-to-end restore process. Integration tests for this change will follow in the next PR.
74 lines
1.8 KiB
Go
74 lines
1.8 KiB
Go
package operations
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/pkg/errors"
|
|
|
|
"github.com/alcionai/corso/internal/connector"
|
|
"github.com/alcionai/corso/internal/kopia"
|
|
"github.com/alcionai/corso/pkg/credentials"
|
|
)
|
|
|
|
// RestoreOperation wraps an operation with restore-specific props.
|
|
type RestoreOperation struct {
|
|
operation
|
|
Version string
|
|
|
|
restorePointID string
|
|
creds credentials.M365
|
|
|
|
Targets []string // something for targets/filter/source/app&users/etc
|
|
}
|
|
|
|
// NewRestoreOperation constructs and validates a restore operation.
|
|
func NewRestoreOperation(
|
|
ctx context.Context,
|
|
opts OperationOpts,
|
|
kw *kopia.KopiaWrapper,
|
|
creds credentials.M365,
|
|
restorePointID string,
|
|
targets []string,
|
|
) (RestoreOperation, error) {
|
|
op := RestoreOperation{
|
|
operation: newOperation(opts, kw),
|
|
Version: "v0",
|
|
creds: creds,
|
|
restorePointID: restorePointID,
|
|
Targets: targets,
|
|
}
|
|
if err := op.validate(); err != nil {
|
|
return RestoreOperation{}, err
|
|
}
|
|
|
|
return op, nil
|
|
}
|
|
|
|
func (op RestoreOperation) validate() error {
|
|
if err := op.creds.Validate(); err != nil {
|
|
return errors.Wrap(err, "invalid credentials")
|
|
}
|
|
return op.operation.validate()
|
|
}
|
|
|
|
// Run begins a synchronous restore operation.
|
|
// todo (keepers): return stats block in first param.
|
|
func (op *RestoreOperation) Run(ctx context.Context) error {
|
|
dc, err := op.kopia.RestoreSingleItem(ctx, op.restorePointID, op.Targets)
|
|
if err != nil {
|
|
return errors.Wrap(err, "retrieving service data")
|
|
}
|
|
|
|
gc, err := connector.NewGraphConnector(op.creds.TenantID, op.creds.ClientID, op.creds.ClientSecret)
|
|
if err != nil {
|
|
return errors.Wrap(err, "connecting to graph api")
|
|
}
|
|
|
|
if err := gc.RestoreMessages(ctx, dc); err != nil {
|
|
return errors.Wrap(err, "restoring service data")
|
|
}
|
|
|
|
op.Status = Successful
|
|
return nil
|
|
}
|