static void Main(string[] args) {
var arguments = ArgumentHelper.ToArguments(args);
if (arguments.ContainsKey(@"command")) {
// Split the command/arguments.
var command = arguments[@"command"];
arguments = ArgumentHelper.ScrubAlphaNumericKeys(arguments);
// Create a service to at least handle output to the console for signals.
ServiceController service = new ServiceController {
Settings = {
// Force no update checks. We're just here to configure.
ServiceUpdateCore = false
},
SignalBegin = (controller, message) => Console.WriteLine(@"Signal: {0}", message.Name),
SignalEnd = (controller, message, seconds) => Console.WriteLine(@"Signal: {0} completed in {1} seconds", message.Name, seconds),
WriteServiceConfigBegin = controller => Console.Write(@"Writing config.. "),
WriteServiceConfigEnd = controller => Console.WriteLine(@"Complete"),
DisposeServiceBegin = controller => Console.Write(@"Disposing service.. "),
DisposeServiceEnd = controller => Console.WriteLine(@"Complete"),
UnloadServiceBegin = controller => Console.Write(@"Unloading service domain.. "),
UnloadServiceEnd = controller => Console.WriteLine(@"Complete"),
Observer = {
StatusChange = (observer, type) => Console.WriteLine(@"Status: {0}", type.ToString())
},
SignalResult = (controller, message) => {
foreach (var item in message.Arguments) {
Console.WriteLine(@"{0}: {1}", item.Key, item.Value);
}
},
Arguments = new List<String>(args)
};
// See if we can process the command without loading up the Potato instance (e.g certificate generaton)
var result = ConfigController.Dispatch(command, arguments);
if (result != null) {
service.SignalMessage(result);
}
// Else we need to start up and execute a message on the Potato instance.
else {
service.ExecuteMessage(new ServiceMessage() {
Name = command,
Arguments = arguments
});
}
// We've executed the command locally or on the Potato instance, which may have altered the config files.
// Now we dispose the service which will write the configs.
service.Dispose();
}
else {
Console.WriteLine(@"Missing command argument");
}
}
protected override void OnStart(string[] args)
{
this.Instance = new ServiceController() {
Arguments = new List<String>(args),
Settings = new ServiceSettings(new List<String>(args))
};
this.Instance.SignalMessage(new ServiceMessage() {
Name = "start"
});
}
/// <summary>
/// Updates the core of Potato. This won't validate that an update
/// is successful, it just goes through the motions of updating.
/// </summary>
public static ServiceMessage ServiceUpdateCore() {
ServiceController service = new ServiceController {
Settings = {
ServiceUpdateCore = true
}
};
service.UpdateCore();
return new ServiceMessage() {
Name = "result",
Arguments = new Dictionary<String, String>() {
{ "Command", "ServiceUpdateCore" },
{ "Success", "True" }
}
};
}
private static void Main(string[] args) {
Console.WriteLine(Resources.ConsoleHeader);
if (args.Length > 0) {
// Support for --help command?
System.Console.WriteLine(@"Starting with arguments: {0}", String.Join(" ", args));
}
ServiceController service = new ServiceController {
Arguments = new List<String>(args),
Settings = new ServiceSettings(new List<String>(args)),
Packages = {
BeforeRepositoryInitialize = () => Console.WriteLine(@"Initializing package repository.."),
BeforeSourcePackageFetch = () => Console.WriteLine(@"Checking source repositories.."),
BeforeLocalPackageFetch = () => Console.WriteLine(@"Checking local repository.."),
PackageInstalling = (sender, packageId, version) => Console.WriteLine(@"Installing {0} version {1}..", packageId, version),
PackageInstalled = (sender, packageId, version) => Console.WriteLine(@"Installed {0} version {1}", packageId, version),
PackageUninstalling = (sender, packageId, version) => Console.WriteLine(@"Uninstalling {0} version {1}..", packageId, version),
PackageUninstalled = (sender, packageId, version) => Console.WriteLine(@"Uninstalled {0} version {1}..", packageId, version),
PackageMissing = packageId => Console.WriteLine(@"Couldn't find package {0}.", packageId),
PackageActionCanceled = packageId => Console.WriteLine(@"Package {0} is up to date.", packageId),
RepositoryException = (hint, exception) => {
if (exception is UnauthorizedAccessException) {
Console.WriteLine(@"Unable to access path {0}", Defines.PackagesDirectory);
Console.WriteLine(@"Ensure all applications and open folders using the packages folder are closed and try again.");
}
ServiceControllerHelpers.LogUnhandledException(hint, exception);
}
},
SignalBegin = (controller, message) => Console.WriteLine(@"Signal: {0}", message.Name),
SignalEnd = (controller, message, seconds) => Console.WriteLine(@"Signal: {0} completed in {1} seconds", message.Name, seconds),
SignalParameterError = (controller, list) => Console.WriteLine(@"Missing or valid parameters: {0}", String.Join(", ", list)),
SignalStatistics = (controller, domain) => {
Console.WriteLine(@"Service Controller");
Console.WriteLine(@"+--------------------------------------------------------+");
Console.WriteLine(@"MonitoringSurvivedMemorySize: {0:N0} K", AppDomain.CurrentDomain.MonitoringSurvivedMemorySize / 1024);
Console.WriteLine(@"MonitoringTotalAllocatedMemorySize: {0:N0} K", AppDomain.CurrentDomain.MonitoringTotalAllocatedMemorySize / 1024);
Console.WriteLine(@"MonitoringTotalProcessorTime: {0}", AppDomain.CurrentDomain.MonitoringTotalProcessorTime);
Console.WriteLine("");
Console.WriteLine(@"Service Domain");
Console.WriteLine(@"+--------------------------------------------------------+");
Console.WriteLine(@"MonitoringSurvivedMemorySize: {0:N0} K", domain.MonitoringSurvivedMemorySize / 1024);
Console.WriteLine(@"MonitoringTotalAllocatedMemorySize: {0:N0} K", domain.MonitoringTotalAllocatedMemorySize / 1024);
Console.WriteLine(@"MonitoringTotalProcessorTime: {0}", domain.MonitoringTotalProcessorTime);
},
SignalHelp = controller => Console.WriteLine(Resources.ConsoleHelp),
WriteServiceConfigBegin = controller => Console.Write(@"Writing config.. "),
WriteServiceConfigEnd = controller => Console.WriteLine(@"Complete"),
DisposeServiceBegin = controller => Console.Write(@"Disposing service.. "),
DisposeServiceEnd = controller => Console.WriteLine(@"Complete"),
UnloadServiceBegin = controller => Console.Write(@"Unloading service domain.. "),
UnloadServiceEnd = controller => Console.WriteLine(@"Complete"),
Observer = {
StatusChange = (observer, type) => Console.WriteLine(@"Status: {0}", type.ToString())
}
};
service.SignalMessage(new ServiceMessage() {
Name = "help"
});
service.SignalMessage(new ServiceMessage() {
Name = "start"
});
AutoResetEvent exitWait = new AutoResetEvent(false);
Task.Factory.StartNew(() => {
var input = String.Empty;
do {
input = Console.ReadLine();
if (input != null) {
var words = Program.Wordify(input);
service.SignalMessage(new ServiceMessage() {
Name = words.FirstOrDefault(),
Arguments = ArgumentHelper.ToArguments(words.Skip(1).ToList())
});
}
} while (String.Compare(input, "exit", StringComparison.OrdinalIgnoreCase) != 0);
exitWait.Set();
});
// Wait until exit() is called.
exitWait.WaitOne();
service.SignalMessage(new ServiceMessage() {
Name = "stop"
});
service.Dispose();
System.Console.WriteLine(@"Closing..");
Thread.Sleep(1000);
}
