列出用户的 sandbox

export class SandboxDirectory extends DurableObject<Env> {
	constructor(ctx: DurableObjectState, env: Env) {
		super(ctx, env);
		ctx.storage.sql.exec(
			"CREATE TABLE IF NOT EXISTS sandboxes (session TEXT PRIMARY KEY, created_at INTEGER NOT NULL)",
		);
	}

	add(session: string) {
		// Keep the time of the first command when the same session runs again
		this.ctx.storage.sql.exec(
			"INSERT OR IGNORE INTO sandboxes (session, created_at) VALUES (?, ?)",
			session,
			Date.now(),
		);
	}

	remove(session: string) {
		this.ctx.storage.sql.exec("DELETE FROM sandboxes WHERE session = ?", session);
	}

	list() {
		return this.ctx.storage.sql
			.exec<{ session: string }>("SELECT session FROM sandboxes ORDER BY created_at")
			.toArray()
			.map((row) => row.session);
	}
}

const INACTIVITY_TIMEOUT_MS = 10 * 60 * 1000;

export class MyContainer extends DurableObject<Env> {
	constructor(ctx: DurableObjectState, env: Env) {
		super(ctx, env);
		const container = ctx.container;
		if (container?.running) {
			void ctx.blockConcurrencyWhile(() =>
				container.setInactivityTimeout(INACTIVITY_TIMEOUT_MS),
			);
		}
	}
}

await container.setInactivityTimeout(INACTIVITY_TIMEOUT_MS);

export class MyContainer extends DurableObject<Env> {
	// ...

	isRunning() {
		// Reading `running` does not start the container
		return this.ctx.container?.running ?? false;
	}

	async destroy() {
		await this.ctx.container?.destroy();
	}
}

export default {
	async fetch(request: Request, env: Env): Promise<Response> {
		const url = new URL(request.url);
		const [, users, owner, sandboxes, session] = url.pathname.split("/");

		if (users !== "users" || !owner || sandboxes !== "sandboxes") {
			return new Response("Not found", { status: 404 });
		}

		const directory = env.SANDBOX_DIRECTORY.getByName(owner);

		if (request.method === "GET" && !session) {
			const sessions = await directory.list();
			// Call the Durable Object of every session in parallel
			const list = await Promise.all(
				sessions.map(async (name) => ({
					session: name,
					running: await env.MY_CONTAINER.getByName(`${owner}/${name}`).isRunning(),
				})),
			);
			return Response.json(list);
		}

		if (!session) {
			return new Response("Not found", { status: 404 });
		}

		const sandbox = env.MY_CONTAINER.getByName(`${owner}/${session}`);

		if (request.method === "POST") {
			const { argv } = (await request.json()) as { argv: string[] };
			await directory.add(session);
			return Response.json(await sandbox.exec(argv));
		}

		if (request.method === "DELETE") {
			await sandbox.destroy();
			await directory.remove(session);
			return new Response(null, { status: 204 });
		}

		return new Response("Not found", { status: 404 });
	},
};