#!/usr/bin/perl -w
#
# Atle Holm 13.12.2013
# Script to store VMWare VLAN names in mongoDB - 
#   uses VMWare SDK for Perl in Linux and the required modules in the use list below:
#
# Test connection: 
# /usr/lib/vmware-vcli/apps/general/connect.pl --url https://vCenter/sdk/vimService --username 'domain\user' --password 'xxx'
# Usage: ./vmware-vlan.pl --url https://vCenter/sdk/vimService --username 'domain\user' --password 'xxx'

use strict;
use warnings;
use MongoDB;
use VMware::VIRuntime;
use VMware::VILib;

Opts::parse();
Opts::validate();
Util::connect();

my $PORT = "insert port here";
my $MONGO_URL = "insert url here";
my $databaseName = "insert database name here";

sub getAllPorts {
   my $portgroup_views = Vim::find_entity_views(view_type => 'DistributedVirtualPortgroup');
   my @portInfo;
   foreach my $portgroup (@$portgroup_views) {
      my ($portgroupName, $portgroupConfigName, $portgroupConfigDescription, $portgroupConfigNumPorts);

      $portgroupName = $portgroup->name;
      $portgroupConfigName = ($portgroup->config->name ? $portgroup->config->name : "N/A");
      $portgroupConfigDescription = ($portgroup->config->description ? $portgroup->config->description : "N/A");
      $portgroupConfigNumPorts = ($portgroup->config->numPorts ? $portgroup->config->numPorts : "N/A");

      my %tempHash = ('portgroupName' => $portgroupName, 'portgroupConfigName' => $portgroupConfigName, 
         'portgroupConfigDescription' => $portgroupConfigDescription, 'portgroupConfigNumPorts' => $portgroupConfigNumPorts);
      push(@portInfo, \%tempHash);
   }
   return @portInfo;
}

#1. First we get all the portgroups from vCenter
my @portInfo = getAllPorts();
#2. Then we connect to the mongoDB database:
my $mongoDBClient = MongoDB::MongoClient->new(host => $MONGO_URL.':'.$PORT);

#FOR TESTING CONNECTION:
#my @dbs = $mongoDBClient->database_names;
#foreach my $db (@dbs) {
   #print $db . "\n";
#}

#3. Get the desired database:
my $db = $mongoDBClient->get_database($databaseName);
#4. Get the desired document collection - called a table in "normal" databases:
my $col_vlan_names = $db->get_collection('vlan_names');

#If we weren't using upsert next we would have to drop the collection 
#so that we wouldn't end up with adding redundant information in it each time this script is run.
#No need to drop, we are using upsert:
#$col_vlan_names->drop;

#4. Loop through the port groups we have collected and update the database if the port group does not exist there already:
foreach my $portgroup (@portInfo) {
   my %pgroup = %$portgroup;
   #Following is commented away in advance of upsert:
   #$col_vlan_names->insert({"name" => $pgroup{'portgroupName'}});
   $col_vlan_names->update({"name" => $pgroup{'portgroupName'}},{'$inc' => {"views" => 1}},{'upsert' => 1});
}

#FOR TESTING DATABASE CONTENT:
#my $all_vlans = $col_vlan_names->find;
#while(my $vlansObj = $all_vlans->next) {
   #my $vlanName = $vlansObj->{'name'};
   #print $vlanName . "\n" if(defined($vlanName));
#}

#! /usr/bin/perl -w

#Atle Holm - 2009
#Et script som generer en HTML side som lister påslåtte VmWare maskiner og 
#gjør det mulig å slå dem av.Dette er testet med VmWare server, og scriptet krever 
#en del moduler. Feks VIX API fra VmWare. 
#Scriptet må også brukes på en web-server som er satt opp med- og har installert Perl.

use CGI;
use CGI::Carp qw(carpout fatalsToBrowser); #Må ha denne for at die og warn skal fungere, og skrive til browser.
use CGI::Pretty qw(:all); #Må ha dette til formene
use Fcntl qw(:flock); #Må ha denne for å kunne låse filer

use VMware::Vix::Simple;
use VMware::Vix::API::Constants;

#VMWARE spesifike variabler:
my $err = VIX_OK;
my $hostHandle = VIX_INVALID_HANDLE;
my $vmHandle = VIX_INVALID_HANDLE;
our @vms;

#Utvikler definerte variabler:
our $counter = 0;
my ($TempVms, $TempChoice, $hostname, $username, $password);
my $cgi=new CGI; #read in parameters
my $msgfile="/tmp/state.msg";

print $cgi->header(); #print en header
#Hent gammel state hvis den finnes
if ($cgi->param('Logoff')) {
   system("rm /tmp/state.msg");
}
if (open (LOAD,$msgfile)) {
   flock LOAD,LOCK_SH; #shared lock
   my $oldcgi=new CGI(LOAD);
   flock LOAD,LOCK_UN; #release lock
   close (LOAD);
   $hostname = $oldcgi->param('Hostname');
   $username = $oldcgi->param('User');
   $password = $oldcgi->param('Password');
   &ShowVMS;
} elsif ($cgi->param('Hostname') and $cgi->param('User') and $cgi->param('Password')) {
   $hostname = $cgi->param('Hostname');
   $username = $cgi->param('User');
   $password = $cgi->param('Password');
   if (open (SAVE,"> $msgfile")) {
      flock SAVE,LOCK_EX; #exclusive lock 
      $cgi->save(SAVE);
      flock SAVE,LOCK_UN; #release lock 
   } else {
      print $cgi->font({-color=>'red'},"<BR>Failed to save: $!");
   }
   &ShowVMS;
} else {
   print $cgi->start_html(-title=>'VMWare Shutdown Service', -bgcolor=>'black', -LINK=>"red", -ALINK=>"red", -VLINK=>"red", 
         -TEXT=>"white"); #Lag HTML dokument start
   print generate_form();
   print "<BR/><BR/><BR/><BR/><BR/><BR/><BR/><B>Syntax for writing servername is:</B><BR/>";
   print "http://servername:port/sdk<BR>https://ipaddress:port/sdk";
   print $cgi->end_html();
}
sub generate_form {
   return start_form,
   h1("<CENTER>VmWare Shutdown Service</CENTER><BR>"),
   h2("Please enter the information below to enter:"),
   p("Server name:", textfield('Hostname')),
   p("User Name: ", textfield('User')),
   p("Password:   ", password_field('Password')),
   p(submit),
   end_form;
}

#Hit kommer vi bare om vi har de rette parametrene satt og/eller vi har den rette informasjonen
sub ShowVMS {
   ($err, $hostHandle) = HostConnect(VIX_API_VERSION,
      VIX_SERVICEPROVIDER_VMWARE_VI_SERVER,
      $hostname, # hostName
      0, # hostPort
      $username, # userName
      $password, # password
      0, # options
      VIX_INVALID_HANDLE); # propertyListHandle

   die "HostConnect() failed, $err ", GetErrorText($err), "\n" if $err != VIX_OK;

   @vms = FindRunningVMs($hostHandle, 100);
   $err = shift @vms;
   die "Error $err finding running VMs ", GetErrorText($err),"\n" if $err != VIX_OK;
   
   #Her starter selve innholdet i HTML dokumentet
   print $cgi->start_html(-title=>'VMWare Shutdown Service', -bgcolor=>'black', -LINK=>"red", -ALINK=>"red", 
         -VLINK=>"red", -TEXT=>"white"); #Lag HTML dokument start
   print "<P>*Running Virtual Machines at $hostname:\n</P> ";
   foreach $TempVms (@vms) { 
      print "           *(", ++$counter, ")   <span style=\"color:red\">", 
            $cgi->a({-href=>"http://ourserver.nodenet.local/cgi-bin/ShutdownList.pl?Shutdown=$counter"}, $TempVms), "\n</span><BR>"; 
   }
   print "<BR>Choose VM to power down(1 - ", $counter, ") by clicking on them.";
   if ($cgi->param('Shutdown')) {
      my $param = $cgi->param('Shutdown');
      &Poweroff($param);
   } 
   print "<BR><P>   *", $cgi->a({-href=>"http://ourserver.nodenet.local/cgi-bin/ShutdownList.pl?Logoff=1"}, "Log Off</P>");
   print $cgi->end_html(); #Gjør ferdig HTML document
}
sub Poweroff {
   my $choice = shift;
   print "<BR>Attempting to power off: <span style=\"color:red\">", $vms[--$choice], "</span>\n";
   ($err, $vmHandle) = VMOpen($hostHandle, $vms[$choice]);
   die "<BR>VMOpen() failed, $err ", GetErrorText($err), "\n" if $err != VIX_OK;

   # Assume this virtual machine is already running
   $err = VMPowerOff($vmHandle, VIX_VMPOWEROP_FROM_GUEST); 
   die "<BR>VMPowerOff() failed, $err ", GetErrorText($err), "\n" if $err != VIX_OK;
}