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