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

   1 package org.asamk.signal.commands;
   2
   3 import net.sourceforge.argparse4j.impl.Arguments;
   4 import net.sourceforge.argparse4j.inf.Namespace;
   5 import net.sourceforge.argparse4j.inf.Subparser;
   6
   7 import org.asamk.signal.DbusReceiveMessageHandler;
   8 import org.asamk.signal.JsonDbusReceiveMessageHandler;
   9 import org.asamk.signal.dbus.DbusSignalImpl;
  10 import org.asamk.signal.manager.Manager;
  11 import org.freedesktop.dbus.connections.impl.DBusConnection;
  12 import org.freedesktop.dbus.exceptions.DBusException;
  13
  14 import java.io.IOException;
  15 import java.util.concurrent.TimeUnit;
  16
  17 import static org.asamk.signal.DbusConfig.SIGNAL_BUSNAME;
  18 import static org.asamk.signal.DbusConfig.SIGNAL_OBJECTPATH;
  19 import static org.asamk.signal.util.ErrorUtils.handleAssertionError;
  20
  21 public class DaemonCommand implements LocalCommand {
  22
  23     @Override
  24     public void attachToSubparser(final Subparser subparser) {
  25         subparser.addArgument("--system")
  26                 .action(Arguments.storeTrue())
  27                 .help("Use DBus system bus instead of user bus.");
  28         subparser.addArgument("--ignore-attachments")
  29                 .help("Don’t download attachments of received messages.")
  30                 .action(Arguments.storeTrue());
  31         subparser.addArgument("--json")
  32                 .help("Output received messages in json format, one json object per line.")
  33                 .action(Arguments.storeTrue());
  34     }
  35
  36     @Override
  37     public int handleCommand(final Namespace ns, final Manager m) {
  38         DBusConnection conn = null;
  39         try {
  40             try {
  41                 DBusConnection.DBusBusType busType;
  42                 if (ns.getBoolean("system")) {
  43                     busType = DBusConnection.DBusBusType.SYSTEM;
  44                 } else {
  45                     busType = DBusConnection.DBusBusType.SESSION;
  46                 }
  47                 conn = DBusConnection.getConnection(busType);
  48                 conn.exportObject(SIGNAL_OBJECTPATH, new DbusSignalImpl(m));
  49                 conn.requestBusName(SIGNAL_BUSNAME);
  50             } catch (UnsatisfiedLinkError e) {
  51                 System.err.println("Missing native library dependency for dbus service: " + e.getMessage());
  52                 return 1;
  53             } catch (DBusException e) {
  54                 e.printStackTrace();
  55                 return 2;
  56             }
  57             boolean ignoreAttachments = ns.getBoolean("ignore_attachments");
  58             try {
  59                 m.receiveMessages(1,
  60                         TimeUnit.HOURS,
  61                         false,
  62                         ignoreAttachments,
  63                         ns.getBoolean("json")
  64                                 ? new JsonDbusReceiveMessageHandler(m, conn, SIGNAL_OBJECTPATH)
  65                                 : new DbusReceiveMessageHandler(m, conn, SIGNAL_OBJECTPATH));
  66                 return 0;
  67             } catch (IOException e) {
  68                 System.err.println("Error while receiving messages: " + e.getMessage());
  69                 return 3;
  70             } catch (AssertionError e) {
  71                 handleAssertionError(e);
  72                 return 1;
  73             }
  74         } finally {
  75             if (conn != null) {
  76                 conn.disconnect();
  77             }
  78         }
  79     }
  80 }