Method Install

core/object.go:973–1020

(ctx context.Context, dag *dagql.Server, opts ...InstallOpts)

}

func (obj *ModuleObject) Install(ctx context.Context, dag *dagql.Server, opts ...InstallOpts) error {
	if obj.Module.Self() == nil {
		return fmt.Errorf("installing object %q too early", obj.TypeDef.Name)
	}

	var opt InstallOpts
	if len(opts) > 0 {
		opt = opts[0]
	}

	classOpts := dagql.ClassOpts[*ModuleObject]{
		Typed: obj,
	}

	installDirectives := []*ast.Directive{}
	if obj.TypeDef.SourceMap.Valid {
		classOpts.SourceMap = obj.TypeDef.SourceMap.Value.Self().TypeDirective()
		installDirectives = append(installDirectives, obj.TypeDef.SourceMap.Value.Self().TypeDirective())
	}

	class := dagql.NewClass(dag, classOpts)
	if obj.isMainObject() && !opt.SkipConstructor && !opt.Entrypoint {
		if err := obj.installConstructor(ctx, dag); err != nil {
			return fmt.Errorf("failed to install constructor: %w", err)
		}
	}
	fields, err := obj.fields(ctx)
	if err != nil {
		return err
	}

	funs, err := obj.functions(ctx, dag)
	if err != nil {
		return err
	}
	fields = append(fields, funs...)

	class.Install(fields...)
	dag.InstallObject(class, installDirectives...)

	if obj.isMainObject() && opt.Entrypoint {
		if err := obj.installEntrypointMethods(ctx, dag); err != nil {
			return fmt.Errorf("failed to install entrypoint methods: %w", err)
		}
	}

	return nil
}

func (obj *ModuleObject) isMainObject() bool {
	return gqlObjectName(obj.TypeDef.OriginalName) == gqlObjectName(obj.Module.Self().OriginalName)