The move!

Old/Ombi.Common/ServiceProvider.cs

@@ -0,0 +1,203 @@
+#region Copyright
+// /************************************************************************
+//    Copyright (c) 2017 Jamie Rees
+//    File: ServiceProvider.cs
+//    Created By: Jamie Rees
+//   
+//    Permission is hereby granted, free of charge, to any person obtaining
+//    a copy of this software and associated documentation files (the
+//    "Software"), to deal in the Software without restriction, including
+//    without limitation the rights to use, copy, modify, merge, publish,
+//    distribute, sublicense, and/or sell copies of the Software, and to
+//    permit persons to whom the Software is furnished to do so, subject to
+//    the following conditions:
+//   
+//    The above copyright notice and this permission notice shall be
+//    included in all copies or substantial portions of the Software.
+//   
+//    THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+//    EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+//    MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+//    NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+//    LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+//    OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+//    WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+//  ************************************************************************/
+#endregion
+
+using System;
+using System.Collections.Specialized;
+using System.Configuration.Install;
+using System.Diagnostics;
+using System.Linq;
+using System.ServiceProcess;
+using NLog;
+using Ombi.Common.Processes;
+
+namespace Ombi.Common
+{
+    public interface IServiceProvider
+    {
+        bool ServiceExist(string name);
+        bool IsServiceRunning(string name);
+        void Install(string serviceName);
+        void Run(ServiceBase service);
+        ServiceController GetService(string serviceName);
+        void Stop(string serviceName);
+        void Start(string serviceName);
+        ServiceControllerStatus GetStatus(string serviceName);
+        void Restart(string serviceName);
+    }
+
+    public class ServiceProvider : IServiceProvider
+    {
+        public const string OmbiServiceName = "Ombi";
+
+        private readonly IProcessProvider _processProvider;
+
+        private static readonly Logger _logger = LogManager.GetCurrentClassLogger();
+
+
+        public ServiceProvider(IProcessProvider processProvider)
+        {
+            _processProvider = processProvider;
+        }
+
+        public virtual bool ServiceExist(string name)
+        {
+            _logger.Debug("Checking if service {0} exists.", name);
+            return
+                ServiceController.GetServices().Any(
+                    s => string.Equals(s.ServiceName, name, StringComparison.InvariantCultureIgnoreCase));
+        }
+
+        public virtual bool IsServiceRunning(string name)
+        {
+            _logger.Debug("Checking if '{0}' service is running", name);
+
+            var service = ServiceController.GetServices()
+                .SingleOrDefault(s => string.Equals(s.ServiceName, name, StringComparison.InvariantCultureIgnoreCase));
+
+            return service != null && (
+                service.Status != ServiceControllerStatus.Stopped ||
+                service.Status == ServiceControllerStatus.StopPending ||
+                service.Status == ServiceControllerStatus.Paused ||
+                service.Status == ServiceControllerStatus.PausePending);
+        }
+
+        public virtual void Install(string serviceName)
+        {
+            _logger.Info("Installing service '{0}'", serviceName);
+
+
+            var installer = new ServiceProcessInstaller
+            {
+                Account = ServiceAccount.LocalSystem
+            };
+
+            var serviceInstaller = new ServiceInstaller();
+
+
+            string[] cmdline = { @"/assemblypath=" + Process.GetCurrentProcess().MainModule.FileName };
+
+            var context = new InstallContext("service_install.log", cmdline);
+            serviceInstaller.Context = context;
+            serviceInstaller.DisplayName = serviceName;
+            serviceInstaller.ServiceName = serviceName;
+            serviceInstaller.Description = "Ombi Application Server";
+            serviceInstaller.StartType = ServiceStartMode.Automatic;
+            serviceInstaller.ServicesDependedOn = new[] { "EventLog", "Tcpip", "http" };
+
+            serviceInstaller.Parent = installer;
+
+            serviceInstaller.Install(new ListDictionary());
+
+            _logger.Info("Service Has installed successfully.");
+        }
+
+        public virtual void Run(ServiceBase service)
+        {
+            ServiceBase.Run(service);
+        }
+
+        public virtual ServiceController GetService(string serviceName)
+        {
+            return ServiceController.GetServices().FirstOrDefault(c => string.Equals(c.ServiceName, serviceName, StringComparison.InvariantCultureIgnoreCase));
+        }
+
+        public virtual void Stop(string serviceName)
+        {
+            _logger.Info("Stopping {0} Service...", serviceName);
+            var service = GetService(serviceName);
+            if (service == null)
+            {
+                _logger.Warn("Unable to stop {0}. no service with that name exists.", serviceName);
+                return;
+            }
+
+            _logger.Info("Service is currently {0}", service.Status);
+
+            if (service.Status != ServiceControllerStatus.Stopped)
+            {
+                service.Stop();
+                service.WaitForStatus(ServiceControllerStatus.Stopped, TimeSpan.FromSeconds(60));
+
+                service.Refresh();
+                if (service.Status == ServiceControllerStatus.Stopped)
+                {
+                    _logger.Info("{0} has stopped successfully.", serviceName);
+                }
+                else
+                {
+                    _logger.Error("Service stop request has timed out. {0}", service.Status);
+                }
+            }
+            else
+            {
+                _logger.Warn("Service {0} is already in stopped state.", service.ServiceName);
+            }
+        }
+
+        public ServiceControllerStatus GetStatus(string serviceName)
+        {
+            return GetService(serviceName).Status;
+        }
+
+        public void Start(string serviceName)
+        {
+            _logger.Info("Starting {0} Service...", serviceName);
+            var service = GetService(serviceName);
+            if (service == null)
+            {
+                _logger.Warn("Unable to start '{0}' no service with that name exists.", serviceName);
+                return;
+            }
+
+            if (service.Status != ServiceControllerStatus.Paused && service.Status != ServiceControllerStatus.Stopped)
+            {
+                _logger.Warn("Service is in a state that can't be started. Current status: {0}", service.Status);
+            }
+
+            service.Start();
+
+            service.WaitForStatus(ServiceControllerStatus.Running, TimeSpan.FromSeconds(60));
+            service.Refresh();
+
+            if (service.Status == ServiceControllerStatus.Running)
+            {
+                _logger.Info("{0} has started successfully.", serviceName);
+            }
+            else
+            {
+                _logger.Error("Service start request has timed out. {0}", service.Status);
+            }
+        }
+
+        public void Restart(string serviceName)
+        {
+            var args = string.Format("/C net.exe stop \"{0}\" && net.exe start \"{0}\"", serviceName);
+
+            _processProvider.Start("cmd.exe", args);
+        }
+    }
+}