zone_automation.rb

#!/usr/local/bin/ruby
require 'date'
require 'json'
require 'creek'
require 'net/ssh'
require 'net/ssh/telnet'

def cmd(ssh_exec,command_string)
  input_prompt = true
  fixed_output = ''
  ssh_exec.cmd(command_string) do |command_input|
    fixed_output = command_input.gsub(/\r/,"")
      if fixed_output =~ /(^.*?)\n(.*)$/m
        if input_prompt
          puts "[SSH]> " + command_string
          input_prompt = false
        else
          puts "[SSH]< " + $1
        end
        fixed_output = $2
      end
    end
    fixed_output.each_line do |last|
      puts "[SSH]< " + last.strip
    end
end

def parsesheets(grep, array)
  @worksheets.each do |worksheet|
    worksheet.rows.each do |row|
      row_cells = row.values
      unless row_cells.grep(/#{grep}/).empty?
        array.push(row_cells)
      end
    end
  end
end

def make_zone(target_wwpn_list, slice, host, field, roundr, hf1, hf2, cell, tgt)
  unless host[field].nil?
    hba_num = 0
    host[field].split("\n").each do |address|
      unless hf1.nil?
        @zone_member_list.push("member #{@shortname.upcase}-#{tgt.upcase}-#{@host_num}-#{hf1}-#{hf2}")
        @zone_file.puts "zone name #{@shortname.upcase}-#{tgt.upcase}-#{@host_num}-#{hf1}-#{hf2} vsan #{@vsan}"
      else
        @zone_member_list.push("member #{@shortname.upcase}-#{tgt.upcase}-#{@host_num}-hba#{hba_num}-#{hf2}")
        @zone_file.puts "zone name #{@shortname.upcase}-#{tgt.upcase}-#{@host_num}-hba#{hba_num}-#{hf2} vsan #{@vsan}"
      end
      if @platform_input == "pvm"
        @zone_file.puts "member pwwn " + address
      else
        @zone_file.puts "member pwwn " + host[cell].gsub(/:/,"")
      end
      unless roundr.nil?
        if @target_port_count % 2 == 0
          target_wwpn_list.each_slice(slice).to_a[1].to_h.each do |target,wwpn|
            @zone_file.puts "member pwwn " + wwpn
          end
        else
          target_wwpn_list.each_slice(slice).to_a[0].to_h.each do |target,wwpn|
            @zone_file.puts "member pwwn " + wwpn
          end
        end
        @target_port_count += 1
      end
      hba_num += 1
    end
  end
end

puts "––––––––––––––––––––––––––––––––––––––––––––––––"
puts "Available plaforms"
puts
validpf = %w(pvm intel opensys)
validpf.each_with_index do |platform, index|
  index += 1
  puts "#{index} = #{platform}"
end
puts
print "Choose your platform: "
@platform_input = gets.strip
puts "––––––––––––––––––––––––––––––––––––––––––––––––"
validpf.each_with_index do |platform, index|
  index += 1
  if index.to_s == @platform_input
    @platform_input = platform
  end
end
puts "Available workbooks"
puts
Dir.entries('.').grep(/xlsx$/).each_with_index do |workbook, index|
  index += 1
  puts "#{index} = #{workbook}"
end
puts
print "Choose your workbook: "
excel = gets.strip
puts "––––––––––––––––––––––––––––––––––––––––––––––––"
Dir.entries('.').grep(/xlsx$/).each_with_index do |workbook, index|
  index += 1
  if index.to_s == excel
    excel = workbook
  end
end

workbook = Creek::Book.new "#{excel}"
@worksheets = workbook.sheets

wwpn_file = File.read("config.json")
wwpn_data = JSON.parse(wwpn_file)

wwpn_data.sort_by! { |name| 
	name["wwpn_id"]
}

puts "Available shortnames"
puts
validshortname = %w(RS CS SUN HP DEC)
validshortname.each_with_index do |shrtnme, index|
  index += 1
  puts "#{index} = #{shrtnme}"
end
puts
print "Choose your shortname: "
@shortname = gets.strip
validshortname.each_with_index do |shrtnme, index|
  index += 1
  if index.to_s == @shortname
    @shortname = shrtnme
  end
end
puts "––––––––––––––––––––––––––––––––––––––––––––––––"
puts "Currently defined targets:"
puts
puts "0 = Parse from workbook (*opensys only*)"
wwpn_data.each { |wwpn_print|
	puts wwpn_print["wwpn_id"] + " = " + wwpn_print["short_name"]
}
puts
print "Choose your target device: "
target = gets.strip
puts "––––––––––––––––––––––––––––––––––––––––––––––––"
tgt_wwpn_list = {}
name_wwpn_list = []
wwpn_data.each do |group|
  name_wwpn_list.push(group["short_name"])
  if (group["wwpn_id"] == "#{target}") || (group["short_name"] == "#{target}")
    tgt_wwpn_list = group["ports"]
    target = group["short_name"]
  end
end

print "Enter the vsan: "
@vsan = gets.strip

pvmf = [3, 7, 11, 15, 19]
host_wwpn_list = []
tmparr1 = []
tmparr2 = []
wrkord = []
start_date = []
hours = []
@host_num = '001'
@target_port_count = 1
@zone_member_list = []
@zone_file = File.open("zone_file.txt", "w:UTF-8")
@zone_file.puts "configure terminal"

if @platform_input == "pvm"
  parsesheets("^c05", host_wwpn_list)
  pvmf.each do |field|
    host_wwpn_list.each do |host|
      if name_wwpn_list.include?(target.upcase)
        make_zone(tgt_wwpn_list, tgt_wwpn_list.length / 2, host, field, true, nil, host[field - 1], nil, target)
      else
        make_zone(nil, nil, host, field, nil, nil, host[field - 1], nil, target)
      end
      @host_num = @host_num.next
    end
  end
end

if @platform_input == "intel"
  parsesheets("vHBA", host_wwpn_list)
  host_wwpn_list.each_with_index do |host, index|
    tmparr1.push(host[1])
    unless host[1] == tmparr1[index - 1]
      @host_num = @host_num.next
    end
    if name_wwpn_list.include?(target.upcase)
      make_zone(tgt_wwpn_list, tgt_wwpn_list.length / 2, host, 5, true, host[3], host[1], 5, target)
    else
      make_zone(nil, nil, host, 5, nil, host[3], host[1], 5, target)
    end
  end
end

if @platform_input == "opensys"
  parsesheets("^100000|^500", host_wwpn_list)
  parsesheets("^200000|^210000", host_wwpn_list)
  host_wwpn_list.each do |host|
    unless host[0] == nil
      tmparr2.push(host)
    end
  end
  hba = 0
  tmparr2.each_with_index do |host, index|
    tmparr1.push(host[1])
    unless host[1] == tmparr1[index - 1]
      @host_num = @host_num.next
      hba = 0
    end
    unless index == 0
      if host[1] == tmparr1[index - 1]
        hba += 1
      end
    end
    if name_wwpn_list.include?(target.upcase)
      make_zone(tgt_wwpn_list, tgt_wwpn_list.length / 2, host, 3, true, "#{host[0]}-#{host[2]}#{hba}", host[1], 3, target)
    elsif target == "0"
      if name_wwpn_list.include?(host[5])
        wwpn_data.each do |group|
          if group["short_name"] == host[5]
            tgt_wwpn_list = group["ports"]
          end
        end
        make_zone(tgt_wwpn_list, tgt_wwpn_list.length / 2, host, 3, true, "#{host[0]}-#{host[2]}#{hba}", host[1], 3, host[5])
      else
        make_zone(nil, nil, host, 3, nil, "#{host[0]}-#{host[2]}#{hba}", host[1], 3, host[5].gsub(/\s/,"-"))
      end
    else
      make_zone(nil, nil, host, 3, nil, "#{host[0]}-#{host[2]}#{hba}", host[1], 3, target)
    end
  end
end

puts "––––––––––––––––––––––––––––––––––––––––––––––––"
print "Enter the customer name: "
customer = gets.strip
if @platform_input == "opensys"
  parsesheets("Work Order", wrkord)
  work_order = "WO" + wrkord[0][8].to_i.to_s
  parsesheets("Start Date", start_date)
  parsesheets("Duration", hours)
  date = Date.parse(start_date[0][5].to_s)
  duration = date.strftime("%a-%b-%d-%Y-") + hours[0][5] + "hrs"
else
  puts "––––––––––––––––––––––––––––––––––––––––––––––––"
  print "Enter the work order number: "
  work_order = gets.strip
  puts "––––––––––––––––––––––––––––––––––––––––––––––––"
  print "Enter the start and duration: "
  duration = gets.strip
end

@zone_file.puts
@zone_file.puts "zoneset name #{customer.upcase}-#{work_order.upcase}-#{duration.upcase} vsan #{@vsan}"
@zone_file.puts @zone_member_list
@zone_file.puts "zoneset activate name #{customer.upcase}-#{work_order.upcase}-#{duration.upcase} vsan #{@vsan}"
@zone_file.puts "zone commit vsan #{@vsan}"
@zone_file.close

puts "––––––––––––––––––––––––––––––––––––––––––––––––"
puts "zone_file.txt created in #{FileUtils.pwd()}, please review it for accuracy."
puts "––––––––––––––––––––––––––––––––––––––––––––––––"
print "Enter switch ip (default = 172.23.79.11): "
server = gets.strip
puts "––––––––––––––––––––––––––––––––––––––––––––––––"
print "Enter your (switch) username: "
user = gets.strip
puts "––––––––––––––––––––––––––––––––––––––––––––––––"
print "Enter your (switch) password: "
pass = STDIN.noecho(&:gets).strip
puts
puts "––––––––––––––––––––––––––––––––––––––––––––––––"

Net::SSH.start(server, user, :password => pass) do |ssh|
  command_file = File.read("zone_file.txt").split("\n")
  ssh_exec = Net::SSH::Telnet.new("Session" => ssh)
  command_file.each do |command|
    cmd(ssh_exec,"#{command}")
    sleep 0.5
  end
end