mirror of https://github.com/databricks/cli.git
54 lines
1.2 KiB
Go
54 lines
1.2 KiB
Go
package files
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"io/fs"
|
|
|
|
"github.com/databricks/cli/bundle"
|
|
"github.com/databricks/cli/bundle/permissions"
|
|
"github.com/databricks/cli/libs/cmdio"
|
|
"github.com/databricks/cli/libs/diag"
|
|
"github.com/databricks/cli/libs/log"
|
|
"github.com/databricks/cli/libs/sync"
|
|
)
|
|
|
|
type upload struct {
|
|
outputHandler sync.OutputHandler
|
|
}
|
|
|
|
func (m *upload) Name() string {
|
|
return "files.Upload"
|
|
}
|
|
|
|
func (m *upload) Apply(ctx context.Context, b *bundle.Bundle) diag.Diagnostics {
|
|
cmdio.LogString(ctx, fmt.Sprintf("Uploading bundle files to %s...", b.Config.Workspace.FilePath))
|
|
opts, err := GetSyncOptions(ctx, bundle.ReadOnly(b))
|
|
if err != nil {
|
|
return diag.FromErr(err)
|
|
}
|
|
|
|
opts.OutputHandler = m.outputHandler
|
|
sync, err := sync.New(ctx, *opts)
|
|
if err != nil {
|
|
return diag.FromErr(err)
|
|
}
|
|
defer sync.Close()
|
|
|
|
b.Files, err = sync.RunOnce(ctx)
|
|
if err != nil {
|
|
if errors.Is(err, fs.ErrPermission) {
|
|
return permissions.ReportPossiblePermissionDenied(ctx, b, b.Config.Workspace.FilePath)
|
|
}
|
|
return diag.FromErr(err)
|
|
}
|
|
|
|
log.Infof(ctx, "Uploaded bundle files")
|
|
return nil
|
|
}
|
|
|
|
func Upload(outputHandler sync.OutputHandler) bundle.Mutator {
|
|
return &upload{outputHandler}
|
|
}
|