nmode's Git Repositories - signal-cli/blob - src/main/java/org/asamk/signal/JsonReceiveMessageHandler.java

   1 package org.asamk.signal;
   2
   3 import com.fasterxml.jackson.annotation.JsonAutoDetect;
   4 import com.fasterxml.jackson.annotation.PropertyAccessor;
   5 import com.fasterxml.jackson.core.JsonGenerator;
   6 import com.fasterxml.jackson.databind.DeserializationFeature;
   7 import com.fasterxml.jackson.databind.ObjectMapper;
   8 import com.fasterxml.jackson.databind.SerializationFeature;
   9 import com.fasterxml.jackson.databind.node.ObjectNode;
  10
  11 import org.asamk.signal.json.JsonError;
  12 import org.asamk.signal.json.JsonMessageEnvelope;
  13 import org.asamk.signal.manager.Manager;
  14 import org.whispersystems.signalservice.api.messages.SignalServiceContent;
  15 import org.whispersystems.signalservice.api.messages.SignalServiceEnvelope;
  16
  17 import java.io.IOException;
  18
  19 public class JsonReceiveMessageHandler implements Manager.ReceiveMessageHandler {
  20
  21     final Manager m;
  22     private final ObjectMapper jsonProcessor;
  23
  24     public JsonReceiveMessageHandler(Manager m) {
  25         this.m = m;
  26         this.jsonProcessor = new ObjectMapper();
  27         jsonProcessor.setVisibility(PropertyAccessor.ALL, JsonAutoDetect.Visibility.ANY); // disable autodetect
  28         jsonProcessor.enable(SerializationFeature.WRITE_NULL_MAP_VALUES);
  29         jsonProcessor.disable(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES);
  30         jsonProcessor.disable(JsonGenerator.Feature.AUTO_CLOSE_TARGET);
  31     }
  32
  33     @Override
  34     public void handleMessage(SignalServiceEnvelope envelope, SignalServiceContent content, Throwable exception) {
  35         ObjectNode result = jsonProcessor.createObjectNode();
  36         if (exception != null) {
  37             result.putPOJO("error", new JsonError(exception));
  38         }
  39         if (envelope != null) {
  40             result.putPOJO("envelope", new JsonMessageEnvelope(envelope, content));
  41         }
  42         try {
  43             jsonProcessor.writeValue(System.out, result);
  44             System.out.println();
  45         } catch (IOException e) {
  46             e.printStackTrace();
  47         }
  48     }
  49 }