/*******************************************************************************
* Copyright (c) 2012-2017 Codenvy, S.A.
* All rights reserved. This program and the accompanying materials
* are made available under the terms of the Eclipse Public License v1.0
* which accompanies this distribution, and is available at
* http://www.eclipse.org/legal/epl-v10.html
*
* Contributors:
* Codenvy, S.A. - initial API and implementation
*******************************************************************************/
package org.eclipse.che.ide.api.event.ng;
import com.google.gwt.user.client.Timer;
import com.google.inject.Inject;
import com.google.web.bindery.event.shared.EventBus;
import org.eclipse.che.api.core.jsonrpc.commons.RequestTransmitter;
import org.eclipse.che.api.machine.shared.dto.execagent.event.DtoWithPid;
import org.eclipse.che.api.project.shared.dto.event.ProjectTreeTrackingOperationDto;
import org.eclipse.che.ide.api.app.AppContext;
import org.eclipse.che.ide.api.machine.DevMachine;
import org.eclipse.che.ide.api.machine.ExecAgentCommandManager;
import org.eclipse.che.ide.api.machine.MachineEntity;
import org.eclipse.che.ide.api.machine.events.WsAgentStateEvent;
import org.eclipse.che.ide.api.machine.events.WsAgentStateHandler;
import org.eclipse.che.ide.api.workspace.event.EnvironmentOutputEvent;
import org.eclipse.che.ide.dto.DtoFactory;
import org.eclipse.che.ide.jsonrpc.JsonRpcInitializer;
import org.eclipse.che.ide.util.loging.Log;
import javax.inject.Singleton;
import java.util.function.Consumer;
import static java.util.Collections.singletonMap;
import static org.eclipse.che.api.project.shared.dto.event.ProjectTreeTrackingOperationDto.Type.START;
/**
* @author Dmitry Kuleshov
*/
@Singleton
public class JsonRpcWebSocketAgentEventListener implements WsAgentStateHandler {
private final JsonRpcInitializer initializer;
private final AppContext appContext;
private final EventBus eventBus;
private final RequestTransmitter requestTransmitter;
private final DtoFactory dtoFactory;
private final ExecAgentCommandManager execAgentCommandManager;
@Inject
public JsonRpcWebSocketAgentEventListener(JsonRpcInitializer initializer,
AppContext appContext, EventBus eventBus,
RequestTransmitter requestTransmitter,
DtoFactory dtoFactory,
ExecAgentCommandManager execAgentCommandManager) {
this.appContext = appContext;
this.initializer = initializer;
this.eventBus = eventBus;
this.requestTransmitter = requestTransmitter;
this.dtoFactory = dtoFactory;
this.execAgentCommandManager = execAgentCommandManager;
eventBus.addHandler(WsAgentStateEvent.TYPE, this);
}
@Override
public void onWsAgentStarted(WsAgentStateEvent event) {
initializeJsonRpc();
initializeTreeExplorerFileWatcher();
initializeGitCheckoutWatcher();
}
private void initializeJsonRpc() {
Log.debug(JsonRpcWebSocketAgentEventListener.class, "Web socket agent started event caught.");
try {
internalInitialize();
} catch (Exception e) {
Log.debug(JsonRpcWebSocketAgentEventListener.class, "Failed, will try one more time.");
new Timer() {
@Override
public void run() {
internalInitialize();
}
}.schedule(1_000);
}
}
private void internalInitialize() {
DevMachine devMachine = appContext.getDevMachine();
String devMachineId = devMachine.getId();
String wsAgentWebSocketUrl = devMachine.getWsAgentWebSocketUrl();
String wsAgentUrl = wsAgentWebSocketUrl.replaceFirst("(api)(/)(ws)", "websocket" + "$2" + appContext.getAppId());
String execAgentUrl = devMachine.getExecAgentUrl();
initializer.initialize("ws-agent", singletonMap("url", wsAgentUrl));
initializer.initialize(devMachineId, singletonMap("url", execAgentUrl));
for (MachineEntity machineEntity : appContext.getActiveRuntime().getMachines()) {
if (!machineEntity.isDev()) {
initializer.initialize(machineEntity.getId(), singletonMap("url", machineEntity.getExecAgentUrl()));
execAgentCommandManager.getProcesses(machineEntity.getId(), false)
.onSuccess(processes -> {
Consumer<Integer> pidConsumer = pid -> execAgentCommandManager
.getProcessLogs(machineEntity.getId(), pid, null, null, 50, 0)
.onSuccess(logs -> logs.forEach(log -> {
String fixedLog = log.getText().replaceAll("\\[STDOUT\\] ", "");
String machineName = machineEntity.getDisplayName();
eventBus.fireEvent(new EnvironmentOutputEvent(fixedLog, machineName));
}));
processes.stream()
.filter(it -> "CheWsAgent".equals(it.getName()))
.map(DtoWithPid::getPid)
.forEach(pidConsumer);
});
}
}
}
private void initializeTreeExplorerFileWatcher() {
ProjectTreeTrackingOperationDto params = dtoFactory.createDto(ProjectTreeTrackingOperationDto.class)
.withPath("/")
.withType(START);
requestTransmitter.newRequest()
.endpointId("ws-agent")
.methodName("track:project-tree")
.paramsAsDto(params)
.sendAndSkipResult();
}
private void initializeGitCheckoutWatcher() {
requestTransmitter.newRequest()
.endpointId("ws-agent")
.methodName("track:git-checkout")
.noParams()
.sendAndSkipResult();
}
@Override
public void onWsAgentStopped(WsAgentStateEvent event) {
DevMachine devMachine = appContext.getDevMachine();
String devMachineId = devMachine.getId();
Log.debug(JsonRpcWebSocketAgentEventListener.class, "Web socket agent stopped event caught.");
initializer.terminate("ws-agent");
initializer.terminate(devMachineId);
}
}