/* 
 * $Id$
 *
 * (c) 2004 by Kalle Wallin <kaw@linux.se>
 *
 * 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *
 */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <ncurses.h>
#include <glib.h>

#include "config.h"
#include "ncmpc.h"
#include "options.h"
#include "support.h"
#include "colors.h"

#define COLOR_BRIGHT_MASK   (1<<7)

#define COLOR_BRIGHT_BLACK    (COLOR_BLACK | COLOR_BRIGHT_MASK)
#define COLOR_BRIGHT_RED      (COLOR_RED   | COLOR_BRIGHT_MASK)
#define COLOR_BRIGHT_GREEN    (COLOR_GREEN | COLOR_BRIGHT_MASK)
#define COLOR_BRIGHT_YELLOW   (COLOR_YELLOW | COLOR_BRIGHT_MASK)
#define COLOR_BRIGHT_BLUE     (COLOR_BLUE | COLOR_BRIGHT_MASK)
#define COLOR_BRIGHT_MAGENTA  (COLOR_MAGENTA | COLOR_BRIGHT_MASK)
#define COLOR_BRIGHT_CYAN     (COLOR_CYAN | COLOR_BRIGHT_MASK)
#define COLOR_BRIGHT_WHITE    (COLOR_WHITE | COLOR_BRIGHT_MASK)

#define IS_BRIGHT(color) (color & COLOR_BRIGHT_MASK)

/* name of the color fields */
#define NAME_TITLE        "title"
#define NAME_TITLE_BOLD   "title-bold"
#define NAME_LINE         "line"
#define NAME_LINE_BOLD    "line-flags"
#define NAME_LIST         "list"
#define NAME_LIST_BOLD    "list-bold"
#define NAME_PROGRESS     "progressbar"
#define NAME_STATUS       "status-song"
#define NAME_STATUS_BOLD  "status-state"
#define NAME_STATUS_TIME  "status-time"
#define NAME_ALERT        "alert"
#define NAME_BGCOLOR      "background"

typedef struct {
	short color;
	short r,g,b;
} color_definition_entry_t;

typedef struct {
	int id;
	const char *name;
	short fg;
	attr_t attrs;
} color_entry_t;

static color_entry_t colors[] = {
	/* color pair,        field name,        color,            mono attribute */
	/*-------------------------------------------------------------------------*/
	{ COLOR_TITLE,        NAME_TITLE,        COLOR_YELLOW,         A_NORMAL },
	{ COLOR_TITLE_BOLD,   NAME_TITLE_BOLD,   COLOR_BRIGHT_YELLOW,  A_BOLD },
	{ COLOR_LINE,         NAME_LINE,         COLOR_WHITE,          A_NORMAL },
	{ COLOR_LINE_BOLD,    NAME_LINE_BOLD,    COLOR_BRIGHT_WHITE,   A_BOLD },
	{ COLOR_LIST,         NAME_LIST,         COLOR_GREEN,          A_NORMAL },
	{ COLOR_LIST_BOLD,    NAME_LIST_BOLD,    COLOR_BRIGHT_GREEN,   A_BOLD },
	{ COLOR_PROGRESSBAR,  NAME_PROGRESS,     COLOR_WHITE,          A_NORMAL },
	{ COLOR_STATUS,       NAME_STATUS,       COLOR_YELLOW,         A_NORMAL },
	{ COLOR_STATUS_BOLD,  NAME_STATUS_BOLD,  COLOR_BRIGHT_YELLOW,  A_BOLD },
	{ COLOR_STATUS_TIME,  NAME_STATUS_TIME,  COLOR_RED,            A_NORMAL },
	{ COLOR_STATUS_ALERT, NAME_ALERT,        COLOR_BRIGHT_RED,     A_BOLD },
	{ 0,                  NULL,              0,                    0 }
};

/* background color */
static short bg = COLOR_BLACK;

static GList *color_definition_list = NULL;

static color_entry_t *
colors_lookup(int id)
{
  int i;

  i=0;
  while( colors[i].name != NULL )
    {
      if( colors[i].id == id )
	return &colors[i];
      i++;
    }
  return NULL;
}

static color_entry_t *
colors_lookup_by_name(const char *name)
{
  int i;

  i=0;
  while( colors[i].name != NULL )
    {
      if( !strcasecmp(colors[i].name, name) )
	return &colors[i];
      i++;
    }
  return NULL;
}

static int
colors_update_pair(int id)
{
  color_entry_t *entry = colors_lookup(id);
  short fg = -1;

  if( !entry )
    return -1;

  if( IS_BRIGHT(entry->fg) )
    {
      entry->attrs = A_BOLD;
      fg = entry->fg & ~COLOR_BRIGHT_MASK;
    }
  else
    {
      entry->attrs = A_NORMAL;
      fg = entry->fg;
    }

  init_pair(entry->id, fg, bg);
  
  return 0;
}

short
colors_str2color(const char *str)
{
  if( !strcasecmp(str,"black") )
    return COLOR_BLACK;
  else if( !strcasecmp(str,"red") )
    return COLOR_RED;
  else if( !strcasecmp(str,"green") )
    return COLOR_GREEN;
  else if( !strcasecmp(str,"yellow") )
    return COLOR_YELLOW;
  else if( !strcasecmp(str,"blue") )
    return COLOR_BLUE;
  else if( !strcasecmp(str,"magenta") )
    return COLOR_MAGENTA;
  else if( !strcasecmp(str,"cyan") )
    return COLOR_CYAN;
  else if( !strcasecmp(str,"white") )
    return COLOR_WHITE;
  else if( !strcasecmp(str,"brightred") )
    return COLOR_BRIGHT_RED;
  else if( !strcasecmp(str,"brightgreen") )
    return COLOR_BRIGHT_GREEN;
  else if( !strcasecmp(str,"brightyellow") )
    return COLOR_BRIGHT_YELLOW;
  else if( !strcasecmp(str,"brightblue") )
    return COLOR_BRIGHT_BLUE;
  else if( !strcasecmp(str,"brightmagenta") )
    return COLOR_BRIGHT_MAGENTA;
  else if( !strcasecmp(str,"brightcyan") )
    return COLOR_BRIGHT_CYAN;
  else if( !strcasecmp(str,"brightwhite") )
    return COLOR_BRIGHT_WHITE;
  else if( !strcasecmp(str,"grey") || !strcasecmp(str,"gray") )
    return COLOR_BRIGHT_BLACK;
  else if( !strcasecmp(str,"none") )
    return -1;
  fprintf(stderr,_("Warning: Unknown color - %s\n"), str);
  return -2;
}

/* This function is called from conf.c before curses have been started,
 * it adds the definition to the color_definition_list and init_color() is 
 * done in colors_start() */
int
colors_define(const char *name, short r, short g, short b)
{
  color_definition_entry_t *entry;
  short color = colors_str2color(name);

  if( color<0 )
    return -1;

  entry = g_malloc(sizeof(color_definition_entry_t));
  entry->color = color;
  entry->r = r;
  entry->g = g;
  entry->b = b;

  color_definition_list = g_list_append(color_definition_list, entry);

  return 0;
}


int
colors_assign(const char *name, const char *value)
{
  color_entry_t *entry = colors_lookup_by_name(name);
  short color;

  if( !entry )
    {
      if( !strcasecmp(NAME_BGCOLOR, name) )
	{
	  if( (color=colors_str2color(value)) < -1 )
	    return -1;
	  if( color > COLORS )
	    color = color & ~COLOR_BRIGHT_MASK;
	  bg = color;
	  return 0;
	}
      fprintf(stderr,_("Warning: Unknown color field - %s\n"), name);
      return -1;
    }

  if( (color=colors_str2color(value)) < -1 )
    return -1;
  entry->fg = color;

  return 0;
}


int
colors_start(void)
{
  if( has_colors() )
    {
      /* initialize color support */
      start_color();
      use_default_colors();
      /* define any custom colors defined in the configuration file */
      if( color_definition_list && can_change_color() )
	{
	  GList *list = color_definition_list;

	  while( list )
	    {
	      color_definition_entry_t *entry = list->data;

	      if( entry->color <= COLORS )
		init_color(entry->color, entry->r, entry->g, entry->b);
	      list=list->next;
	    }
	}
      else if( color_definition_list && !can_change_color() )
	fprintf(stderr, _("Terminal lacks support for changing colors!\n"));

      if( options.enable_colors )
	{
	  int i = 0;

	  while(colors[i].name)
	    {
	      /* update the color pairs */
	      colors_update_pair(colors[i].id);
	      i++;
	    }

	}
    }
  else if( options.enable_colors )
    {
      fprintf(stderr, _("Terminal lacks color capabilities!\n"));
      options.enable_colors = 0;
    }

  /* free the color_definition_list */
  if( color_definition_list )
    {
      GList *list = color_definition_list;

      while( list )
	{
	  g_free(list->data);
	  list=list->next;
	}
      g_list_free(color_definition_list);
      color_definition_list = NULL;
    }

  return 0;
}


int
colors_use(WINDOW *w, int id)
{
  color_entry_t *entry = colors_lookup(id);
  short pair;
  attr_t attrs;

  if( !entry )
    return -1;

  wattr_get(w, &attrs, &pair, NULL);

  if( options.enable_colors )
    {
      /* color mode */
      if( attrs != entry->attrs || id!=pair )
	wattr_set(w, entry->attrs, id, NULL);
    }
  else
    {
      /* mono mode */
      if( attrs != entry->attrs )
	wattrset(w, entry->attrs);
    }
  
  return 0;
}