Function modifyLocalModule

cmd/dagger/module.go:1199–1240

modifyLocalModule is a helper that loads a local module source, applies a transformation, and exports the generated context directory back to disk.

(ctx context.Context, transform func(*dagger.Client, *dagger.ModuleSource) *dagger.ModuleSource)

// modifyLocalModule is a helper that loads a local module source, applies a
// transformation, and exports the generated context directory back to disk.
func modifyLocalModule(ctx context.Context, transform func(*dagger.Client, *dagger.ModuleSource) *dagger.ModuleSource) error {
	return withEngine(ctx, client.Params{
		SkipWorkspaceModules: true,
	}, func(ctx context.Context, engineClient *client.Client) error {
		dag := engineClient.Dagger()

		modRef, err := getModuleSourceRefWithDefault()
		if err != nil {
			return err
		}
		modSrc := dag.ModuleSource(modRef, dagger.ModuleSourceOpts{
			RequireKind: dagger.ModuleSourceKindLocalSource,
		})

		alreadyExists, err := modSrc.ConfigExists(ctx)
		if err != nil {
			return localModuleErrorf("failed to check if module already exists: %w", err)
		}
		if !alreadyExists {
			return fmt.Errorf("module must be fully initialized")
		}

		contextDirPath, err := modSrc.LocalContextDirectoryPath(ctx)
		if err != nil {
			return localModuleErrorf("failed to get local context directory path: %w", err)
		}

		modSrc = transform(dag, modSrc)
		if engineVersion := getCompatVersion(); engineVersion != "" {
			modSrc = modSrc.WithEngineVersion(engineVersion)
		}

		_, err = modSrc.
			GeneratedContextDirectory().
			Export(ctx, contextDirPath)
		if err != nil {
			return fmt.Errorf("failed to update dependencies: %w", err)
		}

		return nil
	})
}

func localModuleErrorf(format string, err error) error {
	if err == nil {