plugins/vagrant/src/Plugin.ps1

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
# Based on the great work by Posh-git!
#
# Vagrant status is slower then is resonable for a prompt to render. I am making some assumptions
# by scraping the .vagrant folder for any id files that should represent active/aborted some sort
# of Vagrant machine activity.
function Global:Get-VagrantFile {
    $fileName = 'Vagrantfile'
    $base = Get-Item -Force .
    if ($base.PSProvider.Name -ne 'FileSystem'){
        return $FALSE
    }
    $vagrantFile = Join-Path $base.FullName $fileName
    if(Test-Path -LiteralPath $vagrantFile)
    {
        return $TRUE
    }
    return $FALSE
}

function Global:Get-VagrantDir {
    $dirName = '.vagrant'
    $base = Get-Item -Force .
    if($base.PSProvider.Name -ne 'FileSystem'){
        return $FALSE
    }
    $vagrantDir = Join-Path $base.FullName $dirName
    if(Test-Path -LiteralPath $vagrantDir)
    {
        return $vagrantDir
    }
    return $FALSE
}

function Global:Get-VagrantEnvIndex {
    $vagrantEnvVar = [environment]::GetEnvironmentVariable("VAGRANT_HOME","User")
    $dirName = '.vagrant.d'
    if($vagrantEnvVar)
    {
        if($vagrantEnvVar -contains $dirName)
        {
          $base = Get-Item -Force $vagrantEnvVar
          $vagrantEnvDir = Join-Path $base.FullName
        }
        else
        {
          $base = Get-Item -Force $vagrantEnvVar
          $vagrantEnvDir = Join-Path $base.FullName $dirName
        }
        if(Test-Path -LiteralPath $vagrantEnvDir)
        {
          $machineIndex = Get-ChildItem -Path $vagrantEnvDir -Recurse -File -Filter 'index' | % { $_.FullName }
        }
        return $machineIndex
    }
    else
    {
        $base = Get-Item -Force $env:USERPROFILE
        $vagrantEnvDir = Join-Path $base.FullName $dirName
        if(Test-Path -LiteralPath $vagrantEnvDir)
        {
           $machineIndex = Get-ChildItem -Path $vagrantEnvDir -Recurse -File -Filter 'index' | % { $_.FullName }
        }
        return $machineIndex
    }
}

function Global:Write-VagrantStatusSimple {
    $vagrantFolder = Get-VagrantDir
    if((Test-Path $vagrantFolder) -and (Get-VagrantFile))
    {
        $vagrantActive = $FALSE
        $items = Get-ChildItem -Path $vagrantFolder -Recurse -File -Filter 'id'
        foreach($item in $items)
        {
            if($item.ToString().Contains("id"))
            {
                $vagrantActive = $TRUE
            }
            else
            {
                $vagrantActive = $FALSE
            }
        }
        if($vagrantActive)
        {
            Write-Host ' [' -NoNewline
            Write-Host '^' -ForegroundColor Green -NoNewline
            Write-Host ']' -NoNewline
        }
        else
        {
            Write-Host ' [' -NoNewline
            Write-Host '-' -ForegroundColor Gray -NoNewline
            Write-Host ']' -NoNewline

        }
    }
    elseif(Get-VagrantFile)
    {
      Write-Host ' [' -NoNewline
      Write-Host '-' -ForegroundColor Gray -NoNewline
      Write-Host ']' -NoNewline
    }
}

function Global:Write-VagrantStatusDetailed {
  $vagrantFolder = Get-VagrantDir
  $vagrantEnvJson = Get-Content(Get-VagrantEnvIndex -Raw) | ConvertFrom-Json
  $machines = @()
  $d = 0
  $r = 0
  $a = 0
  if((Test-Path $vagrantFolder) -and (Get-VagrantFile))
  {
    $items = Get-ChildItem -Path $vagrantFolder -Recurse -File -Filter 'index_uuid' | % { $_.FullName }
    if($items)
    {
      foreach($item in $items)
      {
          if($item.ToString().Contains("index_uuid"))
          {
              $machines = $machines + (get-content $item)
          }
      }

      foreach($machine in $machines)
      {
        foreach($envMachine in $vagrantEnvJson.machines)
        {
          $stateTemp = $envMachine.$machine | select -ExpandProperty state
          switch($stateTemp){
                  'aborted' { $d += 1; break}
                  'running' {$r += 1; break}
                  'poweroff' {$d += 1; break}
                  default { break}
          }
        }
      }
      Write-Host ' [' -NoNewline
      Write-Host "D:${d} " -ForegroundColor Gray -NoNewline
      Write-Host "R:${r}" -ForegroundColor Green -NoNewline
      Write-Host ']' -NoNewline
    }
    else
    {
      Write-Host ' [' -NoNewline
      Write-Host "-" -ForegroundColor Gray -NoNewline
      Write-Host ']' -NoNewline
    }
  }
  elseif(Get-VagrantFile)
  {
    Write-Host ' [' -NoNewline
    Write-Host "-" -ForegroundColor Gray -NoNewline
    Write-Host ']' -NoNewline
  }
}