"""
Library for the robot based system test tool of the OpenDaylight project.
Authors: Vaibhav Bhatnagar@Brocade
Updated: 2015-06-01
"""
import socket

from robot.libraries.BuiltIn import BuiltIn


class CapwapLibrary(object):
    """Provide many methods to simulate WTPs and their functions."""

    def __init__(self):
        self.builtin = BuiltIn()

    def send_discover(self, ac_ip, wtp_ip='', ip='ip', port=5246):
        """Send Discover CAPWAP Packet from a WTP."""
        data = ''.join(chr(x) for x in [0x00, 0x20, 0x01, 0x02, 0x03, 0x04, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16])
        self.builtin.log('Sending Discover Packet to: %s' % ac_ip, 'DEBUG')
        session = socket.socket(socket.AF_INET, socket.SOCK_DGRAM, socket.IPPROTO_UDP)
        session.sendto(data, (ac_ip, port))
        self.builtin.log('Packet Sent', 'DEBUG')

    def get_hostip(self):
        """Get Host IP Address."""
        ip_addr = socket.gethostbyname(socket.gethostname())
        return ip_addr

    def get_simulated_wtpip(self, controller):
        """Get the Simulated WTP ip based on the controller."""
        if controller == '127.0.0.1':
            exp_ip = controller
        else:
            exp_ip = self.get_hostip()
        return exp_ip