use Cognesy\Agents\Hooks\Contracts\HookInterface;
use Cognesy\Agents\Hooks\Data\HookContext;

class LogStepsHook implements HookInterface
{
    public function handle(HookContext $context): HookContext
    {
        $steps = $context->state()->stepCount();
        echo "Current step: {$steps}\n";
        return $context;
    }
}
// @doctest id="b673"

use Cognesy\Agents\Hooks\Collections\HookTriggers;
use Cognesy\Agents\Hooks\Collections\RegisteredHooks;
use Cognesy\Agents\Hooks\Interceptors\HookStack;

$stack = new HookStack(new RegisteredHooks());
$stack = $stack->with(
    hook: new LogStepsHook(),
    triggerTypes: HookTriggers::afterStep(),
    priority: 10,
    name: 'log_steps',
);

$loop = AgentLoop::default()->withInterceptor($stack);
// @doctest id="e52d"

use Cognesy\Agents\Hooks\Defaults\CallableHook;

$hook = new CallableHook(function (HookContext $ctx): HookContext {
    echo "Step done!\n";
    return $ctx;
});

$stack = $stack->with($hook, HookTriggers::afterStep());
// @doctest id="3a72"

class BlockDangerousTools implements HookInterface
{
    public function handle(HookContext $context): HookContext
    {
        if ($context->toolCall()?->name() === 'dangerous_tool') {
            return $context->withToolExecutionBlocked('Not allowed');
        }
        return $context;
    }
}

$stack = $stack->with(new BlockDangerousTools(), HookTriggers::beforeToolUse());
// @doctest id="17c2"

$hook = new CallableHook(function (HookContext $ctx): HookContext {
    $state = $ctx->state()->withMetadata('custom_key', 'value');
    return $ctx->withState($state);
});
// @doctest id="b372"

use Cognesy\Agents\Hooks\Guards\StepsLimitHook;
use Cognesy\Agents\Hooks\Guards\TokenUsageLimitHook;
use Cognesy\Agents\Hooks\Guards\ExecutionTimeLimitHook;

// Stop after 10 steps
$stepsGuard = new StepsLimitHook(
    maxSteps: 10,
    stepCounter: fn($state) => $state->stepCount(),
);

// Stop after 5000 tokens
$tokenGuard = new TokenUsageLimitHook(maxTotalTokens: 5000);

// Stop after 30 seconds
$timeGuard = new ExecutionTimeLimitHook(maxSeconds: 30.0);

$stack = $stack
    ->with($stepsGuard, HookTriggers::beforeStep(), priority: 100)
    ->with($tokenGuard, HookTriggers::beforeStep(), priority: 100)
    ->with($timeGuard, HookTriggers::with(HookTrigger::BeforeExecution, HookTrigger::BeforeStep), priority: 100);
// @doctest id="e4f8"