misc_pterodactyl-panel/app/Extensions/Backups/BackupManager.php

215 lines
4.9 KiB
PHP
Raw Normal View History

<?php
namespace Pterodactyl\Extensions\Backups;
use Closure;
2020-05-04 03:49:09 +00:00
use Aws\S3\S3Client;
use Illuminate\Support\Arr;
use Illuminate\Support\Str;
use Webmozart\Assert\Assert;
use InvalidArgumentException;
use Illuminate\Foundation\Application;
use League\Flysystem\AdapterInterface;
use League\Flysystem\AwsS3v3\AwsS3Adapter;
use League\Flysystem\Memory\MemoryAdapter;
use Illuminate\Contracts\Config\Repository;
class BackupManager
{
/**
* @var \Illuminate\Foundation\Application
*/
protected $app;
/**
* @var \Illuminate\Contracts\Config\Repository
*/
protected $config;
/**
* The array of resolved backup drivers.
*
* @var \League\Flysystem\AdapterInterface[]
*/
protected $adapters = [];
/**
* The registered custom driver creators.
*
* @var array
*/
protected $customCreators;
/**
* BackupManager constructor.
*/
public function __construct(Application $app)
{
$this->app = $app;
$this->config = $app->make(Repository::class);
}
/**
* Returns a backup adapter instance.
*
* @return \League\Flysystem\AdapterInterface
*/
public function adapter(string $name = null)
{
return $this->get($name ?: $this->getDefaultAdapter());
}
/**
* Set the given backup adapter instance.
*
* @param \League\Flysystem\AdapterInterface $disk
2021-01-23 20:33:34 +00:00
*
* @return $this
*/
public function set(string $name, $disk)
{
$this->adapters[$name] = $disk;
return $this;
}
/**
* Gets a backup adapter.
*
* @return \League\Flysystem\AdapterInterface
*/
protected function get(string $name)
{
return $this->adapters[$name] = $this->resolve($name);
}
/**
* Resolve the given backup disk.
*
* @return \League\Flysystem\AdapterInterface
*/
protected function resolve(string $name)
{
$config = $this->getConfig($name);
if (empty($config['adapter'])) {
2021-01-23 20:33:34 +00:00
throw new InvalidArgumentException("Backup disk [{$name}] does not have a configured adapter.");
}
$adapter = $config['adapter'];
if (isset($this->customCreators[$name])) {
return $this->callCustomCreator($config);
}
$adapterMethod = 'create' . Str::studly($adapter) . 'Adapter';
if (method_exists($this, $adapterMethod)) {
$instance = $this->{$adapterMethod}($config);
Assert::isInstanceOf($instance, AdapterInterface::class);
return $instance;
}
throw new InvalidArgumentException("Adapter [{$adapter}] is not supported.");
}
/**
* Calls a custom creator for a given adapter type.
*
* @return \League\Flysystem\AdapterInterface
*/
protected function callCustomCreator(array $config)
{
$adapter = $this->customCreators[$config['adapter']]($this->app, $config);
Assert::isInstanceOf($adapter, AdapterInterface::class);
return $adapter;
}
/**
* Creates a new wings adapter.
*
* @return \League\Flysystem\AdapterInterface
*/
public function createWingsAdapter(array $config)
{
return new MemoryAdapter(null);
}
/**
* Creates a new S3 adapter.
*
* @return \League\Flysystem\AdapterInterface
*/
public function createS3Adapter(array $config)
{
$config['version'] = 'latest';
2021-01-23 20:33:34 +00:00
if (!empty($config['key']) && !empty($config['secret'])) {
$config['credentials'] = Arr::only($config, ['key', 'secret', 'token']);
}
2020-05-04 03:49:09 +00:00
$client = new S3Client($config);
return new AwsS3Adapter($client, $config['bucket'], $config['prefix'] ?? '', $config['options'] ?? []);
}
/**
* Returns the configuration associated with a given backup type.
*
* @return array
*/
protected function getConfig(string $name)
{
return $this->config->get("backups.disks.{$name}") ?: [];
}
/**
* Get the default backup driver name.
*
* @return string
*/
public function getDefaultAdapter()
{
return $this->config->get('backups.default');
}
/**
* Set the default session driver name.
*/
public function setDefaultAdapter(string $name)
{
$this->config->set('backups.default', $name);
}
/**
* Unset the given adapter instances.
*
* @param string|string[] $adapter
2021-01-23 20:33:34 +00:00
*
* @return $this
*/
public function forget($adapter)
{
foreach ((array) $adapter as $adapterName) {
unset($this->adapters[$adapter]);
}
return $this;
}
/**
* Register a custom adapter creator closure.
*
* @return $this
*/
public function extend(string $adapter, Closure $callback)
{
$this->customCreators[$adapter] = $callback;
return $this;
}
}