mirror of
https://github.com/wpilibsuite/allwpilib
synced 2026-06-19 00:41:43 +00:00
75 lines
2.2 KiB
C
75 lines
2.2 KiB
C
|
/*----------------------------------------------------------------------------*/
|
||
|
|
/* Copyright (c) 2020 FIRST. All Rights Reserved. */
|
||
|
|
/* Open Source Software - may be modified and shared by FRC teams. The code */
|
||
|
|
/* must be accompanied by the FIRST BSD license file in the root directory of */
|
||
|
|
/* the project. */
|
||
|
|
/*----------------------------------------------------------------------------*/
|
||
|
|
|
||
|
|
#pragma once
|
||
|
|
|
||
|
|
#include <memory>
|
||
|
|
#include <string>
|
||
|
|
|
||
|
|
#include <WSProviderContainer.h>
|
||
|
|
#include <WSProvider_SimDevice.h>
|
||
|
|
#include <wpi/json.h>
|
||
|
|
#include <wpi/uv/AsyncFunction.h>
|
||
|
|
#include <wpi/uv/Loop.h>
|
||
|
|
#include <wpi/uv/Tcp.h>
|
||
|
|
#include <wpi/uv/Timer.h>
|
||
|
|
|
||
|
|
namespace wpilibws {
|
||
|
|
|
||
|
|
class HALSimWSClientConnection;
|
||
|
|
|
||
|
|
class HALSimWS {
|
||
|
|
public:
|
||
|
|
static std::shared_ptr<HALSimWS> GetInstance() { return g_instance; }
|
||
|
|
static void SetInstance(std::shared_ptr<HALSimWS> inst) { g_instance = inst; }
|
||
|
|
|
||
|
|
explicit HALSimWS(ProviderContainer& providers,
|
||
|
|
HALSimWSProviderSimDevices& simDevicesProvider)
|
||
|
|
: m_providers(providers), m_simDevicesProvider(simDevicesProvider) {}
|
||
|
|
HALSimWS(const HALSimWS&) = delete;
|
||
|
|
HALSimWS& operator=(const HALSimWS&) = delete;
|
||
|
|
|
||
|
|
bool Initialize();
|
||
|
|
static void Main(void*);
|
||
|
|
static void Exit(void*);
|
||
|
|
|
||
|
|
bool RegisterWebsocket(std::shared_ptr<HALSimBaseWebSocketConnection> hws);
|
||
|
|
void CloseWebsocket(std::shared_ptr<HALSimBaseWebSocketConnection> hws);
|
||
|
|
|
||
|
|
void OnNetValueChanged(const wpi::json& msg);
|
||
|
|
|
||
|
|
std::string GetTargetHost() const { return m_host; }
|
||
|
|
std::string GetTargetUri() const { return m_uri; }
|
||
|
|
int GetTargetPort() { return m_port; }
|
||
|
|
std::shared_ptr<wpi::uv::Loop> GetLoop() { return m_loop; }
|
||
|
|
|
||
|
|
private:
|
||
|
|
static std::shared_ptr<HALSimWS> g_instance;
|
||
|
|
|
||
|
|
void MainLoop();
|
||
|
|
|
||
|
|
void AttemptConnect();
|
||
|
|
|
||
|
|
bool m_tcp_connected = false;
|
||
|
|
std::shared_ptr<wpi::uv::Timer> m_connect_timer;
|
||
|
|
int m_connect_attempts = 0;
|
||
|
|
|
||
|
|
std::weak_ptr<HALSimBaseWebSocketConnection> m_hws;
|
||
|
|
|
||
|
|
ProviderContainer& m_providers;
|
||
|
|
HALSimWSProviderSimDevices& m_simDevicesProvider;
|
||
|
|
|
||
|
|
std::shared_ptr<wpi::uv::Loop> m_loop;
|
||
|
|
std::shared_ptr<wpi::uv::Tcp> m_tcp_client;
|
||
|
|
|
||
|
|
std::string m_host;
|
||
|
|
std::string m_uri;
|
||
|
|
int m_port;
|
||
|
|
};
|
||
|
|
|
||
|
|
} // namespace wpilibws
|