/* $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.1 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 Lesser 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.
***/

#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#include <avahi-core/core.h>

#define DBUS_API_SUBJECT_TO_CHANGE
#include <dbus/dbus.h>
#include <dbus/dbus-glib-lowlevel.h>

#include "main.h"
#include "simple-protocol.h"
#include "static-services.h"

#define DBUS_SERVICE_AVAHI "org.freedesktop.Avahi"

AvahiServer *avahi_server = NULL;

static DBusHandlerResult
do_register (DBusConnection *conn, DBusMessage *message)
{
    DBusError error;
    char *s;

    dbus_error_init (&error);

    dbus_message_get_args (message, &error,
                           DBUS_TYPE_STRING, &s,
                           DBUS_TYPE_INVALID);

    if (dbus_error_is_set (&error))
    {
        g_warning ("Error parsing register attempt");
        dbus_error_free (&error);

        return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
    }

    g_message ("Register received from: %s", s);

    return DBUS_HANDLER_RESULT_HANDLED;
}

static DBusHandlerResult
signal_filter (DBusConnection *conn, DBusMessage *message, void *user_data)
{
    GMainLoop *loop = user_data;
    DBusError error;

    dbus_error_init (&error);

    g_message ("dbus: interface=%s, path=%s, member=%s",
               dbus_message_get_interface (message),
               dbus_message_get_path (message),
               dbus_message_get_member (message));

    if (dbus_message_is_signal (message,
                                DBUS_INTERFACE_ORG_FREEDESKTOP_LOCAL,
                                "Disconnected"))
    {
        /* No, we shouldn't quit, but until we get somewhere
         * usefull such that we can restore our state, we will */
        g_warning ("Disconnnected from d-bus");

        g_main_loop_quit (loop);
        return DBUS_HANDLER_RESULT_HANDLED;
    } else if (dbus_message_is_method_call (message, DBUS_SERVICE_AVAHI,
                                            "Register"))
    {
        return do_register (conn, message);
    } else if (dbus_message_is_signal (message,
                                       DBUS_INTERFACE_ORG_FREEDESKTOP_DBUS,
                                       "ServiceAcquired"))
    {
        char *name;

        dbus_message_get_args (message, &error,
                               DBUS_TYPE_STRING, &name,
                               DBUS_TYPE_INVALID);

        if (dbus_error_is_set (&error))
        {
            g_warning ("Error parsing NameAcquired message");
            dbus_error_free (&error);

            return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
        }

        g_message ("dbus: ServiceAcquired (%s)", name);

        return DBUS_HANDLER_RESULT_HANDLED;
    }

    g_message ("dbus: missed event");

    return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
}

static void server_callback(AvahiServer *s, AvahiServerState state, gpointer userdata) {
    g_assert(s);

    if (state == AVAHI_SERVER_RUNNING) {
        g_message("Server startup complete.  Host name is <%s>", avahi_server_get_host_name_fqdn(s));
        static_service_add_to_server();
    } else if (state == AVAHI_SERVER_COLLISION) {
        gchar *n;

        static_service_remove_from_server();
        
        n = avahi_alternative_host_name(avahi_server_get_host_name(s));
        g_message("Host name conflict, retrying with <%s>", n);
        avahi_server_set_host_name(s, n);
        g_free(n);
    }
}

int main(int argc, char *argv[]) {
    GMainLoop *loop = NULL;
    DBusConnection *bus = NULL;
    DBusError error;
    gint r = 255;
    AvahiServerConfig config;

    avahi_server_config_init(&config);

    loop = g_main_loop_new(NULL, FALSE);

    dbus_error_init (&error);

    bus = dbus_bus_get (DBUS_BUS_SYSTEM, &error);

    if (bus == NULL)
    {
        g_warning ("dbus_bus_get(): %s", error.message);
        dbus_error_free (&error);

        goto finish;
    }

    dbus_connection_setup_with_g_main (bus, NULL);
    dbus_connection_set_exit_on_disconnect (bus, FALSE);

    dbus_bus_acquire_service (bus, DBUS_SERVICE_AVAHI, 0, &error);

    if (dbus_error_is_set (&error))
    {
        g_warning ("dbus_error_is_set (): %s", error.message);
        dbus_error_free (&error);

        goto finish;
    }

    dbus_connection_add_filter (bus, signal_filter, loop, NULL);
    dbus_bus_add_match (bus,
                        "type='method_call',interface='org.freedesktop.Avahi'",
                        &error);

    if (dbus_error_is_set (&error))
    {
        g_warning ("dbus_bus_add_match (): %s", error.message);
        dbus_error_free (&error);

        goto finish;
    }

    if (simple_protocol_setup(NULL) < 0)
        goto finish;

    if (!(avahi_server = avahi_server_new(NULL, &config, server_callback, NULL)))
        goto finish;

    static_service_load();

    g_main_loop_run(loop);

    r = 0;
    
finish:

    static_service_remove_from_server();
    static_service_free_all();
    
    simple_protocol_shutdown();

    if (bus) {
        dbus_connection_disconnect(bus);
        dbus_connection_unref(bus);
    }

    if (avahi_server)
        avahi_server_free(avahi_server);
    
    if (loop)
        g_main_loop_unref(loop);

    avahi_server_config_free(&config);

    return r;
}