Method moduleSourceWithClient

core/schema/modulesource.go:3343–3399

(
	ctx context.Context,
	src *core.ModuleSource,
	args struct {
		Generator dagql.String
		OutputDir dagql.String
	},
)

}

func (s *moduleSourceSchema) moduleSourceWithClient(
	ctx context.Context,
	src *core.ModuleSource,
	args struct {
		Generator dagql.String
		OutputDir dagql.String
	},
) (*core.ModuleSource, error) {
	src = src.Clone()

	if src.ConfigClients == nil {
		src.ConfigClients = []*modules.ModuleConfigClient{}
	}

	moduleConfigClient := &modules.ModuleConfigClient{
		Generator: args.Generator.String(),
		Directory: args.OutputDir.String(),
	}

	for _, client := range src.ConfigClients {
		if filepath.Clean(client.Directory) == filepath.Clean(moduleConfigClient.Directory) {
			return nil, fmt.Errorf("a client is already generated in the %s directory", client.Directory)
		}
	}

	// Verify that the generator can be loaded as a module and clean
	// the generator path if it's a local path.
	if !sdk.IsModuleSDKBuiltin(moduleConfigClient.Generator) {
		dag, err := core.CurrentDagqlServer(ctx)
		if err != nil {
			return nil, fmt.Errorf("failed to get dag server: %w", err)
		}

		query, err := core.CurrentQuery(ctx)
		if err != nil {
			return nil, fmt.Errorf("failed to get current query: %w", err)
		}

		bk, err := query.Engine(ctx)
		if err != nil {
			return nil, fmt.Errorf("failed to get engine client: %w", err)
		}

		clientModule, err := core.ResolveDepToSource(ctx, bk, dag, src, moduleConfigClient.Generator, "", "")
		if err != nil {
			return nil, fmt.Errorf("failed to resolve client module source: %w", err)
		}

		if clientModule.Self().Kind == core.ModuleSourceKindLocal {
			moduleConfigClient.Generator = filepath.Clean(moduleConfigClient.Generator)
		}
	}

	src.ConfigClients = append(src.ConfigClients, moduleConfigClient)

	return src, nil
}