Function getPubWorkspaceFolderOrPackageFolderPath

src/shared/vscode/pub.ts:311–362

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

 * containing folder).
 */
export function getPubWorkspaceFolderOrPackageFolderPath(
	packageFolderOrPubspecPath: string,
	existsSync: (itemPath: string) => boolean = fs.existsSync,
	readFileSync: (itemPath: string) => string = (p) => fs.readFileSync(p, "utf8").toString(),
): string {
	const pubspecPath = path.basename(packageFolderOrPubspecPath) === "pubspec.yaml"
		? packageFolderOrPubspecPath
		: path.join(packageFolderOrPubspecPath, "pubspec.yaml");
	const packageFolderPath = path.dirname(pubspecPath);

	// We shouldn't really have gotten here without a pubspec because that means this isn't
	// a package, but in that case just return the containing folder.
	if (!existsSync(pubspecPath))
		return packageFolderPath;

	let pubspecContent: string;
	try {
		pubspecContent = readFileSync(pubspecPath);
	} catch {
		return packageFolderPath;
	}

	// If this is not a Pub Workspace project, just return the package itself.
	if (!pubspecIsWorkspaceProjectRegex.test(pubspecContent))
		return packageFolderPath;

	// Walk up the tree to find the workspace root.
	let currentFolder = path.dirname(packageFolderPath);
	while (true) {
		// Check if the current folder is the workspace root
		try {
			const currentPubspecPath = path.join(currentFolder, "pubspec.yaml");
			const currentPubspecContent = readFileSync(currentPubspecPath);

			// We've found the root.
			if (pubspecIsWorkspaceRootRegex.test(currentPubspecContent))
				return currentFolder;
		} catch {
			// A pubspec may not exist at every level.
		}

		// Otherwise, continue up.
		const parent = path.dirname(currentFolder);

		// If we get to the root of the filesystem without finding, use the original package.
		if (parent === currentFolder)
			return packageFolderPath;

		// Otherwise, loop.
		currentFolder = parent;
	}
}