Add MapFlag (#375)

* Add MapFlag

Signed-off-by: Mark <33119mark@gmail.com>

* Remove apache import

Signed-off-by: Mark <33119mark@gmail.com>

* Add copyright

Signed-off-by: Mark <33119mark@gmail.com>
This commit is contained in:
Mark Vogel 2018-12-30 00:20:12 -07:00 committed by Matthew Miller
parent b48e145508
commit cd39387e7f

View File

@ -0,0 +1,123 @@
/*
* WorldGuard, a suite of tools for Minecraft
* Copyright (C) sk89q <http://www.sk89q.com>
* Copyright (C) WorldGuard team and contributors
*
* 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.
*
* 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 Lesser General Public License
* for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
package com.sk89q.worldguard.protection.flags;
import com.google.common.collect.Maps;
import javax.annotation.Nullable;
import java.util.Map;
import java.util.Map.Entry;
/**
* Stores a key value map of typed {@link Flag}s.
*/
public class MapFlag<K, V> extends Flag<Map<K, V>> {
private final Flag<K> keyFlag;
private final Flag<V> valueFlag;
public MapFlag(final String name, final Flag<K> keyFlag, final Flag<V> valueFlag) {
super(name);
this.keyFlag = keyFlag;
this.valueFlag = valueFlag;
}
public MapFlag(final String name, @Nullable final RegionGroup defaultGroup, final Flag<K> keyFlag, final Flag<V> valueFlag) {
super(name, defaultGroup);
this.keyFlag = keyFlag;
this.valueFlag = valueFlag;
}
/**
* Get the flag that is stored as the key flag type.
*
* @return The key flag type.
*/
public Flag<K> getKeyFlag() {
return this.keyFlag;
}
/**
* Get the flag type that is stored as values.
*
* @return The value flag type.
*/
public Flag<V> getValueFlag() {
return this.valueFlag;
}
@Override
public Map<K, V> parseInput(final FlagContext context) throws InvalidFlagFormat {
final String input = context.getUserInput();
if (input.isEmpty()) {
return Maps.newHashMap();
}
final Map<K, V> items = Maps.newHashMap();
for (final String str : input.split(",")) {
final char split = str.indexOf('=') == -1 ? ':' : '=';
final String[] keyVal = str.split(String.valueOf(split));
if (keyVal.length != 2) {
throw new InvalidFlagFormat("Input must be in a 'key:value,key1=value1' format. Either ':' or '=' can be used.");
}
final FlagContext key = context.copyWith(null, keyVal[0], null);
final FlagContext value = context.copyWith(null, keyVal[1], null);
items.put(this.keyFlag.parseInput(key), this.valueFlag.parseInput(value));
}
return items;
}
@Override
public Map<K, V> unmarshal(@Nullable final Object o) {
if (o instanceof Map<?, ?>) {
final Map<?, ?> map = (Map<?, ?>) o;
final Map<K, V> items = Maps.newHashMap();
for (final Entry<?, ?> entry : map.entrySet()) {
final K keyItem = this.keyFlag.unmarshal(entry.getKey());
final V valueItem = this.valueFlag.unmarshal(entry.getValue());
if (keyItem != null && valueItem != null) {
items.put(keyItem, valueItem);
}
}
return items;
}
return null;
}
@Override
public Object marshal(final Map<K, V> o) {
final Map<Object, Object> map = Maps.newHashMap();
for (final Entry<K, V> entry : o.entrySet()) {
map.put(this.keyFlag.marshal(entry.getKey()), this.valueFlag.marshal(entry.getValue()));
}
return map;
}
}