/// <summary>
/// Process user driven requests
/// </summary>
/// <param name="json">Json recieved form the external client shown here as a string.</param>
public static void ProcessRequest(string json)
{
//Check if the json is in the HeartBeat class
if (json.Contains("heartbeat"))
{
//Deserialize the json to a HeatBeat class
HeartBeat beat = JsonConvert.DeserializeObject <HeartBeat>(json);
int HighestID = 0;
foreach (var beats in Program.Clients)
{
//Check if the client is already in the HeatBeat list, if it is not add it to the list
if (beats.ipSocket == beat.ipSocket)
{
ConsoleView.AddLog("Client already in database");
return;
}
if (beats.pcId > HighestID)
{
HighestID = beats.pcId;
}
}
Program.Clients.Add(
new HeartBeat
{
pcId = HighestID + 1,
heartbeat = beat.heartbeat,
ipSocket = beat.ipSocket,
lastUpdate = beat.lastUpdate,
pcName = beat.pcName
}
);
ConsoleView.AddLog("Added client on " + beat.ipSocket + " with ID:" + (HighestID + 1).ToString());
}
//If the json is a restart command tell the server application that the client is restarting
if (json.Contains("restart"))
{
ConsoleView.AddLog("Restarting remote computer");
}
ConsoleView.Update();
}
/// <summary>
///
/// </summary>
static void ClientHeartbeatTimer_Elapsed(object sender, ElapsedEventArgs e)
{
int clientIndex = 0;
int clientId = 0;
if (Clients != null)
{
try {
foreach (var client in Clients)
{
if (!WebPost.SendRequest(client.ipSocket, true, "{\"heartbeat\":true, \"pcId\":" + client.pcId.ToString() + "}"))
{
//Remove client from Client
clientId = client.pcId;
Clients.RemoveAt(clientIndex);
//Remove client from ClientUsage
clientIndex = 0;
foreach (var clientU in ClientUsage)
{
if (clientU.pcId == clientId)
{
ClientUsage.RemoveAt(clientIndex);
}
clientIndex++;
}
ConsoleView.AddLog("Heartbeat failed, removing client: " + client.ipSocket);
return;
}
clientIndex++;
}
}catch (Exception)
{
}
}
ConsoleData.CurrentConnectedClients = clientIndex;
ConsoleView.Update();
}
static void Main(string[] args)
{
//First update of the console to draw UI
ConsoleView.Update();
//Set timer up initially and activate it
ClientHeartbeatTimer = new Timer(HeartbeatInterval * 1000);
ClientHeartbeatTimer.Elapsed += new ElapsedEventHandler(ClientHeartbeatTimer_Elapsed);
ClientHeartbeatTimer.Enabled = true;
while (true)
{
//Wait for user input
string ConsoleString = Console.ReadLine();
//Split console commands via a space or " " represented as a null
String[] ConsoleCommand = ConsoleString.Split(null);
switch (ConsoleCommand[0])
{
//The add command to add a new client to the Clients list
case "add":
if (!WebPost.SendRequest(ConsoleCommand[1]))
{
ConsoleView.AddLog("Failed to add client on socket: " + ConsoleCommand[1]);
}
else
{
ConsoleData.CurrentConnectedClients += 1;
}
break;
//Set the interval between
case "update":
HeartbeatInterval = Convert.ToInt32(ConsoleCommand[1]);
ClientHeartbeatTimer.Interval = (HeartbeatInterval * 1000);
ConsoleView.AddLog("Update rate modifier: " + ConsoleCommand[1]);
break;
//Restart the remote computer
case "restart":
case "reboot":
if (!WebPost.SendRequest(ConsoleCommand[1], false, "{\"restartPc\" : true}"))
{
ConsoleView.AddLog("Failed to reboot remote computer");
}
else
{
ConsoleView.AddLog("Restarting remote computer " + ConsoleCommand[1]);
}
break;
//show the next 5 clients in the list on the console ui
case "list":
ConsoleView.WriteClients(Convert.ToInt32(ConsoleCommand[1]));
break;
//Basic commands
case "help":
case "?":
Console.WriteLine("Commands Available:");
Console.WriteLine("add <socket>, clear");
break;
}
ConsoleView.ClearInput();
ConsoleView.Update();
}
}