#include <e.h>
#include "e_mod_main.h"
//#include "e_mod_config.h"
#include "config.h"

static Cpu *_cpu_init                   (E_Module *m);
static void _cpu_shutdown               (Cpu *n);
static void _cpu_config_menu_new        (Cpu *n);
static int  _cpu_face_init              (Cpu_Face *cf);
static void _cpu_face_menu_new          (Cpu_Face *cf);
static void _cpu_face_enable            (Cpu_Face *cf);
static void _cpu_face_disable           (Cpu_Face *cf);
static void _cpu_face_free              (Cpu_Face *cf);
static void _cpu_face_cb_gmc_change     (void *data, E_Gadman_Client *gmc, E_Gadman_Change change);
static void _cpu_face_cb_mouse_down     (void *data, Evas *evas, Evas_Object *obj, void *event_info);
static void _cpu_face_cb_menu_edit      (void *data, E_Menu *mn, E_Menu_Item *mi);
static void _cpu_face_cb_menu_configure (void *data, E_Menu *mn, E_Menu_Item *mi);
static int  _cpu_face_update_values     (void *data);
static int  _cpu_face_get_cpu_count     (Cpu_Face *cf); 
static int  _cpu_face_get_load          (Cpu_Face *cf); 

static int cpu_count;

EAPI E_Module_Api e_modapi = 
{
   E_MODULE_API_VERSION,
     "Cpu"
};

EAPI void *
e_modapi_init(E_Module *m) 
{
   Cpu *c;
   
   c = _cpu_init(m);
   if (!c)
     return NULL;
   
   m->config_menu = c->config_menu;
   return c;
}

EAPI int
e_modapi_shutdown(E_Module *m) 
{
   Cpu *c;
   
   c = m->data;
   if (!c)
     return 0;
   
   if (m->config_menu) 
     {
	e_menu_deactivate(m->config_menu);
	e_object_del(E_OBJECT(m->config_menu));
	m->config_menu = NULL;
     }
   if (c->cfd) 
     {
	e_object_del(E_OBJECT(c->cfd));
	c->cfd = NULL;
     }
   _cpu_shutdown(c);
   return 1;
}

EAPI int
e_modapi_save(E_Module *m) 
{
   Cpu *c;
   
   c = m->data;
   if (!c)
     return 0;
   e_config_domain_save("module.cpu", c->conf_edd, c->conf);
   return 1;
}

EAPI int
e_modapi_info(E_Module *m) 
{
   m->icon_file = strdup(PACKAGE_DATA_DIR"/module_icon.png");
   return 1;
}

EAPI int
e_modapi_about(E_Module *m) 
{
   e_module_dialog_show(_("Enlightenment Cpu Monitor Module"),
			_("This module is used to monitor a cpu."));
   return 1;
}

EAPI int
e_modapi_config(E_Module *m) 
{
   Cpu *c;
   E_Container *con;
   
   c = m->data;
   if (!c)
     return 0;
   if (!c->face)
     return 0;
   
   con = e_container_current_get(e_manager_current_get());

/*   if (c->face->con == con)
     _configure_cpu_module(con, c);
*/   
   return 1;
}

static Cpu *
_cpu_init(E_Module *m) 
{
   Cpu *c;
   E_Menu_Item *mi;
   Evas_List *mans, *l, *l2;
   
   c = E_NEW(Cpu, 1);
   if (!c)
     return NULL;
   
   c->conf_edd = E_CONFIG_DD_NEW("Cpu_Config", Config);
   #undef T
   #undef D
   #define T Config
   #define D c->conf_edd
   E_CONFIG_VAL(D, T, check_interval, INT);
   
   c->conf = e_config_domain_load("module.cpu", c->conf_edd);
   if (!c->conf) 
     {
	c->conf = E_NEW(Config, 1);
	c->conf->check_interval = 1;
     }
   E_CONFIG_LIMIT(c->conf->check_interval, 0, 60);
   
   _cpu_config_menu_new(c);
   
   mans = e_manager_list();
   for (l = mans; l; l = l->next) 
     {
	E_Manager *man;
	
	man = l->data;
	for (l2 = man->containers; l2; l2 = l2->next) 
	  {
	     E_Container *con;
	     Cpu_Face *cf;
	     
	     con = l2->data;
	     cf = E_NEW(Cpu_Face, 1);
	     if (cf) 
	       {
		  cf->conf_face_edd = E_CONFIG_DD_NEW("Cpu_Face_Config", Config_Face);
		  #undef T
		  #undef D
		  #define T Config_Face
		  #define D cf->conf_face_edd
		  E_CONFIG_VAL(D, T, enabled, UCHAR);
		  
		  c->face = cf;
		  cf->cpu = c;		  
		  cf->con = con;
		  cf->evas = con->bg_evas;
		  
		  cf->conf = E_NEW(Config_Face, 1);
		  cf->conf->enabled = 1;
		  
		  if (!_cpu_face_init(cf))
		    return NULL;
		  
		  _cpu_face_menu_new(cf);
		  
		  mi = e_menu_item_new(c->config_menu);
		  e_menu_item_label_set(mi, _("Configuration"));
		  //e_menu_item_callback_set(mi, _cpu_face_cb_menu_configure, cf);
		  
		  mi = e_menu_item_new(c->config_menu);
		  e_menu_item_label_set(mi, con->name);
		  e_menu_item_submenu_set(mi, cf->menu);
		 
		  if (!cf->conf->enabled)
		    _cpu_face_disable(cf);
		  else
		    _cpu_face_enable(cf);
	       }
	  }
     }
   return c;
}

static void
_cpu_shutdown(Cpu *c) 
{
   _cpu_face_free(c->face);
   
   E_FREE(c->conf);
   E_CONFIG_DD_FREE(c->conf_edd);
   E_FREE(c);
}

static void
_cpu_config_menu_new(Cpu *c) 
{
   E_Menu *mn;
   
   mn = e_menu_new();
   c->config_menu = mn;
}

static int
_cpu_face_init(Cpu_Face *cf) 
{
   Evas_Object *o;
   char buf[4096];
   
   evas_event_freeze(cf->evas);
   
   o = edje_object_add(cf->evas);
   cf->cpu_obj = o;
   
   if (!e_theme_edje_object_set(o, "base/theme/modules/cpu", 
				"modules/cpu/main")) 
     {
	snprintf(buf, sizeof(buf), PACKAGE_DATA_DIR"/cpu.edj");	
	edje_object_file_set(o, buf, "modules/cpu/main");
     }
   
   evas_object_show(o);

   /*
   o = evas_object_rectangle_add(cf->evas);
   cf->chart_obj = o;
   evas_object_layer_set(o, 2);
   evas_object_repeat_events_set(o, 0);
   evas_object_color_set(o, 0, 0, 0, 255);
   evas_object_show(o);
   */
   
   o = evas_object_rectangle_add(cf->evas);
   cf->event_obj = o;
   evas_object_layer_set(o, 2);
   evas_object_repeat_events_set(o, 1);
   evas_object_color_set(o, 0, 0, 0, 0);
   evas_object_event_callback_add(o, EVAS_CALLBACK_MOUSE_DOWN,
				  _cpu_face_cb_mouse_down, cf);
   evas_object_show(o);
   
   cf->gmc = e_gadman_client_new(cf->con->gadman);
   e_gadman_client_domain_set(cf->gmc, "module.cpu", cpu_count++);
   e_gadman_client_policy_set(cf->gmc,
			      E_GADMAN_POLICY_ANYWHERE |
			      E_GADMAN_POLICY_HMOVE |
			      E_GADMAN_POLICY_HSIZE |
			      E_GADMAN_POLICY_VMOVE |
			      E_GADMAN_POLICY_VSIZE);
   e_gadman_client_min_size_set(cf->gmc, 4, 4);
   e_gadman_client_max_size_set(cf->gmc, 128, 128);
   e_gadman_client_auto_size_set(cf->gmc, 40, 40);
   e_gadman_client_align_set(cf->gmc, 1.0, 1.0);
   e_gadman_client_aspect_set(cf->gmc, 1.0, 1.0);
   e_gadman_client_resize(cf->gmc, 40, 40);
   e_gadman_client_change_func_set(cf->gmc, _cpu_face_cb_gmc_change, cf);
   e_gadman_client_load(cf->gmc);
   evas_event_thaw(cf->evas);

   cf->monitor = ecore_timer_add((double)cf->cpu->conf->check_interval, _cpu_face_update_values, cf);
   return 1;
}

static void
_cpu_face_menu_new(Cpu_Face *cf) 
{
   E_Menu *mn;
   E_Menu_Item *mi;
   
   mn = e_menu_new();
   cf->menu = mn;
   
   mi = e_menu_item_new(mn);
   e_menu_item_label_set(mi, _("Configuration"));
   //e_menu_item_callback_set(mi, _cpu_face_cb_menu_configure, cf);
   
   mi = e_menu_item_new(mn);
   e_menu_item_label_set(mi, _("Edit Mode"));
   e_menu_item_callback_set(mi, _cpu_face_cb_menu_edit, cf);
}

static void
_cpu_face_enable(Cpu_Face *cf) 
{
   cf->conf->enabled = 1;
   e_config_save_queue();
   evas_object_show(cf->cpu_obj);
   /* evas_object_show(cf->chart_obj); */
   evas_object_show(cf->event_obj);
}

static void
_cpu_face_disable(Cpu_Face *cf) 
{
   cf->conf->enabled = 0;
   e_config_save_queue();
   evas_object_hide(cf->event_obj);
   /* evas_object_hide(cf->chart_obj); */
   evas_object_hide(cf->cpu_obj);
}

static void 
_cpu_face_free(Cpu_Face *cf) 
{
   if (cf->monitor)
     ecore_timer_del(cf->monitor);
   if (cf->menu)
     e_object_del(E_OBJECT(cf->menu));
   if (cf->event_obj)
     evas_object_del(cf->event_obj);
   if (cf->cpu_obj)
     evas_object_del(cf->cpu_obj);

   /*
   if (cf->chart_obj)
     evas_object_del(cf->chart_obj);
    */
   
   if (cf->gmc) 
     {
	e_gadman_client_save(cf->gmc);
	e_object_del(E_OBJECT(cf->gmc));
     }
   
   E_FREE(cf->conf);
   E_FREE(cf);
   cpu_count--;
}

static void 
_cpu_face_cb_gmc_change(void *data, E_Gadman_Client *gmc, E_Gadman_Change change) 
{
   Cpu_Face *cf;
   Evas_Coord x, y, w, h;
   
   cf = data;
   switch (change) 
     {
      case E_GADMAN_CHANGE_MOVE_RESIZE:
	e_gadman_client_geometry_get(cf->gmc, &x, &y, &w, &h);
	evas_object_move(cf->cpu_obj, x, y);
	/* evas_object_move(cf->chart_obj, x, y); */
	evas_object_move(cf->event_obj, x, y);
	evas_object_resize(cf->cpu_obj, w, h);
	/* evas_object_resize(cf->chart_obj, w, h); */
	evas_object_resize(cf->event_obj, w, h);
	break;
      case E_GADMAN_CHANGE_RAISE:
	evas_object_raise(cf->cpu_obj);
	/* evas_object_raise(cf->chart_obj); */
	evas_object_raise(cf->event_obj);
	break;
      default:
	break;
     }   
}

static void 
_cpu_face_cb_mouse_down(void *data, Evas *evas, Evas_Object *obj, void *event_info) 
{
   Cpu_Face *cf;
   Evas_Event_Mouse_Down *ev;
   
   ev = event_info;
   cf = data;
   if (ev->button == 3) 
     {
	e_menu_activate_mouse(cf->menu, e_zone_current_get(cf->con),
			      ev->output.x, ev->output.y, 1, 1,
			      E_MENU_POP_DIRECTION_DOWN, ev->timestamp);
	e_util_container_fake_mouse_up_all_later(cf->con);
     }
}

static void 
_cpu_face_cb_menu_edit(void *data, E_Menu *mn, E_Menu_Item *mi) 
{
   Cpu_Face *cf;
   
   cf = data;
   e_gadman_mode_set(cf->gmc->gadman, E_GADMAN_MODE_EDIT);
}

static void 
_cpu_face_cb_menu_configure(void *data, E_Menu *mn, E_Menu_Item *mi) 
{
   Cpu_Face *cf;

   cf = data;
   //_configure_cpu_module(cf->con, cf->cpu);
}

static int 
_cpu_face_update_values(void *data) 
{
   Cpu_Face *cf;
   int val;
   Edje_Message_String_Set *msg;
   char str[100];
   
   cf = data;
   val = _cpu_face_get_load(cf);
   if (val == -1)
     return 1;

   snprintf(str, sizeof(str), "%d %%", val);
   msg = malloc(sizeof(Edje_Message_String_Set) - sizeof(char *) + (1 + sizeof(char *)));
   msg->count = 1;
   msg->str[0] = str;
   edje_object_message_send(cf->cpu_obj, EDJE_MESSAGE_STRING_SET, 1, msg);
   free(msg);
   return 1;
}

static int
_cpu_face_get_cpu_count(Cpu_Face *cf) 
{
   char tmp[4];
   FILE *f;
   int cpu = -1;
   
   if (!(f = fopen("/proc/stat", "r")))
     return -1;

   while (fscanf(f, "cp%s %*u %*u %*u %*u %*u %*u %*u %*u\n", (char *)&tmp) == 1)
     cpu++;
   
   fclose(f);
   return cpu;   
}

static int
_cpu_face_get_load(Cpu_Face *cf) 
{
   static unsigned long old_u, old_n, old_s, old_i, old_wa, old_hi, old_si;
   unsigned long new_u, new_n, new_s, new_i, new_wa = 0, new_hi = 0, new_si = 0, ticks_past;
   int tmp_u, tmp_n, tmp_s, tmp_i;
   char dummy[16];
   FILE *stat;
   int load, cpu_count;
   
   cpu_count = _cpu_face_get_cpu_count(cf);
   if (cpu_count == -1)
     return -1;

   if (!(stat = fopen("/proc/stat", "r")))
     return -1;

   if (fscanf(stat, "%s %lu %lu %lu %lu %lu %lu %lu", dummy,
              &new_u, &new_n, &new_s, &new_i, &new_wa, &new_hi, &new_si) < 5)
     {
        fclose(stat);
        return -1;
     }

   fclose(stat);

   ticks_past = ((new_u + new_n + new_s + new_i + new_wa + new_hi + new_si) -
                 (old_u + old_n + old_s + old_i + old_wa + old_hi + old_si));

   if (ticks_past)
     {
        tmp_u = ((new_u - old_u));
        tmp_n = ((new_n - old_n));
        tmp_s = ((new_s - old_s));
        tmp_i = ((new_i - old_i));
     }
   else
     {
        tmp_u = 0;
        tmp_n = 0;
        tmp_s = 0;
        tmp_i = 0;
     }

   /* Update the values */   
   load = (tmp_u + tmp_n + tmp_s) / cpu_count;
   
   old_u = new_u;
   old_n = new_n;
   old_s = new_s;
   old_i = new_i;
   old_wa = new_wa;
   old_hi = new_hi;
   old_si = new_si;

   if (load >= 100)
     return 100;   
   return load;   
}