Method runGeneratedContext

core/schema/modulesource.go:2656–2715

runGeneratedContext runs codegen, client generation, and dagger.json writing for the given module source, returning the original context directory and the fully-generated context directory. Callers can use these two directories to produce either a diff or a Changeset.

(
	ctx context.Context,
	srcInst dagql.ObjectResult[*core.ModuleSource],
)

// module source, returning the original context directory and the fully-generated context
// directory. Callers can use these two directories to produce either a diff or a Changeset.
func (s *moduleSourceSchema) runGeneratedContext(
	ctx context.Context,
	srcInst dagql.ObjectResult[*core.ModuleSource],
) (originalCtxDir dagql.ObjectResult[*core.Directory], genDirInst dagql.ObjectResult[*core.Directory], _ error) {
	dag, err := core.CurrentDagqlServer(ctx)
	if err != nil {
		return originalCtxDir, genDirInst, fmt.Errorf("failed to get dag server: %w", err)
	}

	modCfg, err := s.loadModuleSourceConfig(srcInst.Self())
	if err != nil {
		return originalCtxDir, genDirInst, fmt.Errorf("failed to load module source config: %w", err)
	}

	originalCtxDir = srcInst.Self().ContextDirectory

	// run codegen too if we have a name and SDK
	genDirInst = originalCtxDir
	if modCfg.Name != "" && modCfg.SDK != nil && modCfg.SDK.Source != "" {
		updatedGenDirInst, err := s.runCodegen(ctx, srcInst)
		var missingImplErr ErrSDKCodegenNotImplemented
		if err != nil && !errors.As(err, &missingImplErr) {
			return originalCtxDir, genDirInst, fmt.Errorf("failed to run codegen: %w", err)
		}
		if err == nil {
			genDirInst = updatedGenDirInst
		}
	}

	// Generate clients
	for _, client := range modCfg.Clients {
		genDirInst, err = s.runClientGenerator(ctx, srcInst, genDirInst, client)
		if err != nil {
			return originalCtxDir, genDirInst, fmt.Errorf("failed to generate client %s: %w", client.Generator, err)
		}
	}

	// write dagger.json to the generated context directory
	modCfgBytes, err := json.MarshalIndent(modCfg, "", "  ")
	if err != nil {
		return originalCtxDir, genDirInst, fmt.Errorf("failed to encode module config: %w", err)
	}
	modCfgBytes = append(modCfgBytes, '\n')
	modCfgPath := filepath.Join(srcInst.Self().SourceRootSubpath, modules.Filename)
	err = dag.Select(ctx, genDirInst, &genDirInst,
		dagql.Selector{
			Field: "withNewFile",
			Args: []dagql.NamedInput{
				{Name: "path", Value: dagql.String(modCfgPath)},
				{Name: "contents", Value: dagql.String(modCfgBytes)},
				{Name: "permissions", Value: dagql.Int(0o644)},
			},
		},
	)
	if err != nil {
		return originalCtxDir, genDirInst, fmt.Errorf("failed to add updated dagger.json to context dir: %w", err)
	}