<?php

namespace RoosterLMS\UI;

defined( 'ABSPATH' ) || exit;

final class Output {

	public static function capture( callable $callback ): string {
		$level = ob_get_level();
		ob_start();
		try {
			$callback();
		} catch ( \Throwable $e ) {
			while ( ob_get_level() > $level ) {
				ob_end_clean();
			}
			throw $e;
		}
		$inner = '';
		while ( ob_get_level() > $level + 1 ) {
			$inner = (string) ob_get_clean() . $inner;
		}
		return ( ob_get_level() > $level ? (string) ob_get_clean() : '' ) . $inner;
	}
}