Examples with DisallowedUsageException com.sk89q.worldedit.extension.input.DisallowedUsageException used on opensource projects

Example 1 with DisallowedUsageException

use of com.sk89q.worldedit.extension.input.DisallowedUsageException in project FastAsyncWorldEdit by IntellectualSites.

the class DefaultBlockParser method parseProperties.

// FAWE start - make public
public static Map<Property<?>, Object> parseProperties(// FAWE end
BlockType type, String[] stateProperties, ParserContext context, // FAWE start - if null should be returned instead of throwing an error
boolean nullNotError) throws // FAWE end
NoMatchException {
    Map<Property<?>, Object> blockStates = new HashMap<>();
    // FAWE start - disallowed states
    if (context != null && context.getActor() != null && !context.getActor().getLimit().isUnlimited()) {
        for (String input : context.getActor().getLimit().DISALLOWED_BLOCKS) {
            if (input.indexOf('[') == -1 && input.indexOf(']') == -1) {
                continue;
            }
            if (!type.getId().equalsIgnoreCase(input.substring(0, input.indexOf('[')))) {
                continue;
            }
            String[] properties = input.substring(input.indexOf('[') + 1, input.indexOf(']')).split(",");
            Set<String> blocked = Arrays.stream(properties).filter(s -> {
                for (String in : stateProperties) {
                    if (in.equalsIgnoreCase(s)) {
                        return true;
                    }
                }
                return false;
            }).collect(Collectors.toSet());
            if (!blocked.isEmpty()) {
                throw new DisallowedUsageException(Caption.of("fawe.error.limit.disallowed-block", TextComponent.of(input)));
            }
        }
    }
    if (stateProperties.length > 0) {
        // Parse the block data (optional)
        for (String parseableData : stateProperties) {
            try {
                String[] parts = parseableData.split("=");
                if (parts.length != 2) {
                    // FAWE start - if null should be returned instead of throwing an error
                    if (nullNotError) {
                        return null;
                    }
                    // FAWE end
                    throw new InputParseException(Caption.of("worldedit.error.parser.bad-state-format", TextComponent.of(parseableData)));
                }
                @SuppressWarnings("unchecked") Property<Object> propertyKey = (Property<Object>) type.getPropertyMap().get(parts[0]);
                if (propertyKey == null) {
                    // FAWE start - nullable context
                    if (context != null && context.getActor() != null) {
                        // FAWE start - if null should be returned instead of throwing an error
                        if (nullNotError) {
                            return null;
                        }
                        // FAWE end
                        throw new NoMatchException(Caption.of("worldedit.error.parser.unknown-property", TextComponent.of(parts[0]), TextComponent.of(type.getId())));
                    } else {
                        WorldEdit.logger.debug("Unknown property " + parts[0] + " for block " + type.getId());
                    }
                    return Maps.newHashMap();
                }
                if (blockStates.containsKey(propertyKey)) {
                    // FAWE start - if null should be returned instead of throwing an error
                    if (nullNotError) {
                        return null;
                    }
                    // FAWE end
                    throw new InputParseException(Caption.of("worldedit.error.parser.duplicate-property", TextComponent.of(parts[0])));
                }
                Object value;
                try {
                    value = propertyKey.getValueFor(parts[1]);
                } catch (IllegalArgumentException e) {
                    // FAWE start - if null should be returned instead of throwing an error
                    if (nullNotError) {
                        return null;
                    }
                    // FAWE end
                    throw new NoMatchException(Caption.of("worldedit.error.parser.unknown-value", TextComponent.of(parts[1]), TextComponent.of(propertyKey.getName())));
                }
                // FAWE start - blocked states
                if (context != null && context.getActor() != null && !context.getActor().getLimit().isUnlimited()) {
                    if (context.getActor().getLimit().REMAP_PROPERTIES != null && !context.getActor().getLimit().REMAP_PROPERTIES.isEmpty()) {
                        for (PropertyRemap remap : context.getActor().getLimit().REMAP_PROPERTIES) {
                            Object newValue = remap.apply(type, value);
                            if (newValue != value) {
                                value = newValue;
                                break;
                            }
                        }
                    }
                }
                // FAWE end
                blockStates.put(propertyKey, value);
            } catch (NoMatchException | DisallowedUsageException e) {
                // Pass-through
                throw e;
            } catch (Exception e) {
                // FAWE start - if null should be returned instead of throwing an error
                if (nullNotError) {
                    return null;
                }
                // FAWE end
                throw new InputParseException(Caption.of("worldedit.error.parser.bad-state-format", TextComponent.of(parseableData)));
            }
        }
    }
    return blockStates;
}

Example 2 with DisallowedUsageException

use of com.sk89q.worldedit.extension.input.DisallowedUsageException in project FastAsyncWorldEdit by IntellectualSites.

the class DefaultBlockParser method parseLogic.

private BaseBlock parseLogic(String input, ParserContext context) throws InputParseException {
    // FAWE start
    String[] blockAndExtraData = input.trim().split("\\|", 2);
    blockAndExtraData[0] = woolMapper(blockAndExtraData[0]);
    Map<Property<?>, Object> blockStates = new HashMap<>();
    // FAWE end
    BlockState state = null;
    // Legacy matcher
    if (context.isTryingLegacy()) {
        try {
            String[] split = blockAndExtraData[0].split(":", 2);
            if (split.length == 0) {
                throw new InputParseException(Caption.of("worldedit.error.parser.invalid-colon"));
            } else if (split.length == 1) {
                state = LegacyMapper.getInstance().getBlockFromLegacy(Integer.parseInt(split[0]));
            } else if (MathMan.isInteger(split[0])) {
                int id = Integer.parseInt(split[0]);
                int data = Integer.parseInt(split[1]);
                // FAWE start
                if (data < 0 || data >= 16) {
                    throw new InputParseException(Caption.of("fawe.error.parser.invalid-data", TextComponent.of(data)));
                }
                state = LegacyMapper.getInstance().getBlockFromLegacy(id, data);
            } else {
                BlockType type = BlockTypes.get(split[0].toLowerCase(Locale.ROOT));
                if (type != null) {
                    int data = Integer.parseInt(split[1]);
                    if (data < 0 || data >= 16) {
                        throw new InputParseException(Caption.of("fawe.error.parser.invalid-data", TextComponent.of(data)));
                    }
                    state = LegacyMapper.getInstance().getBlockFromLegacy(type.getLegacyCombinedId() >> 4, data);
                }
            }
        } catch (NumberFormatException ignored) {
        }
    }
    CompoundTag nbt = null;
    // FAWE end
    if (state == null) {
        String typeString;
        String stateString = null;
        int stateStart = blockAndExtraData[0].indexOf('[');
        if (stateStart == -1) {
            typeString = blockAndExtraData[0];
        } else {
            typeString = blockAndExtraData[0].substring(0, stateStart);
            if (stateStart + 1 >= blockAndExtraData[0].length()) {
                throw new InputParseException(Caption.of("worldedit.error.parser.hanging-lbracket", TextComponent.of(stateStart)));
            }
            int stateEnd = blockAndExtraData[0].lastIndexOf(']');
            if (stateEnd < 0) {
                throw new InputParseException(Caption.of("worldedit.error.parser.missing-rbracket"));
            }
            stateString = blockAndExtraData[0].substring(stateStart + 1, blockAndExtraData[0].length() - 1);
        }
        String[] stateProperties = EMPTY_STRING_ARRAY;
        if (stateString != null) {
            stateProperties = stateString.split(",");
        }
        if (typeString.isEmpty()) {
            throw new InputParseException(Caption.of("worldedit.error.parser.bad-state-format", TextComponent.of(blockAndExtraData[0])));
        }
        if ("hand".equalsIgnoreCase(typeString)) {
            // Get the block type from the item in the user's hand.
            final BaseBlock blockInHand = getBlockInHand(context.requireActor(), HandSide.MAIN_HAND);
            // FAWE start
            state = blockInHand.toBlockState();
            nbt = blockInHand.getNbtData();
        // FAWE end
        } else if ("offhand".equalsIgnoreCase(typeString)) {
            // Get the block type from the item in the user's off hand.
            final BaseBlock blockInHand = getBlockInHand(context.requireActor(), HandSide.OFF_HAND);
            // FAWE start
            state = blockInHand.toBlockState();
            nbt = blockInHand.getNbtData();
        // FAWE end
        } else if (typeString.matches("pos[0-9]+")) {
            int index = Integer.parseInt(typeString.replaceAll("[a-z]+", ""));
            // Get the block type from the "primary position"
            final World world = context.requireWorld();
            final BlockVector3 primaryPosition;
            try {
                primaryPosition = context.requireSession().getRegionSelector(world).getVertices().get(index - 1);
            } catch (IncompleteRegionException e) {
                throw new InputParseException(Caption.of("worldedit.error.incomplete-region"));
            }
            state = world.getBlock(primaryPosition);
            nbt = state.getNbtData();
        // FAWE start
        } else if (typeString.matches("slot[0-9]+")) {
            int slot = Integer.parseInt(typeString.substring(4)) - 1;
            Actor actor = context.requireActor();
            if (!(actor instanceof Player)) {
                throw new InputParseException(Caption.of("worldedit.command.player-only"));
            }
            Player player = (Player) actor;
            BlockBag bag = player.getInventoryBlockBag();
            if (!(bag instanceof SlottableBlockBag)) {
                throw new InputParseException(Caption.of("fawe.error.unsupported"));
            }
            SlottableBlockBag slottable = (SlottableBlockBag) bag;
            BaseItem item = slottable.getItem(slot);
            if (!item.getType().hasBlockType()) {
                throw new InputParseException(Caption.of("worldedit.error.not-a-block"));
            }
            state = item.getType().getBlockType().getDefaultState();
            nbt = item.getNbtData();
        } else {
            BlockType type = BlockTypes.parse(typeString.toLowerCase(Locale.ROOT));
            if (type != null) {
                state = type.getDefaultState();
            }
            if (state == null) {
                throw new NoMatchException(Caption.of("fawe.error.invalid-block-type", TextComponent.of(input)));
            }
        }
        // FAWE end
        // FAWE start -  Not null if nullNotError false.
        blockStates.putAll(parseProperties(state.getBlockType(), stateProperties, context, false));
        // FAWE end
        if (context.isPreferringWildcard()) {
            if (stateString == null || stateString.isEmpty()) {
                state = new FuzzyBlockState(state);
            } else {
                FuzzyBlockState.Builder fuzzyBuilder = FuzzyBlockState.builder();
                fuzzyBuilder.type(state.getBlockType());
                for (Map.Entry<Property<?>, Object> blockState : blockStates.entrySet()) {
                    @SuppressWarnings("unchecked") Property<Object> objProp = (Property<Object>) blockState.getKey();
                    fuzzyBuilder.withProperty(objProp, blockState.getValue());
                }
                state = fuzzyBuilder.build();
            }
        } else {
            for (Map.Entry<Property<?>, Object> blockState : blockStates.entrySet()) {
                @SuppressWarnings("unchecked") Property<Object> objProp = (Property<Object>) blockState.getKey();
                state = state.with(objProp, blockState.getValue());
            }
        }
    }
    // this should be impossible but IntelliJ isn't that smart
    if (state == null) {
        throw new NoMatchException(Caption.of("worldedit.error.unknown-block", TextComponent.of(input)));
    }
    // FAWE start
    if (blockAndExtraData.length > 1 && blockAndExtraData[1].startsWith("{")) {
        String joined = StringMan.join(Arrays.copyOfRange(blockAndExtraData, 1, blockAndExtraData.length), "|");
        try {
            nbt = JSON2NBT.getTagFromJson(joined);
        } catch (NBTException e) {
            throw new NoMatchException(TextComponent.of(e.getMessage()));
        }
    }
    // FAWE end
    // Check if the item is allowed
    BlockType blockType = state.getBlockType();
    if (context.isRestricted()) {
        Actor actor = context.requireActor();
        // FAWE start - per-limit disallowed blocks
        if (actor != null) {
            if (!actor.hasPermission("worldedit.anyblock") && worldEdit.getConfiguration().disallowedBlocks.contains(blockType.getId().toLowerCase(Locale.ROOT))) {
                throw new DisallowedUsageException(Caption.of("worldedit.error.disallowed-block", TextComponent.of(blockType.getId())));
            }
            FaweLimit limit = actor.getLimit();
            if (!limit.isUnlimited()) {
                // during contains.
                if (limit.DISALLOWED_BLOCKS.contains(blockType.getId().toLowerCase(Locale.ROOT))) {
                    throw new DisallowedUsageException(Caption.of("fawe.error.limit.disallowed-block", TextComponent.of(blockType.getId())));
                }
            }
        }
    // FAWE end
    }
    if (nbt != null) {
        BaseBlock result = blockStates.size() > 0 ? state.toBaseBlock(nbt) : new BlanketBaseBlock(state, nbt);
        return validate(context, result);
    }
    if (blockType == BlockTypes.SIGN || blockType == BlockTypes.WALL_SIGN || BlockCategories.SIGNS.contains(blockType)) {
        // Allow special sign text syntax
        String[] text = new String[4];
        text[0] = blockAndExtraData.length > 1 ? blockAndExtraData[1] : "";
        text[1] = blockAndExtraData.length > 2 ? blockAndExtraData[2] : "";
        text[2] = blockAndExtraData.length > 3 ? blockAndExtraData[3] : "";
        text[3] = blockAndExtraData.length > 4 ? blockAndExtraData[4] : "";
        return validate(context, new SignBlock(state, text));
    } else if (blockType == BlockTypes.SPAWNER) {
        // Allow setting mob spawn type
        if (blockAndExtraData.length > 1) {
            String mobName = blockAndExtraData[1];
            EntityType ent = EntityTypes.get(mobName.toLowerCase(Locale.ROOT));
            if (ent == null) {
                throw new NoMatchException(Caption.of("worldedit.error.unknown-entity", TextComponent.of(mobName)));
            }
            mobName = ent.getId();
            if (!worldEdit.getPlatformManager().queryCapability(Capability.USER_COMMANDS).isValidMobType(mobName)) {
                throw new NoMatchException(Caption.of("worldedit.error.unknown-mob", TextComponent.of(mobName)));
            }
            return validate(context, new MobSpawnerBlock(state, mobName));
        } else {
            // noinspection ConstantConditions
            return validate(context, new MobSpawnerBlock(state, EntityTypes.PIG.getId()));
        }
    } else if (blockType == BlockTypes.PLAYER_HEAD || blockType == BlockTypes.PLAYER_WALL_HEAD) {
        // allow setting type/player/rotation
        if (blockAndExtraData.length <= 1) {
            return validate(context, new SkullBlock(state));
        }
        String type = blockAndExtraData[1];
        // valid MC usernames
        return validate(context, new SkullBlock(state, type.replace(" ", "_")));
    } else {
        // FAWE start
        nbt = state.getNbtData();
        BaseBlock result;
        if (nbt != null) {
            result = blockStates.size() > 0 ? state.toBaseBlock(nbt) : new BlanketBaseBlock(state, nbt);
        } else {
            result = blockStates.size() > 0 ? new BaseBlock(state) : state.toBaseBlock();
        }
        return validate(context, result);
    // FAWE end
    }
}

Example 3 with DisallowedUsageException

use of com.sk89q.worldedit.extension.input.DisallowedUsageException in project FastAsyncWorldEdit by IntellectualSites.

the class DefaultBlockParser method validate.

// FAWE Start
private <T extends BlockStateHolder> T validate(ParserContext context, T holder) {
    if (context.isRestricted()) {
        Actor actor = context.requireActor();
        if (!actor.hasPermission("worldedit.anyblock") && worldEdit.getConfiguration().checkDisallowedBlocks(holder)) {
            throw new DisallowedUsageException(Caption.of("worldedit.error.disallowed-block", TextComponent.of(String.valueOf(holder))));
        }
        CompoundTag nbt = holder.getNbtData();
        if (nbt != null) {
            if (!actor.hasPermission("worldedit.anyblock.nbt")) {
                throw new DisallowedUsageException(Caption.of("fawe.error.nbt.forbidden", TextComponent.of("worldedit.anyblock.nbt")));
            }
        }
    }
    return holder;
}