use of com.bluenimble.platform.cli.impls.YamlObject in project serverless by bluenimble.
the class RemoteUtils method processRequest.
public static CommandResult processRequest(Tool tool, JsonObject source, final Map<String, String> options) throws CommandExecutionException {
if (Lang.isDebugMode()) {
tool.printer().content("Remote Command", source.toString());
}
JsonObject oKeys = null;
Keys keys = BlueNimble.keys();
if (keys != null) {
oKeys = keys.json();
} else {
oKeys = new JsonObject();
}
@SuppressWarnings("unchecked") Map<String, Object> vars = (Map<String, Object>) tool.getContext(Tool.ROOT_CTX).get(ToolContext.VARS);
Object oTrustAll = vars.get(DefaultVars.SslTrust);
if (oTrustAll == null) {
Http.setTrustAll(false);
} else {
Http.setTrustAll(Lang.TrueValues.contains(String.valueOf(oTrustAll)));
}
boolean isOutFile = AbstractTool.CMD_OUT_FILE.equals(vars.get(AbstractTool.CMD_OUT));
List<Object> streams = new ArrayList<Object>();
HttpResponse response = null;
try {
HttpRequest request = request(oKeys, Json.getObject(source, Spec.request.class.getSimpleName()), tool, BlueNimble.Config, options, streams);
response = Http.send(request);
String contentType = response.getContentType();
if (contentType == null) {
contentType = ApiContentTypes.Text;
}
int indexOfSemi = contentType.indexOf(Lang.SEMICOLON);
if (indexOfSemi > 0) {
contentType = contentType.substring(0, indexOfSemi).trim();
}
OutputStream out = System.out;
if (Printable.contains(contentType) && !isOutFile) {
out = new ByteArrayOutputStream();
response.getBody().dump(out, Encodings.UTF8, null);
}
List<HttpHeader> rHeaders = response.getHeaders();
if (rHeaders != null && !rHeaders.isEmpty()) {
JsonObject oHeaders = new JsonObject();
for (HttpHeader h : rHeaders) {
oHeaders.set(h.getName(), Lang.join(h.getValues(), Lang.COMMA));
}
vars.put(RemoteResponseHeaders, oHeaders);
}
if (contentType.startsWith(ApiContentTypes.Json)) {
JsonObject result = new JsonObject(new String(((ByteArrayOutputStream) out).toByteArray()));
String trace = null;
if (response.getStatus() >= 400) {
trace = result.getString("trace");
result.remove("trace");
}
if (trace != null && Lang.isDebugMode()) {
vars.put(RemoteResponseError, trace);
}
if (response.getStatus() < 400) {
return new DefaultCommandResult(CommandResult.OK, result);
} else {
return new DefaultCommandResult(CommandResult.KO, result);
}
} else if (contentType.startsWith(ApiContentTypes.Yaml)) {
Yaml yaml = new Yaml();
String ys = new String(((ByteArrayOutputStream) out).toByteArray());
ys = Lang.replace(ys, Lang.TAB, " ");
@SuppressWarnings("unchecked") Map<String, Object> map = yaml.loadAs(ys, Map.class);
Object trace = null;
if (response.getStatus() >= 400) {
trace = map.get("trace");
map.remove("trace");
}
if (trace != null && Lang.isDebugMode()) {
vars.put(RemoteResponseError, trace);
}
if (response.getStatus() < 400) {
return new DefaultCommandResult(CommandResult.OK, new YamlObject(map));
} else {
return new DefaultCommandResult(CommandResult.KO, new YamlObject(map));
}
} else if (contentType.startsWith(ApiContentTypes.Text) || contentType.startsWith(ApiContentTypes.Html)) {
String content = new String(((ByteArrayOutputStream) out).toByteArray());
if (response.getStatus() < 400) {
return new DefaultCommandResult(CommandResult.OK, content);
} else {
return new DefaultCommandResult(CommandResult.KO, content);
}
} else {
if (response.getStatus() < 400) {
if (isOutFile) {
return new DefaultCommandResult(CommandResult.OK, response.getBody().get(0).toInputStream());
} else {
ByteArrayOutputStream baos = new ByteArrayOutputStream();
response.getBody().dump(baos, Encodings.UTF8, null);
return new DefaultCommandResult(CommandResult.OK, new String(((ByteArrayOutputStream) baos).toByteArray()));
}
} else {
ByteArrayOutputStream baos = new ByteArrayOutputStream();
response.getBody().dump(baos, Encodings.UTF8, null);
return new DefaultCommandResult(CommandResult.KO, new String(((ByteArrayOutputStream) baos).toByteArray()));
}
}
} catch (Exception e) {
throw new CommandExecutionException(e.getMessage(), e);
} finally {
if (streams != null) {
for (Object s : streams) {
if (s instanceof InputStream) {
IOUtils.closeQuietly((InputStream) s);
} else if (s instanceof StreamPointer) {
StreamPointer sp = (StreamPointer) s;
IOUtils.closeQuietly(sp.getStream());
if (sp.shouldDelete()) {
try {
FileUtils.delete(sp.getPointer());
} catch (IOException e) {
// IGNORE
}
}
}
}
}
}
}
use of com.bluenimble.platform.cli.impls.YamlObject in project serverless by bluenimble.
the class SetCommand method execute.
@SuppressWarnings("unchecked")
@Override
public CommandResult execute(Tool tool, Map<String, CommandOption> options) throws CommandExecutionException {
String cmd = (String) tool.currentContext().get(ToolContext.CommandLine);
if (Lang.isNullOrEmpty(cmd)) {
throw new CommandExecutionException("command '" + name + "' missing arguments").showUsage();
}
String[] nameValue = Lang.split(cmd, Lang.SPACE, true);
String varName = nameValue[0];
Object value = Lang.BLANK;
if (nameValue.length > 1) {
value = nameValue[1];
}
if (value == null) {
value = Lang.BLANK;
}
final Map<String, Object> vars = (Map<String, Object>) tool.getContext(Tool.ROOT_CTX).get(ToolContext.VARS);
try {
if (varName.equals(Tool.ParaPhraseVar)) {
tool.setParaphrase((String) value, true);
tool.printer().content("Security", "Paraphase Updated");
}
if (value instanceof YamlObject) {
value = ((YamlObject) value).toJson();
}
value = varName.equals(Tool.ParaPhraseVar) ? tool.getParaphrase(false) : value;
vars.put(varName, value);
tool.saveVariable(varName, value);
} catch (Exception e) {
throw new CommandExecutionException(e.getMessage(), e);
}
tool.printer().content(varName, String.valueOf(vars.get(varName)));
return null;
}
use of com.bluenimble.platform.cli.impls.YamlObject in project serverless by bluenimble.
the class VarsCommand method execute.
@SuppressWarnings("unchecked")
@Override
public CommandResult execute(Tool tool, Map<String, CommandOption> options) throws CommandExecutionException {
Map<String, Object> vars = (Map<String, Object>) tool.getContext(Tool.ROOT_CTX).get(ToolContext.VARS);
String varName = (String) tool.currentContext().get(ToolContext.CommandLine);
Object result = "";
if (!Lang.isNullOrEmpty(varName)) {
Object value = vars.get(varName);
if (value != null) {
if (value instanceof JsonObject) {
result = value;
} else {
result = varName + Lang.EQUALS + value;
}
} else {
result = "variable " + varName + " not found";
}
} else {
if (vars != null && !vars.isEmpty()) {
result = new YamlObject(vars);
}
}
return new DefaultCommandResult(CommandResult.OK, result);
}
Aggregations