static void Main(string[] args)
{
if (Environment.UserInteractive)
{
if (args.Length == 0 || args[0] == "-?" || args[0] == "-h")
{
DisplayUsage();
return;
}
switch (args[0])
{
case "--install":
case "-i":
new ServiceManager(args.Length == 1 ? DefaultServicName : args[1]).Install();
break;
case "--remove":
case "-r":
new ServiceManager(args.Length == 1 ? DefaultServicName : args[1]).Remove();
break;
case "--start":
new ServiceManager(args.Length == 1 ? DefaultServicName : args[1]).Start();
break;
case "--stop":
new ServiceManager(args.Length == 1 ? DefaultServicName : args[1]).Stop();
break;
case "mock":
new LogRotaterMock().Execute(args);
break;
case "daemon":
var evt = new ManualResetEvent(false);
Console.CancelKeyPress += (s, e) =>
{
evt.Set();
e.Cancel = true;
};
ThreadPool.UnsafeQueueUserWorkItem(_ =>
{
using (var daemon = new Program("console"))
{
daemon.OnStart(new string[0]);
Console.WriteLine("logrotate is running, press Ctrl+C to stop.");
((ManualResetEvent)_).WaitOne();
daemon.OnStop();
Console.WriteLine("logrotate is stopped.");
}
}, evt);
evt.WaitOne();
Thread.Sleep(100);
break;
case InternalIoRedirectCommand:
if (args.Length == 1 || !File.Exists(args[1]))
{
Environment.Exit(1);
}
using (var stream = File.Open(args[1], FileMode.Open, FileAccess.Write, FileShare.Read))
{
using (var io = new StreamWriter(stream)
{
AutoFlush = true
})
{
Console.SetError(io);
Console.SetOut(io);
var trimArgs = new string[args.Length - 2];
Array.Copy(args, 2, trimArgs, 0, trimArgs.Length);
Main(trimArgs);
}
}
break;
default:
// service command
using (var client = PipeFactory.CreateClient())
{
try
{
client.Connect(1);
using (var writer = new StreamWriter(client)
{
AutoFlush = true
})
{
writer.WriteLine(string.Join(" ", args));
client.WaitForPipeDrain();
Console.Write(new StreamReader(client).ReadToEnd());
}
}
catch (IOException) { }
catch (System.TimeoutException)
{
Console.WriteLine("Connect logrotate service timeout.");
}
}
break;
}
}
else
{
using (var service = new Program(args.Length == 0 ? DefaultServicName : args[0]))
{
Environment.CurrentDirectory = AppDomain.CurrentDomain.BaseDirectory;
Run(service);
}
}
}
public LogRotaterDaemon()
{
configPipe = PipeFactory.CreateServer();
rotateTimer = new Timer(OnTick);
}