/*
 * Conky, a system monitor, based on torsmo
 *
 * Any original torsmo code is licensed under the BSD license
 *
 * All code written since the fork of torsmo is licensed under the GPL
 *
 * Please see COPYING for details
 *
 * Copyright (c) 2005-2007 Brenden Matthews, Philip Kovacs, et. al. (see AUTHORS)
 * All rights reserved.
 *
 * 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 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 General Public License for more details.
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>. 
 *
 *  $Id$
 */

#include "conky.h"
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <xmmsclient/xmmsclient.h>

#define CONN_INIT    0
#define CONN_OK      1
#define CONN_NO      2

/* callbacks */

static void xmms_alloc(struct information *ptr)
{
	if (ptr->xmms2.status == NULL) {
		ptr->xmms2.status = malloc(TEXT_BUFFER_SIZE);
		ptr->xmms2.status[0] = '\0';
	}

	if (ptr->xmms2.artist == NULL) {
		ptr->xmms2.artist = malloc(TEXT_BUFFER_SIZE);
		ptr->xmms2.artist[0] = '\0';
	}

	if (ptr->xmms2.album == NULL) {
		ptr->xmms2.album = malloc(TEXT_BUFFER_SIZE);
		ptr->xmms2.album[0] = '\0';
	}

	if (ptr->xmms2.title == NULL) {
		ptr->xmms2.title = malloc(TEXT_BUFFER_SIZE);
		ptr->xmms2.title[0] = '\0';
	}

	if (ptr->xmms2.genre == NULL) {
		ptr->xmms2.genre = malloc(TEXT_BUFFER_SIZE);
		ptr->xmms2.genre[0] = '\0';
	}

	if (ptr->xmms2.comment == NULL) {
		ptr->xmms2.comment = malloc(TEXT_BUFFER_SIZE);
		ptr->xmms2.comment[0] = '\0';
	}

	if (ptr->xmms2.decoder == NULL) {
		ptr->xmms2.decoder = malloc(TEXT_BUFFER_SIZE);
		ptr->xmms2.decoder[0] = '\0';
	}

	if (ptr->xmms2.transport == NULL) {
		ptr->xmms2.transport = malloc(TEXT_BUFFER_SIZE);
		ptr->xmms2.transport[0] = '\0';
	}

	if (ptr->xmms2.url == NULL) {
		ptr->xmms2.url = malloc(TEXT_BUFFER_SIZE);
		ptr->xmms2.url[0] = '\0';
	}

	if (ptr->xmms2.date == NULL) {
		ptr->xmms2.date = malloc(TEXT_BUFFER_SIZE);
		ptr->xmms2.date[0] = '\0';
	}
}

static void xmms_clear(struct information *ptr) {
	xmms_alloc(ptr);
	ptr->xmms2.status[0] = '\0';
	ptr->xmms2.artist[0] = '\0';
	ptr->xmms2.album[0] = '\0';
	ptr->xmms2.title[0] = '\0';
	ptr->xmms2.genre[0] = '\0';
	ptr->xmms2.comment[0] = '\0';
	ptr->xmms2.decoder[0] = '\0';
	ptr->xmms2.transport[0] = '\0';
	ptr->xmms2.url[0] = '\0';
	ptr->xmms2.date[0] = '\0';
}

void connection_lost(void *p)
{
	struct information *ptr = p;
	ptr->xmms2_conn_state = CONN_NO;

	xmms_clear(ptr);
	ptr->xmms2.tracknr = 0;
	ptr->xmms2.id = 0;
	ptr->xmms2.bitrate = 0;
	ptr->xmms2.duration = 0;
	ptr->xmms2.elapsed = 0;
	ptr->xmms2.size = 0;
	ptr->xmms2.progress = 0;
}

void handle_curent_id(xmmsc_result_t *res, void *p)
{
    uint current_id;
    struct information *ptr = p;

    if ( xmmsc_result_get_uint( res, &current_id ) ) {

        xmmsc_result_t *res2;
        res2 = xmmsc_medialib_get_info(ptr->xmms2_conn, current_id);
        xmmsc_result_wait( res2 );
		
       	xmms_clear(ptr);

        ptr->xmms2.id = current_id;

        char *temp;
        xmmsc_result_get_dict_entry_string( res2, "artist", &temp );
        if ( temp != NULL ) {
            strncpy(ptr->xmms2.artist, temp, TEXT_BUFFER_SIZE - 1);
        } else {
            strncpy(ptr->xmms2.artist, "[Unknown]", TEXT_BUFFER_SIZE - 1);
        }


        xmmsc_result_get_dict_entry_string( res2, "title", &temp );
        if ( temp != NULL ) {
            strncpy(ptr->xmms2.title, temp, TEXT_BUFFER_SIZE - 1);
        } else {
            strncpy(ptr->xmms2.title, "[Unknown]", TEXT_BUFFER_SIZE - 1);
        }

        xmmsc_result_get_dict_entry_string( res2, "album", &temp );
        if ( temp != NULL ) {
            strncpy(ptr->xmms2.album, temp, TEXT_BUFFER_SIZE - 1);
        } else {
            strncpy(ptr->xmms2.album, "[Unknown]", TEXT_BUFFER_SIZE - 1);
        }


        xmmsc_result_get_dict_entry_string( res2, "genre", &temp );
        if ( temp != NULL ) {

            strncpy(ptr->xmms2.genre, temp, TEXT_BUFFER_SIZE - 1);
        } else {
            strncpy(ptr->xmms2.genre, "[Unknown]", TEXT_BUFFER_SIZE - 1);
        }


        xmmsc_result_get_dict_entry_string( res2, "comment", &temp );
        if ( temp != NULL ) {
            strncpy(ptr->xmms2.comment, temp, TEXT_BUFFER_SIZE - 1);
        } else {
            strncpy(ptr->xmms2.comment, "", TEXT_BUFFER_SIZE - 1);
        }


        xmmsc_result_get_dict_entry_string( res2, "decoder", &temp );
        if ( temp != NULL ) {
            strncpy(ptr->xmms2.decoder, temp, TEXT_BUFFER_SIZE - 1);
        } else {
            strncpy(ptr->xmms2.decoder, "[Unknown]", TEXT_BUFFER_SIZE - 1);
        }


        xmmsc_result_get_dict_entry_string( res2, "transport", &temp );
        if ( temp != NULL ) {
            strncpy(ptr->xmms2.transport, temp, TEXT_BUFFER_SIZE - 1);
        } else {
            strncpy(ptr->xmms2.transport, "[Unknown]", TEXT_BUFFER_SIZE - 1);
        }


        xmmsc_result_get_dict_entry_string( res2, "url", &temp );
        if ( temp != NULL ) {
            strncpy(ptr->xmms2.url, temp, TEXT_BUFFER_SIZE - 1);
        } else {
            strncpy(ptr->xmms2.url, "[Unknown]", TEXT_BUFFER_SIZE - 1);
        }


        xmmsc_result_get_dict_entry_string( res2, "date", &temp );
        if ( temp != NULL ) {
            strncpy(ptr->xmms2.date, temp, TEXT_BUFFER_SIZE - 1);
        } else {
            strncpy(ptr->xmms2.date, "????", TEXT_BUFFER_SIZE - 1);
        }


        int itemp;
        xmmsc_result_get_dict_entry_int( res2, "tracknr", &itemp );
        ptr->xmms2.tracknr = itemp;

        xmmsc_result_get_dict_entry_int( res2, "duration", &itemp );
        ptr->xmms2.duration = itemp;

        xmmsc_result_get_dict_entry_int( res2, "bitrate", &itemp );
        ptr->xmms2.bitrate = itemp / 1000;

        xmmsc_result_get_dict_entry_int( res2, "size", &itemp );
        ptr->xmms2.size = (float)itemp / 1048576;

        xmmsc_result_unref( res2 );
    }
}

void handle_playtime(xmmsc_result_t *res, void *p) {
	struct information *ptr = p;
    xmmsc_result_t * res2;
    uint play_time;

    if ( xmmsc_result_iserror( res ) )
        return;

    if ( !xmmsc_result_get_uint( res, &play_time ) )
        return;

    res2 = xmmsc_result_restart( res );
    xmmsc_result_unref( res2 );

    ptr->xmms2.elapsed = play_time;
    ptr->xmms2.progress = (float) play_time / ptr->xmms2.duration;
}

void handle_playback_state_change(xmmsc_result_t *res, void *p) {
	struct information *ptr = p;
    uint pb_state = 0;
    if ( xmmsc_result_iserror( res ) )
        return;

    if ( !xmmsc_result_get_uint( res, &pb_state ) )
        return;

    switch (pb_state) {
    case XMMS_PLAYBACK_STATUS_PLAY:
        strncpy(ptr->xmms2.status, "Playing", TEXT_BUFFER_SIZE - 1);
        break;
    case XMMS_PLAYBACK_STATUS_PAUSE:
        strncpy(ptr->xmms2.status, "Paused", TEXT_BUFFER_SIZE - 1);
        break;
    case XMMS_PLAYBACK_STATUS_STOP:
        strncpy(ptr->xmms2.status, "Stopped", TEXT_BUFFER_SIZE - 1);
        break;
    default:
        strncpy(ptr->xmms2.status, "Unknown", TEXT_BUFFER_SIZE - 1);
    }
}


void update_xmms2() {
    struct information * current_info = &info;

    /* initialize connection */
    if ( current_info->xmms2_conn_state == CONN_INIT ) {

        if ( current_info->xmms2_conn == NULL ) {
            current_info->xmms2_conn = xmmsc_init( "conky" );
        }

        /* did init fail? */
        if ( current_info->xmms2_conn == NULL ) {
            fprintf(stderr,"Conky: xmms2 init failed. %s\n", xmmsc_get_last_error ( current_info->xmms2_conn ));
            fflush(stderr);
            return;
        }

        /* init ok but not connected yet.. */
        current_info->xmms2_conn_state = CONN_NO;

        /* clear all values */
		xmms_clear(current_info);

        current_info->xmms2.tracknr = 0;
        current_info->xmms2.id = 0;
        current_info->xmms2.bitrate = 0;
        current_info->xmms2.duration = 0;
        current_info->xmms2.elapsed = 0;
        current_info->xmms2.size = 0;
        current_info->xmms2.progress = 0;

        /*    fprintf(stderr,"Conky: xmms2 init ok.\n");
            fflush(stderr); */
    }

    /* connect */
    if ( current_info->xmms2_conn_state == CONN_NO ) {

        char *path = getenv ( "XMMS_PATH" );
        if ( !xmmsc_connect( current_info->xmms2_conn, path ) ) {
            fprintf(stderr,"Conky: xmms2 connection failed. %s\n",
                    xmmsc_get_last_error ( current_info->xmms2_conn ));
            fflush(stderr);
            current_info->xmms2_conn_state = CONN_NO;
            return;
        }

        /* set callbacks */
        xmmsc_disconnect_callback_set( current_info->xmms2_conn, connection_lost, current_info );
        XMMS_CALLBACK_SET( current_info->xmms2_conn, xmmsc_playback_current_id, handle_curent_id, current_info );
        XMMS_CALLBACK_SET( current_info->xmms2_conn, xmmsc_broadcast_playback_current_id, handle_curent_id, current_info );
        XMMS_CALLBACK_SET( current_info->xmms2_conn, xmmsc_signal_playback_playtime, handle_playtime, current_info );
        XMMS_CALLBACK_SET( current_info->xmms2_conn, xmmsc_broadcast_playback_status, handle_playback_state_change, current_info );

	/* get playback status, it wont be broadcasted untill it chages */
	xmmsc_result_t * res = xmmsc_playback_status( current_info->xmms2_conn );
        xmmsc_result_wait ( res );
        unsigned int pb_state;

        xmmsc_result_get_uint( res, &pb_state );
        switch (pb_state) {
        case XMMS_PLAYBACK_STATUS_PLAY:
            strncpy(current_info->xmms2.status,
                     "Playing", TEXT_BUFFER_SIZE - 1 );
            break;
        case XMMS_PLAYBACK_STATUS_PAUSE:
            strncpy( current_info->xmms2.status,
                     "Paused", TEXT_BUFFER_SIZE - 1 );
            break;
        case XMMS_PLAYBACK_STATUS_STOP:
            strncpy( current_info->xmms2.status,
                     "Stopped", TEXT_BUFFER_SIZE - 1 );
            break;
        default:
            strncpy( current_info->xmms2.status,
                     "Unknown", TEXT_BUFFER_SIZE - 1 );
        }
        xmmsc_result_unref ( res );
	
        /* everything seems to be ok */
        current_info->xmms2_conn_state = CONN_OK;

        /*   fprintf(stderr,"Conky: xmms2 connected.\n");
              fflush(stderr);  */
    }


    /* handle callbacks */
    if ( current_info->xmms2_conn_state == CONN_OK ) {
        struct timeval tmout;
        tmout.tv_sec = 0;
        tmout.tv_usec = 100;

        select( current_info->xmms2_fd + 1, &current_info->xmms2_fdset, NULL, NULL, &tmout );

        xmmsc_io_in_handle(current_info->xmms2_conn);
        if (xmmsc_io_want_out(current_info->xmms2_conn)) {
            xmmsc_io_out_handle(current_info->xmms2_conn);
        }
    }
}