divineniiquaye / rade-di

src/Facade/FacadeProxy.php

<?php

declare(strict_types=1);

/*
 * This file is part of DivineNii opensource projects.
 *
 * PHP version 7.4 and above required
 *
 * @author    Divine Niiquaye Ibok <divineibok@gmail.com>
 * @copyright 2021 DivineNii (https://divinenii.com/)
 * @license   https://opensource.org/licenses/BSD-3-Clause License
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

namespace Rade\DI\Facade;

use PhpParser\BuilderFactory;
use PhpParser\Node\{
    Expr\StaticPropertyFetch,
    Name,
    Stmt\Declare_,
    Stmt\DeclareDeclare,
    Stmt\Return_,
    UnionType
};
use Psr\Container\ContainerInterface;
use Rade\DI\Builder\CodePrinter;
use Rade\DI\{ContainerBuilder, Definition};

/**
 * A Proxy manager for implementing laravel like facade system.
 *
 * @author Divine Niiquaye Ibok <divineibok@gmail.com>
 */
class FacadeProxy
{
    private ContainerInterface $container;

    /** @var array<string,string> */
    private array $proxies = [];

    public function __construct(ContainerInterface $container)
    {
        $this->container = $container;
    }

    /**
     * Register(s) service{s) found in container as shared proxy facade(s).
     */
    public function proxy(string ...$services): void
    {
        foreach ($services as $service) {
            $id = \str_replace(['.', '_', '\\'], '', \lcfirst(\ucwords($service, '._')));

            if (!$this->container instanceof ContainerBuilder) {
                Facade::$proxies[$id] = $service;

                continue;
            }

            $this->proxies[$id] = $service;
        }
    }

    /**
     * This build method works with container builder.
     *
     * @param string $className for compiled facade class
     */
    public function build(string $className = 'Facade'): ?string
    {
        /** @var ContainerBuilder */
        $container = $this->container;

        if ([] !== $proxiedServices = $this->proxies) {
            $astNodes = [];
            $builder = $container->getBuilder();

The method getBuilder() does not exist on Psr\Container\ContainerInterface. It seems like you code against a sub-type of Psr\Container\ContainerInterface such as Rade\DI\AbstractContainer.

            \ksort($proxiedServices);

            $astNodes[] = new Declare_([new DeclareDeclare('strict_types', $builder->val(1))]);
            $classNode = $builder->class($className)->extend('\Rade\DI\Facade\Facade')->setDocComment(CodePrinter::COMMENT);

            $classNode->addStmts($this->resolveProxies($container, $builder, $proxiedServices));
            $astNodes[] = $classNode->getNode();

            return CodePrinter::print($astNodes);
        }

        return null;
    }

    /**
     * This method resolves the proxies from container builder.
     *
     * @param string[] $proxiedServices
     *
     * @return \PhpParser\Builder\Method[]
     */
    protected function resolveProxies(ContainerBuilder $container, BuilderFactory $builder, array $proxiedServices): array
    {
        $builtProxies = [];

        foreach ($proxiedServices as $method => $proxy) {
            if (!$container->has($proxy)) {
                continue;
            }

            $definition = $container->service($proxy);
            $proxyNode = $builder->method($method)->makePublic()->makeStatic();

            if ($definition instanceof Definition) {
                if (!$definition->isPublic()) {
                    continue;
                }

                if (!empty($type = $definition->getType())) {
                    $proxyNode->setReturnType(\is_array($type) ? new UnionType(\array_map(fn ($type) => new Name($type), $type)) : $type);

It seems like is_array($type) ? new PhpParser\Node\UnionType(array_map(function(...) { /* ... */ }, $type)) : $type can also be of type PhpParser\Node\UnionType; however, parameter $type of PhpParser\Builder\FunctionLike::setReturnType() does only seem to accept PhpParser\Node\Name|PhpParser\Node\NullableType|string, maybe add an additional type check?

                }
            }

            $body = $builder->methodCall(new StaticPropertyFetch(new Name('self'), 'container'), 'get', [$proxy]);
            $builtProxies[] = $proxyNode->addStmt(new Return_($body));
        }

        return $builtProxies;
    }
}