🌟[Major]: Implement Module Contexts and User Contexts

src/functions/private/Get-ActiveModuleContext.ps1

@@ -0,0 +1,38 @@
+function Get-ActiveModuleContext {
+ <#
+ .SYNOPSIS
+ Gets the name of the active module context for a vault.
+
+ .DESCRIPTION
+ Reads the active module context name from the 'active-context' file in the module directory.
+ If the file doesn't exist or is invalid, returns 'default'.
+
+ .EXAMPLE
+ Get-ActiveModuleContext -VaultPath 'C:\Users\John\.contextvaults\MyVault'
+
+ Returns the name of the active module context, or 'default' if none is set.
+ #>
+ [CmdletBinding()]
+ param(
+ # The path to the vault directory.
+ [Parameter(Mandatory)]
+ [string] $VaultPath
+ )
+
+ $activeContextFile = Join-Path -Path $VaultPath -ChildPath 'module' | Join-Path -ChildPath 'active-context'
+
+ if (Test-Path $activeContextFile) {
+ try {
+ $contextName = (Get-ContentNonLocking -Path $activeContextFile).Trim()
+ if ([string]::IsNullOrWhiteSpace($contextName)) {
+ return 'default'
+ }
+ return $contextName
+ } catch {
+ Write-Warning "Failed to read active context file: $_"
+ return 'default'
+ }
+ }
+
+ return 'default'
+}

src/functions/private/Get-ContextDirectory.ps1

@@ -0,0 +1,35 @@
+function Get-ContextDirectory {
+ <#
+ .SYNOPSIS
+ Gets the appropriate directory path for a context based on its type.
+
+ .DESCRIPTION
+ Returns the directory path where contexts of the specified type should be stored.
+ For User contexts, returns the 'user' subdirectory.
+ For Module contexts, returns the 'module' subdirectory.
+
+ .EXAMPLE
+ Get-ContextDirectory -VaultPath 'C:\Users\John\.contextvaults\MyVault' -Type 'User'
+
+ Returns: C:\Users\John\.contextvaults\MyVault\user
+
+ .EXAMPLE
+ Get-ContextDirectory -VaultPath 'C:\Users\John\.contextvaults\MyVault' -Type 'Module'
+
+ Returns: C:\Users\John\.contextvaults\MyVault\module
+ #>
+ [CmdletBinding()]
+ param(
+ # The path to the vault directory.
+ [Parameter(Mandatory)]
+ [string] $VaultPath,
+
+ # The type of context: 'User' or 'Module'.
+ [Parameter(Mandatory)]
+ [ValidateSet('User', 'Module')]
+ [string] $Type
+ )
+
+ $subdirectory = if ($Type -eq 'Module') { 'module' } else { 'user' }
+ return Join-Path -Path $VaultPath -ChildPath $subdirectory
+}

src/functions/private/Set-ActiveModuleContext.ps1

@@ -0,0 +1,39 @@
+function Set-ActiveModuleContext {
+ <#
+ .SYNOPSIS
+ Sets the active module context for a vault.
+
+ .DESCRIPTION
+ Writes the active module context name to the 'active-context' file in the module directory.
+
+ .EXAMPLE
+ Set-ActiveModuleContext -VaultPath 'C:\Users\John\.contextvaults\MyVault' -ContextName 'staging'
+
+ Sets 'staging' as the active module context.
+ #>
+ [CmdletBinding()]
+ param(
+ # The path to the vault directory.
+ [Parameter(Mandatory)]
+ [string] $VaultPath,
+
+ # The name of the context to set as active.
+ [Parameter(Mandatory)]
+ [string] $ContextName
+ )
+
+ $moduleDir = Join-Path -Path $VaultPath -ChildPath 'module'
+ $activeContextFile = Join-Path -Path $moduleDir -ChildPath 'active-context'
+
+ # Ensure module directory exists
+ if (-not (Test-Path $moduleDir)) {
+ $null = New-Item -Path $moduleDir -ItemType Directory -Force
+ }
+
+ try {
+ Set-Content -Path $activeContextFile -Value $ContextName -NoNewline
+ Write-Verbose "Set active module context to: $ContextName"
+ } catch {
+ throw "Failed to set active context: $_"
+ }
+}

src/functions/public/Get-ActiveModuleContextName.ps1

@@ -0,0 +1,48 @@
+function Get-ActiveModuleContextName {
+ <#
+ .SYNOPSIS
+ Gets the name of the currently active module context for a vault.
+
+ .DESCRIPTION
+ Returns the name of the currently active module context for the specified vault.
+ If no active context is set or the file is missing, returns 'default'.
+
+ .EXAMPLE
+ Get-ActiveModuleContextName -Vault 'MyModule'
+
+ Returns the name of the active module context for 'MyModule' vault.
+
+ .OUTPUTS
+ [System.String]
+
+ .NOTES
+ This function reads the active context name from the vault's metadata,
+ falling back to 'default' if not set.
+
+ .LINK
+ https://psmodule.io/Context/Functions/Get-ActiveModuleContextName/
+ #>
+ [OutputType([string])]
+ [CmdletBinding()]
+ param(
+ # The name of the vault to check.
+ [Parameter(Mandatory)]
+ [string] $Vault
+ )
+
+ begin {
+ $stackPath = Get-PSCallStackPath
+ Write-Debug "[$stackPath] - Begin"
+ }
+
+ process {
+ $vaultObject = Get-ContextVault -Name $Vault
+ $activeContextName = Get-ActiveModuleContext -VaultPath $vaultObject.Path
+ Write-Verbose "[$stackPath] - Active module context for vault '$Vault': '$activeContextName'"
+ return $activeContextName
+ }
+
+ end {
+ Write-Debug "[$stackPath] - End"
+ }
+}

src/functions/public/Get-Context.ps1

@@ -77,12 +77,26 @@ function Get-Context {
 
  Retrieves all contexts that start with 'My' from the context vault (directly from disk).
 
+ .EXAMPLE
+ Get-Context -Vault 'MyModule' -ID 'default' -Type 'Module'
+
+ Retrieves the default module context from the 'MyModule' vault.
+
+ .EXAMPLE
+ Get-Context -Vault 'MyModule' -Type 'Module'
+
+ Retrieves all module contexts from the 'MyModule' vault.
+
  .OUTPUTS
  [System.Object]
 
  .NOTES
  Returns a list of contexts matching the specified ID or all contexts if no ID is specified.
  Each context object contains its ID and corresponding stored properties.
+
+ The -Type parameter determines whether to retrieve User contexts (default) or Module contexts.
+ Module contexts are stored in the 'module' subdirectory, while User contexts are stored 
+ in the 'user' subdirectory of each vault.
 
  .LINK
  https://psmodule.io/Context/Functions/Get-Context/
@@ -101,7 +115,12 @@ function Get-Context {
  # The name of the vault to store the context in.
  [Parameter()]
  [SupportsWildcards()]
- [string[]] $Vault = '*'
+ [string[]] $Vault = '*',
+
+ # The type of context to retrieve: 'User' or 'Module'.
+ [Parameter()]
+ [ValidateSet('User', 'Module')]
+ [string] $Type = 'User'
  )
 
  begin {
@@ -110,9 +129,25 @@ function Get-Context {
  }
 
  process {
- $contextInfos = Get-ContextInfo -ID $ID -Vault $Vault -ErrorAction Stop
+ # Special handling for Module contexts: if no specific ID is given, get the active context
+ if ($Type -eq 'Module' -and $ID.Count -eq 1 -and $ID[0] -eq '*') {
+ foreach ($vaultName in $Vault) {
+ $vaultObject = Get-ContextVault -Name $vaultName -ErrorAction SilentlyContinue
+ if ($vaultObject) {
+ $activeContextName = Get-ActiveModuleContext -VaultPath $vaultObject.Path
+ Write-Verbose "[$stackPath] - Getting active module context '$activeContextName' for vault '$vaultName'"
+ $activeContext = Get-Context -ID $activeContextName -Vault $vaultName -Type 'Module'
+ if ($activeContext) {
+ $activeContext
+ }
+ }
+ }
+ return
+ }
+
+ $contextInfos = Get-ContextInfo -ID $ID -Vault $Vault -Type $Type -ErrorAction Stop
  foreach ($contextInfo in $contextInfos) {
- Write-Verbose "Retrieving context - ID: [$($contextInfo.ID)], Vault: [$($contextInfo.Vault)]"
+ Write-Verbose "Retrieving context - ID: [$($contextInfo.ID)], Vault: [$($contextInfo.Vault)], Type: [$Type]"
  try {
  if (-not (Test-Path -Path $contextInfo.Path)) {
  Write-Warning "Context file does not exist: $($contextInfo.Path)"

src/functions/public/Get-ContextInfo.ps1

@@ -70,7 +70,12 @@
 
  # The name of the vault to retrieve context info from. Supports wildcards.
  [Parameter()]
- [string[]] $Vault = '*'
+ [string[]] $Vault = '*',
+
+ # The type of context to retrieve: 'User' or 'Module'.
+ [Parameter()]
+ [ValidateSet('User', 'Module')]
+ [string] $Type = 'User'
  )
 
  begin {
@@ -85,9 +90,12 @@
  Write-Verbose "[$stackPath] - Found $($vaults.Count) vault(s) matching '$($Vault -join ', ')'."
 
  $files = foreach ($vaultObject in $vaults) {
- Get-ChildItem -Path $vaultObject.Path -Filter *.json -File
+ $contextDir = Get-ContextDirectory -VaultPath $vaultObject.Path -Type $Type
+ if (Test-Path $contextDir) {
+ Get-ChildItem -Path $contextDir -Filter *.json -File
+ }
  }
- Write-Verbose "[$stackPath] - Found $($files.Count) context file(s) in vault(s)."
+ Write-Verbose "[$stackPath] - Found $($files.Count) context file(s) in vault(s) for type '$Type'."
 
  foreach ($file in $files) {
  # Use non-locking file reading to allow concurrent access

src/functions/public/Remove-Context.ps1

@@ -87,7 +87,12 @@
 
  # The name of the vault to remove contexts from.
  [Parameter()]
- [string] $Vault
+ [string] $Vault,
+
+ # The type of context to remove: 'User' or 'Module'.
+ [Parameter()]
+ [ValidateSet('User', 'Module')]
+ [string] $Type = 'User'
  )
 
  begin {
@@ -96,14 +101,34 @@
  }
 
  process {
- $contextInfo = Get-ContextInfo -ID $ID -Vault $Vault
+ $contextInfo = Get-ContextInfo -ID $ID -Vault $Vault -Type $Type
  foreach ($contextInfo in $contextInfo) {
  $contextId = $contextInfo.ID
 
  if ($PSCmdlet.ShouldProcess("Context '$contextId'", 'Remove')) {
- Write-Verbose "[$stackPath] - Removing context [$contextId]"
+ Write-Verbose "[$stackPath] - Removing context [$contextId] of type [$Type]"
+
+ # Special handling for module contexts - don't allow removing 'default' if it's the only one
+ if ($Type -eq 'Module' -and $contextId -eq 'default') {
+ $allModuleContexts = Get-ContextInfo -ID '*' -Vault $Vault -Type 'Module'
+ if ($allModuleContexts.Count -le 1) {
+ Write-Warning "Cannot remove the default module context when it's the only module context."
+ continue
+ }
+ }
+
  $contextInfo.Path | Remove-Item -Force -ErrorAction Stop
  Write-Verbose "[$stackPath] - Removed item: $contextId"
+
+ # If we removed the active module context, reset to 'default'
+ if ($Type -eq 'Module') {
+ $vaultObject = Get-ContextVault -Name $Vault
+ $activeContext = Get-ActiveModuleContext -VaultPath $vaultObject.Path
+ if ($activeContext -eq $contextId -and $contextId -ne 'default') {
+ Write-Verbose "[$stackPath] - Resetting active module context to 'default'"
+ Set-ActiveModuleContext -VaultPath $vaultObject.Path -ContextName 'default'
+ }
+ }
  }
  }
  }

src/functions/public/Rename-Context.ps1

@@ -61,6 +61,11 @@
  [Parameter()]
  [string] $Vault,
 
+ # The type of context to rename: 'User' or 'Module'.
+ [Parameter()]
+ [ValidateSet('User', 'Module')]
+ [string] $Type = 'User',
+
  # Pass the context through the pipeline.
  [Parameter()]
  [switch] $PassThru
@@ -72,19 +77,31 @@
  }
 
  process {
- $context = Get-Context -ID $ID -Vault $Vault
+ $context = Get-Context -ID $ID -Vault $Vault -Type $Type
  if (-not $context) {
- throw "Context with ID '$ID' not found in vault '$Vault'"
+ throw "Context with ID '$ID' not found in vault '$Vault' for type '$Type'"
  }
 
- $existingContext = Get-Context -ID $NewID -Vault $Vault
+ $existingContext = Get-Context -ID $NewID -Vault $Vault -Type $Type
  if ($existingContext -and -not $Force) {
- throw "Context with ID '$NewID' already exists in vault '$Vault'"
+ throw "Context with ID '$NewID' already exists in vault '$Vault' for type '$Type'"
  }
 
- if ($PSCmdlet.ShouldProcess("Renaming context '$ID' to '$NewID' in vault '$Vault'")) {
- $context | Set-Context -ID $NewID -Vault $Vault -PassThru:$PassThru
- Remove-Context -ID $ID -Vault $Vault
+ if ($PSCmdlet.ShouldProcess("Renaming context '$ID' to '$NewID' in vault '$Vault' for type '$Type'")) {
+ # Special handling for module contexts and active context tracking
+ if ($Type -eq 'Module') {
+ $vaultObject = Get-ContextVault -Name $Vault
+ $activeContext = Get-ActiveModuleContext -VaultPath $vaultObject.Path
+
+ # If we're renaming the active context, update the active context reference
+ if ($activeContext -eq $ID) {
+ Write-Verbose "[$stackPath] - Updating active module context from '$ID' to '$NewID'"
+ Set-ActiveModuleContext -VaultPath $vaultObject.Path -ContextName $NewID
+ }
+ }
+
+ $context | Set-Context -ID $NewID -Vault $Vault -Type $Type -PassThru:$PassThru
+ Remove-Context -ID $ID -Vault $Vault -Type $Type
  }
  }