org.pkl.thirdparty.truffle.polyglot.OptionValuesImpl Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of pkl-tools Show documentation
Show all versions of pkl-tools Show documentation
Fat Jar containing pkl-cli, pkl-codegen-java, pkl-codegen-kotlin, pkl-config-java, pkl-core, pkl-doc, and their shaded third-party dependencies.
/*
* Copyright (c) 2017, 2022, Oracle and/or its affiliates. All rights reserved.
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
*
* The Universal Permissive License (UPL), Version 1.0
*
* Subject to the condition set forth below, permission is hereby granted to any
* person obtaining a copy of this software, associated documentation and/or
* data (collectively the "Software"), free of charge and under any and all
* copyright rights in the Software, and any and all patent rights owned or
* freely licensable by each licensor hereunder covering either (i) the
* unmodified Software as contributed to or provided by such licensor, or (ii)
* the Larger Works (as defined below), to deal in both
*
* (a) the Software, and
*
* (b) any piece of software and/or hardware listed in the lrgrwrks.txt file if
* one is included with the Software each a "Larger Work" to which the Software
* is contributed by such licensors),
*
* without restriction, including without limitation the rights to copy, create
* derivative works of, display, perform, and distribute the Software and make,
* use, sell, offer for sale, import, export, have made, and have sold the
* Software and the Larger Work(s), and to sublicense the foregoing rights on
* either these or other terms.
*
* This license is subject to the following condition:
*
* The above copyright notice and either this complete permission notice or at a
* minimum a reference to the UPL must be included in all copies or substantial
* portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
*/
package org.pkl.thirdparty.truffle.polyglot;
import java.util.ArrayList;
import java.util.Formatter;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Objects;
import org.pkl.thirdparty.truffle.api.TruffleOptionDescriptors;
import org.pkl.thirdparty.graalvm.options.OptionDescriptor;
import org.pkl.thirdparty.graalvm.options.OptionDescriptors;
import org.pkl.thirdparty.graalvm.options.OptionKey;
import org.pkl.thirdparty.graalvm.options.OptionStability;
import org.pkl.thirdparty.graalvm.options.OptionValues;
import org.pkl.thirdparty.graalvm.polyglot.SandboxPolicy;
final class OptionValuesImpl implements OptionValues {
private static final float FUZZY_MATCH_THRESHOLD = 0.7F;
private final OptionDescriptors descriptors;
private final SandboxPolicy sandboxPolicy;
private final Map, Object> values;
private final Map, String> unparsedValues;
OptionValuesImpl(OptionDescriptors descriptors, SandboxPolicy sandboxPolicy, boolean preserveUnparsedValues) {
Objects.requireNonNull(descriptors);
Objects.requireNonNull(sandboxPolicy);
this.descriptors = descriptors;
this.sandboxPolicy = sandboxPolicy;
this.values = new HashMap<>();
this.unparsedValues = preserveUnparsedValues ? new HashMap<>() : null;
}
@Override
public int hashCode() {
int result = 31 + descriptors.hashCode();
result = 31 * result + values.hashCode();
return result;
}
@Override
public boolean equals(Object obj) {
if (!(obj instanceof OptionValues)) {
return super.equals(obj);
} else {
if (this == obj) {
return true;
}
OptionValues other = ((OptionValues) obj);
if (!getDescriptors().equals(other.getDescriptors())) {
return false;
}
if (!hasSetOptions() && !other.hasSetOptions()) {
return true;
}
if (other instanceof OptionValuesImpl) {
// faster comparison that only depends on the set values
OptionValuesImpl otherOptions = (OptionValuesImpl) other;
if (!values.equals(otherOptions.values)) {
return false;
}
} else {
// slow comparison for arbitrary option values
for (OptionDescriptor descriptor : getDescriptors()) {
OptionKey key = descriptor.getKey();
if (!slowCompareKey(key, other)) {
return false;
}
}
}
return true;
}
}
private boolean slowCompareKey(OptionKey key, OptionValues other) {
boolean set = hasBeenSet(key);
if (set != other.hasBeenSet(key)) {
return false;
}
if (set && !get(key).equals(other.get(key))) {
return false;
}
return true;
}
public void putAll(PolyglotEngineImpl engine, Map providedValues, boolean allowExperimentalOptions) {
for (String key : providedValues.keySet()) {
put(engine, key, providedValues.get(key), allowExperimentalOptions);
}
}
public void put(PolyglotEngineImpl engine, String key, String value, boolean allowExperimentalOptions) {
OptionDescriptor descriptor = findDescriptor(engine, key, allowExperimentalOptions);
if (sandboxPolicy != SandboxPolicy.TRUSTED) {
SandboxPolicy optionSandboxPolicy = descriptors instanceof TruffleOptionDescriptors ? ((TruffleOptionDescriptors) descriptors).getSandboxPolicy(key) : SandboxPolicy.TRUSTED;
if (sandboxPolicy.isStricterThan(optionSandboxPolicy)) {
throw PolyglotEngineException.illegalArgument(PolyglotImpl.sandboxPolicyException(sandboxPolicy,
String.format("The option %s can only be used up to the %s sandbox policy.", descriptor.getName(), optionSandboxPolicy),
String.format("do not set the %s option by removing Builder.option(\"%s\", \"%s\")", descriptor.getName(), descriptor.getName(), value)));
}
}
OptionKey optionKey = descriptor.getKey();
Object previousValue;
if (values.containsKey(optionKey)) {
previousValue = values.get(optionKey);
} else {
previousValue = optionKey.getDefaultValue();
}
String name = descriptor.getName();
String suffix = null;
if (descriptor.isOptionMap()) {
suffix = key.substring(name.length());
assert suffix.isEmpty() || suffix.startsWith(".");
if (suffix.startsWith(".")) {
suffix = suffix.substring(1);
}
}
Object convertedValue;
try {
convertedValue = optionKey.getType().convert(previousValue, suffix, value);
} catch (IllegalArgumentException e) {
throw PolyglotEngineException.illegalArgument(e);
}
values.put(descriptor.getKey(), convertedValue);
if (unparsedValues != null) {
unparsedValues.put(descriptor.getKey(), value);
}
}
private OptionValuesImpl(OptionValuesImpl copy) {
this.values = new HashMap<>(copy.values);
this.descriptors = copy.descriptors;
this.sandboxPolicy = copy.sandboxPolicy;
this.unparsedValues = copy.unparsedValues;
}
private boolean contains(OptionKey optionKey) {
for (OptionDescriptor descriptor : descriptors) {
if (descriptor.getKey() == optionKey) {
return true;
}
}
return false;
}
@Override
public boolean hasBeenSet(OptionKey optionKey) {
assert contains(optionKey);
return values.containsKey(optionKey);
}
OptionValuesImpl copy() {
return new OptionValuesImpl(this);
}
void copyInto(OptionValuesImpl target) {
if (!target.values.isEmpty()) {
throw new IllegalStateException("Values must be empty.");
}
if (sandboxPolicy != target.sandboxPolicy) {
throw new AssertionError("Source and target must have the same SandboxPolicy.");
}
target.values.putAll(values);
}
public OptionDescriptors getDescriptors() {
return descriptors;
}
@SuppressWarnings("unchecked")
@Override
public T get(OptionKey optionKey) {
assert contains(optionKey);
Object value = values.get(optionKey);
if (value == null) {
return optionKey.getDefaultValue();
}
return (T) value;
}
@SuppressWarnings("deprecation")
@Override
public void set(OptionKey optionKey, T value) {
throw new UnsupportedOperationException("OptionValues#set() is no longer supported");
}
@Override
public boolean hasSetOptions() {
return !values.isEmpty();
}
String getUnparsedOptionValue(OptionKey key) {
if (unparsedValues == null) {
throw new IllegalStateException("Unparsed values are not supported");
}
return unparsedValues.get(key);
}
private OptionDescriptor findDescriptor(PolyglotEngineImpl engine, String key, boolean allowExperimentalOptions) {
OptionDescriptor descriptor = descriptors.get(key);
if (descriptor == null) {
throw failNotFound(engine, key);
}
if (!allowExperimentalOptions && descriptor.getStability() == OptionStability.EXPERIMENTAL) {
throw failExperimental(key);
}
return descriptor;
}
private static RuntimeException failExperimental(String key) {
final String message = String.format("Option '%s' is experimental and must be enabled with allowExperimentalOptions(boolean) in Context.Builder or Engine.Builder. ", key) +
"Do not use experimental options in production environments.";
return PolyglotEngineException.illegalArgument(message);
}
private RuntimeException failNotFound(PolyglotEngineImpl engine, String key) {
OptionDescriptors allOptions;
Exception errorOptions = null;
try {
allOptions = engine == null ? this.descriptors : engine.getAllOptions();
} catch (Exception e) {
errorOptions = e;
allOptions = this.descriptors;
}
RuntimeException error = failNotFound(allOptions, key);
if (errorOptions != null) {
error.addSuppressed(errorOptions);
}
throw error;
}
static RuntimeException failNotFound(OptionDescriptors allOptions, String key) {
Iterable matches = fuzzyMatch(allOptions, key);
Formatter msg = new Formatter();
msg.format("Could not find option with name %s.", key);
Iterator iterator = matches.iterator();
if (iterator.hasNext()) {
msg.format("%nDid you mean one of the following?");
for (OptionDescriptor match : matches) {
msg.format("%n %s=<%s>", match.getName(), match.getKey().getType().getName());
}
}
throw PolyglotEngineException.illegalArgument(msg.toString());
}
/**
* Returns the set of options that fuzzy match a given option name.
*/
static List fuzzyMatch(OptionDescriptors descriptors, String optionKey) {
List matches = new ArrayList<>();
for (org.pkl.thirdparty.graalvm.options.OptionDescriptor option : descriptors) {
float score = stringSimiliarity(option.getName(), optionKey);
if (score >= FUZZY_MATCH_THRESHOLD) {
matches.add(option);
}
}
return matches;
}
/**
* Compute string similarity based on Dice's coefficient.
*
* Ported from str_similar() in globals.cpp.
*/
private static float stringSimiliarity(String str1, String str2) {
int hit = 0;
for (int i = 0; i < str1.length() - 1; ++i) {
for (int j = 0; j < str2.length() - 1; ++j) {
if ((str1.charAt(i) == str2.charAt(j)) && (str1.charAt(i + 1) == str2.charAt(j + 1))) {
++hit;
break;
}
}
}
return 2.0f * hit / (str1.length() + str2.length());
}
@Override
public String toString() {
Map, ? extends Object> options;
if (unparsedValues != null) {
options = this.unparsedValues;
} else {
options = this.values;
}
StringBuilder b = new StringBuilder("{");
String sep = "";
for (OptionDescriptor descriptor : getDescriptors()) {
OptionKey key = descriptor.getKey();
if (hasBeenSet(key)) {
b.append(sep);
b.append(descriptor.getName());
b.append("=");
b.append(options.get(key));
sep = ", ";
}
}
b.append("}");
return b.toString();
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy