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.DbusConfig;
   8 import org.asamk.signal.DbusReceiveMessageHandler;
   9 import org.asamk.signal.JsonReceiveMessageHandler;
  10 import org.asamk.signal.JsonWriter;
  11 import org.asamk.signal.OutputType;
  12 import org.asamk.signal.OutputWriter;
  13 import org.asamk.signal.PlainTextWriter;
  14 import org.asamk.signal.ReceiveMessageHandler;
  15 import org.asamk.signal.commands.exceptions.CommandException;
  16 import org.asamk.signal.commands.exceptions.UnexpectedErrorException;
  17 import org.asamk.signal.dbus.DbusSignalControlImpl;
  18 import org.asamk.signal.dbus.DbusSignalImpl;
  19 import org.asamk.signal.manager.Manager;
  20 import org.freedesktop.dbus.connections.impl.DBusConnection;
  21 import org.freedesktop.dbus.exceptions.DBusException;
  22 import org.slf4j.Logger;
  23 import org.slf4j.LoggerFactory;
  24
  25 import java.io.IOException;
  26 import java.util.List;
  27
  28 public class DaemonCommand implements MultiLocalCommand {
  29
  30     private final static Logger logger = LoggerFactory.getLogger(DaemonCommand.class);
  31
  32     @Override
  33     public String getName() {
  34         return "daemon";
  35     }
  36
  37     @Override
  38     public void attachToSubparser(final Subparser subparser) {
  39         subparser.help("Run in daemon mode and provide an experimental dbus interface.");
  40         subparser.addArgument("--system")
  41                 .action(Arguments.storeTrue())
  42                 .help("Use DBus system bus instead of user bus.");
  43         subparser.addArgument("--ignore-attachments")
  44                 .help("Don’t download attachments of received messages.")
  45                 .action(Arguments.storeTrue());
  46     }
  47
  48     @Override
  49     public List<OutputType> getSupportedOutputTypes() {
  50         return List.of(OutputType.PLAIN_TEXT, OutputType.JSON);
  51     }
  52
  53     @Override
  54     public void handleCommand(
  55             final Namespace ns, final Manager m, final OutputWriter outputWriter
  56     ) throws CommandException {
  57         boolean ignoreAttachments = Boolean.TRUE.equals(ns.getBoolean("ignore-attachments"));
  58         m.setIgnoreAttachments(ignoreAttachments);
  59
  60         DBusConnection.DBusBusType busType;
  61         if (Boolean.TRUE.equals(ns.getBoolean("system"))) {
  62             busType = DBusConnection.DBusBusType.SYSTEM;
  63         } else {
  64             busType = DBusConnection.DBusBusType.SESSION;
  65         }
  66
  67         try (var conn = DBusConnection.getConnection(busType)) {
  68             var objectPath = DbusConfig.getObjectPath();
  69             var t = run(conn, objectPath, m, outputWriter);
  70
  71             conn.requestBusName(DbusConfig.getBusname());
  72
  73             try {
  74                 t.join();
  75                 synchronized (this) {
  76                     wait();
  77                 }
  78             } catch (InterruptedException ignored) {
  79             }
  80         } catch (DBusException | IOException e) {
  81             logger.error("Dbus command failed", e);
  82             throw new UnexpectedErrorException("Dbus command failed", e);
  83         }
  84     }
  85
  86     @Override
  87     public void handleCommand(
  88             final Namespace ns, final List<Manager> managers, final SignalCreator c, final OutputWriter outputWriter
  89     ) throws CommandException {
  90         boolean ignoreAttachments = Boolean.TRUE.equals(ns.getBoolean("ignore-attachments"));
  91
  92         DBusConnection.DBusBusType busType;
  93         if (Boolean.TRUE.equals(ns.getBoolean("system"))) {
  94             busType = DBusConnection.DBusBusType.SYSTEM;
  95         } else {
  96             busType = DBusConnection.DBusBusType.SESSION;
  97         }
  98
  99         try (var conn = DBusConnection.getConnection(busType)) {
 100             final var signalControl = new DbusSignalControlImpl(c, m -> {
 101                 m.setIgnoreAttachments(ignoreAttachments);
 102                 try {
 103                     final var objectPath = DbusConfig.getObjectPath(m.getSelfNumber());
 104                     return run(conn, objectPath, m, outputWriter);
 105                 } catch (DBusException e) {
 106                     logger.error("Failed to export object", e);
 107                     return null;
 108                 }
 109             }, DbusConfig.getObjectPath());
 110             conn.exportObject(signalControl);
 111
 112             for (var m : managers) {
 113                 signalControl.addManager(m);
 114             }
 115
 116             conn.requestBusName(DbusConfig.getBusname());
 117
 118             signalControl.run();
 119         } catch (DBusException | IOException e) {
 120             logger.error("Dbus command failed", e);
 121             throw new UnexpectedErrorException("Dbus command failed", e);
 122         }
 123     }
 124
 125     private Thread run(
 126             DBusConnection conn, String objectPath, Manager m, OutputWriter outputWriter
 127     ) throws DBusException {
 128         final var signal = new DbusSignalImpl(m, conn, objectPath);
 129         conn.exportObject(signal);
 130         final var initThread = new Thread(signal::initObjects);
 131         initThread.start();
 132
 133         logger.info("Exported dbus object: " + objectPath);
 134
 135         final var handler = outputWriter instanceof JsonWriter ? new JsonReceiveMessageHandler(m,
 136                 (JsonWriter) outputWriter) : new ReceiveMessageHandler(m, (PlainTextWriter) outputWriter);
 137         m.addReceiveHandler(handler);
 138
 139         final var dbusMessageHandler = new DbusReceiveMessageHandler(m, conn, objectPath);
 140         m.addReceiveHandler(dbusMessageHandler);
 141
 142         return initThread;
 143     }
 144 }