X-Git-Url: https://git.nmode.ca/signal-cli/blobdiff_plain/4a1af0786c938f887a109a17dcc879da21704a8b..1d77153a2bc5c6c7a8d7bc7b72fe8fc7db1208c8:/src/main/java/org/asamk/signal/commands/VersionCommand.java

diff --git a/src/main/java/org/asamk/signal/commands/VersionCommand.java b/src/main/java/org/asamk/signal/commands/VersionCommand.java
index b8fb1fe9..52099ff3 100644
--- a/src/main/java/org/asamk/signal/commands/VersionCommand.java
+++ b/src/main/java/org/asamk/signal/commands/VersionCommand.java
@@ -1,35 +1,65 @@
 package org.asamk.signal.commands;
 
+import com.fasterxml.jackson.core.type.TypeReference;
+
+import net.sourceforge.argparse4j.inf.Namespace;
+import net.sourceforge.argparse4j.inf.Subparser;
+
 import org.asamk.signal.BaseConfig;
-import org.asamk.signal.JsonWriter;
+import org.asamk.signal.OutputType;
 import org.asamk.signal.commands.exceptions.CommandException;
 import org.asamk.signal.manager.Manager;
 import org.asamk.signal.manager.MultiAccountManager;
+import org.asamk.signal.output.JsonWriter;
+import org.asamk.signal.output.OutputWriter;
+import org.asamk.signal.output.PlainTextWriter;
 
+import java.util.List;
 import java.util.Map;
 
-public class VersionCommand implements JsonRpcSingleCommand<Void>, JsonRpcMultiCommand<Void> {
+public class VersionCommand implements JsonRpcLocalCommand, JsonRpcMultiLocalCommand {
 
     @Override
     public String getName() {
         return "version";
     }
 
+    @Override
+    public List<OutputType> getSupportedOutputTypes() {
+        return List.of(OutputType.PLAIN_TEXT, OutputType.JSON);
+    }
+
+    @Override
+    public void attachToSubparser(final Subparser subparser) {
+    }
+
     @Override
     public void handleCommand(
-            final Void request, final Manager m, final JsonWriter jsonWriter
+            final Namespace ns, final Manager m, final OutputWriter outputWriter
     ) throws CommandException {
-        outputVersion(jsonWriter);
+        outputVersion(outputWriter);
     }
 
     @Override
     public void handleCommand(
-            final Void request, final MultiAccountManager c, final JsonWriter jsonWriter
+            final Namespace ns, final MultiAccountManager c, final OutputWriter outputWriter
     ) throws CommandException {
-        outputVersion(jsonWriter);
+        outputVersion(outputWriter);
+    }
+
+    @Override
+    public TypeReference<Map<String, Object>> getRequestType() {
+        return new TypeReference<>() {};
     }
 
-    private void outputVersion(final JsonWriter jsonWriter) {
-        jsonWriter.write(Map.of("version", BaseConfig.PROJECT_VERSION));
+    private void outputVersion(final OutputWriter outputWriter) {
+        final var projectName = BaseConfig.PROJECT_NAME == null ? "signal-cli" : BaseConfig.PROJECT_NAME;
+        final var version = BaseConfig.PROJECT_VERSION == null ? "unknown" : BaseConfig.PROJECT_VERSION;
+
+        if (outputWriter instanceof JsonWriter jsonWriter) {
+            jsonWriter.write(Map.of("version", version));
+        } else if (outputWriter instanceof PlainTextWriter plainTextWriter) {
+            plainTextWriter.println("{} {}", projectName, version);
+        }
     }
 }