PlotSquared/Core/src/main/java/com/plotsquared/core/plot/world/SinglePlotAreaManager.java

/*
 * PlotSquared, a land and world management plugin for Minecraft.
 * Copyright (C) IntellectualSites <https://intellectualsites.com>
 * Copyright (C) IntellectualSites team and contributors
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */
package com.plotsquared.core.plot.world;

import com.google.inject.Inject;
import com.google.inject.Singleton;
import com.plotsquared.core.collection.ArrayUtil;
import com.plotsquared.core.configuration.file.YamlConfiguration;
import com.plotsquared.core.generator.SingleWorldGenerator;
import com.plotsquared.core.inject.annotations.WorldConfig;
import com.plotsquared.core.listener.PlotListener;
import com.plotsquared.core.location.Location;
import com.plotsquared.core.plot.PlotArea;
import com.plotsquared.core.queue.GlobalBlockQueue;
import com.plotsquared.core.util.EventDispatcher;
import com.plotsquared.core.util.SetupUtils;
import com.sk89q.worldedit.regions.CuboidRegion;
import org.checkerframework.checker.nullness.qual.NonNull;
import org.checkerframework.checker.nullness.qual.Nullable;

@Singleton
public class SinglePlotAreaManager extends DefaultPlotAreaManager {

    private final SinglePlotArea[] array;
    private SinglePlotArea area;
    private PlotArea[] all;

    @Inject
    public SinglePlotAreaManager(
            final @NonNull EventDispatcher eventDispatcher,
            final @NonNull PlotListener plotListener,
            @WorldConfig final @NonNull YamlConfiguration worldConfiguration,
            final @NonNull GlobalBlockQueue blockQueue
    ) {
        this.area = new SinglePlotArea(this, eventDispatcher, plotListener,
                worldConfiguration, blockQueue
        );
        this.array = new SinglePlotArea[]{area};
        this.all = new PlotArea[]{area};
        SetupUtils.generators.put(
                "PlotSquared:single",
                new SingleWorldGenerator(this).specify("CheckingPlotSquaredGenerator")
        );
    }

    public SinglePlotArea getArea() {
        return area;
    }

    public void setArea(final @NonNull SinglePlotArea area) {
        this.area = area;
        array[0] = area;
        all = ArrayUtil.concatAll(super.getAllPlotAreas(), array);
    }

    public boolean isWorld(final @NonNull String id) {
        char[] chars = id.toCharArray();
        if (chars.length == 1 && chars[0] == '*') {
            return true;
        }
        int mode = 0;
        for (char c : chars) {
            switch (mode) {
                case 0:
                    mode = 1;
                    if (c == '-') {
                        continue;
                    }
                case 1:
                    if ((c <= '/') || (c >= ':')) {
                        if (c == '_') {
                            mode = 2;
                            continue;
                        }
                        return false;
                    } else {
                        continue;
                    }
                case 2:
                    mode = 3;
                    if (c == '-') {
                        continue;
                    }
                case 3:
                    if ((c <= '/') || (c >= ':')) {
                        return false;
                    }
            }
        }
        return mode == 3;
    }

    @Override
    public @Nullable PlotArea getApplicablePlotArea(final @Nullable Location location) {
        if (location == null) {
            return null;
        }
        String world = location.getWorldName();
        return isWorld(world) || world.equals("*") || super.getAllPlotAreas().length == 0 ?
                area :
                super.getApplicablePlotArea(location);
    }

    @Override
    public @Nullable PlotArea getPlotArea(final @NonNull String world, final @NonNull String id) {
        PlotArea found = super.getPlotArea(world, id);
        if (found != null) {
            return found;
        }
        return isWorld(world) || world.equals("*") ? area : super.getPlotArea(world, id);
    }

    @Override
    public @Nullable PlotArea getPlotArea(final @NonNull Location location) {
        PlotArea found = super.getPlotArea(location);
        if (found != null) {
            return found;
        }
        return isWorld(location.getWorldName()) || location.getWorldName().equals("*") ? area : null;
    }

    @Override
    public @NonNull PlotArea[] getPlotAreas(final @NonNull String world, final @NonNull CuboidRegion region) {
        PlotArea[] found = super.getPlotAreas(world, region);
        if (found != null && found.length != 0) {
            return found;
        }
        return isWorld(world) || world.equals("*") ? array : all.length == 0 ? noPlotAreas : found;
    }

    @Override
    public @NonNull PlotArea[] getAllPlotAreas() {
        return all;
    }

    @Override
    public @NonNull String[] getAllWorlds() {
        return super.getAllWorlds();
    }

    @Override
    public void addPlotArea(final @NonNull PlotArea area) {
        if (area == this.area) {
            return;
        }
        super.addPlotArea(area);
        all = ArrayUtil.concatAll(super.getAllPlotAreas(), array);
    }

    @Override
    public void removePlotArea(final @NonNull PlotArea area) {
        if (area == this.area) {
            throw new UnsupportedOperationException("Cannot remove base area!");
        }
        super.removePlotArea(area);
    }

    @Override
    public void addWorld(final @NonNull String worldName) {
        super.addWorld(worldName);
    }

    @Override
    public void removeWorld(final @NonNull String worldName) {
        super.removeWorld(worldName);
    }

}