import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.node.ContainerNode;
+import com.fasterxml.jackson.databind.node.ObjectNode;
import org.asamk.signal.JsonReceiveMessageHandler;
import org.asamk.signal.JsonWriter;
-import org.asamk.signal.OutputWriter;
import org.asamk.signal.commands.Command;
import org.asamk.signal.commands.Commands;
-import org.asamk.signal.commands.JsonRpcCommand;
-import org.asamk.signal.commands.SignalCreator;
+import org.asamk.signal.commands.JsonRpcMultiCommand;
+import org.asamk.signal.commands.JsonRpcRegistrationCommand;
+import org.asamk.signal.commands.JsonRpcSingleCommand;
import org.asamk.signal.commands.exceptions.CommandException;
import org.asamk.signal.commands.exceptions.IOErrorException;
import org.asamk.signal.commands.exceptions.UntrustedKeyErrorException;
import org.asamk.signal.commands.exceptions.UserErrorException;
import org.asamk.signal.manager.Manager;
+import org.asamk.signal.manager.MultiAccountManager;
+import org.asamk.signal.manager.RegistrationManager;
import org.asamk.signal.util.Util;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
private final JsonRpcReader jsonRpcReader;
private final boolean noReceiveOnStart;
- private SignalCreator c;
+ private MultiAccountManager c;
private final Map<Manager, Manager.ReceiveMessageHandler> receiveHandlers = new HashMap<>();
private Manager m;
public SignalJsonRpcDispatcherHandler(
- final JsonWriter outputWriter, final Supplier<String> lineSupplier, final boolean noReceiveOnStart
+ final JsonWriter jsonWriter, final Supplier<String> lineSupplier, final boolean noReceiveOnStart
) {
this.noReceiveOnStart = noReceiveOnStart;
this.objectMapper = Util.createJsonObjectMapper();
- this.jsonRpcSender = new JsonRpcSender(outputWriter);
+ this.jsonRpcSender = new JsonRpcSender(jsonWriter);
this.jsonRpcReader = new JsonRpcReader(jsonRpcSender, lineSupplier);
}
- public void handleConnection(final SignalCreator c) {
+ public void handleConnection(final MultiAccountManager c) {
this.c = c;
if (!noReceiveOnStart) {
final ObjectMapper objectMapper, final String method, ContainerNode<?> params
) throws JsonRpcException {
var command = getCommand(method);
- // TODO implement listAccounts, register, verify, link
- if (command instanceof JsonRpcCommand<?> jsonRpcCommand) {
+ if (c != null) {
+ if (command instanceof JsonRpcMultiCommand<?> jsonRpcCommand) {
+ return runCommand(objectMapper, params, new MultiCommandRunnerImpl<>(c, jsonRpcCommand));
+ }
+ if (command instanceof JsonRpcRegistrationCommand<?> jsonRpcCommand) {
+ try (var manager = getRegistrationManagerFromParams(params)) {
+ if (manager != null) {
+ return runCommand(objectMapper,
+ params,
+ new RegistrationCommandRunnerImpl<>(manager, c, jsonRpcCommand));
+ } else {
+ throw new JsonRpcException(new JsonRpcResponse.Error(JsonRpcResponse.Error.INVALID_PARAMS,
+ "Method requires valid account parameter",
+ null));
+ }
+ } catch (IOException e) {
+ logger.warn("Failed to close registration manager", e);
+ }
+ }
+ }
+ if (command instanceof JsonRpcSingleCommand<?> jsonRpcCommand) {
if (m != null) {
return runCommand(objectMapper, params, new CommandRunnerImpl<>(m, jsonRpcCommand));
}
- if (params.has("account")) {
- Manager manager = c.getManager(params.get("account").asText());
- if (manager != null) {
- return runCommand(objectMapper, params, new CommandRunnerImpl<>(manager, jsonRpcCommand));
- }
+ final var manager = getManagerFromParams(params);
+ if (manager != null) {
+ return runCommand(objectMapper, params, new CommandRunnerImpl<>(manager, jsonRpcCommand));
} else {
throw new JsonRpcException(new JsonRpcResponse.Error(JsonRpcResponse.Error.INVALID_PARAMS,
"Method requires valid account parameter",
null));
}
+ private Manager getManagerFromParams(final ContainerNode<?> params) {
+ if (params != null && params.has("account")) {
+ final var manager = c.getManager(params.get("account").asText());
+ ((ObjectNode) params).remove("account");
+ return manager;
+ }
+ return null;
+ }
+
+ private RegistrationManager getRegistrationManagerFromParams(final ContainerNode<?> params) {
+ if (params != null && params.has("account")) {
+ try {
+ final var registrationManager = c.getNewRegistrationManager(params.get("account").asText());
+ ((ObjectNode) params).remove("account");
+ return registrationManager;
+ } catch (IOException | IllegalStateException e) {
+ logger.warn("Failed to load registration manager", e);
+ return null;
+ }
+ }
+ return null;
+ }
+
private Command getCommand(final String method) {
if ("subscribeReceive".equals(method)) {
return new SubscribeReceiveCommand();
return Commands.getCommand(method);
}
- private record CommandRunnerImpl<T>(Manager m, JsonRpcCommand<T> command) implements CommandRunner<T> {
+ private record CommandRunnerImpl<T>(Manager m, JsonRpcSingleCommand<T> command) implements CommandRunner<T> {
+
+ @Override
+ public void handleCommand(final T request, final JsonWriter jsonWriter) throws CommandException {
+ command.handleCommand(request, m, jsonWriter);
+ }
+
+ @Override
+ public TypeReference<T> getRequestType() {
+ return command.getRequestType();
+ }
+ }
+
+ private record RegistrationCommandRunnerImpl<T>(
+ RegistrationManager m, MultiAccountManager c, JsonRpcRegistrationCommand<T> command
+ ) implements CommandRunner<T> {
+
+ @Override
+ public void handleCommand(final T request, final JsonWriter jsonWriter) throws CommandException {
+ command.handleCommand(request, m, jsonWriter);
+ }
+
+ @Override
+ public TypeReference<T> getRequestType() {
+ return command.getRequestType();
+ }
+ }
+
+ private record MultiCommandRunnerImpl<T>(
+ MultiAccountManager c, JsonRpcMultiCommand<T> command
+ ) implements CommandRunner<T> {
@Override
- public void handleCommand(final T request, final OutputWriter outputWriter) throws CommandException {
- command.handleCommand(request, m, outputWriter);
+ public void handleCommand(final T request, final JsonWriter jsonWriter) throws CommandException {
+ command.handleCommand(request, c, jsonWriter);
}
@Override
interface CommandRunner<T> {
- void handleCommand(T request, OutputWriter outputWriter) throws CommandException;
+ void handleCommand(T request, JsonWriter jsonWriter) throws CommandException;
TypeReference<T> getRequestType();
}
final ObjectMapper objectMapper, final ContainerNode<?> params, final CommandRunner<?> command
) throws JsonRpcException {
final Object[] result = {null};
- final JsonWriter commandOutputWriter = s -> {
+ final JsonWriter commandJsonWriter = s -> {
if (result[0] != null) {
throw new AssertionError("Command may only write one json result");
}
};
try {
- parseParamsAndRunCommand(objectMapper, params, commandOutputWriter, command);
+ parseParamsAndRunCommand(objectMapper, params, commandJsonWriter, command);
} catch (JsonMappingException e) {
throw new JsonRpcException(new JsonRpcResponse.Error(JsonRpcResponse.Error.INVALID_REQUEST,
e.getMessage(),
private <T> void parseParamsAndRunCommand(
final ObjectMapper objectMapper,
final TreeNode params,
- final OutputWriter outputWriter,
+ final JsonWriter jsonWriter,
final CommandRunner<T> command
) throws CommandException, JsonMappingException {
T requestParams = null;
throw new AssertionError(e);
}
}
- command.handleCommand(requestParams, outputWriter);
+ command.handleCommand(requestParams, jsonWriter);
}
- private class SubscribeReceiveCommand implements JsonRpcCommand<Void> {
+ private class SubscribeReceiveCommand implements JsonRpcSingleCommand<Void> {
@Override
public String getName() {
@Override
public void handleCommand(
- final Void request, final Manager m, final OutputWriter outputWriter
+ final Void request, final Manager m, final JsonWriter jsonWriter
) throws CommandException {
subscribeReceive(m);
}
}
- private class UnsubscribeReceiveCommand implements JsonRpcCommand<Void> {
+ private class UnsubscribeReceiveCommand implements JsonRpcSingleCommand<Void> {
@Override
public String getName() {
@Override
public void handleCommand(
- final Void request, final Manager m, final OutputWriter outputWriter
+ final Void request, final Manager m, final JsonWriter jsonWriter
) throws CommandException {
unsubscribeReceive(m);
}
git.nmode.ca / signal-cli / blobdiff
