Out-Daemon.ps1

function Out-Daemon
{
    <#
    .Synopsis
        Creates small services that run PowerShell
    .Description
        Creates small services that run PowerShell on a regular basis
    .Example
        Out-Daemon -ScriptBlock {    
            Write-Warning "Warning"
            Write-Error "Error"
            $VerbosePreference = 'continue'
            Write-Verbose "Verbose"
            $debugPreference = 'continue'
            Write-Debug "Debug"
            Write-Progress "a" "b" -PercentComplete 1
            1
            New-Object PSObject -Property @{"A" = "b";"c" = "d" }
        } -Interval "00:00:15" -Name streamtest
    .Link
        Watch-Daemon
    .Link
        Remove-Daemon

    #>
    [OutputType([Nullable])]
    param(
    # The short name of the service.  
    [Parameter(Mandatory=$true,ValueFromPipelineByPropertyName=$true)]    
    [string]
    $Name,

    # The display name of the service.  If no displayname is provided, it will be set to the name
    [string]
    $DisplayName,

    # A list of scripts the deamon will run.
    [Parameter(Mandatory=$true)]
    [ScriptBlock[]]
    $ScriptBlock,

    # The interval the scripts will use to run.  If no interval is provided, the scripts will run constantly.
    [Timespan[]]
    $Interval,

    # Where the service will be stored. If not provided, the service will be stored in $env:AppData\PowerShellDaemons\
    [string]
    $ServicePath
    )

    process {
        $servicePath = if (-not $PSBoundParameters.ServicePath) {
            $psNodeRoot = Join-Path $env:APPDATA "PowerShellDaemons"
            if (-not (Test-Path $psNodeRoot)) {
                $null = New-Item -ItemType Directory -Path $psNodeRoot -Force
            }
            "$(Join-Path $psNodeRoot "$Name.exe")"                    
        } else {
            $servicePath
        }

        if (-not $DisplayName) {
            $DisplayName = $Name
        }


        $elapsedCode = ""
        $psDeclare = ""
        $psInit = ""
        $psRun = ""
        $psStop = ""
        $psStart = ""
        $timerDeclare = ""

        $dataHandlers = ""

        $timerInit = @"
        bool constant  = $(if (-not $psboundParameters.Interval) { "true;"} else {"false;"}) 
        $(
        $inc = 0
        for ($inc = 0; $inc -lt $scriptBLock.Count; $inc++) {
            if ($Interval -and $Interval[$inc]){
                $lastInterval = $interval[$inc]
            
            } 
            if ($lastInterval) {
            "System.Timers.Timer Timer$inc = new System.Timers.Timer();
        Timer$inc.Interval = $([uint32]$LastInterval.TotalMilliseconds);
        Timer$inc.Elapsed += new ElapsedEventHandler(Time${inc}Elapsed);
        Timer$inc.Start();
        "
                
            $elapsedCode += "
    void Time${inc}Elapsed(object sender, ElapsedEventArgs e) {
        RunScript${inc}();
    }
"
                
            }
            $psDeclare += 
            
        "PowerShell powerShellScript$inc;
        PSDataCollection<PSObject> powerShellScriptOutput$inc = new PSDataCollection<PSObject>();

    "
            
            $psInit += 
        "
        powerShellScript$inc = PowerShell.Create();
        powerShellScript$inc.Streams.Error.DataAdded += new EventHandler<DataAddedEventArgs>(Error${Inc}_DataAdded);
        powerShellScript$inc.Streams.Verbose.DataAdded += new EventHandler<DataAddedEventArgs>(Verbose${Inc}_DataAdded);
        powerShellScript$inc.Streams.Debug.DataAdded += new EventHandler<DataAddedEventArgs>(Debug${inc}_DataAdded);
        powerShellScript$inc.Streams.Warning.DataAdded += new EventHandler<DataAddedEventArgs>(Warning${inc}_DataAdded);
        powerShellScript$inc.Streams.Progress.DataAdded += new EventHandler<DataAddedEventArgs>(Progress${inc}_DataAdded);
        powerShellScriptOutput$inc.DataAdded += new EventHandler<DataAddedEventArgs>(Output${inc}_DataAdded);
        "
            $ps64 = [Convert]::ToBase64String([Text.Encoding]::Unicode.getbytes($ScriptBlock[$inc]))
            $psRun +=         
        "
    void RunScript${inc}() {
        if (powerShellScript${inc}.InvocationStateInfo.State != PSInvocationState.Running) {
            byte[] byteArr = Convert.FromBase64String(`"$ps64`");
            string powerShellScript = Encoding.Unicode.GetString(byteArr);
            powerShellScript$inc.Commands.Clear();
            powerShellScript$inc.AddScript(powerShellScript, false);
            powerShellScript$inc.BeginInvoke<Object, PSObject>(null, powerShellScriptOutput$inc);
        } 
    }

"
            $psStop += "
            powerShellScript$inc.BeginStop(StopDone, null);
"
            $psStart += "
            RunScript${inc}();            
"
        
            $dataHandlers +=  @"
    void Debug${Inc}_DataAdded(object sender, DataAddedEventArgs e) {
        System.Diagnostics.Process process = System.Diagnostics.Process.GetCurrentProcess();

        PSDataCollection<DebugRecord> collection = sender as PSDataCollection<DebugRecord>;
        DebugRecord err = collection[e.Index];

        
        string errorText = "@{OutputType='Debug';Message='" + err.ToString().Replace("'", "''") + "';Timestamp='" + DateTime.Now.ToString("o") + "'}" + Environment.NewLine;
        string outputFile = process.MainModule.FileName.Replace(".exe", "") + ".$inc.out";
        System.IO.File.AppendAllText(outputFile,errorText);
   }

    void Warning${Inc}_DataAdded(object sender, DataAddedEventArgs e) {
        System.Diagnostics.Process process = System.Diagnostics.Process.GetCurrentProcess();

        PSDataCollection<WarningRecord> collection = sender as PSDataCollection<WarningRecord>;
        WarningRecord err = collection[e.Index];

        
        string errorText = "@{OutputType='Warning';Message='" + err.ToString().Replace("'", "''") + "';Timestamp='" + DateTime.Now.ToString("o") + "'}" + Environment.NewLine;
        string outputFile = process.MainModule.FileName.Replace(".exe", "") + ".$inc.out";
        System.IO.File.AppendAllText(outputFile,errorText);
   }

   void Verbose${Inc}_DataAdded(object sender, DataAddedEventArgs e) {
        System.Diagnostics.Process process = System.Diagnostics.Process.GetCurrentProcess();

        PSDataCollection<VerboseRecord> collection = sender as PSDataCollection<VerboseRecord>;
        VerboseRecord err = collection[e.Index];

        
        string errorText = "@{OutputType='Verbose';Message='" + err.ToString().Replace("'", "''") + "';Timestamp='" + DateTime.Now.ToString("o") + "'}" + Environment.NewLine;
        string outputFile = process.MainModule.FileName.Replace(".exe", "") + ".$inc.out";
        System.IO.File.AppendAllText(outputFile,errorText);
   }

    void Error${Inc}_DataAdded(object sender, DataAddedEventArgs e) {
        System.Diagnostics.Process process = System.Diagnostics.Process.GetCurrentProcess();

        PSDataCollection<ErrorRecord> collection = sender as PSDataCollection<ErrorRecord>;
        ErrorRecord err = collection[e.Index];

        
        string errorText = "@{OutputType='Error';Message='" + err.Exception.Message.Replace("'", "''").ToString() + "';Timestamp='" + DateTime.Now.ToString("o")+"'}" + Environment.NewLine;
        string outputFile = process.MainModule.FileName.Replace(".exe", "") + ".$inc.out";
        System.IO.File.AppendAllText(outputFile,errorText);
   }
   
   void Progress${Inc}_DataAdded(object sender, DataAddedEventArgs e) {
        System.Diagnostics.Process process = System.Diagnostics.Process.GetCurrentProcess();

        PSDataCollection<ProgressRecord> collection = sender as PSDataCollection<ProgressRecord>;
        ProgressRecord err = collection[e.Index];
       
        
        string errorText = "@{OutputType='Progress';StatusDescription='" + 
            err.StatusDescription.Replace("'", "''") + 
            "';Activity='" + err.Activity.Replace("'", "''") +
            "';ActivityId='" + err.ActivityId.ToString() +
            "';ParentActivityId='" + err.ParentActivityId.ToString() +
            "';RecordType='" + err.RecordType.ToString() +
            "';PercentComplete='" + err.PercentComplete.ToString() +
            "';Timestamp='" + DateTime.Now.ToString("o")+"'}" + Environment.NewLine;
        string outputFile = process.MainModule.FileName.Replace(".exe", "") + ".$inc.out";
        System.IO.File.AppendAllText(outputFile,errorText);
   }


   void Output${Inc}_DataAdded(object sender, DataAddedEventArgs e) {
        System.Diagnostics.Process process = System.Diagnostics.Process.GetCurrentProcess();

        PSDataCollection<PSObject> collection = sender as PSDataCollection<PSObject>;
        PSObject err = collection[e.Index];

        string objStr = (string)LanguagePrimitives.ConvertTo(err, typeof(string));

        if (objStr.StartsWith("@{")) {
            objStr = objStr.Replace(";", "';").Replace("=", "='").TrimEnd('}') + "'}";
        }

        string errorText = "@{OutputType='Output';Output='" +
            objStr.Replace("'", "''") +
            "';Timestamp='" + DateTime.Now.ToString("o")+"'}" + Environment.NewLine;
        string outputFile = process.MainModule.FileName.Replace(".exe", "") + ".$inc.out";
        System.IO.File.AppendAllText(outputFile,errorText);
   }
"@
        }
        )

"@
    $serviceCode = @"
using System;
using System.Timers;
using System.ServiceProcess;
using System.Threading;
using System.ComponentModel;
using System.Collections.Generic;
using System.Configuration.Install;
using System.Management.Automation;
using System.Text;
 
public class PSNodeService : ServiceBase
{
   
   public PSNodeService()
   {
      this.ServiceName = "$Name";
      this.CanStop = true;
      this.CanPauseAndContinue = false;
      this.AutoLog = true;
   }

      
    $psDeclare

   protected override void OnStart(string [] args)
   {
        $timerInit
        $psInit

        if (constant) {
            while (true) {
                $psStart
            }
        } else {
            $psStart
        }
        
        
   }

   $elapsedCode
   $psRun   
 
   protected override void OnStop()
   {
      $psStop
   }

   public void StopDone(IAsyncResult result) {
   
   }
 
   public static void Main()
   {
      System.ServiceProcess.ServiceBase.Run(new PSNodeService());
   }

   $dataHandlers
}
"@


        $s = $serviceCode

        $svcexists = Get-Service -Name $name -ErrorAction SilentlyContinue
        $useCred = if ($Credential) {
            @{Credential = $Credential}
        } else {
            @{}
        }
        if ($svcexists) {
            Stop-Service -Name $Name
            $svcExists = Get-WmiObject -Class Win32_Service -Filter "Name = '$Name'"
            if ($svcExists) {                
                $null = $svcExists.Delete()
                    
            }
            Add-Type -TypeDefinition $serviceCode -ReferencedAssemblies System.Management.Automation, System.ServiceProcess, System.Configuration.Install -OutputAssembly $servicePath -OutputType WindowsApplication -IgnoreWarnings
            
            $null = New-Service -Name $Name -DisplayName $DisplayName -BinaryPathName $ServicePath -StartupType Automatic @useCred
            Start-Service -Name $Name
        } else {
            Add-Type -TypeDefinition $serviceCode -ReferencedAssemblies System.Management.Automation, System.ServiceProcess, System.Configuration.Install -OutputAssembly $servicePath -OutputType WindowsApplication -IgnoreWarnings
            $null = New-Service -Name $Name -DisplayName $DisplayName -BinaryPathName $ServicePath -StartupType Automatic @useCred
            Start-Service -Name $Name
        }
    }
}