#!@PYTHON@
# -*-python-*-
# $Id$

# This file is part of avahi.
#
# avahi is free software; you can redistribute it and/or modify it
# under the terms of the GNU Lesser General Public License as
# published by the Free Software Foundation; either version 2 of the
# License, or (at your option) any later version.
#
# avahi 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 Lesser General Public
# License along with avahi; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
# USA.

import sys, getopt, string

try:
    import avahi, gobject, dbus
except ImportError:
    print "Sorry, to use this tool you need to install Avahi, pygtk and python-dbus."
    sys.exit(1)

try:
    import dbus.glib
except ImportError, e:
    pass

def usage(retval = 0):
    print "%s <name> <address>" % sys.argv[0]
    sys.exit(retval)

if len(sys.argv) != 3:
    usage(2)

name = sys.argv[1]
address = sys.argv[2]

group = None
n_rename = 0

def remove_address():
    global group
    
    if not (group is None):
        group.Free()
        group = None

def add_address():
    global server, group, name, address
    assert group is None

    print "Adding address '%s' for '%s' ..." % (name, address)
    group = dbus.Interface(bus.get_object(avahi.DBUS_NAME, server.EntryGroupNew()), avahi.DBUS_INTERFACE_ENTRY_GROUP)
    group.connect_to_signal('StateChanged', entry_group_state_changed)
    group.AddAddress(avahi.IF_UNSPEC, avahi.PROTO_UNSPEC, name, address)
    group.Commit()

def entry_group_state_changed(state):
    global name, server, n_rename, main_loop
    
    if state == avahi.ENTRY_GROUP_ESTABLISHED:
        print "Address established."
    elif state == avahi.ENTRY_GROUP_COLLISION:

        n_rename = n_rename + 1
        if n_rename >= 12:
            print "ERROR: No suitable name found after %i retries, exiting." % n_rename
            main_loop.quit()
        else:
            hn = name.split('.')
            hn[0] = server.GetAlternativeHostName(hn[0])
            name = string.join(hn, '.')
            print "WARNING: Address/host name collision, changing name to '%s' ..." % name
            remove_address()
            add_address()

main_loop = gobject.MainLoop()

bus = dbus.SystemBus()
server = dbus.Interface(bus.get_object(avahi.DBUS_NAME, avahi.DBUS_PATH_SERVER), avahi.DBUS_INTERFACE_SERVER)

add_address()

try:
    main_loop.run()
except KeyboardInterrupt, k:
    pass

remove_address()