com.github.fonimus.ssh.shell.postprocess.provided.JsonPointerPostProcessor Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of spring-boot-ssh-shell-starter Show documentation

The newest version!

package com.github.fonimus.ssh.shell.postprocess.provided;

import java.io.IOException;
import java.util.List;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.github.fonimus.ssh.shell.postprocess.PostProcessor;

/**
 * Json pointer post processor
 */
public class JsonPointerPostProcessor
		implements PostProcessor {

	private static final Logger LOGGER = LoggerFactory.getLogger(JsonPointerPostProcessor.class);

	private static final ObjectMapper MAPPER = new ObjectMapper();

	@Override
	public String getName() {
		return "json";
	}

	@Override
	public String process(String result, List parameters) {
		if (parameters == null || parameters.isEmpty()) {
			LOGGER.debug("Cannot use [{}] post processor without any parameters", getName());
		} else {
			if (parameters.size() != 1) {
				LOGGER.debug("[{}] post processor only need one parameter, rest will be ignored", getName());
			}
			String path = parameters.get(0);
			try {
				JsonNode node = MAPPER.readTree(result).at(path);
				if (node.isMissingNode()) {
					return "No node found with json path expression: " + path;
				} else {
					if (node.isTextual()) {
						return node.asText();
					} else {
						return MAPPER.writerWithDefaultPrettyPrinter().writeValueAsString(node);
					}
				}
			} catch (IOException e) {
				LOGGER.debug("Unable to read tree", e);
			} catch (IllegalArgumentException e) {
				LOGGER.debug("Illegal argument: " + path, e);
				return e.getMessage();
			}
		}
		return result;
	}
}