upstream-tldraw/internal/scripts/lib/workspace.ts

import path from 'path'
import { glob } from 'glob'
import { REPO_ROOT, readJsonIfExists } from './file'
import { PackageJson } from './types'

export interface Package {
	packageJson: PackageJson
	relativePath: string
	path: string
	name: string
}

async function readPackage(packageJsonFile: string): Promise<Package> {
	const packageJsonPath = path.resolve(packageJsonFile)
	const packagePath = path.dirname(packageJsonPath)
	const relativePath = path.relative(REPO_ROOT, packagePath)

	const packageJsonRaw = await readJsonIfExists(packageJsonFile)
	if (!packageJsonRaw) {
		throw new Error(`No package.json found at ${packageJsonPath}`)
	}

	let packageJson: PackageJson
	try {
		packageJson = PackageJson.validate(packageJsonRaw)
	} catch (e: any) {
		throw new Error(`Invalid package.json in ${relativePath}: ${e.message}`)
	}

	return {
		packageJson,
		relativePath,
		path: packagePath,
		name: packageJson.name,
	}
}

async function getChildWorkspaces(parent: Package): Promise<Package[]> {
	if (!parent.packageJson.workspaces) return []

	const foundPackages = []
	for (const workspace of parent.packageJson.workspaces) {
		const workspacePath = path.join(parent.path, workspace)
		for (const packageJsonFilePath of glob.sync(path.join(workspacePath, 'package.json'))) {
			const child = await readPackage(packageJsonFilePath)
			foundPackages.push(child)
			if (child.packageJson.workspaces) {
				foundPackages.push(...(await getChildWorkspaces(child)))
			}
		}
	}

	return foundPackages
}

export async function getRootPackage() {
	return await readPackage(path.join(REPO_ROOT, 'package.json'))
}

export async function getAllWorkspacePackages() {
	return await getChildWorkspaces(await getRootPackage())
}