<?php

/**
 * Mockery (https://docs.mockery.io/en/stable/)
 *
 * @copyright https://github.com/mockery/mockery/blob/HEAD/COPYRIGHT.md
 * @license   https://github.com/mockery/mockery/blob/HEAD/LICENSE BSD 3-Clause License
 * @see       https://github.com/mockery/mockery for the canonical source repository
 */

namespace Mockery\Generator\StringManipulation\Pass;

use Mockery\Generator\MockConfiguration;
use Override;

use function strrpos;
use function substr;

class InstanceMockPass implements Pass
{
    public const INSTANCE_MOCK_CODE = <<<MOCK

            protected \$_mockery_ignoreVerification = true;

            public function __construct()
            {
                \$this->_mockery_ignoreVerification = false;
                \$associatedRealObject = \Mockery::fetchMock(__CLASS__);

                foreach (get_object_vars(\$this) as \$attr => \$val) {
                    if (\$attr !== "_mockery_ignoreVerification" && \$attr !== "_mockery_expectations") {
                        \$this->\$attr = \$associatedRealObject->\$attr;
                    }
                }

                \$directors = \$associatedRealObject->mockery_getExpectations();
                foreach (\$directors as \$method=>\$director) {
                    // get the director method needed
                    \$existingDirector = \$this->mockery_getExpectationsFor(\$method);
                    if (!\$existingDirector) {
                        \$existingDirector = new \Mockery\ExpectationDirector(\$method, \$this);
                        \$this->mockery_setExpectationsFor(\$method, \$existingDirector);
                    }
                    \$expectations = \$director->getExpectations();
                    foreach (\$expectations as \$expectation) {
                        \$clonedExpectation = clone \$expectation;
                        \$existingDirector->addExpectation(\$clonedExpectation);
                    }
                    \$defaultExpectations = \$director->getDefaultExpectations();
                    foreach (array_reverse(\$defaultExpectations) as \$expectation) {
                        \$clonedExpectation = clone \$expectation;
                        \$existingDirector->addExpectation(\$clonedExpectation);
                        \$existingDirector->makeExpectationDefault(\$clonedExpectation);
                    }
                }
                \Mockery::getContainer()->rememberMock(\$this);

                \$this->_mockery_constructorCalled(func_get_args());
            }
        MOCK;

    /**
     * @param  string $code
     * @return string
     */
    #[Override]
    public function apply($code, MockConfiguration $config)
    {
        if ($config->isInstanceMock()) {
            return $this->appendToClass($code, static::INSTANCE_MOCK_CODE);
        }

        return $code;
    }

    protected function appendToClass($class, $code)
    {
        $lastBrace = strrpos($class, '}');

        return substr($class, 0, $lastBrace) . $code . "\n    }\n";
    }
}
