Remotely/Agent/Services/DeviceInformationServiceMac.cs
2021-07-29 07:58:05 -07:00

116 lines
3.6 KiB
C#

using Remotely.Agent.Interfaces;
using Remotely.Shared.Models;
using Remotely.Shared.Services;
using Remotely.Shared.Utilities;
using Remotely.Shared.Win32;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
namespace Remotely.Agent.Services
{
public class DeviceInformationServiceMac : DeviceInformationServiceBase, IDeviceInformationService
{
private readonly IProcessInvoker _processInvoker;
public DeviceInformationServiceMac(IProcessInvoker processInvoker)
{
_processInvoker = processInvoker;
}
public async Task<Device> CreateDevice(string deviceId, string orgId)
{
var device = GetDeviceBase(deviceId, orgId);
try
{
var (usedStorage, totalStorage) = GetSystemDriveInfo();
var (usedMemory, totalMemory) = GetMemoryInGB();
device.CurrentUser = Environment.UserName;
device.Drives = GetAllDrives();
device.UsedStorage = usedStorage;
device.TotalStorage = totalStorage;
device.UsedMemory = usedMemory;
device.TotalMemory = totalMemory;
device.CpuUtilization = await GetCpuUtilization();
device.AgentVersion = GetAgentVersion();
}
catch (Exception ex)
{
Logger.Write(ex, "Error getting device info.");
}
return device;
}
public new Task<double> GetCpuUtilization()
{
try
{
var cpuPercentStrings = _processInvoker.InvokeProcessOutput("zsh", "-c \"ps -A -o %cpu\"");
double cpuPercent = 0;
cpuPercentStrings
.Split(Environment.NewLine)
.ToList()
.ForEach(x => {
if (double.TryParse(x, out var result))
{
cpuPercent += result;
}
});
return Task.FromResult(cpuPercent / Environment.ProcessorCount / 100);
}
catch (Exception ex)
{
Logger.Write(ex, "Error while getting CPU utilization.");
}
return Task.FromResult((double)0);
}
public (double usedGB, double totalGB) GetMemoryInGB()
{
try
{
double totalGB = default;
var totalMemoryString = _processInvoker.InvokeProcessOutput("zsh", "-c \"sysctl -n hw.memsize\"");
if (double.TryParse(totalMemoryString, out var totalMemory))
{
totalGB = (double)Math.Round(totalMemory / 1024 / 1024 / 1024, 2);
}
double usedGB = default;
var memPercentStrings = _processInvoker.InvokeProcessOutput("zsh", $"-c \"ps -A -o %mem\"");
double usedMemPercent = 0;
memPercentStrings
.Split(Environment.NewLine)
.ToList()
.ForEach(x => {
if (double.TryParse(x, out var result))
{
usedMemPercent += result;
}
});
usedMemPercent = usedMemPercent / 4 / 100;
usedGB = usedMemPercent * totalGB;
return (usedGB,totalGB);
}
catch
{
return (0, 0);
}
}
}
}