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);
							 | 
						||
| 
								 | 
							
								    }
							 | 
						||
| 
								 | 
							
								}
							 |