Method Patch

core/module.go:1835–1999

Patch is called after all types have been loaded - here we can update any definitions as required, and attempt to resolve references. nolint:gocyclo // intrinsically long state machine; refactoring would hurt clarity

(ctx context.Context)

//
//nolint:gocyclo // intrinsically long state machine; refactoring would hurt clarity
func (mod *Module) Patch(ctx context.Context) error {
	dag, err := CurrentDagqlServer(ctx)
	if err != nil {
		return fmt.Errorf("current dagql server: %w", err)
	}

	patchFunctionEnumDefaults := func(fn dagql.ObjectResult[*Function]) (dagql.ObjectResult[*Function], error) {
		updated := fn
		for _, arg := range fn.Self().Args {
			argSelf := arg.Self()
			if argSelf.DefaultValue == nil {
				continue
			}
			if argSelf.TypeDef.Self().Kind != TypeDefKindEnum {
				continue
			}
			var enumTypeDef *EnumTypeDef
			for _, enum := range mod.EnumDefs {
				if enum.Self().AsEnum.Value.Self().Name == argSelf.TypeDef.Self().AsEnum.Value.Self().Name {
					enumTypeDef = enum.Self().AsEnum.Value.Self()
					break
				}
			}
			if enumTypeDef == nil {
				continue
			}

			var val string
			dec := json.NewDecoder(bytes.NewReader(argSelf.DefaultValue.Bytes()))
			dec.UseNumber()
			if err := dec.Decode(&val); err != nil {
				return updated, fmt.Errorf("failed to decode default value for arg %q: %w", argSelf.Name, err)
			}

			found := false
			for _, member := range enumTypeDef.Members {
				memberSelf := member.Self()
				if val == memberSelf.OriginalName {
					val = memberSelf.Name
					found = true
					break
				}
			}
			if !found {
				return updated, fmt.Errorf("enum name %q not found", val)
			}

			res, err := json.Marshal(val)
			if err != nil {
				return updated, err
			}
			var updatedArg dagql.ObjectResult[*FunctionArg]
			if err := dag.Select(ctx, arg, &updatedArg, dagql.Selector{
				Field: "__withDefaultValue",
				Args:  []dagql.NamedInput{{Name: "defaultValue", Value: JSON(res)}},
			}); err != nil {
				return updated, fmt.Errorf("patch enum default arg %q: %w", argSelf.Name, err)
			}