From 2fe4baec0a3889589bc48ec1ee9adf0ca1e724ac Mon Sep 17 00:00:00 2001
From: KnifMelti <gaj@knifmelti.se>
Date: Thu, 12 Jan 2023 21:16:03 +0100
Subject: [PATCH] LogRotation (max 3, max 1MB)

---
 .../functions/Invoke-LogRotation.ps1          | 79 +++++++++++++++++++
 Winget-AutoUpdate/functions/Start-Init.ps1    | 11 +++
 2 files changed, 90 insertions(+)
 create mode 100644 Winget-AutoUpdate/functions/Invoke-LogRotation.ps1

diff --git a/Winget-AutoUpdate/functions/Invoke-LogRotation.ps1 b/Winget-AutoUpdate/functions/Invoke-LogRotation.ps1
new file mode 100644
index 0000000..ca33fdf
--- /dev/null
+++ b/Winget-AutoUpdate/functions/Invoke-LogRotation.ps1
@@ -0,0 +1,79 @@
+#Function rotate the logs
+
+function Invoke-LogRotation ($LogFile, $MaxLogFiles, $MaxLogSize) {
+    <#
+        .SYNOPSIS
+            Handle log rotation.
+        .DESCRIPTION
+            Invoke-LogRotation handles log rotation
+        .NOTES
+            Author: Øyvind Kallstad (Minimized and changed for WAU 12.01.2023 by Göran Axel Johannesson)
+            URL: https://www.powershellgallery.com/packages/Communary.Logger/1.1
+            Date: 21.11.2014
+            Version: 1.0
+    #>
+
+    try {
+        # get current size of log file
+        $currentSize = (Get-Item $LogFile).Length
+
+        # get log name
+        $logFileName = Split-Path $LogFile -Leaf
+        $logFilePath = Split-Path $LogFile
+        $logFileNameWithoutExtension = [System.IO.Path]::GetFileNameWithoutExtension($logFileName)
+        $logFileNameExtension = [System.IO.Path]::GetExtension($logFileName)
+
+        # if MaxLogFiles is 1 just keep the original one and let it grow
+        if (-not($MaxLogFiles -eq 1)) {
+            if ($currentSize -ge $MaxLogSize) {
+
+                # construct name of archived log file
+                $newLogFileName = $logFileNameWithoutExtension + (Get-Date -Format 'yyyyMMddHHmmss').ToString() + $logFileNameExtension
+
+                # copy old log file to new using the archived name constructed above
+                Copy-Item -Path $LogFile -Destination (Join-Path (Split-Path $LogFile) $newLogFileName)
+
+                # Create a new log file
+                try {
+                    Remove-Item -Path $LogFile -Force
+                    New-Item -ItemType File -Path $LogFile -Force
+                    #Set ACL for users on logfile
+                    $NewAcl = Get-Acl -Path $LogFile
+                    $identity = New-Object System.Security.Principal.SecurityIdentifier S-1-5-11
+                    $fileSystemRights = "Modify"
+                    $type = "Allow"
+                    $fileSystemAccessRuleArgumentList = $identity, $fileSystemRights, $type
+                    $fileSystemAccessRule = New-Object -TypeName System.Security.AccessControl.FileSystemAccessRule -ArgumentList $fileSystemAccessRuleArgumentList
+                    $NewAcl.SetAccessRule($fileSystemAccessRule)
+                    Set-Acl -Path $LogFile -AclObject $NewAcl
+                }
+                catch {
+                    Write-Log $_.Exception.Message
+                }
+
+                # if MaxLogFiles is 0 don't delete any old archived log files
+                if (-not($MaxLogFiles -eq 0)) {
+
+                    # set filter to search for archived log files
+                    $archivedLogFileFilter = $logFileNameWithoutExtension + '??????????????' + $logFileNameExtension
+
+                    # get archived log files
+                    $oldLogFiles = Get-Item -Path "$(Join-Path -Path $logFilePath -ChildPath $archivedLogFileFilter)"
+
+                    if ([bool]$oldLogFiles) {
+                        # compare found log files to MaxLogFiles parameter of the log object, and delete oldest until we are
+                        # back to the correct number
+                        if (($oldLogFiles.Count + 1) -gt $MaxLogFiles) {
+                            [int]$numTooMany = (($oldLogFiles.Count) + 1) - $MaxLogFiles
+                            $oldLogFiles | Sort-Object 'LastWriteTime' | Select-Object -First $numTooMany | Remove-Item
+                        }
+                    }
+                }
+            }
+        }
+    }
+    catch {
+        Write-Log "`n`n##################################################`n#     CHECK FOR LOG ROTATION - $(Get-Date -Format (Get-culture).DateTimeFormat.ShortDatePattern)`n##################################################"
+        Write-Log $_.Exception.Message
+    }
+}
diff --git a/Winget-AutoUpdate/functions/Start-Init.ps1 b/Winget-AutoUpdate/functions/Start-Init.ps1
index 280ecbb..5de8bfc 100644
--- a/Winget-AutoUpdate/functions/Start-Init.ps1
+++ b/Winget-AutoUpdate/functions/Start-Init.ps1
@@ -5,6 +5,12 @@ function Start-Init {
     #Config console output encoding
     [Console]::OutputEncoding = [System.Text.Encoding]::UTF8
 
+    # Maximum number of log files to keep. Default is 3. Setting MaxLogFiles to 0 will keep all log files.
+    [int32] $MaxLogFiles = 3
+    
+    # Maximum size of log file.
+    [int64] $MaxLogSize = 1048576 # in bytes, default is 1048576 = 1 MB
+
     #Log Header
     $Log = "`n##################################################`n#     CHECK FOR APP UPDATES - $(Get-Date -Format (Get-culture).DateTimeFormat.ShortDatePattern)`n##################################################"
     $Log | Write-host
@@ -27,6 +33,11 @@ function Start-Init {
         Set-Acl -Path $LogFile -AclObject $NewAcl
     }
 
+    #LogRotation if System
+    if ($IsSystem) {
+        Invoke-LogRotation $LogFile $MaxLogFiles $MaxLogSize
+    }
+
     #Log file
     $Log | out-file -filepath $LogFile -Append