GeneratorManager.php

<?php
 
/*
 *
 *  ____            _        _   __  __ _                  __  __ ____
 * |  _ \ ___   ___| | _____| |_|  \/  (_)_ __   ___      |  \/  |  _ \
 * | |_) / _ \ / __| |/ / _ \ __| |\/| | | '_ \ / _ \_____| |\/| | |_) |
 * |  __/ (_) | (__|   <  __/ |_| |  | | | | | |  __/_____| |  | |  __/
 * |_|   \___/ \___|_|\_\___|\__|_|  |_|_|_| |_|\___|     |_|  |_|_|
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * @author PocketMine Team
 * @link http://www.pocketmine.net/
 *
 *
 */
 
declare(strict_types=1);
 
namespace pocketmine\world\generator;
 
use pocketmine\math\Vector3;
use pocketmine\utils\SingletonTrait;
use pocketmine\utils\Utils;
use pocketmine\world\generator\hell\Nether;
use pocketmine\world\generator\normal\Normal;
use function array_keys;
use function strtolower;
 
final class GeneratorManager{
    use SingletonTrait;
 
    private array $list = [];
 
    public function __construct(){
        $this->addGenerator(Flat::class, "flat", function(string $preset) : ?InvalidGeneratorOptionsException{
            if($preset === ""){
                return null;
            }
            try{
                FlatGeneratorOptions::parsePreset($preset);
                return null;
            }catch(InvalidGeneratorOptionsException $e){
                return $e;
            }
        }, fast: true);
        $this->addGenerator(Normal::class, "normal", fn() => null);
        $this->addAlias("normal", "default");
        $this->addGenerator(Nether::class, "nether", fn() => null);
        $this->addAlias("nether", "hell");
        $this->addGenerator(VoidGenerator::class, "void", fn() => null, fast: true, spawnPositionProvider: fn(int $seed) => new Vector3(0, -63, 0));
    }
 
    public function addGenerator(string $class, string $name, \Closure $presetValidator, bool $overwrite = false, bool $fast = false, ?\Closure $spawnPositionProvider = null) : void{
        Utils::testValidInstance($class, Generator::class);
 
        $name = strtolower($name);
        if(!$overwrite && isset($this->list[$name])){
            throw new \InvalidArgumentException("Alias \"$name\" is already assigned");
        }
 
        $this->list[$name] = new GeneratorManagerEntry($class, $presetValidator, $fast, $spawnPositionProvider);
    }
 
    public function addAlias(string $name, string $alias) : void{
        $name = strtolower($name);
        $alias = strtolower($alias);
        if(!isset($this->list[$name])){
            throw new \InvalidArgumentException("Alias \"$name\" is not assigned");
        }
        if(isset($this->list[$alias])){
            throw new \InvalidArgumentException("Alias \"$alias\" is already assigned");
        }
        $this->list[$alias] = $this->list[$name];
    }
 
    public function getGeneratorList() : array{
        return array_keys($this->list);
    }
 
    public function getGenerator(string $name) : ?GeneratorManagerEntry{
        return $this->list[strtolower($name)] ?? null;
    }
 
    public function getGeneratorName(string $class) : string{
        Utils::testValidInstance($class, Generator::class);
        foreach(Utils::stringifyKeys($this->list) as $name => $c){
            if($c->getGeneratorClass() === $class){
                return $name;
            }
        }
 
        throw new \InvalidArgumentException("Generator class $class is not registered");
    }
}