PSReminderLite.psm1
|
#region Main # used for culture debugging # write-host "Importing with culture $(Get-Culture)" if ((Get-Culture).Name -match '\w+') { Import-LocalizedData -BindingVariable strings } else { #force using En-US if no culture found, which might happen on non-Windows systems. Import-LocalizedData -BindingVariable strings -FileName PSReminderLite.psd1 -BaseDirectory $PSScriptRoot\en-us } #dot-source module functions Get-ChildItem -Path $PSScriptRoot\functions\*.ps1 | ForEach-Object { . $_.FullName } #endregion #region Define module variables $ExportPath = Join-Path -Path $HOME -ChildPath '.psreminder.json' If (Test-Path -Path $ExportPath) { #use the preference file Get-Content -Path $ExportPath | ConvertFrom-Json | ForEach-Object { if ($_.Value.GetType().Name -ne 'PSCustomObject') { Set-Variable -Name $_.Name -Value $_.Value -Force } elseif ($_.name -eq 'PSReminderTag') { $Hash = @{} $global:t = $_ $_.Value.PSObject.Properties | ForEach-Object { $Hash[$_.Name] = $_.Value } Set-Variable -Name PSReminderTag -Value $Hash -Force } } <# Get-Content -Path $ExportPath | ConvertFrom-Json | ForEach-Object { Set-Variable -Name $_.Name -Value $_.Value -Force } #> } else { #the default number of days to display for Show-TickleEvents $PSReminderDefaultDays = 7 #database defaults $PSReminderDB = Join-Path -Path $HOME -ChildPath PSReminder.db $PSReminderTable = 'EventData' $PSReminderArchiveTable = 'ArchivedEvent' #define a default tag list and style settings $PSReminderTag = @{ 'Work' = "`e[38;5;192m" 'Personal' = "`e[96m" 'Priority' = "`e[1;3;38;5;199m" } } If (-Not (Test-Path -Path $PSReminderDB)) { Write-Warning "The database file $PSReminderDB does not exist or could not be found. Please run Initialize-PSReminderDatabase to create the database." } #endregion #region Class definitions Class PSReminder { [String]$Event [DateTime]$Date [String]$Comment [int32]$ID [string[]]$Tags [boolean]$Expired = $False #add a hidden property that captures the database path hidden [string]$Source #add a hidden property to capture the computer name' hidden [string]$ComputerName = [System.Environment]::MachineName #constructor PSReminder([int32]$ID, [String]$Event, [DateTime]$Date, [String]$Comment, [String[]]$Tags) { $this.ID = $ID $this.Event = $Event $this.Date = $Date $this.Comment = $Comment $this.Tags = $Tags if ($Date -lt (Get-Date)) { $this.Expired = $True } } } #close PSReminder class Class ArchivePSReminder { [String]$Event [DateTime]$Date [String]$Comment [int32]$ID [string[]]$Tags [DateTime]$ArchivedDate #add a hidden property that captures the database path hidden [string]$Source #add a hidden property to capture the computer name' hidden [string]$ComputerName = [System.Environment]::MachineName ArchivePSReminder([int32]$ID, [String]$Event, [DateTime]$Date, [String]$Comment, [String[]]$Tags, [DateTime]$ArchivedDate) { $this.ID = $ID $this.Event = $Event $this.Date = $Date $this.Comment = $Comment $this.ArchivedDate = $ArchivedDate $this.Tags = $Tags } } #close ArchivePSReminder class Class PSReminderDBInfo { [string]$Name [string]$Path [int32]$PageSize [int32]$PageCount [int32]$Reminders [int32]$Expired [int32]$Archived [int32]$Size [DateTime]$Created [DateTime]$Modified [string]$Author [string]$Comment [string]$Encoding [version]$SQLiteVersion [DateTime]$Date = (Get-Date) [string]$Computername = [System.Environment]::MachineName #methods hidden [void]GetDBInfo () { Try { _verbose ($script:strings.GetDBInfo -f $this.Path) $r = Get-MySQLiteDB -Path $this.Path -ErrorAction Stop $this.Name = Split-Path -Path $r.path -Leaf $this.PageSize = $r.PageSize $this.PageCount = $r.PageCount $this.Size = $r.Size $this.Created = $r.Created $this.Modified = $r.Modified $this.Encoding = $r.Encoding $this.SQLiteVersion = $r.SQLiteVersion } Catch { Throw $_ } } [PSReminderDBInfo]RefreshInfo () { _verbose $script:strings.OpenDB $this.GetDBInfo() $conn = Open-MySQLiteDB -Path $this.Path $InvokeParams = @{ Connection = $conn KeepAlive = $True Query = '' } $InvokeParams.Query = 'select Author,comment from metadata' _verbose ($script:strings.InvokeQuery -f $InvokeParams.Query) $meta = Invoke-MySQLiteQuery @InvokeParams $this.Author = $meta.Author $this.Comment = $meta.Comment $now = Get-Date -Format 'yyyy-MM-dd HH:mm:ss' $InvokeParams.Query = "Select count(*) AS Count from $global:PSReminderTable Where EventDate>= '$now'" _verbose ($script:strings.InvokeQuery -f $InvokeParams.Query) $this.Reminders = $(Invoke-MySQLiteQuery @InvokeParams).count _verbose ($script:strings.InvokeQuery -f $InvokeParams.Query) $InvokeParams.Query = "Select count(*) AS Count from $global:PSReminderArchiveTable" $this.Archived = (Invoke-MySQLiteQuery @InvokeParams).Count _verbose ($script:strings.InvokeQuery -f $InvokeParams.Query) $InvokeParams.Query = "Select count(*) AS Count from $global:PSReminderTable Where EventDate< '$((Get-Date).ToString('yyyy-MM-dd HH:mm:ss'))'" $this.Expired = (Invoke-MySQLiteQuery @InvokeParams).count If ($conn.state -eq 'Open') { _verbose $script:strings.CloseDB $conn.Close() } return $this } #Constructor PSReminderDBInfo([string]$DatabasePath) { $this.Path = $DatabasePath $this.RefreshInfo() } } #close PSReminderDBInfo class Class PSReminderPreference { [string]$PSReminderDB = $global:PSReminderDB [string]$PSReminderDefaultDays = $global:PSReminderDefaultDays [string]$PSReminderTable = $global:PSReminderTable [string]$PSReminderArchiveTable = $global:PSReminderArchiveTable [hashtable]$PSReminderTag = $global:PSReminderTag [object]ShowTags () { $r = $this.PSReminderTag.GetEnumerator() | Foreach-Object { $stringValue = $_.Value.Replace("$([char]27)", '`e') [PSCustomObject]@{ PSTypeName = 'PSReminderTag' Tag = $_.Key Style = '{0}{1}{2}' -f $($_.Value), $stringValue, $("`e[0m") } } return $r } } #close PSReminderPreference class #custom type extensions. This might be exported to a ps1xml file in a future release Update-TypeData -TypeName PSReminder -DefaultDisplayPropertySet ID, Date, Event, Comment -Force Update-TypeData -TypeName PSReminder -MemberType AliasProperty -MemberName Name -Value Event -Force Update-TypeData -TypeName PSReminder -MemberType ScriptProperty -MemberName Countdown -Value { $ts = $this.Date - (Get-Date) if ($ts.TotalMinutes -lt 0) { $ts = New-TimeSpan -Minutes 0 } $ts } -Force Update-TypeData -TypeName PSReminderDBInfo -MemberType ScriptProperty -MemberName Age -Value { $ts = New-TimeSpan -Start $this.Modified -End (Get-Date) $ts } -Force #endregion #region auto completers Register-ArgumentCompleter -CommandName Add-PSReminder, Set-PSReminder -ParameterName Tags -ScriptBlock { param($commandName, $parameterName, $wordToComplete, $commandAst, $fakeBoundParameter) (Get-PSReminderTag).Where({ $_.Tag -like "$WordToComplete*" }).ForEach({ [System.Management.Automation.CompletionResult]::new($_.Tag.Trim(), $_.Tag.Trim(), 'ParameterValue', $_.Tag) }) } Register-ArgumentCompleter -CommandName Get-PSReminder -ParameterName Tag -ScriptBlock { param($commandName, $parameterName, $wordToComplete, $commandAst, $fakeBoundParameter) (Get-PSReminderTag).Where({ $_.Tag -like "$WordToComplete*" }).ForEach({ [System.Management.Automation.CompletionResult]::new($_.Tag.Trim(), $_.Tag.Trim(), 'ParameterValue', $_.Tag) }) } #endregion $export = @{ Variable = @('PSReminderDefaultDays', 'PSReminderDB', 'PSReminderTable', 'PSReminderArchiveTable', 'PSReminderTag') Function = @('Export-PSReminderPreference', 'Initialize-PSReminderDatabase', 'Add-PSReminder', 'Get-PSReminder', 'Get-PSReminderDBInformation', 'Set-PSReminder', 'Remove-PSReminder', 'Export-PSReminderDatabase', 'Import-PSReminderDatabase', 'Move-PSReminder', 'Get-AboutPSReminder', 'Get-PSReminderTag', 'Import-FromTickleDatabase','Get-PSReminderPreference') Alias = @('apsr', 'gpsr', 'spsr', 'rpsr', 'Archive-PSReminder', 'gprt','New-PSReminder') } Export-ModuleMember @export |