#!/usr/bin/expect -f
# BEGIN_ICS_COPYRIGHT8 ****************************************
# 
# Copyright (c) 2015, Intel Corporation
# 
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
# 
#     * Redistributions of source code must retain the above copyright notice,
#       this list of conditions and the following disclaimer.
#     * Redistributions in binary form must reproduce the above copyright
#       notice, this list of conditions and the following disclaimer in the
#       documentation and/or other materials provided with the distribution.
#     * Neither the name of Intel Corporation nor the names of its contributors
#       may be used to endorse or promote products derived from this software
#       without specific prior written permission.
# 
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
# DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE
# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
# SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
# CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
# OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
# 
# END_ICS_COPYRIGHT8   ****************************************

#[ICS VERSION STRING: unknown]
#########################################################################################
# Name        : get_switch_lids_to_ports_map
# Input       : cmd_host (where can opa commands be run to query SM)
# Return      . A dict mapping LID to list of ports.
# Description : Queries fabric return a dict with all lids of switches and ports that are enabled on them.  Only ports with LINK will be returned (no disconnected ports)
#########################################################################################


proc get_switch_lids_to_ports_map { cmd_host }  {

    # Build DOM from XML
    set lids_dom [ run_opareport $cmd_host "-o lids" ]
    set links_dom [ run_opareport $cmd_host "-o links" ]

    # Get all switches 
    set lidsList [ $lids_dom selectNodes {/Report/LIDSummary/LIDs/Value[NodeType='SW']} ]
    #puts "found [ llength $lidsList ] switch lids"


    set lid_port_map [dict create]

    # For each switch node
    foreach item $lidsList {
	set lid [ $item getAttribute "LID"]
	set guid [ [ $item selectNodes NodeGUID/text() ] nodeValue ]
	#puts "lid:$lid; guid:$guid"	

	# Now, using guid as the key, poke into linksDOM to find the ports for this guid.
	set portList [ $links_dom selectNodes /Report/LinkSummary/Link/Port\[NodeGUID='$guid'\]/PortNum/text() ]   
	# Now you have a list of ports for specific LID/GUID.
	#puts "found [ llength $portList ] ports"

	# List of ports per lid
	set ports {} 
	foreach port $portList {
	    set portNum [ $port nodeValue ]

	    # If don't care about port 1 - skip it
	    #` if { $portNum == 1 } {
	    #    continue
	    # }
	    #puts "lid:$lid->port num:$portNum"

	    lappend ports $portNum

	    # I would have a data structure that is a map of lists, i.e.  
	    #	LID1: port1, port2, port3
	    #	LID2: port5,port6,port7
	    #	…….                                        	    
	}
	#puts "Added [ llength $ports ] ports"

	dict set lid_port_map $lid $ports
	#puts "lid_port_map size: [dict size $lid_port_map]"
    }
    
    # Free memory
    $lids_dom delete 
    $links_dom delete

    return $lid_port_map
}
#########################################################################################
# Name        : get_switch_lids_to_hfi_ports_map
# Input       : cmd_host (where can opa commands be run to query SM)
# Return      . A dict mapping LID to list of ports that are connect to HFIs.
# Description : Queries fabric return a dict with all lids of switches and ports that are enabled and connected to HFIs on them.  Only ports with LINK will be returned (no disconnected ports)
#########################################################################################


proc get_switch_lids_to_hfi_ports_map { cmd_host }  {
    # Build DOM from XML
    set lids_dom [ run_opareport $cmd_host "-o lids" ]
    set links_dom [ run_opareport $cmd_host "-o links" ]

    set lidsList [ $lids_dom selectNodes {/Report/LIDSummary/LIDs/Value[NodeType='SW']} ]
#    puts "found [ llength $lidsList ] switch lids"
#    puts "[$lidsList selectNodes NodeGUID/text()]"

    set lid_port_map [dict create]

    # For each switch node
    foreach item $lidsList {
	set lid [ $item getAttribute "LID"]
	set guid [ [ $item selectNodes NodeGUID/text() ] nodeValue ]
	#puts "lid:$lid; guid:$guid"	

	# Now, using guid as the key, poke into linksDOM to find the ports for this guid.
        
#	set portList [ $links_dom selectNodes /Report/LinkSummary/Link/Port\[NodeGUID='$guid'\]/PortNum/text() ]          
	set linkList [ $links_dom selectNodes /Report/LinkSummary/Link ]
        #puts "found [ llength $linkList ] links"
        foreach link $linkList {
          set portNodes [ $link getElementsByTagName Port ]
          set port1 [lindex $portNodes 0]
          set port2 [lindex $portNodes 1]
          set port1_NodeType [[$port1 child all NodeType] asText]
          set port2_NodeType [[$port2 child all NodeType] asText]
          set port1_GUID [[$port1 child all NodeGUID] asText]
          set port2_GUID [[$port2 child all NodeGUID] asText]
          set port1_PortNum [[$port1 child all PortNum] asText]
          set port2_PortNum [[$port2 child all PortNum] asText]
          if { ([string compare $port1_GUID $guid]==0) || ([string compare $port2_GUID $guid]==0) } {
            if { ([string compare $port1_NodeType SW]==0) && ([string compare $port2_NodeType FI]==0) } {
	      lappend ports $port1_PortNum
            } elseif {([string compare $port1_NodeType FI] == 0) && ([string compare $port2_NodeType SW] == 0)} {
	      lappend ports $port2_PortNum
            } 
          } 
           
        }
	#puts "Added [ llength $ports ] ports"

	dict set lid_port_map $lid $ports
	#puts "lid_port_map size: [dict size $lid_port_map]"
    }
    
    # Free memory
    $lids_dom delete 
    $links_dom delete

    return $lid_port_map
}



#########################################################################################
# Name		: get_lid_from_guid
# Input		: guid
# Return	: output of opasaquery -o lid -n $guid
# Description	: get lid from guid
#########################################################################################


proc get_lid_from_guid { guid } {
##
## get_lid_from_guid
## --------------
## Obtain my guid
## assumes that target_root_sh has already been called for the host
        set cmd "lid=`opasaquery -o lid -n $guid`"
        #debug_message "running: $cmd"
        unix_cmd 60 "" $cmd
        send_unix_cmd { echo "XXX${lid}XXX"; echo DONE }
        expect_list 60 { "DONE" "XXX" "DONE" } { "exiting" "No such" "Error" "ERROR" "VAPI" "THH" } tmp
        check_exit_status 60 0
        regexp {DONE.*XXX *(.*) *XXX.*DONE} $tmp trash lid

        return "$lid"
}

#########################################################################################
# Name        : get_switchport_from_lid
# Input       : cmd_host lid  *** LID must be in format 0x0001 (the same format as provided by opareport -o lids)
# Return      . A list with element 1 being the SWITCH LID and element 2 being the SWITCH PORT where HFI with that LID is directly connected.  If not found the list [ none none ] will be returned
# Description : For the lid of an HFI return the switch lid and port that that HFI is connected to
#########################################################################################
proc get_switchport_from_lid { cmd_host mylid }  {
    set lids_dom [ run_opareport $cmd_host "-o lids" ]
    set links_dom [ run_opareport $cmd_host "-o links" ]

    #################################################
    #debug/dev only
    #set lids_file  "lids.xml"
    #set links_file "links.xml"
    #################################################
    # Open lids XML files
    #set size [file size $lids_file]
    #set fp [open $lids_file r]
    #set lidsXML [read $fp $size]
    #################################################
    # Open links XML files
    #set size [file size $links_file]
    #set fp [open $links_file r]
    #set linksXML [read $fp $size]
    ##################################################
    #set lids_dom [dom parse $lidsXML]
    #set links_dom [dom parse $linksXML]

    set switchLidsList [ $lids_dom selectNodes {/Report/LIDSummary/LIDs/Value[NodeType='SW']} ]
    set hfiLidsList [ $lids_dom selectNodes {/Report/LIDSummary/LIDs/Value[NodeType='FI']} ]

    set myguid "none"
    set return_lid "none"
    set return_port "none"

    foreach item $hfiLidsList {
        set lid [ $item getAttribute "LID"]
        set guid [ [ $item selectNodes NodeGUID/text() ] nodeValue ]
        if { [string compare $lid $mylid] == 0 } {
          set myguid $guid
        }
    }

    if { [string compare $myguid "none"] == 0 } {
       error "Lid $mylid not found"
    }

    # For each switch node
    foreach item $switchLidsList {
        set lid [ $item getAttribute "LID"]
        set guid [ [ $item selectNodes NodeGUID/text() ] nodeValue ]

        set linkList [ $links_dom selectNodes /Report/LinkSummary/Link ]
        #puts "found [ llength $linkList ] links"
        foreach link $linkList {
#          puts "processing link [$link asXML] [$link asText]"
          set portNodes [ $link getElementsByTagName Port ]
          set port1 [lindex $portNodes 0]
          set port2 [lindex $portNodes 1]
          set port1_NodeType [[$port1 child all NodeType] asText]
          set port2_NodeType [[$port2 child all NodeType] asText]
          set port1_GUID [[$port1 child all NodeGUID] asText]
          set port2_GUID [[$port2 child all NodeGUID] asText]
          set port1_PortNum [[$port1 child all PortNum] asText]
          set port2_PortNum [[$port2 child all PortNum] asText]
          target_root_sh $cmd_host
          if { ([string compare $port1_GUID $myguid]==0) } {
            #set return_lid $lid
	    set return_lid [get_lid_from_guid $port2_GUID]
            set return_port $port2_PortNum
          } elseif { ([string compare $port2_GUID $myguid]==0) } {
            #set return_lid $lid
	    set return_lid [get_lid_from guid $port1_GUID]
            set return_port $port1_PortNum
          }

        }

    }
    $lids_dom delete
    $links_dom delete
    return [list $return_lid $return_port]
}


#########################################################################################
# Name        : get_switchport_from_guid
# Input       : cmd_host guid
# Return      . A list with element 1 being the SWITCH LID and element 2 being the SWITCH PORT where HFI with that LID is directly connected.  If not found the list [ none none ] will be returned
# Description : For the lid of an HFI return the switch lid and port that that HFI is connected to
#########################################################################################
proc get_switchport_from_guid { cmd_host myguid }  {
    set lids_dom [ run_opareport $cmd_host "-o lids" ]
    set links_dom [ run_opareport $cmd_host "-o links" ]

    #################################################
    #debug/dev only
    #set lids_file  "lids.xml"
    #set links_file "links.xml"
    #################################################
    # Open lids XML files
    #set size [file size $lids_file]
    #set fp [open $lids_file r]
    #set lidsXML [read $fp $size]
    #################################################
    # Open links XML files
    #set size [file size $links_file]
    #set fp [open $links_file r]
    #set linksXML [read $fp $size]
    ##################################################
    #set lids_dom [dom parse $lidsXML]
    #set links_dom [dom parse $linksXML]

    set switchLidsList [ $lids_dom selectNodes {/Report/LIDSummary/LIDs/Value[NodeType='SW']} ]
    set hfiLidsList [ $lids_dom selectNodes {/Report/LIDSummary/LIDs/Value[NodeType='FI']} ]

    set return_lid "none"
    set return_port "none"

    # For each switch node
    foreach item $switchLidsList {
        set lid [ $item getAttribute "LID"]
        set guid [ [ $item selectNodes NodeGUID/text() ] nodeValue ]

        set linkList [ $links_dom selectNodes /Report/LinkSummary/Link ]
        #puts "found [ llength $linkList ] links"
        foreach link $linkList {
          #puts "processing link [$link asXML] [$link asText]"
          set portNodes [ $link getElementsByTagName Port ]
          set port1 [lindex $portNodes 0]
          set port2 [lindex $portNodes 1]
          set port1_NodeType [[$port1 child all NodeType] asText]
          set port2_NodeType [[$port2 child all NodeType] asText]
          set port1_GUID [[$port1 child all NodeGUID] asText]
          set port2_GUID [[$port2 child all NodeGUID] asText]
          set port1_PortNum [[$port1 child all PortNum] asText]
          set port2_PortNum [[$port2 child all PortNum] asText]
          #puts "myguid is $myguid port1guid = $port1_GUID port2guid= $port2_GUID lid = $lid"
          if { ([string compare $port1_GUID $myguid]==0) } {
            set return_lid $lid
            set return_lid [get_lid_from_guid $port2_GUID]

            set return_port $port2_PortNum
          } elseif { ([string compare $port2_GUID $myguid]==0) } {
            set return_lid $lid
            set return_lid [get_lid_from_guid $port1_GUID]

            set return_port $port1_PortNum
          }
        }
    }
   
    $lids_dom delete
    $links_dom delete
    return [list $return_lid $return_port]
}