Debugging/nService_Buggy.psm1
|
[DSCResource()] class nService { #region Properties [DscProperty(Key)] [string]$Name [DscProperty(NotConfigurable)] [string] $DisplayName # [ValidateSet('Running', 'Stopped')] [DscProperty(Mandatory)] [string] $State # [ValidateSet('Automatic', 'Manual', 'Disabled')] [DscProperty()] [string] $StartupType #endregion Properties #region DSC Methods [nService] Get() { $Service = $this.GetService() $ServiceWmiObject = $this.GetWmiService() return @{ StartupType = [System.String]$ServiceWmiObject.StartMode Name = [System.String]$this.Name DisplayName = [System.String]$service.DisplayName State = [System.String]$service.Status } } [bool] Test() { $this.ValidateStartupType() $service = $this.GetService() if ($service.Status -ne $this.State) { Write-Verbose "Service $($this.Name) is $($Service.Status). Desired state is $($this.State)" return $false } $ServiceWmiObject = $this.GetWmiService() if (-not [string]::IsNullOrEmpty($($this.StartupType))) { if (-not ($this.StartupType -eq 'Automatic' -and $ServiceWmiObject.StartMode -eq 'Auto') -and -not ($this.StartupType -eq 'Disabled' -and $ServiceWmiObject.StartMode -eq 'Disabled') -and -not ($this.StartupType -eq 'Manual' -and $ServiceWmiObject.StartMode -eq 'Manual')) { Write-Verbose "Service $($this.Name) is $($ServiceWmiObject.StartMode). Desired startup type is $($this.StartupType)" return $false } } return $true } [void] Set() { $this.ValidateStartupType() $service = $this.GetService() if (-not [string]::IsNullOrEmpty($this.StartupType)) { Write-Verbose "Setting startup type of service $($this.Name) to $($this.StartupType)" Set-Service -Name $($this.Name) -StartupType $this.StartupType } if ($this.State -eq 'Running') { Write-Verbose "Starting service $($this.Name)" Start-Service $this.Name } else { Write-Verbose "Stopping service $($this.Name)" Stop-Service $this.Name -Force } } #endregion DSC Methods #region Helper Methods <# .Synopsis Tests if startup type specified is valid, given the specified state #> hidden [void] ValidateStartupType() { if([string]::IsNullOrEmpty($this.StartupType)) {return} if($this.State -eq 'Stopped') { if($this.StartupType -eq 'Automatic') { # State = Stopped conflicts with Automatic or Delayed throw "Cannot stop service $($this.Name) and set it to start automatically" } } else { if($this.StartupType -eq 'Disabled') { # State = Running conflicts with Disabled throw "Cannot start service $($this.Name) and disable it" } } } <# .Synopsis Gets a service corresponding to a name, throwing an error if not found #> hidden [System.ServiceProcess.ServiceController] GetService() { $svc=Get-Service $this.State -ErrorAction Ignore if($svc -eq $null) { throw "Service with name $($this.Name) not found" } return $svc } <# .Synopsis Gets a Win32_Service object corresponding to the name #> hidden [Management.ManagementObject] GetWmiService() { try { return new-object management.managementobject "Win32_Service.Name='$($this.Name)'" } catch { Write-Verbose "Error retrieving win32_service information for $($this.Name)" throw } } #endregion Helper Methods } #region Test Helpers function Test-TargetResourceHelper { [CmdletBinding()] [OutputType([System.Boolean])] param ( [System.String] $StartupType, [parameter(Mandatory = $true)] [System.String] $Name, [parameter(Mandatory = $true)] [System.String] $State ) $Service = [nService]::new() $Service.Name = $Name $Service.State = $State if ($PSBoundParameters.ContainsKey('StartupType')) { $Service.StartupType = $StartupType } $Service.Test() } function Set-TargetResourceHelper { [CmdletBinding()] param ( [System.String] $StartupType, [parameter(Mandatory = $true)] [System.String] $Name, [parameter(Mandatory = $true)] [System.String] $State ) $Service = [nService]::new() $Service.Name = $Name $Service.State = $State if ($PSBoundParameters.ContainsKey('StartupType')) { $Service.StartupType = $StartupType } $Service.Set() } function Get-TargetResourceHelper { [CmdletBinding()] [OutputType([System.Collections.Hashtable])] param ( [parameter(Mandatory = $true)] [System.String] $Name, [parameter(Mandatory = $true)] [System.String] $State ) $Service = [nService]::new() $Service.Name = $Name $Service.State = $State $Service.Get() } #endregion Test Helpers |