"""
Network CLI module.

Copyright 2009, Red Hat, Inc
John Eckersberg <jeckersb@redhat.com>

This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 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 General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
02110-1301  USA
"""

import distutils.sysconfig
import sys

plib = distutils.sysconfig.get_python_lib()
mod_path="%s/cobbler" % plib
sys.path.insert(0, mod_path)

from utils import _, _IP, _CIDR
import cobbler.commands as commands
from cexceptions import *


class NetworkFunction(commands.CobblerFunction):

    def help_me(self):
        return commands.HELP_FORMAT % ("cobbler network","<add|copy|edit|find|list|remove|rename|report> [ARGS]")

    def command_name(self):
        return "network"

    def subcommands(self):
        return [ "add", "copy", "dumpvars", "edit", "find", "list", "remove", "rename", "report" ]

    def add_options(self, p, args):
        if not self.matches_args(args,["dumpvars","remove","report","list"]):
            p.add_option("--cidr",             dest="cidr",             help="CIDR representation of the network (REQUIRED)")
            p.add_option("--address",          dest="address",          help="Network address")
            p.add_option("--broadcast",        dest="broadcast",        help="Broadcast address")
            p.add_option("--gateway",          dest="gateway",          help="Gateway address")
            p.add_option("--ns",               dest="ns",               help="comma-delimited list of nameservers")
            p.add_option("--reserved",         dest="reserved",         help="comma-delimited list of IP/CIDR to reserve")
            p.add_option("--comment",          dest="comment",          help="user field")

        p.add_option("--name",                 dest="name",             help="ex: 'vlan001' (REQUIRED)")

        if self.matches_args(args,["add"]):
            p.add_option("--clobber", dest="clobber", help="allow add to overwrite existing objects", action="store_true")
        if self.matches_args(args,["copy","rename"]):
            p.add_option("--newname",          dest="newname",          help="used for copy/edit")
        if not self.matches_args(args,["dumpvars","find","remove","report","list"]):
            p.add_option("--no-sync",     action="store_true", dest="nosync", help="suppress sync for speed")
        if not self.matches_args(args,["dumpvars","find","report","list"]):
            p.add_option("--no-triggers", action="store_true", dest="notriggers", help="suppress trigger execution")
        if not self.matches_args(args,["dumpvars","remove","report","list"]):
            p.add_option("--owners", dest="owners", help="specify owners for authz_ownership module")


    def run(self):
        if self.args and "find" in self.args:
            items = self.api.find_network(return_list=True, no_errors=True, **self.options.__dict__)
            for x in items:
                print x.name
            return True

        obj = self.object_manipulator_start(self.api.new_network,self.api.networks)
        if obj is None:
            return True
        if self.matches_args(self.args,["dumpvars"]):
            return self.object_manipulator_finish(obj, self.api.profiles, self.options)

        if self.options.cidr is not None:
            obj.set_cidr(self.options.cidr)
        elif self.matches_args(self.args, ['add']):
            raise CX(_("cidr is required"))

        if self.options.address is not None:
            obj.set_address(self.options.address)
        elif self.matches_args(self.args, ["add"]):
            obj.set_address(obj.cidr[0])

        if self.options.broadcast is not None:
            obj.set_broadcast(self.options.broadcast)
        elif self.matches_args(self.args, ["add"]):
            obj.set_broadcast(_CIDR(self.options.cidr)[-1])

        if self.options.gateway is not None:
            obj.set_gateway(self.options.gateway)
        elif self.matches_args(self.args, ["add"]):
            obj.set_gateway(_CIDR(self.options.cidr)[-2])

        if self.options.ns is not None:
            obj.set_nameservers(self.options.ns)
        if self.options.reserved is not None:
            obj.set_reserved(self.options.reserved)
        if self.options.owners is not None:
            obj.set_owners(self.options.owners)
        if self.options.comment is not None:
            obj.set_comment(self.options.comment)

        return self.object_manipulator_finish(obj, self.api.networks, self.options)



########################################################
# MODULE HOOKS

def register():
    """
    The mandatory cobbler module registration hook.
    """
    return "cli"

def cli_functions(api):
    return [
       NetworkFunction(api)
    ]
    return []