rusefi/java_tools/ts_plugin/src/main/java/com/rusefi/ts_plugin/RemoteTab.java

package com.rusefi.ts_plugin;

import com.rusefi.NamedThreadFactory;
import com.rusefi.core.SignatureHelper;
import com.rusefi.Timeouts;
import com.rusefi.core.ui.AutoupdateUtil;
import com.rusefi.core.Pair;
import com.rusefi.io.serial.StreamStatistics;
import com.rusefi.io.tcp.ServerSocketReference;
import com.rusefi.io.tcp.TcpIoStream;
import com.rusefi.proxy.NetworkConnector;
import com.rusefi.proxy.client.LocalApplicationProxy;
import com.rusefi.proxy.client.LocalApplicationProxyContextImpl;
import com.rusefi.proxy.client.UpdateType;
import com.rusefi.core.rusEFIVersion;
import com.rusefi.server.ApplicationRequest;
import com.rusefi.server.ControllerInfo;
import com.rusefi.server.SessionDetails;
import com.rusefi.server.UserDetails;
import com.rusefi.tools.online.HttpUtil;
import com.rusefi.tools.online.ProxyClient;
import com.rusefi.tools.online.PublicSession;
import com.rusefi.ts_plugin.auth.InstanceAuthContext;
import com.rusefi.ts_plugin.auth.SelfInfo;
import com.rusefi.ui.AuthTokenPanel;
import com.rusefi.ui.util.URLLabel;
import org.jetbrains.annotations.NotNull;
import org.putgemin.VerticalFlowLayout;

import javax.swing.*;
import java.awt.*;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.io.IOException;
import java.util.List;
import java.util.concurrent.Executor;
import java.util.concurrent.Executors;
import java.util.concurrent.atomic.AtomicReference;

import static com.rusefi.core.preferences.storage.PersistentConfiguration.getConfig;

/**
 * remote ECU access & control
 *
 * @see RemoteTabSandbox
 * @see PluginEntry
 */
public class RemoteTab {
    private static final String APPLICATION_PORT = "application_port";
    public static final String HOWTO_REMOTE_TUNING = "https://github.com/rusefi/rusefi/wiki/HOWTO-Remote-Tuning";
    private final JComponent content = new JPanel(new BorderLayout());
    private final JScrollPane scroll = new JScrollPane(content, JScrollPane.VERTICAL_SCROLLBAR_ALWAYS, JScrollPane.HORIZONTAL_SCROLLBAR_AS_NEEDED);

    private final JPanel list = new JPanel(new VerticalFlowLayout());
    private final JTextField oneTimePasswordControl = new JTextField("0") {
        @Override
        public Dimension getPreferredSize() {
            Dimension size = super.getPreferredSize();
            // todo: dynamic calculation of desired with based on String width?
            return new Dimension(100, size.height);
        }
    };

    private StreamStatusControl streamStatusControl = null;

    private final JButton disconnect = new JButton("Disconnect");

    private final Executor listDownloadExecutor = Executors.newSingleThreadExecutor(new NamedThreadFactory("online list downloader", true));

    public RemoteTab() {
        JButton refresh = new JButton("Refresh Remote Controllers List");
        refresh.addActionListener(e -> requestControllersList());

        disconnect.addActionListener(e -> {
            LocalApplicationProxy localApplicationProxy = RemoteTabController.INSTANCE.getLocalApplicationProxy();
            if (localApplicationProxy != null)
                localApplicationProxy.close();
            RemoteTabController.INSTANCE.setState(RemoteTabController.State.NOT_CONNECTED);
            requestControllersList();
        });


        Timer timer = new Timer(Timeouts.SECOND, new ActionListener() {
            @Override
            public void actionPerformed(ActionEvent e) {
                if (streamStatusControl != null)
                    streamStatusControl.update();
            }
        });
        timer.start();

        JTextField applicationPort = new JTextField() {
            @Override
            public Dimension getPreferredSize() {
                Dimension size = super.getPreferredSize();
                // todo: dynamic calculation of desired with based on String width?
                return new Dimension(100, size.height);
            }
        };
        IntegerDocumentFilter.install(applicationPort);
        IntegerDocumentFilter.install(oneTimePasswordControl);
        String portProperty = getLocalPort();
        applicationPort.setText(portProperty);

        JPanel topLines = new JPanel(new VerticalFlowLayout());

        topLines.add(new URLLabel(HOWTO_REMOTE_TUNING));
        topLines.add(new SelfInfo().getContent());
        topLines.add(refresh);
        topLines.add(new JLabel("Local Port for tuning software"));
        topLines.add(applicationPort);
        topLines.add(new JLabel("One time password:"));
        topLines.add(oneTimePasswordControl);

        content.add(topLines, BorderLayout.NORTH);
        content.add(list, BorderLayout.CENTER);
        list.add(new JLabel("Requesting list of ECUs"));

        InstanceAuthContext.listeners.add(userDetails -> requestControllersList());

        LocalApplicationProxy currentState = RemoteTabController.INSTANCE.getLocalApplicationProxy();
        if (currentState == null) {
            requestControllersList();
        } else {
            setConnectedStatus(currentState.getApplicationRequest().getVehicleOwner(), null,
                    currentState.getApplicationRequest().getSessionDetails().getControllerInfo());
        }
    }

    private String getLocalPort() {
        return getConfig().getRoot().getProperty(APPLICATION_PORT, "29001");
    }

    private void requestControllersList() {
        listDownloadExecutor.execute(() -> {
            try {
                List<PublicSession> userDetails = ProxyClient.getOnlineApplications(HttpUtil.PROXY_JSON_API_HTTP_PORT);
                SwingUtilities.invokeLater(() -> showList(userDetails));
            } catch (IOException e) {
                e.printStackTrace();
            }
        });
    }

    private void showList(List<PublicSession> userDetails) {
        if (RemoteTabController.INSTANCE.getState() != RemoteTabController.State.NOT_CONNECTED)
            return;
        list.removeAll();
        if (userDetails.isEmpty()) {
            list.add(new JLabel("No ECUs are broadcasting at the moment :("));
        } else {

            JPanel verticalPanel = new JPanel(new VerticalFlowLayout());
            list.add(verticalPanel);

            for (PublicSession user : userDetails) {
                verticalPanel.add(createControllerRow(user));
            }
        }
        AutoupdateUtil.trueLayout(list);
    }

    private JComponent createControllerRow(PublicSession publicSession) {
        ControllerInfo controllerInfo = publicSession.getControllerInfo();

        JComponent topLine = new JPanel(new FlowLayout());
        topLine.add(new JLabel(publicSession.getVehicleOwner().getUserName()));
        topLine.add(new JLabel(controllerInfo.getVehicleName() + " " + controllerInfo.getEngineMake() + " " + controllerInfo.getEngineCode()));

        JPanel bottomPanel = new JPanel(new VerticalFlowLayout());

        if (publicSession.isUsed()) {
            bottomPanel.add(new JLabel(" Used by " + publicSession.getTunerName()));
        } else {
            JButton connect = new JButton("Connect to " + publicSession.getVehicleOwner().getUserName() + " ECU");
            connect.addActionListener(event -> connectToProxy(publicSession));
            bottomPanel.add(connect);

            if (InstanceAuthContext.isOurController(publicSession.getVehicleOwner().getUserId())) {

                if (publicSession.getImplementation().equals(NetworkConnector.Implementation.SBC.name())) {

                    JPanel updateSoftwarePanel = new JPanel(new FlowLayout());


                    JButton updateSoftwareLatest = new JButton("Update Connector to Latest");
                    updateSoftwareLatest.addActionListener(new AbstractAction() {
                        @Override
                        public void actionPerformed(ActionEvent e) {
                            requestUpdate(publicSession, updateSoftwareLatest, UpdateType.CONTROLLER);
                        }
                    });

                    JButton updateSoftwareRelease = new JButton("Update Connector to Release");
                    updateSoftwareRelease.addActionListener(new AbstractAction() {
                        @Override
                        public void actionPerformed(ActionEvent e) {
                            requestUpdate(publicSession, updateSoftwareRelease, UpdateType.CONTROLLER_RELEASE);
                        }
                    });
                    updateSoftwarePanel.add(updateSoftwareLatest);
                    updateSoftwarePanel.add(updateSoftwareRelease);

                    bottomPanel.add(updateSoftwarePanel);
                }

                JPanel updateFirmwarePanel = createUpdateFirmwarePanel(publicSession);

                bottomPanel.add(updateFirmwarePanel);
            }
        }

        JPanel userPanel = new JPanel(new BorderLayout());

        JPanel infoLine = new JPanel(new FlowLayout());
        infoLine.add(new JLabel("Age " + publicSession.getAge()));
        infoLine.add(getSignatureDownload(controllerInfo));

        userPanel.add(topLine, BorderLayout.NORTH);
        userPanel.add(infoLine, BorderLayout.CENTER);
        userPanel.add(bottomPanel, BorderLayout.SOUTH);

        userPanel.setBorder(BorderFactory.createLineBorder(Color.BLACK));

        return userPanel;
    }

    @NotNull
    private JPanel createUpdateFirmwarePanel(PublicSession publicSession) {
        JButton updateFirmwareLatest = new JButton("Update ECU to latest");
        updateFirmwareLatest.addActionListener(new AbstractAction() {
            @Override
            public void actionPerformed(ActionEvent e) {
                requestUpdate(publicSession, updateFirmwareLatest, UpdateType.FIRMWARE);
            }
        });

        JButton updateFirmwareRelease = new JButton("Update ECU to release");
        updateFirmwareRelease.addActionListener(new AbstractAction() {
            @Override
            public void actionPerformed(ActionEvent e) {
                requestUpdate(publicSession, updateFirmwareRelease, UpdateType.FIRMWARE_RELEASE);
            }
        });


        JPanel updateFirmwarePanel = new JPanel(new FlowLayout());
        updateFirmwarePanel.add(updateFirmwareLatest);
        updateFirmwarePanel.add(updateFirmwareRelease);
        return updateFirmwarePanel;
    }

    private void requestUpdate(PublicSession publicSession, JButton updateSoftware, UpdateType type) {
        try {
            LocalApplicationProxy.requestSoftwareUpdate(HttpUtil.PROXY_JSON_API_HTTP_PORT,
                    getApplicationRequest(publicSession), type);
            updateSoftware.setText("Update requested");
        } catch (IOException ioException) {
            ioException.printStackTrace();
        }
    }

    @NotNull
    private URLLabel getSignatureDownload(ControllerInfo controllerInfo) {
        Pair<String, String> url = SignatureHelper.getUrl(controllerInfo.getSignature());

        return new URLLabel(url.second, url.first);
    }

    private void connectToProxy(PublicSession publicSession) {
        RemoteTabController.INSTANCE.setState(RemoteTabController.State.CONNECTING);
        setStatus("Connecting to " + publicSession.getVehicleOwner().getUserName());

        LocalApplicationProxy.ConnectionListener connectionListener = (localApplicationProxy, authenticatorToProxyStream) -> {
            RemoteTabController.INSTANCE.setConnected(localApplicationProxy);
            SwingUtilities.invokeLater(new Runnable() {
                @Override
                public void run() {
                    setConnectedStatus(publicSession.getVehicleOwner(), authenticatorToProxyStream, publicSession.getControllerInfo());
                }
            });
        };

        new Thread(() -> {
            runAuthenticator(publicSession, connectionListener);
        }, "Authenticator").start();
    }

    private void setConnectedStatus(UserDetails userDetails, StreamStatistics authenticatorToProxyStream, ControllerInfo controllerInfo) {
        if (authenticatorToProxyStream != null) {
            streamStatusControl = new StreamStatusControl(authenticatorToProxyStream);
        }

        setStatus("Connected to " + userDetails.getUserName(),
                new JLabel("You can now connect your TunerStudio to IP address localhost and port " + getLocalPort()),
                new URLLabel(SignatureHelper.getUrl(controllerInfo.getSignature()).first),
                disconnect, streamStatusControl == null ? null : streamStatusControl.getContent());
    }

    private void setStatus(String text, JComponent... extra) {
        list.removeAll();
        list.add(new JLabel(text));
        for (JComponent component : extra) {
            if (component != null) {
                list.add(component);
            }
        }
        AutoupdateUtil.trueLayout(list);
    }

    private void runAuthenticator(PublicSession publicSession, LocalApplicationProxy.ConnectionListener connectionListener) {
        ApplicationRequest applicationRequest = getApplicationRequest(publicSession);

        try {
            AtomicReference<ServerSocketReference> serverHolderAtomicReference = new AtomicReference<>();

            TcpIoStream.DisconnectListener disconnectListener = message -> SwingUtilities.invokeLater(() -> {
                System.out.println("Disconnected " + message);
                setStatus("Disconnected: " + message);
                RemoteTabController.INSTANCE.setState(RemoteTabController.State.NOT_CONNECTED);
                ServerSocketReference serverHolder = serverHolderAtomicReference.get();
                if (serverHolder != null)
                    serverHolder.close();
            });

            LocalApplicationProxyContextImpl context = new LocalApplicationProxyContextImpl() {
                @Override
                public int authenticatorPort() {
                    return Integer.parseInt(getLocalPort());
                }
            };
            ServerSocketReference serverHolder = LocalApplicationProxy.startAndRun(
                    context,
                    applicationRequest,
                    HttpUtil.PROXY_JSON_API_HTTP_PORT, disconnectListener, connectionListener);
            serverHolderAtomicReference.set(serverHolder);
        } catch (IOException e) {
            setStatus("IO error: " + e);
        }
    }

    @NotNull
    private ApplicationRequest getApplicationRequest(PublicSession publicSession) {
        SessionDetails sessionDetails = new SessionDetails(NetworkConnector.Implementation.Plugin,
                publicSession.getControllerInfo(), AuthTokenPanel.getAuthToken(),
                Integer.parseInt(oneTimePasswordControl.getText()), rusEFIVersion.CONSOLE_VERSION);

        return new ApplicationRequest(sessionDetails, publicSession.getVehicleOwner());
    }

    public JComponent getContent() {
        return scroll;
    }

    interface Listener {
        void onConnected();
    }
}