Atom

$dest = $(Split-Path -parent $MyInvocation.MyCommand.Definition)

. $(Join-Path $dest "functions.ps1")

# Install zip
$packageName = 'Atom' # arbitrary name for the package, used in messages
$url = 'https://github.com/atom/atom/releases/download/v0.140.0/atom-windows.zip' # download url
$url64 = $url # 64bit URL here or just use the same as $url
Install-ChocolateyZipPackage "$packageName" "$url" "$dest" "$url64"
$exePath = $dest + '\Atom\atom.exe'

# Install desktop shortcut
$desktop = $([System.Environment]::GetFolderPath([System.Environment+SpecialFolder]::DesktopDirectory))
$desktopLink = Join-Path $desktop "$packageName.lnk"
Install-ChocolateyShortcut -shortcutFilePath $desktopLink -targetPath $exePath

# Install start menu shortcut
$startMenu = $([System.Environment]::GetFolderPath([System.Environment+SpecialFolder]::Programs))
$githubStartMenuGroup = Join-Path $startMenu "GitHub, Inc"
New-Item $githubStartMenuGroup -type directory -force
$startMenuLink = Join-Path $githubStartMenuGroup "$packageName.lnk"
Install-ChocolateyShortcut -shortcutFilePath $startMenuLink -targetPath $exePath

$iconPath = '"' + $exePath + '"'
# Install explorer menu file, directory, and background shortcuts
$elevated = @()
  $elevated += $(Install-ChocolateyExplorerMenuItemEx "Atom" "Open with Atom" "$dest\Atom\atom.exe" -iconLocation $iconPath -type "file")
  $elevated += $(Install-ChocolateyExplorerMenuItemEx "Atom" "Open Folder with Atom" "$dest\Atom\atom.exe" -iconLocation $iconPath -type "directory")
  $elevated += $(Install-ChocolateyExplorerMenuItemEx "Atom" "Open Folder with Atom" "$dest\Atom\atom.exe" -iconLocation $iconPath -type "background")
$cmd = $($elevated -join ';')
Start-ChocolateyProcessAsAdmin $cmd

# Install apm shortcut
Install-BinFile "apm" "$dest\Atom\resources\app\apm\node_modules\atom-package-manager\bin\apm.cmd"

Write-Host "explorer menu items created"

function Install-ChocolateyShortcut {
  param(
    [string] $shortcutFilePath,
    [string] $targetPath,
    [string] $workingDirectory,
    [string] $arguments,
    [string] $iconLocation,
    [string] $description
  )

  Write-Debug "Running 'Install-ChocolateyShortcut' with parameters ShortcutFilePath: `'$shortcutFilePath`', TargetPath: `'$targetPath`', WorkingDirectory: `'$workingDirectory`', Arguments: `'$arguments`', IconLocation: `'$iconLocation`', Description: `'$description`'";

  if(!$shortcutFilePath) {
    Write-ChocolateyFailure "Install-ChocolateyShortcut" "Missing ShortCutFilePath input parameter."
    return
  }

  if(!$targetPath) {
    Write-ChocolateyFailure "Install-ChocolateyShortcut" "Missing TargetPath input parameter."
    return
  }

  if(!(Test-Path($targetPath))) {
    Write-ChocolateyFailure "Install-ChocolateyShortcut" "TargetPath does not exist, so can't create shortcut."
    return
  }

  if($iconLocation) {
    if(!(Test-Path($iconLocation))) {
      Write-ChocolateyFailure "Install-ChocolateyShortcut" "IconLocation does not exist, so can't create shortcut."
      return
    }
  }

  if($workingDirectory) {
    if(!(Test-Path($workingDirectory))) {
      Write-ChocolateyFailure "Install-ChocolateyShortcut" "WorkingDirectory does not exist, so can't create shortcut."
      return
    }
  }

  Write-Debug "Creating Shortcut..."

  try {
    $global:WshShell = New-Object -com "WScript.Shell"
      $lnk = $global:WshShell.CreateShortcut($shortcutFilePath)
      $lnk.TargetPath = $targetPath
    $lnk.WorkingDirectory = $workingDirectory
      $lnk.Arguments = $arguments
      if($iconLocation) {
          $lnk.IconLocation = $iconLocation
      }
      $lnk.Description = $description
      $lnk.Save()

    Write-Debug "Shortcut created."

    Write-ChocolateySuccess "Install-ChocolateyShortcut completed"

  }
  catch {
    Write-ChocolateyFailure "Install-ChocolateyShortcut" "There were errors attempting to create shortcut. The error message was '$_'."
  }
}

function Install-BinFile {
  param(
    [string] $name,
    [string] $path
  )

  Write-Debug "Running 'Install-BinFile' for $name with path:`'$path`'";

  $packageBatchFileName = Join-Path $nugetExePath "$name.bat"

  if (Test-Path ($packageBatchFileName)) {Remove-Item $packageBatchFileName -force}
  $path = $path.ToLower().Replace($nugetPath.ToLower(), "..\").Replace("\\","\")
  $path = "%DIR%$($path)"
  $pathBash = $path.Replace("%DIR%..\","`$DIR/../").Replace("\","/")
  Write-Host "Adding $packageBatchFileName and pointing to `'$path`'." -ForegroundColor $Note
"@echo off
SET DIR=%~dp0%
cmd /c ""$path %*""
exit /b %ERRORLEVEL%" | Out-File $packageBatchFileName -encoding ASCII
}

function Install-ChocolateyExplorerMenuItemEx {
param(
  [string]$menuKey,
  [string]$menuLabel,
  [string]$command,
  [string]$iconLocation,
  [ValidateSet('file','directory','background')]
  [string]$type = "file"
)
try {
  Write-Debug "Running 'Install-ChocolateyExplorerMenuItem' with menuKey:'$menuKey', menuLabel:'$menuLabel', command:'$command', type '$type', iconLocation:'$iconLocation'"
  $argName = "%1"
  if($type -eq "file") {
    $key = "*"
  }
  elseif($type -eq "directory") {
    $key="directory"
  }
  elseif($type -eq "background") {
    $key="directory\background"
    $argName="%V"
  } else{
    return 1
  }

  $iconCmd =""
  if($iconLocation) {
    $iconCmd="Set-ItemProperty -LiteralPath 'HKCR:\$key\shell\$menuKey' -Name 'Icon'  -Value '$iconLocation'"
  }
  $elevated = "`
    if( -not (Test-Path -path HKCR:) ) {New-PSDrive -Name HKCR -PSProvider registry -Root Hkey_Classes_Root};`
    if(!(test-path -LiteralPath 'HKCR:\$key\shell\$menuKey')) { new-item -Path 'HKCR:\$key\shell\$menuKey' };`
    Set-ItemProperty -LiteralPath 'HKCR:\$key\shell\$menuKey' -Name '(Default)'  -Value '$menuLabel';`
    if(!(test-path -LiteralPath 'HKCR:\$key\shell\$menuKey\command')) { new-item -Path 'HKCR:\$key\shell\$menuKey\command' };`
    Set-ItemProperty -LiteralPath 'HKCR:\$key\shell\$menuKey\command' -Name '(Default)' -Value '$command \`"$argName\`"';`
	$iconCmd;"

  return $elevated
}
catch {
    $errorMessage = "'$menuKey' explorer menu item was not created $($_.Exception.Message)"
    Write-Error $errorMessage
    throw $errorMessage
  }
}