# Copyright 2016 Canonical, Ltd.
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as
# published by the Free Software Foundation, either version 3 of the
# License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

from probert.network import NetworkInfo

IGNORED_DEVICE_TYPES = ['lo', 'tun', 'tap']

class Device:
    # In general, an element of config['ethernets'] (or
    # config['wifis']) matches an arbitrary set of interfaces. In
    # subiquity, though, it just matches a single interface by
    # name.
    def __init__(self, name, vendor):
        self.name = name
        self.vendor = vendor
        self.dhcp4 = False
        self.dhcp6 = False
        self.addresses = []
        self.ignored = False

    @classmethod
    def from_probe_data(cls, data):
        name = data['hardware']['INTERFACE']
        info = NetworkInfo({name:data})
        return cls(name, info.vendor)

    def render(self):
        addresses = []
        for address in self.addresses:
            if address.version == 4 and not self.dhcp4:
                addresses.append(addresses.with_prefixlen)
            if address.version == 6 and not self.dhcp6:
                addresses.append(addresses.with_prefixlen)
        data = {
            'dhcp4': str(self.dhcp4).lower(),
            'dhcp4': str(self.dhcp4).lower(),
            }
        if addresses:
            data['addresses'] = addresses
        return data

class PhysicalDevice(Device):
    pass

class EthernetDevice(PhysicalDevice):
    pass

## class WifiDevice(PhysicalDevice):
##     ## lots of things go here

## class BridgeDevice(Device):
##     def __init__(self, name):
##         Device.__init__(name)
##         self.interfaces = []

class NetworkConfig:
    def __init__(self):
        self.ethernets = {}
        ## self.wifis = {}
        ## self.bridges = {}

    def render(self):
        ethernets = {}
        for ethernet in ethernets:
            if not ethernet.ignored:
                ethernets[ethernet.name] = ethernet.render()
        data = {
            'version': 2,
            }
        if ethernets:
            data['ethernets'] = ethernets
        return data

    @classmethod
    def from_probe_data(self, results):
        network = results['network']
        for name, data in network.items():
            type = data['type']
            if type in IGNORED_DEVICE_TYPES:
                continue
            if data['bridge']['is_bridge']:
                pass ## bridges not yet supported
            elif data['bridge']['bond']['is_master']:
                pass ## bridges not yet supported
            if type == 'eth':
                self.ethernets[name] = EthernetDevice.from_probe_data(data)