X-Git-Url: https://git.nmode.ca/signal-cli/blobdiff_plain/6c000544072fb0be012dafeea5761fa9e0744ee4..a8bbdb54d006f157a009ece0cae5bf72fb636ced:/src/main/java/org/asamk/signal/jsonrpc/JsonRpcReader.java

diff --git a/src/main/java/org/asamk/signal/jsonrpc/JsonRpcReader.java b/src/main/java/org/asamk/signal/jsonrpc/JsonRpcReader.java
new file mode 100644
index 00000000..67fced0d
--- /dev/null
+++ b/src/main/java/org/asamk/signal/jsonrpc/JsonRpcReader.java
@@ -0,0 +1,216 @@
+package org.asamk.signal.jsonrpc;
+
+import com.fasterxml.jackson.core.JsonParseException;
+import com.fasterxml.jackson.databind.JsonMappingException;
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.node.ContainerNode;
+import com.fasterxml.jackson.databind.node.ValueNode;
+
+import org.asamk.signal.util.Util;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import java.io.IOException;
+import java.util.Objects;
+import java.util.function.Consumer;
+import java.util.function.Supplier;
+import java.util.stream.Collectors;
+import java.util.stream.StreamSupport;
+
+public class JsonRpcReader {
+
+    private final static Logger logger = LoggerFactory.getLogger(JsonRpcReader.class);
+
+    private final JsonRpcSender jsonRpcSender;
+    private final ObjectMapper objectMapper;
+    private final Supplier<String> lineSupplier;
+
+    public JsonRpcReader(
+            final JsonRpcSender jsonRpcSender, final Supplier<String> lineSupplier
+    ) {
+        this.jsonRpcSender = jsonRpcSender;
+        this.lineSupplier = lineSupplier;
+        this.objectMapper = Util.createJsonObjectMapper();
+    }
+
+    public void readRequests(
+            final RequestHandler requestHandler, final Consumer<JsonRpcResponse> responseHandler
+    ) {
+        while (!Thread.interrupted()) {
+            JsonRpcMessage message = readMessage();
+            if (message == null) break;
+
+            if (message instanceof JsonRpcRequest) {
+                final var response = handleRequest(requestHandler, (JsonRpcRequest) message);
+                if (response != null) {
+                    jsonRpcSender.sendResponse(response);
+                }
+            } else if (message instanceof JsonRpcResponse) {
+                responseHandler.accept((JsonRpcResponse) message);
+            } else {
+                final var responseList = ((JsonRpcBulkMessage) message).getMessages().stream().map(jsonNode -> {
+                    final JsonRpcRequest request;
+                    try {
+                        request = parseJsonRpcRequest(jsonNode);
+                    } catch (JsonRpcException e) {
+                        return JsonRpcResponse.forError(e.getError(), getId(jsonNode));
+                    }
+
+                    return handleRequest(requestHandler, request);
+                }).filter(Objects::nonNull).collect(Collectors.toList());
+
+                jsonRpcSender.sendBulkResponses(responseList);
+            }
+        }
+    }
+
+    private JsonRpcResponse handleRequest(final RequestHandler requestHandler, final JsonRpcRequest request) {
+        try {
+            final var result = requestHandler.apply(request.getMethod(), request.getParams());
+            if (request.getId() != null) {
+                return JsonRpcResponse.forSuccess(result, request.getId());
+            }
+        } catch (JsonRpcException e) {
+            if (request.getId() != null) {
+                return JsonRpcResponse.forError(e.getError(), request.getId());
+            }
+        }
+        return null;
+    }
+
+    private JsonRpcMessage readMessage() {
+        while (!Thread.interrupted()) {
+            String input = lineSupplier.get();
+
+            if (input == null) {
+                // Reached end of input stream
+                break;
+            }
+
+            JsonRpcMessage message = parseJsonRpcMessage(input);
+            if (message == null) continue;
+
+            return message;
+        }
+
+        return null;
+    }
+
+    private JsonRpcMessage parseJsonRpcMessage(final String input) {
+        final JsonNode jsonNode;
+        try {
+            jsonNode = objectMapper.readTree(input);
+        } catch (JsonParseException e) {
+            jsonRpcSender.sendResponse(JsonRpcResponse.forError(new JsonRpcResponse.Error(JsonRpcResponse.Error.PARSE_ERROR,
+                    e.getMessage(),
+                    null), null));
+            return null;
+        } catch (IOException e) {
+            throw new AssertionError(e);
+        }
+
+        if (jsonNode == null) {
+            jsonRpcSender.sendResponse(JsonRpcResponse.forError(new JsonRpcResponse.Error(JsonRpcResponse.Error.INVALID_REQUEST,
+                    "invalid request",
+                    null), null));
+            return null;
+        } else if (jsonNode.isArray()) {
+            if (jsonNode.size() == 0) {
+                jsonRpcSender.sendResponse(JsonRpcResponse.forError(new JsonRpcResponse.Error(JsonRpcResponse.Error.INVALID_REQUEST,
+                        "invalid request",
+                        null), null));
+                return null;
+            }
+            return new JsonRpcBulkMessage(StreamSupport.stream(jsonNode.spliterator(), false)
+                    .collect(Collectors.toList()));
+        } else if (jsonNode.isObject()) {
+            if (jsonNode.has("result") || jsonNode.has("error")) {
+                return parseJsonRpcResponse(jsonNode);
+            } else {
+                try {
+                    return parseJsonRpcRequest(jsonNode);
+                } catch (JsonRpcException e) {
+                    jsonRpcSender.sendResponse(JsonRpcResponse.forError(e.getError(), getId(jsonNode)));
+                    return null;
+                }
+            }
+        } else {
+            jsonRpcSender.sendResponse(JsonRpcResponse.forError(new JsonRpcResponse.Error(JsonRpcResponse.Error.INVALID_REQUEST,
+                    "unexpected type: " + jsonNode.getNodeType().name(),
+                    null), null));
+            return null;
+        }
+    }
+
+    private ValueNode getId(JsonNode jsonNode) {
+        final var id = jsonNode.get("id");
+        return id instanceof ValueNode ? (ValueNode) id : null;
+    }
+
+    private JsonRpcRequest parseJsonRpcRequest(final JsonNode input) throws JsonRpcException {
+        JsonRpcRequest request;
+        try {
+            request = objectMapper.treeToValue(input, JsonRpcRequest.class);
+        } catch (JsonMappingException e) {
+            throw new JsonRpcException(new JsonRpcResponse.Error(JsonRpcResponse.Error.INVALID_REQUEST,
+                    e.getMessage(),
+                    null));
+        } catch (IOException e) {
+            throw new AssertionError(e);
+        }
+
+        if (!"2.0".equals(request.getJsonrpc())) {
+            throw new JsonRpcException(new JsonRpcResponse.Error(JsonRpcResponse.Error.INVALID_REQUEST,
+                    "only jsonrpc version 2.0 is supported",
+                    null));
+        }
+
+        if (request.getMethod() == null) {
+            throw new JsonRpcException(new JsonRpcResponse.Error(JsonRpcResponse.Error.INVALID_REQUEST,
+                    "method field must be set",
+                    null));
+        }
+
+        return request;
+    }
+
+    private JsonRpcResponse parseJsonRpcResponse(final JsonNode input) {
+        JsonRpcResponse response;
+        try {
+            response = objectMapper.treeToValue(input, JsonRpcResponse.class);
+        } catch (JsonParseException | JsonMappingException e) {
+            logger.debug("Received invalid jsonrpc response {}", e.getMessage());
+            return null;
+        } catch (IOException e) {
+            throw new AssertionError(e);
+        }
+
+        if (!"2.0".equals(response.getJsonrpc())) {
+            logger.debug("Received invalid jsonrpc response with invalid version {}", response.getJsonrpc());
+            return null;
+        }
+
+        if (response.getResult() != null && response.getError() != null) {
+            logger.debug("Received invalid jsonrpc response with both result and error");
+            return null;
+        }
+
+        if (response.getResult() == null && response.getError() == null) {
+            logger.debug("Received invalid jsonrpc response without result and error");
+            return null;
+        }
+
+        if (response.getId() == null || response.getId().isNull()) {
+            logger.debug("Received invalid jsonrpc response without id");
+            return null;
+        }
+
+        return response;
+    }
+
+    public interface RequestHandler {
+
+        JsonNode apply(String method, ContainerNode<?> params) throws JsonRpcException;
+    }
+}