org.apache.tinkerpop.gremlin.console.Preferences.groovy Maven / Gradle / Ivy
/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
package org.apache.tinkerpop.gremlin.console;
import java.util.prefs.PreferenceChangeEvent
import java.util.prefs.PreferenceChangeListener
import org.codehaus.groovy.tools.shell.Groovysh
import org.codehaus.groovy.tools.shell.IO
public class Preferences {
private static final java.util.prefs.Preferences STORE = java.util.prefs.Preferences.userRoot().node("/org/apache/tinkerpop/gremlin/console");
public static final String PREFERENCE_ITERATION_MAX = "max-iteration"
private static final int DEFAULT_ITERATION_MAX = 100
public static int maxIteration = DEFAULT_ITERATION_MAX
public static final String PREF_GREMLIN_COLOR = "gremlin.color"
public static final String PREF_GREMLIN_COLOR_DEFAULT = "reset"
public static String gremlinColor = PREF_GREMLIN_COLOR_DEFAULT
public static final String PREF_VERTEX_COLOR = "vertex.color"
public static final String PREF_VERTEX_COLOR_DEFAULT = "reset"
public static String vertexColor = PREF_VERTEX_COLOR_DEFAULT
public static final String PREF_EDGE_COLOR = "edge.color"
public static final String PREF_EDGE_COLOR_DEFAULT = "reset"
public static String edgeColor = PREF_EDGE_COLOR_DEFAULT
public static final String PREF_ERROR_COLOR = "error.color"
public static final String PREF_ERROR_COLOR_DEFAULT = "reset"
public static String errorColor = PREF_ERROR_COLOR_DEFAULT
public static final String PREF_INFO_COLOR = "info.color"
public static final String PREF_INFO_COLOR_DEFAULT = "reset"
public static String infoColor = PREF_INFO_COLOR_DEFAULT
public static final String PREF_STRING_COLOR = "string.color"
public static final String PREF_STRING_COLOR_DEFAULT = "reset"
public static String stringColor = PREF_STRING_COLOR_DEFAULT
public static final String PREF_NUMBER_COLOR = "number.color"
public static final String PREF_NUMBER_COLOR_DEFAULT = "reset"
public static String numberColor = PREF_NUMBER_COLOR_DEFAULT
public static final String PREF_T_COLOR = "T.color"
public static final String PREF_T_COLOR_DEFAULT = "reset"
public static String tColor = PREF_T_COLOR_DEFAULT
public static final String PREF_INPUT_PROMPT_COLOR = "input.prompt.color"
public static final String PREF_INPUT_PROMPT_COLOR_DEFAULT = "reset"
public static String inputPromptColor = PREF_INPUT_PROMPT_COLOR_DEFAULT
public static final String PREF_RESULT_PROMPT_COLOR = "result.prompt.color"
public static final String PREF_RESULT_PROMPT_COLOR_DEFAULT = "reset"
public static String resultPromptColor = PREF_RESULT_PROMPT_COLOR_DEFAULT
public static final String PREF_RESULT_IND_NULL = "result.indicator.null"
public static final String PREF_RESULT_IND_NULL_DEFAULT = "null"
public static String emptyResult = PREF_RESULT_IND_NULL_DEFAULT
public static final String PREF_INPUT_PROMPT = "input.prompt"
public static final String PREF_INPUT_PROMPT_DEFAULT = "gremlin>"
public static String inputPrompt = PREF_INPUT_PROMPT_DEFAULT
public static final String PREF_RESULT_PROMPT = "result.prompt"
public static final String PREF_RESULT_PROMPT_DEFAULT = "==>"
public static String resultPrompt = PREF_RESULT_PROMPT_DEFAULT
public static final String PREF_COLORS = Groovysh.COLORS_PREFERENCE_KEY
public static final Boolean PREF_COLORS_DEFAULT = true;
public static boolean colors = PREF_COLORS_DEFAULT
public static final String PREF_WARNINGS = "warnings"
public static final Boolean PREF_WARNINGS_DEFAULT = true
public static boolean warnings = PREF_WARNINGS_DEFAULT
public static void expandoMagic() {
// Override all GroovySH Preference methods
org.codehaus.groovy.tools.shell.util.Preferences.getMetaClass().'static'.getShowLastResult = {
return STORE.getBoolean(org.codehaus.groovy.tools.shell.util.Preferences.SHOW_LAST_RESULT_KEY, true);
}
org.codehaus.groovy.tools.shell.util.Preferences.getMetaClass().'static'.getSanitizeStackTrace = {
return STORE.getBoolean(org.codehaus.groovy.tools.shell.util.Preferences.SANITIZE_STACK_TRACE_KEY, true);
}
org.codehaus.groovy.tools.shell.util.Preferences.getMetaClass().'static'.getEditor = {
return STORE.get(org.codehaus.groovy.tools.shell.util.Preferences.EDITOR_KEY, System.getenv("EDITOR"));
}
org.codehaus.groovy.tools.shell.util.Preferences.getMetaClass().'static'.getParserFlavor = {
return STORE.get(org.codehaus.groovy.tools.shell.util.Preferences.PARSER_FLAVOR_KEY, org.codehaus.groovy.tools.shell.util.Preferences.PARSER_RIGID);
}
//
// Store Access
//
org.codehaus.groovy.tools.shell.util.Preferences.getMetaClass().'static'.keys = {
return STORE.keys();
}
org.codehaus.groovy.tools.shell.util.Preferences.getMetaClass().'static'.get = { String name, String defaultValue ->
return STORE.get(name, defaultValue);
}
org.codehaus.groovy.tools.shell.util.Preferences.getMetaClass().'static'.get = { String name ->
return get(name, null);
}
org.codehaus.groovy.tools.shell.util.Preferences.getMetaClass().'static'.put = { String name, String value ->
STORE.put(name, value);
}
org.codehaus.groovy.tools.shell.util.Preferences.getMetaClass().'static'.clear = { STORE.clear(); }
org.codehaus.groovy.tools.shell.util.Preferences.getMetaClass().'static'.addChangeListener = { PreferenceChangeListener listener ->
STORE.addPreferenceChangeListener(listener);
}
// reinstall change handler
String tmp = STORE.get(org.codehaus.groovy.tools.shell.util.Preferences.VERBOSITY_KEY, IO.Verbosity.INFO.name);
try {
org.codehaus.groovy.tools.shell.util.Preferences.verbosity = IO.Verbosity.forName(tmp);
}
catch (IllegalArgumentException e) {
org.codehaus.groovy.tools.shell.util.Preferences.verbosity = IO.Verbosity.INFO;
STORE.remove(org.codehaus.groovy.tools.shell.util.Preferences.VERBOSITY_KEY);
}
org.codehaus.groovy.tools.shell.util.Preferences.addChangeListener(new PreferenceChangeListener() {
public void preferenceChange(final PreferenceChangeEvent event) {
if (event.getKey().equals(org.codehaus.groovy.tools.shell.util.Preferences.VERBOSITY_KEY)) {
String name = event.getNewValue();
if (name == null) {
name = IO.Verbosity.INFO.name;
}
try {
org.codehaus.groovy.tools.shell.util.Preferences.verbosity = IO.Verbosity.forName(name);
}
catch (Exception e) {
event.getNode().put(event.getKey(), org.codehaus.groovy.tools.shell.util.Preferences.verbosity.name);
}
}
}
});
// Gremlin Handlers
// Initial Load
loadDefaultValues()
// Listeners
installPropertyListeners()
}
private static installPropertyListeners() {
org.codehaus.groovy.tools.shell.util.Preferences.addChangeListener(new PreferenceChangeListener() {
@Override
void preferenceChange(PreferenceChangeEvent evt) {
if (evt.key == PREFERENCE_ITERATION_MAX && null != evt.newValue) {
try {
maxIteration = Integer.parseInt(evt.newValue)
} catch (NumberFormatException e) {
println(Colorizer.render(errorColor,"Unable to convert '${evt.newValue}' to integer. Using default ${DEFAULT_ITERATION_MAX}"))
maxIteration = DEFAULT_ITERATION_MAX
}
} else if (evt.key == PREFERENCE_ITERATION_MAX){
maxIteration = DEFAULT_ITERATION_MAX
}
}
})
org.codehaus.groovy.tools.shell.util.Preferences.addChangeListener(new PreferenceChangeListener() {
@Override
void preferenceChange(PreferenceChangeEvent evt) {
if (evt.key == PREF_GREMLIN_COLOR) {
gremlinColor = getValidColor(PREF_GREMLIN_COLOR, evt.newValue, PREF_GREMLIN_COLOR_DEFAULT)
} else if (evt.key == PREF_VERTEX_COLOR) {
vertexColor = getValidColor(PREF_VERTEX_COLOR, evt.newValue, PREF_VERTEX_COLOR_DEFAULT)
} else if (evt.key == PREF_EDGE_COLOR) {
edgeColor = getValidColor(PREF_EDGE_COLOR, evt.newValue, PREF_EDGE_COLOR_DEFAULT)
} else if (evt.key == PREF_ERROR_COLOR) {
errorColor = getValidColor(PREF_ERROR_COLOR, evt.newValue, PREF_ERROR_COLOR_DEFAULT)
} else if (evt.key == PREF_INFO_COLOR) {
infoColor = getValidColor(PREF_INFO_COLOR, evt.newValue, PREF_INFO_COLOR_DEFAULT)
} else if (evt.key == PREF_STRING_COLOR) {
stringColor = getValidColor(PREF_STRING_COLOR, evt.newValue, PREF_STRING_COLOR_DEFAULT)
} else if (evt.key == PREF_NUMBER_COLOR) {
numberColor = getValidColor(PREF_NUMBER_COLOR, evt.newValue, PREF_NUMBER_COLOR_DEFAULT)
} else if (evt.key == PREF_T_COLOR) {
tColor = getValidColor(PREF_T_COLOR, evt.newValue, PREF_T_COLOR_DEFAULT)
} else if (evt.key == PREF_INPUT_PROMPT_COLOR) {
inputPromptColor = getValidColor(PREF_INPUT_PROMPT_COLOR, evt.newValue, PREF_INPUT_PROMPT_COLOR_DEFAULT)
} else if (evt.key == PREF_RESULT_PROMPT_COLOR) {
resultPromptColor = getValidColor(PREF_RESULT_PROMPT_COLOR, evt.newValue, PREF_RESULT_PROMPT_COLOR_DEFAULT)
} else if (evt.key == PREF_RESULT_IND_NULL) {
if (null == evt.newValue) {
emptyResult = STORE.get(PREF_RESULT_IND_NULL, PREF_RESULT_IND_NULL_DEFAULT)
} else {
emptyResult = evt.newValue
}
} else if (evt.key == PREF_INPUT_PROMPT) {
if (null == evt.newValue) {
inputPrompt = STORE.get(PREF_INPUT_PROMPT, PREF_INPUT_PROMPT_DEFAULT)
} else {
inputPrompt = evt.newValue
}
} else if (evt.key == PREF_RESULT_PROMPT) {
if (null == evt.newValue) {
resultPrompt = STORE.get(PREF_RESULT_PROMPT, PREF_RESULT_PROMPT_DEFAULT)
} else {
resultPrompt = evt.newValue
}
} else if (evt.key == PREF_COLORS) {
if (null == evt.newValue) {
colors = Boolean.valueOf(STORE.get(PREF_COLORS, PREF_COLORS_DEFAULT.toString()))
} else {
colors = Boolean.valueOf(evt.newValue)
}
} else if (evt.key == PREF_WARNINGS) {
if (null == evt.newValue) {
warnings = Boolean.valueOf(STORE.get(PREF_WARNINGS, PREF_WARNINGS_DEFAULT.toString()))
} else {
warnings = Boolean.valueOf(evt.newValue)
}
}
}
})
}
private static String getValidColor(String key, Object desired, String defaultValue) {
String result = desired
if (null == result) {
result = STORE.get(key, defaultValue)
}
try {
Colorizer.render(result, "test")
} catch (Exception e) {
println(Colorizer.render(errorColor, "Invalid color option for ${key}: ${result}"))
result = defaultValue
}
return result
}
private static loadDefaultValues() {
try {
maxIteration = STORE.get(PREFERENCE_ITERATION_MAX, DEFAULT_ITERATION_MAX.toString()).toInteger()
}catch (NumberFormatException e) {
String maxIterationString = STORE.get(PREFERENCE_ITERATION_MAX, DEFAULT_ITERATION_MAX.toString())
println(Colorizer.render(Preferences.errorColor,"Unable to convert '${maxIterationString}' to integer. Using default ${DEFAULT_ITERATION_MAX}"))
maxIteration = DEFAULT_ITERATION_MAX
}
gremlinColor = getValidColor(PREF_GREMLIN_COLOR, null, PREF_GREMLIN_COLOR_DEFAULT)
vertexColor = getValidColor(PREF_VERTEX_COLOR, null, PREF_VERTEX_COLOR_DEFAULT)
edgeColor = getValidColor(PREF_EDGE_COLOR, null, PREF_EDGE_COLOR_DEFAULT)
errorColor = getValidColor(PREF_ERROR_COLOR, null, PREF_ERROR_COLOR_DEFAULT)
infoColor = getValidColor(PREF_INFO_COLOR, null, PREF_INFO_COLOR_DEFAULT)
stringColor = getValidColor(PREF_STRING_COLOR, null, PREF_STRING_COLOR_DEFAULT)
numberColor = getValidColor(PREF_NUMBER_COLOR, null, PREF_NUMBER_COLOR_DEFAULT)
tColor = getValidColor(PREF_T_COLOR, null, PREF_T_COLOR_DEFAULT)
inputPromptColor = getValidColor(PREF_INPUT_PROMPT_COLOR, null, PREF_INPUT_PROMPT_COLOR_DEFAULT)
resultPromptColor = getValidColor(PREF_RESULT_PROMPT_COLOR, null, PREF_RESULT_PROMPT_COLOR_DEFAULT)
emptyResult = STORE.get(PREF_RESULT_IND_NULL, PREF_RESULT_IND_NULL_DEFAULT)
inputPrompt = STORE.get(PREF_INPUT_PROMPT, PREF_INPUT_PROMPT_DEFAULT)
resultPrompt = STORE.get(PREF_RESULT_PROMPT, PREF_RESULT_PROMPT_DEFAULT)
colors = Boolean.valueOf(STORE.get(PREF_COLORS, PREF_COLORS_DEFAULT.toString()))
warnings = Boolean.valueOf(STORE.get(PREF_WARNINGS, PREF_WARNINGS_DEFAULT.toString()))
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy