cliHelper.env

0.1.2

Public/Add-Env.ps1

function Add-Env {
  # .DESCRIPTION
  # Saves an environment variable.
  # .EXAMPLE
  # Read-Env | Set-Env
  # .EXAMPLE
  # Add-Env -Name 'path' -scope 'Machine' -value $rscriptPath
  # .NOTES
  # **NOTE:** Administrative Access Required when using `-Scope 'Machine'.` On windows
  # You can check in C:\Windows\System32\SystemPropertiesAdvanced.exe to see if it took effect.
  # .LINK
  # https://github.com/alainQtec/cliHelper.env/Public/Add-Env.ps1
  #
  [CmdletBinding(SupportsShouldProcess = $true, DefaultParameterSetName = 'session')]
  [OutputType([void])]
  [Alias('Set-Env')]
  param (
    [Parameter(Mandatory = $true, Position = 0, ValueFromPipeline = $true, ParameterSetName = 'en')]
    [AllowNull()]
    [dotEntry[]]$Entries,

    [Parameter(Mandatory = $true, Position = 0, ParameterSetName = 'session')]
    [ValidateNotNullOrWhiteSpace()]
    [string]$Name,

    [Parameter(Mandatory = $true, Position = 0, ParameterSetName = 'fromfile')]
    [ValidateNotNullOrEmpty()][Alias('File')]
    [IO.FileInfo]$source,

    [parameter(Mandatory = $false, Position = 1, ParameterSetName = 'session')]
    [AllowEmptyString()]
    [AllowNull()]
    [string]$Value,

    [Parameter(Mandatory = $false, Position = 2, ParameterSetName = '__AllparameterSets')]
    [System.EnvironmentVariableTarget]$Scope = "Process",

    [Parameter(Mandatory = $false, Position = 3, ParameterSetName = '__AllParameterSets')]
    [ValidateNotNullOrWhiteSpace()]
    [string]$OutFile,

    [Parameter(Mandatory = $false, ParameterSetName = '__AllParameterSets')]
    [switch]$Force
  )

  begin {
    [System.Management.Automation.ActionPreference]$eap = $ErrorActionPreference; $ErrorActionPreference = "SilentlyContinue"
    [System.Management.Automation.ActionPreference]$Ifp = $InformationPreference; $InformationPreference = "Continue"
    if ([xcrypt]::Get_Host_Os() -eq "Windows" -and ![dotEnv]::IsAdmin()) {
      Write-Warning "$fxn [!] It seems You're not Admin"
      # exit
    }
  }

  process {
    switch ($PSCmdlet.ParameterSetName) {
      "fromfile" {
        [dotEnv]::Read($source.FullName).ForEach({
            if ($PSCmdlet.ShouldProcess("$($_.Name)@$Scope", "SetEnvironmentVariable")) {
              [dotEnv]::SetEnvironmentVariable($_.Name, $_.Value, $Scope)
              if ($OutFile) {
                [dotEnv]::Update($OutFile, $_.Name, $_.Value)
              }
            }
          }
        )
        break
      }
      "session" {
        if ($PSCmdlet.ShouldProcess("$Name@$Scope", "SetEnvironmentVariable")) {
          [dotEnv]::SetEnvironmentVariable($Name, $Value, $Scope)
          if ($OutFile) {
            [dotEnv]::Update($OutFile, $Name, $Value)
          }
        }
        break
      }
      "en" {
        $Entries.ForEach({
            if ($PSCmdlet.ShouldProcess("$($_.Name)@$($Scope)", "SetEnvironmentVariable")) {
              [dotEnv]::SetEnvironmentVariable($_.Name, $_.Value, $Scope)
              if ($OutFile) {
                [dotEnv]::Update($OutFile, $_.Name, $_.Value)
              }
            }
          }
        )
      }
      Default {
        $PSCmdlet.ThrowTerminatingError("Could not resolve parameterSet")
      }
    }
  }

  end {
    [System.GC]::Collect()
    $ErrorActionPreference = $eap;
    $InformationPreference = $Ifp;
  }
}