private static ISafeguardConnection CreateConnection(ToolOptions opts)
{
ISafeguardConnection connection;
if (!string.IsNullOrEmpty(opts.Username))
{
var password = HandlePassword(opts.ReadPassword);
connection = Safeguard.Connect(opts.Appliance, opts.IdentityProvider, opts.Username, password,
opts.ApiVersion, opts.Insecure);
}
else if (!string.IsNullOrEmpty(opts.CertificateFile))
{
var password = HandlePassword(opts.ReadPassword);
connection = Safeguard.Connect(opts.Appliance, opts.CertificateFile, password, opts.ApiVersion,
opts.Insecure);
}
else if (!string.IsNullOrEmpty(opts.Thumbprint))
{
connection = Safeguard.Connect(opts.Appliance, opts.Thumbprint, opts.ApiVersion, opts.Insecure);
}
else
{
throw new Exception("Must specify Username, CertificateFile, or Thumbprint");
}
return(connection);
}
private static ISafeguardEventListener CreatePersistentListener(ToolOptions opts)
{
if (!string.IsNullOrEmpty(opts.ApiKey))
{
void A2AHandler(string name, string body)
{
Log.Information("Received A2AHandler Event: {EventBody}", body);
}
if (!string.IsNullOrEmpty(opts.CertificateFile))
{
var password = HandlePassword(opts.ReadPassword);
return(Safeguard.A2A.Event.GetPersistentA2AEventListener(opts.ApiKey.ToSecureString(), A2AHandler,
opts.Appliance, opts.CertificateFile, password, opts.ApiVersion, opts.Insecure));
}
if (!string.IsNullOrEmpty(opts.Thumbprint))
{
return(Safeguard.A2A.Event.GetPersistentA2AEventListener(opts.ApiKey.ToSecureString(), A2AHandler,
opts.Appliance, opts.Thumbprint, opts.ApiVersion, opts.Insecure));
}
throw new Exception("Must specify CertificateFile or Thumbprint");
}
if (string.IsNullOrEmpty(opts.Event))
{
throw new Exception("You must specify the event name when not using A2A.");
}
void Handler(string name, string body)
{
Log.Information("Received Safeguard Event: {EventBody}", body);
}
ISafeguardEventListener listener;
if (!string.IsNullOrEmpty(opts.Username))
{
var password = HandlePassword(opts.ReadPassword);
listener = Safeguard.Event.GetPersistentEventListener(opts.Appliance, opts.IdentityProvider, opts.Username,
password, opts.ApiVersion, opts.Insecure);
}
else if (!string.IsNullOrEmpty(opts.CertificateFile))
{
var password = HandlePassword(opts.ReadPassword);
listener = Safeguard.Event.GetPersistentEventListener(opts.Appliance, opts.CertificateFile, password,
opts.ApiVersion, opts.Insecure);
}
else if (!string.IsNullOrEmpty(opts.Thumbprint))
{
listener = Safeguard.Event.GetPersistentEventListener(opts.Appliance, opts.Thumbprint, opts.ApiVersion,
opts.Insecure);
}
else
{
throw new Exception("Must specify Username, CertificateFile, or Thumbprint");
}
listener.RegisterEventHandler(opts.Event, Handler);
return(listener);
}
private static ISafeguardEventListener CreateA2AListener(ToolOptions opts)
{
using (var context = CreateA2AContext(opts))
{
return(context.GetA2AEventListener(opts.ApiKey.ToSecureString(),
(name, body) => { Log.Information("Received A2A Event: {EventBody}", body); }));
}
}
private static ISafeguardEventListener CreateA2AListener(ToolOptions opts)
{
using (var context = CreateA2AContext(opts))
{
void A2AHandler(string name, string body)
{
Log.Information("Received A2AHandler Event: {EventBody}", body);
}
if (!opts.ApiKey.Contains(','))
{
return(context.GetA2AEventListener(opts.ApiKey.ToSecureString(), A2AHandler));
}
return(context.GetA2AEventListener(opts.ApiKey.Split(',').Select(k => k.ToSecureString()), A2AHandler));
}
}
private static void Execute(ToolOptions opts)
{
try
{
var config = new LoggerConfiguration();
config.WriteTo.ColoredConsole(outputTemplate: "{Message:lj}{NewLine}{Exception}");
if (opts.Verbose)
{
config.MinimumLevel.Debug();
}
else
{
config.MinimumLevel.Information();
}
Log.Logger = config.CreateLogger();
void RunListener(ISafeguardEventListener listener)
{
listener.Start();
Log.Information("Press any key to shut down the event listener...");
Console.ReadKey();
listener.Stop();
}
if (opts.Persistent)
{
using (var listener = CreatePersistentListener(opts))
RunListener(listener);
}
else if (!string.IsNullOrEmpty(opts.ApiKey))
{
using (var listener = CreateA2AListener(opts))
RunListener(listener);
}
else
{
using (var listener = CreateEventListener(opts))
RunListener(listener);
}
}
catch (Exception ex)
{
Log.Error(ex, "Fatal exception occurred");
Environment.Exit(1);
}
}
private static ISafeguardEventListener CreateEventListener(ToolOptions opts)
{
if (string.IsNullOrEmpty(opts.Event))
{
throw new Exception("You must specify the event name when not using A2A.");
}
using (var connection = CreateConnection(opts))
{
Log.Information(
$"Access Token Lifetime Remaining: {connection.GetAccessTokenLifetimeRemaining()}");
var listener = connection.GetEventListener();
listener.RegisterEventHandler(opts.Event, (name, body) =>
{
Log.Information("Received Event: {EventName}", name);
Log.Information("Details: {EventBody}", body);
});
return(listener);
}
}
private static ISafeguardA2AContext CreateA2AContext(ToolOptions opts)
{
ISafeguardA2AContext context;
if (!string.IsNullOrEmpty(opts.CertificateFile))
{
var password = HandlePassword(opts.ReadPassword);
context = Safeguard.A2A.GetContext(opts.Appliance, opts.CertificateFile, password, opts.ApiVersion,
opts.Insecure);
}
else if (!string.IsNullOrEmpty(opts.Thumbprint))
{
context = Safeguard.A2A.GetContext(opts.Appliance, opts.Thumbprint, opts.ApiVersion, opts.Insecure);
}
else
{
throw new Exception("Must specify CertificateFile or Thumbprint");
}
return(context);
}
