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

   1 package org.asamk.signal;
   2
   3 import org.asamk.signal.json.JsonError;
   4 import org.asamk.signal.json.JsonMessageEnvelope;
   5 import org.asamk.signal.manager.Manager;
   6 import org.slf4j.Logger;
   7 import org.slf4j.LoggerFactory;
   8 import org.whispersystems.signalservice.api.messages.SignalServiceContent;
   9 import org.whispersystems.signalservice.api.messages.SignalServiceEnvelope;
  10
  11 import java.util.HashMap;
  12
  13 public class JsonReceiveMessageHandler implements Manager.ReceiveMessageHandler {
  14
  15     private final static Logger logger = LoggerFactory.getLogger(JsonReceiveMessageHandler.class);
  16
  17     protected final Manager m;
  18     private final JsonWriter jsonWriter;
  19
  20     public JsonReceiveMessageHandler(Manager m, JsonWriter jsonWriter) {
  21         this.m = m;
  22         this.jsonWriter = jsonWriter;
  23     }
  24
  25     @Override
  26     public void handleMessage(SignalServiceEnvelope envelope, SignalServiceContent content, Throwable exception) {
  27         final var object = new HashMap<String, Object>();
  28         if (exception != null) {
  29             object.put("error", new JsonError(exception));
  30         }
  31
  32         if (envelope != null) {
  33             object.put("envelope", new JsonMessageEnvelope(envelope, content, exception, m));
  34         }
  35
  36         jsonWriter.write(object);
  37     }
  38 }