PlotSquared/Core/src/main/java/com/plotsquared/core/plot/world/DefaultPlotAreaManager.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.Singleton;
import com.plotsquared.core.location.Location;
import com.plotsquared.core.plot.PlotArea;
import com.plotsquared.core.plot.PlotAreaType;
import com.plotsquared.core.plot.PlotWorld;
import com.plotsquared.core.util.StringMan;
import com.sk89q.worldedit.regions.CuboidRegion;
import org.checkerframework.checker.nullness.qual.NonNull;
import org.checkerframework.checker.nullness.qual.Nullable;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;

@Singleton
public class DefaultPlotAreaManager implements PlotAreaManager {

    final PlotArea[] noPlotAreas = new PlotArea[0];
    private final Map<String, PlotWorld> plotWorlds = new HashMap<>();

    @Override
    public @NonNull PlotArea[] getAllPlotAreas() {
        final Set<PlotArea> area = new HashSet<>();
        for (final PlotWorld world : plotWorlds.values()) {
            area.addAll(world.getAreas());
        }
        return area.toArray(new PlotArea[0]);
    }

    @Override
    public @Nullable PlotArea getApplicablePlotArea(final @Nullable Location location) {
        if (location == null) {
            return null;
        }
        final PlotWorld world = this.plotWorlds.get(location.getWorldName());
        if (world == null) {
            return null;
        }
        return world.getArea(location);
    }

    @Override
    public void addPlotArea(final @NonNull PlotArea plotArea) {
        PlotWorld world = this.plotWorlds.get(plotArea.getWorldName());
        if (world != null) {
            if (world instanceof StandardPlotWorld && world.getAreas().isEmpty()) {
                this.plotWorlds.remove(plotArea.getWorldName());
            } else {
                world.addArea(plotArea);
                return;
            }
        }
        if (plotArea.getType() != PlotAreaType.PARTIAL) {
            world = new StandardPlotWorld(plotArea.getWorldName(), plotArea);
        } else {
            world = new ScatteredPlotWorld(plotArea.getWorldName());
            world.addArea(plotArea);
        }
        this.plotWorlds.put(plotArea.getWorldName(), world);
    }

    @Override
    public void removePlotArea(final @NonNull PlotArea area) {
        final PlotWorld world = this.plotWorlds.get(area.getWorldName());
        if (world == null) {
            return;
        }
        if (world instanceof StandardPlotWorld) {
            this.plotWorlds.remove(world.getWorld());
        } else {
            world.removeArea(area);
            if (world.getAreas().isEmpty()) {
                this.plotWorlds.remove(world.getWorld());
            }
        }
    }

    @Override
    public PlotArea getPlotArea(final @NonNull String world, final @Nullable String id) {
        final PlotWorld plotWorld = this.plotWorlds.get(world);
        if (plotWorld == null) {
            return null;
        }
        final List<PlotArea> areas = new ArrayList<>(plotWorld.getAreas());
        if (areas.size() == 1) {
            return areas.get(0);
        }
        if (id == null) {
            return null;
        }
        for (final PlotArea area : areas) {
            if (StringMan.isEqual(id, area.getId())) {
                return area;
            }
        }
        return null;
    }

    @Override
    public @Nullable PlotArea getPlotArea(final @NonNull Location location) {
        return this.getApplicablePlotArea(location);
    }

    @Override
    public @NonNull PlotArea[] getPlotAreas(final @NonNull String world, final @Nullable CuboidRegion region) {
        final PlotWorld plotWorld = this.plotWorlds.get(world);
        if (plotWorld == null) {
            return noPlotAreas;
        }
        if (region == null) {
            return plotWorld.getAreas().toArray(new PlotArea[0]);
        }
        return plotWorld.getAreasInRegion(region).toArray(new PlotArea[0]);
    }

    @Override
    public void addWorld(final @NonNull String worldName) {
        PlotWorld world = this.plotWorlds.get(worldName);
        if (world != null) {
            return;
        }
        // Create a new empty world. When a new area is added
        // the world will be re-recreated with the correct type
        world = new StandardPlotWorld(worldName, null);
        this.plotWorlds.put(worldName, world);
    }

    @Override
    public void removeWorld(final @NonNull String worldName) {
        this.plotWorlds.remove(worldName);
    }

    @Override
    public @NonNull String[] getAllWorlds() {
        return this.plotWorlds.keySet().toArray(new String[0]);
    }

}