Function getPubWorkspaceOrPackageFolderInfo

src/shared/vscode/pub.ts:252–276

(
	packageFolderOrPubspecPaths: string[],
	existsSync: (itemPath: string) => boolean = fs.existsSync,
	readFileSync: (itemPath: string) => string = (p) => fs.readFileSync(p, "utf8").toString(),
)

 * of a Pub Workspace root and not the child packages.
 */
export function getPubWorkspaceOrPackageFolderInfo(
	packageFolderOrPubspecPaths: string[],
	existsSync: (itemPath: string) => boolean = fs.existsSync,
	readFileSync: (itemPath: string) => string = (p) => fs.readFileSync(p, "utf8").toString(),
): PubWorkspaceOrPackageFolderInfo[] {
	const pubWorkspacePackageCounts = new Map<string, number>();

	for (const packageFolderOrPubspecPath of packageFolderOrPubspecPaths) {
		const packageFolderPath = path.basename(packageFolderOrPubspecPath) === "pubspec.yaml"
			? path.dirname(packageFolderOrPubspecPath)
			: packageFolderOrPubspecPath;
		const workspaceRootPath = getPubWorkspaceFolderOrPackageFolderPath(packageFolderOrPubspecPath, existsSync, readFileSync);

		let count = pubWorkspacePackageCounts.get(workspaceRootPath) ?? 0;
		const isRoot = workspaceRootPath === packageFolderPath;
		if (!isRoot)
			count++;

		pubWorkspacePackageCounts.set(workspaceRootPath, count);
	}

	return [...pubWorkspacePackageCounts.entries()]
		.map((p) => p[1] ? { path: p[0], workspacePackageCount: p[1] } : { path: p[0] })
		.sort((a, b) => a.path.localeCompare(b.path));
}

/**
 * Returns the uris of "pub roots" for a set of package folders/pubspecs, where "pub root" is