/**
* Copyright (C) 2010-2017 Structr GmbH
*
* This file is part of Structr <http://structr.org>.
*
* Structr is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as
* published by the Free Software Foundation, either version 3 of the
* License, or (at your option) any later version.
*
* Structr is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with Structr. If not, see <http://www.gnu.org/licenses/>.
*/
package org.structr.cloud.websocket;
import java.util.Map;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.structr.cloud.CloudService;
import org.structr.cloud.HostInfo;
import org.structr.cloud.WebsocketProgressListener;
import org.structr.cloud.transmission.PushTransmission;
import org.structr.common.error.FrameworkException;
import org.structr.core.GraphObject;
import org.structr.core.app.App;
import org.structr.core.app.StructrApp;
import org.structr.core.graph.Tx;
import org.structr.websocket.StructrWebSocket;
import org.structr.websocket.command.AbstractCommand;
import org.structr.websocket.message.MessageBuilder;
import org.structr.websocket.message.WebSocketMessage;
//~--- classes ----------------------------------------------------------------
/**
*
*
*/
public class PushNodesCommand extends AbstractCommand {
private static final Logger logger = LoggerFactory.getLogger(PushNodesCommand.class.getName());
static {
StructrWebSocket.addCommand(PushNodesCommand.class);
}
@Override
public void processMessage(final WebSocketMessage webSocketData) {
final Map<String, Object> properties = webSocketData.getNodeData();
final String sourceId = webSocketData.getId();
final Object recursiveSource = properties.get("recursive");
final String username = (String)properties.get("username");
final String password = (String)properties.get("password");
final String host = (String)properties.get("host");
final Long port = (Long)properties.get("port");
final String key = (String)properties.get("key");
if (sourceId != null && host != null && port != null && username != null && password != null && key != null) {
final App app = StructrApp.getInstance();
try (final Tx tx = app.tx()) {
final GraphObject root = app.getNodeById(sourceId);
if (root != null) {
boolean recursive = false;
if (recursiveSource != null) {
recursive = "true".equals(recursiveSource.toString());
}
CloudService.doRemote(
webSocket.getSecurityContext(),
new PushTransmission(root, recursive),
new HostInfo(username, password, host, port.intValue()),
new WebsocketProgressListener(getWebSocket(), key, callback)
);
} else {
getWebSocket().send(MessageBuilder.status().code(404).message("Entity with ID " + sourceId + " not found.").build(), true);
}
tx.success();
} catch (FrameworkException fex) {
logger.warn("", fex);
getWebSocket().send(MessageBuilder.status().code(400).message(fex.getMessage()).build(), true);
}
} else {
getWebSocket().send(MessageBuilder.status().code(400).message("The PUSH command needs id, username, password, host, port and key!").build(), true);
}
}
//~--- get methods ----------------------------------------------------
@Override
public String getCommand() {
return "PUSH";
}
}