148 lines
		
	
	
		
			4.6 KiB
		
	
	
	
		
			PHP
		
	
	
	
			
		
		
	
	
			148 lines
		
	
	
		
			4.6 KiB
		
	
	
	
		
			PHP
		
	
	
	
<?php
 | 
						|
 | 
						|
namespace BookStack\Access;
 | 
						|
 | 
						|
use BookStack\Exceptions\SocialDriverNotConfigured;
 | 
						|
use Illuminate\Support\Facades\Event;
 | 
						|
use Illuminate\Support\Str;
 | 
						|
use SocialiteProviders\Manager\SocialiteWasCalled;
 | 
						|
 | 
						|
class SocialDriverManager
 | 
						|
{
 | 
						|
    /**
 | 
						|
     * The default built-in social drivers we support.
 | 
						|
     *
 | 
						|
     * @var string[]
 | 
						|
     */
 | 
						|
    protected array $validDrivers = [
 | 
						|
        'google',
 | 
						|
        'github',
 | 
						|
        'facebook',
 | 
						|
        'slack',
 | 
						|
        'twitter',
 | 
						|
        'azure',
 | 
						|
        'okta',
 | 
						|
        'gitlab',
 | 
						|
        'twitch',
 | 
						|
        'discord',
 | 
						|
    ];
 | 
						|
 | 
						|
    /**
 | 
						|
     * Callbacks to run when configuring a social driver
 | 
						|
     * for an initial redirect action.
 | 
						|
     * Array is keyed by social driver name.
 | 
						|
     * Callbacks are passed an instance of the driver.
 | 
						|
     *
 | 
						|
     * @var array<string, callable>
 | 
						|
     */
 | 
						|
    protected array $configureForRedirectCallbacks = [];
 | 
						|
 | 
						|
    /**
 | 
						|
     * Check if the current config for the given driver allows auto-registration.
 | 
						|
     */
 | 
						|
    public function isAutoRegisterEnabled(string $driver): bool
 | 
						|
    {
 | 
						|
        return $this->getDriverConfigProperty($driver, 'auto_register') === true;
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Check if the current config for the given driver allow email address auto-confirmation.
 | 
						|
     */
 | 
						|
    public function isAutoConfirmEmailEnabled(string $driver): bool
 | 
						|
    {
 | 
						|
        return $this->getDriverConfigProperty($driver, 'auto_confirm') === true;
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Gets the names of the active social drivers, keyed by driver id.
 | 
						|
     * @returns array<string, string>
 | 
						|
     */
 | 
						|
    public function getActive(): array
 | 
						|
    {
 | 
						|
        $activeDrivers = [];
 | 
						|
 | 
						|
        foreach ($this->validDrivers as $driverKey) {
 | 
						|
            if ($this->checkDriverConfigured($driverKey)) {
 | 
						|
                $activeDrivers[$driverKey] = $this->getName($driverKey);
 | 
						|
            }
 | 
						|
        }
 | 
						|
 | 
						|
        return $activeDrivers;
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Get the configure-for-redirect callback for the given driver.
 | 
						|
     * This is a callable that allows modification of the driver at redirect time.
 | 
						|
     * Commonly used to perform custom dynamic configuration where required.
 | 
						|
     * The callback is passed a \Laravel\Socialite\Contracts\Provider instance.
 | 
						|
     */
 | 
						|
    public function getConfigureForRedirectCallback(string $driver): callable
 | 
						|
    {
 | 
						|
        return $this->configureForRedirectCallbacks[$driver] ?? (fn() => true);
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Add a custom socialite driver to be used.
 | 
						|
     * Driver name should be lower_snake_case.
 | 
						|
     * Config array should mirror the structure of a service
 | 
						|
     * within the `Config/services.php` file.
 | 
						|
     * Handler should be a Class@method handler to the SocialiteWasCalled event.
 | 
						|
     */
 | 
						|
    public function addSocialDriver(
 | 
						|
        string $driverName,
 | 
						|
        array $config,
 | 
						|
        string $socialiteHandler,
 | 
						|
        callable $configureForRedirect = null
 | 
						|
    ) {
 | 
						|
        $this->validDrivers[] = $driverName;
 | 
						|
        config()->set('services.' . $driverName, $config);
 | 
						|
        config()->set('services.' . $driverName . '.redirect', url('/login/service/' . $driverName . '/callback'));
 | 
						|
        config()->set('services.' . $driverName . '.name', $config['name'] ?? $driverName);
 | 
						|
        Event::listen(SocialiteWasCalled::class, $socialiteHandler);
 | 
						|
        if (!is_null($configureForRedirect)) {
 | 
						|
            $this->configureForRedirectCallbacks[$driverName] = $configureForRedirect;
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Get the presentational name for a driver.
 | 
						|
     */
 | 
						|
    protected function getName(string $driver): string
 | 
						|
    {
 | 
						|
        return $this->getDriverConfigProperty($driver, 'name') ?? '';
 | 
						|
    }
 | 
						|
 | 
						|
    protected function getDriverConfigProperty(string $driver, string $property): mixed
 | 
						|
    {
 | 
						|
        return config("services.{$driver}.{$property}");
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Ensure the social driver is correct and supported.
 | 
						|
     *
 | 
						|
     * @throws SocialDriverNotConfigured
 | 
						|
     */
 | 
						|
    public function ensureDriverActive(string $driverName): void
 | 
						|
    {
 | 
						|
        if (!in_array($driverName, $this->validDrivers)) {
 | 
						|
            abort(404, trans('errors.social_driver_not_found'));
 | 
						|
        }
 | 
						|
 | 
						|
        if (!$this->checkDriverConfigured($driverName)) {
 | 
						|
            throw new SocialDriverNotConfigured(trans('errors.social_driver_not_configured', ['socialAccount' => Str::title($driverName)]));
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Check a social driver has been configured correctly.
 | 
						|
     */
 | 
						|
    protected function checkDriverConfigured(string $driver): bool
 | 
						|
    {
 | 
						|
        $lowerName = strtolower($driver);
 | 
						|
        $configPrefix = 'services.' . $lowerName . '.';
 | 
						|
        $config = [config($configPrefix . 'client_id'), config($configPrefix . 'client_secret'), config('services.callback_url')];
 | 
						|
 | 
						|
        return !in_array(false, $config) && !in_array(null, $config);
 | 
						|
    }
 | 
						|
}
 |