package generate import ( "fmt" "os" "path/filepath" "github.com/databricks/cli/bundle" "github.com/databricks/cli/bundle/config/generate" "github.com/databricks/cli/cmd/root" "github.com/databricks/cli/libs/cmdio" "github.com/databricks/cli/libs/dyn" "github.com/databricks/cli/libs/dyn/yamlsaver" "github.com/databricks/cli/libs/textutil" "github.com/databricks/databricks-sdk-go/service/jobs" "github.com/spf13/cobra" ) func NewGenerateJobCommand() *cobra.Command { var configDir string var sourceDir string var jobId int64 var force bool cmd := &cobra.Command{ Use: "job", Short: "Generate bundle configuration for a job", PreRunE: root.MustConfigureBundle, } cmd.Flags().Int64Var(&jobId, "existing-job-id", 0, `Job ID of the job to generate config for`) cmd.MarkFlagRequired("existing-job-id") wd, err := os.Getwd() if err != nil { wd = "." } cmd.Flags().StringVarP(&configDir, "config-dir", "d", filepath.Join(wd, "resources"), `Dir path where the output config will be stored`) cmd.Flags().StringVarP(&sourceDir, "source-dir", "s", filepath.Join(wd, "src"), `Dir path where the downloaded files will be stored`) cmd.Flags().BoolVarP(&force, "force", "f", false, `Force overwrite existing files in the output directory`) cmd.RunE = func(cmd *cobra.Command, args []string) error { ctx := cmd.Context() b := bundle.Get(ctx) w := b.WorkspaceClient() job, err := w.Jobs.Get(ctx, jobs.GetJobRequest{JobId: jobId}) if err != nil { return err } downloader := newDownloader(w, sourceDir, configDir) for _, task := range job.Settings.Tasks { err := downloader.MarkTaskForDownload(ctx, &task) if err != nil { return err } } v, err := generate.ConvertJobToValue(job) if err != nil { return err } jobKey := cmd.Flag("key").Value.String() if jobKey == "" { jobKey = textutil.NormalizeString(job.Settings.Name) } result := map[string]dyn.Value{ "resources": dyn.V(map[string]dyn.Value{ "jobs": dyn.V(map[string]dyn.Value{ jobKey: v, }), }), } err = downloader.FlushToDisk(ctx, force) if err != nil { return err } filename := filepath.Join(configDir, fmt.Sprintf("%s.yml", jobKey)) err = yamlsaver.SaveAsYAML(result, filename, force) if err != nil { return err } cmdio.LogString(ctx, fmt.Sprintf("Job configuration successfully saved to %s", filename)) return nil } return cmd }