enlightenment/src/modules/everything-files/e_mod_main.c

/*
 * vim:ts=8:sw=3:sts=8:noexpandtab:cino=>5n-3f0^-2{2
 */

#include "Evry.h"
#include "e_mod_main.h"

#define MAX_ITEMS 50
#define MAX_SHOWN 500
#define TERM_ACTION_DIR "%s"

typedef struct _Plugin Plugin;
typedef struct _Data Data;
typedef struct _Module_Config Module_Config;

struct _Plugin
{
  Evry_Plugin base;

  Eina_List *files;
  const char *directory;
  const char *input;
  Eina_Bool command;
  Eina_Bool parent;
  Eina_List *hist_added;

  Ecore_Thread *thread;
  Eina_Bool cleanup;
};

struct _Data
{
  Plugin *plugin;
  long id;
  int level;
  int cnt;
  Eina_List *files;
  Eina_List *list;
};

struct _Module_Config
{
  int version;

  unsigned char show_homedir;
  unsigned char show_recent;
  unsigned char search_recent;

  // TODO
  int sort_by;
  Eina_List *search_dirs;

  E_Config_Dialog *cfd;
  E_Module *module;
};

static Module_Config *_conf;

static Evry_Plugin *p1 = NULL;
static Evry_Plugin *p2 = NULL;
static Evry_Action *act1 = NULL;
static Evry_Action *act2 = NULL;

static E_Module *module = NULL;

static void _cleanup(Evry_Plugin *plugin);


static void
_item_fill(Evry_Item_File *file)
{
   const char *mime;

   if (file->mime) return;

   if ((mime = efreet_mime_type_get(file->path)))
     {
	file->mime = eina_stringshare_add(mime);
	EVRY_ITEM(file)->context = eina_stringshare_ref(file->mime);

	if ((!strcmp(mime, "inode/directory")) ||
	    (!strcmp(mime, "inode/mount-point")))
	  EVRY_ITEM(file)->browseable = EINA_TRUE;

	return;
     }

   file->mime = eina_stringshare_add("None");
}

static int
_cb_sort(const void *data1, const void *data2)
{
   const Evry_Item *it1 = data1;
   const Evry_Item *it2 = data2;

   if (it1->browseable && !it2->browseable)
     return -1;

   if (!it1->browseable && it2->browseable)
     return 1;

   if (it1->fuzzy_match && it2->fuzzy_match)
     if (it1->fuzzy_match - it2->fuzzy_match)
       return (it1->fuzzy_match - it2->fuzzy_match);

   return strcasecmp(it1->label, it2->label);
}

static void
_item_free(Evry_Item *it)
{
   ITEM_FILE(file, it);
   if (file->path) eina_stringshare_del(file->path);
   if (file->mime) eina_stringshare_del(file->mime);

   E_FREE(file);
}

static void
_scan_func(void *data)
{
   Data *d = data;
   Plugin *p = d->plugin;
   char *filename;
   const char *mime;
   
   Evry_Item_File *file;
   char buf[4096];
   int cnt = 0;

   if (!d->list)
     d->list = ecore_file_ls(p->directory);

   while(d->list)
     {
	filename = d->list->data;
	d->list = eina_list_remove_list(d->list, d->list);

	if (filename[0] == '.')
	  {
	     free(filename);
	     continue;
	  }

	file = E_NEW(Evry_Item_File, 1);
	if (!file)
	  {
	     free(filename);
	     continue;
	  }

	evry_item_new(EVRY_ITEM(file), EVRY_PLUGIN(p), NULL, _item_free);

	EVRY_ITEM(file)->label = filename;

	snprintf(buf, sizeof(buf), "%s/%s", p->directory, filename);
	file->path = strdup(buf);

	if (ecore_file_is_dir(file->path))
	  {
	     file->mime = "inode/directory";
	     EVRY_ITEM(file)->browseable = EINA_TRUE;
	  }
	else if ((mime = efreet_mime_type_get(file->path)))
	  {
	     file->mime = mime;
	  }

	d->files = eina_list_append(d->files, file);

	if (cnt++ > MAX_ITEMS) break;
     }
}

static int
_append_file(Plugin *p, Evry_Item_File *file)
{
   int match;

   if (p->input && (match = evry_fuzzy_match(EVRY_ITEM(file)->label, p->input)))
     {
	EVRY_ITEM(file)->fuzzy_match = match;
	EVRY_PLUGIN_ITEM_APPEND(p, file);
	return 1;
     }
   else if (!p->input)
     {
	EVRY_PLUGIN_ITEM_APPEND(p, file);
	return 1;
     }

   return 0;
}

static void
_scan_cancel_func(void *data)
{
   Data *d = data;
   Plugin *p = d->plugin;
   Evry_Item_File *file;
   char *filename;

   EINA_LIST_FREE(d->files, file)
     {
	free((char *)EVRY_ITEM(file)->label);
	free((char *)file->path);
	free(file);
     }

   EINA_LIST_FREE(d->list, filename)
     free(filename);

   E_FREE(d);

   if (p->directory)
     eina_stringshare_del(p->directory);

   p->thread = NULL;

   if (p->cleanup)
     E_FREE(p);
}

static void
_scan_end_func(void *data)
{
   Data *d = data;
   Plugin *p = d->plugin;
   Evry_Item *item;
   Evry_Item_File *f;
   char *filename, *path, *mime;
   int cnt = 0;
   Eina_List *l;

   p->thread = NULL;
   printf("scan end\n");

   EINA_LIST_FREE(d->files, item)
     {
	ITEM_FILE(file, item);

	filename = (char *)item->label;
	path = (char *) file->path;
	mime = (char *) file->mime;

	file->path = eina_stringshare_add(path);

	/* filter out files that we already have from history */
	EINA_LIST_FOREACH(p->files, l, f)
	  {
	     if (f->path == file->path)
	       {
		  free(filename);
		  free(path);
		  E_FREE(file);
		  file = NULL;
		  break;
	       }
	  }
	if (!file) continue;

	if (mime)
	  file->mime = eina_stringshare_add(mime);
	else
	  file->mime = eina_stringshare_add("None");
	item->context = eina_stringshare_ref(file->mime);
	item->id = eina_stringshare_ref(file->path);
	item->label = eina_stringshare_add(filename);

	free(filename);
	free(path);

	p->files = eina_list_append(p->files, file);

	evry_util_file_detail_set(file);

	/* if (cnt >= MAX_ITEMS) continue;
	 * cnt += _append_file(p, file);
	 * item->priority = cnt; */
     }
   if (d->list)
     p->thread = ecore_thread_run(_scan_func, _scan_end_func, _scan_cancel_func, d);
   else
     E_FREE(d);

   p->files = eina_list_sort(p->files, -1, _cb_sort);

   EVRY_PLUGIN_ITEMS_CLEAR(p);

   EINA_LIST_FOREACH(p->files, l, f)
     {
	if (cnt >= MAX_SHOWN) break;
	cnt += _append_file(p, f);
	EVRY_ITEM(f)->priority = cnt;
     }

   evry_plugin_async_update(EVRY_PLUGIN(p), EVRY_ASYNC_UPDATE_ADD);
}

static void
_read_directory(Plugin *p)
{
   Data *d = E_NEW(Data, 1);
   d->plugin = p;

   p->thread = ecore_thread_run(_scan_func, _scan_end_func, _scan_cancel_func, d);
}

static Evry_Plugin *
_begin(Evry_Plugin *plugin, const Evry_Item *it)
{
   Plugin *p = NULL;

   /* is FILE ? */
   if (it && (it->plugin->type_out == plugin->type_in))
     {
	ITEM_FILE(file, it);

	if (!file->path || !ecore_file_is_dir(file->path))
	  return NULL;

	p = E_NEW(Plugin, 1);
	p->base = *plugin;
	p->base.items = NULL;

	p->directory = eina_stringshare_add(file->path);
	p->parent = EINA_TRUE;
     }
   else
     {
	p = E_NEW(Plugin, 1);
	p->base = *plugin;
	p->base.items = NULL;
	if (_conf->show_homedir)
	  p->directory = eina_stringshare_add(e_user_homedir_get());
	p->parent = EINA_FALSE;
     }

   if (p->directory)
     _read_directory(p);

   return EVRY_PLUGIN(p);
}

static int
_hist_add(Evry_Plugin *plugin, Evry_Item_File *file)
{
   Eina_List *l;
   History_Item *hi;
   History_Entry *he;

   he = eina_hash_find(evry_hist->subjects, file->path);

   if (!he) return 0;

   EINA_LIST_FOREACH(he->items, l, hi)
     {
	if (hi->type != plugin->type_out)
	  continue;

	if (hi->data)
	  eina_stringshare_del(hi->data);

	hi->data = eina_stringshare_ref(file->mime);
     }

   return 1;
}

static void
_cleanup(Evry_Plugin *plugin)
{
   PLUGIN(p, plugin);

   Evry_Item_File *file;

   if (p->directory)
     eina_stringshare_del(p->directory);

   EINA_LIST_FREE(p->files, file)
     {
	_hist_add(plugin, file);
	evry_item_free(EVRY_ITEM(file));
     }
   
   EVRY_PLUGIN_ITEMS_CLEAR(p);

   if (p->input)
     eina_stringshare_del(p->input);

   if (p->thread)
     {
	ecore_thread_cancel(p->thread);
	p->cleanup = EINA_TRUE;
	return;
     }

   E_FREE(p);
}

static Eina_Bool
_hist_items_add_cb(const Eina_Hash *hash, const void *key, void *data, void *fdata)
{
   History_Entry *he = data;
   History_Item *hi;
   Plugin *p = fdata;
   Eina_List *l, *ll;
   Evry_Item_File *file;
   const char *label;

   EINA_LIST_FOREACH(he->items, l, hi)
     {
	if (hi->type != p->base.type_out)
	  continue;

	/* filter out files that we already have from history */
	EINA_LIST_FOREACH(p->files, ll, file)
	  if (!strcmp(file->path, key))
	    return EINA_TRUE;

	/* XXX this blocks ui when drive is sleeping */
	/* if (!ecore_file_exists(key))
	 *   continue; */

	label = ecore_file_file_get(key);
	if (!label)
	  continue;

	file = E_NEW(Evry_Item_File, 1);
	if (!file)
	  continue;

	evry_item_new(EVRY_ITEM(file), EVRY_PLUGIN(p), label, _item_free);

	file->path = eina_stringshare_add(key);
	if (hi->data)
	  file->mime = eina_stringshare_add(hi->data);
	if (hi->data)
	  EVRY_ITEM(file)->context = eina_stringshare_ref(file->mime);

	EVRY_ITEM(file)->id = eina_stringshare_ref(file->path);

	evry_util_file_detail_set(file);

	if (file->mime)
	  {
	     if ((!strcmp(file->mime, "inode/directory")) ||
		 (!strcmp(file->mime, "inode/mount-point")))
	       EVRY_ITEM(file)->browseable = EINA_TRUE;
	  }
	/* else */
	/* if (ecore_file_is_dir(file->path))
	 * EVRY_ITEM(file)->browseable = EINA_TRUE; */

	p->files = eina_list_append(p->files, file);
	p->hist_added = eina_list_append(p->hist_added, file);
	break;
     }
   return EINA_TRUE;
}

static void
_folder_item_add(Plugin *p, const char *path)
{
   Evry_Item_File *file = E_NEW(Evry_Item_File, 1);

   if (!file) return;

   evry_item_new(EVRY_ITEM(file), EVRY_PLUGIN(p), path, _item_free);
   file->path = eina_stringshare_add(path);
   file->mime = eina_stringshare_add("inode/directory");
   EVRY_ITEM(file)->browseable = EINA_TRUE;
   p->files = eina_list_append(p->files, file);
   EVRY_PLUGIN_ITEM_APPEND(p, file);
}

static int
_fetch(Evry_Plugin *plugin, const char *input)
{
   PLUGIN(p, plugin);
   Evry_Item_File *file;
   Eina_List *l;
   int cnt = 0;
   int cmd = 0;
   
   if (!p->command)
     EVRY_PLUGIN_ITEMS_CLEAR(p);

   if (p->input)
     eina_stringshare_del(p->input);
   p->input = NULL;

   if (!p->parent)
     {
	/* input is command ? */
   	if (input && !strncmp(input, "/", 1))
   	  {
	     if (!p->command)
	       {
		  /* browse root */
		  EINA_LIST_FREE(p->files, file)
		    evry_item_free(EVRY_ITEM(file));

		  eina_stringshare_del(p->directory);
		  p->directory = eina_stringshare_add("/");
		  _read_directory(p);

		  p->command = EINA_TRUE;
	       }

	     cmd = 1;
	     return 0;
   	  }
     }

   if (p->directory  && input && !strncmp(input, "#", 1))
     {
	if (!p->command)
	  {
	     char *end;
	     char dir[4096];
	     char *tmp;
	     int prio = 0;

	     /* browse root */
	     EINA_LIST_FREE(p->hist_added, file)
	       {
		  p->files = eina_list_remove(p->files, file);
		  evry_item_free(EVRY_ITEM(file));
	       }
	     EINA_LIST_FREE(p->files, file)
	       evry_item_free(EVRY_ITEM(file));

	     if (strncmp(p->directory, "/", 1)) return 0;
	     if (!strcmp(p->directory, "/")) return 0;
	
	     snprintf(dir, 4096, "%s", p->directory);
	     end = strrchr(dir, '/');

	     while (end != dir)
	       {
		  tmp = strdup(dir);
		  snprintf(dir, (end - dir) + 1, "%s", tmp);

		  _folder_item_add(p, dir);

		  end = strrchr(dir, '/');
		  free(tmp);
		  prio--;
	       }

	     _folder_item_add(p, "/");
	     p->command = EINA_TRUE;
	  }
	cmd = 1;
	return 1;
     }

   /* clear command items */
   if (!cmd && p->command)
     {
   	p->command = EINA_FALSE;

	EINA_LIST_FREE(p->files, file)
	  evry_item_free(EVRY_ITEM(file));

	if (p->directory)
	  eina_stringshare_del(p->directory);
	
	p->directory = eina_stringshare_add(e_user_homedir_get());
	_read_directory(p);
	return 0;
     }

   /* add recent files */
   if ((!p->parent && !p->command && !p->hist_added) &&
       (_conf->show_recent || (input && _conf->search_recent)))
     {
        eina_hash_foreach(evry_hist->subjects, _hist_items_add_cb, p);
     }
   else if (p->hist_added && !input && _conf->search_recent)
     {
        EINA_LIST_FREE(p->hist_added, file)
          {
	     p->files = eina_list_remove(p->files, file);
	     evry_item_free(EVRY_ITEM(file));
          }
     }

   if (input)
     {
	/* skip command prefix FIXME */
	if (p->command)
	  p->input = eina_stringshare_add(input + 1);
	else
	  p->input = eina_stringshare_add(input);
     }

   EINA_LIST_FOREACH(p->files, l, file)
     {
	if (cnt >= MAX_SHOWN) break;
	cnt += _append_file(p, file);
	EVRY_ITEM(file)->priority = cnt;
     }

   if (!EVRY_PLUGIN(p)->items)
     return 0;

   if (!p->parent && _conf->show_recent)
     EVRY_PLUGIN_ITEMS_SORT(p, _cb_sort);

   return 1;
}

static Evas_Object *
_icon_get(Evry_Plugin *p __UNUSED__, const Evry_Item *it, Evas *e)
{
   Evas_Object *o = NULL;
   ITEM_FILE(file, it);

   if (!file->mime)
     _item_fill(file);

   if (!file->mime) return NULL;

   if (it->browseable)
     o = evry_icon_theme_get("folder", e);
   else
     o = evry_icon_mime_get(file->mime, e);

   return o;
}

static int
_open_folder_check(Evry_Action *act __UNUSED__, const Evry_Item *it)
{
   return (it->browseable && e_action_find("fileman"));
}

static int
_open_folder_action(Evry_Action *act)
{
   E_Action *action = e_action_find("fileman");
   char *path;
   Eina_List *m;

   if (!action) return 0;

   m = e_manager_list();

   ITEM_FILE(file, act->item1);

   if (!act->item1->browseable)
     {
	path = ecore_file_dir_get(file->path);
	if (!path) return 0;
	action->func.go(E_OBJECT(m->data), path);
	free(path);
     }
   else
     {
	action->func.go(E_OBJECT(m->data), file->path);
     }

   return 1;
}

static int
_open_term_action(Evry_Action *act)
{
   ITEM_FILE(file, act->item1);
   Evry_Item_App *tmp;
   char cwd[4096];
   char *dir;
   int ret = 0;

   if (act->item1->browseable)
     dir = strdup(file->path);
   else
     dir = ecore_file_dir_get(file->path);

   if (dir)
     {
	getcwd(cwd, sizeof(cwd));
	chdir(dir);

	tmp = E_NEW(Evry_Item_App, 1);
	tmp->file = evry_conf->cmd_terminal;

	ret = evry_util_exec_app(EVRY_ITEM(tmp), NULL);
	E_FREE(tmp);
	E_FREE(dir);
	chdir(cwd);
     }

   return ret;
}

static Eina_Bool
module_init(void)
{
   if (!evry_api_version_check(EVRY_API_VERSION))
     return EINA_FALSE;

   p1 = EVRY_PLUGIN_NEW(NULL, N_("Files"), type_subject, "FILE", "FILE",
			_begin, _cleanup, _fetch, _icon_get, NULL);
   EVRY_PLUGIN(p1)->config_path = "extensions/everything-files";
   evry_plugin_register(p1, 3);
   
   p2 = EVRY_PLUGIN_NEW(NULL, N_("Files"), type_object, "FILE", "FILE",
			_begin, _cleanup, _fetch, _icon_get, NULL);
   EVRY_PLUGIN(p2)->config_path = "extensions/everything-files";
   evry_plugin_register(p2, 1);

   act1 = EVRY_ACTION_NEW(N_("Open Folder (EFM)"), "FILE", NULL, "folder-open",
			  _open_folder_action, _open_folder_check);
   evry_action_register(act1, 0);

   act2 = EVRY_ACTION_NEW(N_("Open Terminal here"), "FILE", NULL, "system-run",
			  _open_term_action, NULL);
   evry_action_register(act2, 2);

   return EINA_TRUE;
}

static void
module_shutdown(void)
{
   EVRY_PLUGIN_FREE(p1);
   EVRY_PLUGIN_FREE(p2);

   evry_action_free(act1);
   evry_action_free(act2);
}


/***************************************************************************/

static E_Config_DD *conf_edd = NULL;

struct _E_Config_Dialog_Data
{
  int show_homedir;
  int show_recent;
  int search_recent;
};

static void *_create_data(E_Config_Dialog *cfd);
static void _free_data(E_Config_Dialog *cfd, E_Config_Dialog_Data *cfdata);
static void _fill_data(E_Config_Dialog_Data *cfdata);
static Evas_Object *_basic_create(E_Config_Dialog *cfd, Evas *evas, E_Config_Dialog_Data *cfdata);
static int _basic_apply(E_Config_Dialog *cfd, E_Config_Dialog_Data *cfdata);

static E_Config_Dialog *
_conf_dialog(E_Container *con, const char *params)
{
   E_Config_Dialog *cfd = NULL;
   E_Config_Dialog_View *v = NULL;
   char buf[4096];

   if (e_config_dialog_find("everything-files", "extensions/everything-files")) return NULL;

   v = E_NEW(E_Config_Dialog_View, 1);
   if (!v) return NULL;

   v->create_cfdata = _create_data;
   v->free_cfdata = _free_data;
   v->basic.create_widgets = _basic_create;
   v->basic.apply_cfdata = _basic_apply;

   snprintf(buf, sizeof(buf), "%s/e-module.edj", _conf->module->dir);

   cfd = e_config_dialog_new(con, _("Everything Files"), "everything-files",
			     "extensions/everything-files", buf, 0, v, NULL);

   /* e_dialog_resizable_set(cfd->dia, 1); */
   _conf->cfd = cfd;
   return cfd;
}

static Evas_Object *
_basic_create(E_Config_Dialog *cfd, Evas *evas, E_Config_Dialog_Data *cfdata)
{
   Evas_Object *o = NULL, *of = NULL, *ow = NULL;

   o = e_widget_list_add(evas, 0, 0);

   of = e_widget_framelist_add(evas, _("General"), 0);
   e_widget_framelist_content_align_set(of, 0.0, 0.0);

   ow = e_widget_check_add(evas, _("Show Home Directory"),
			   &(cfdata->show_homedir));
   e_widget_framelist_object_append(of, ow);

   ow = e_widget_check_add(evas, _("Show Recent Files"),
			   &(cfdata->show_recent));
   e_widget_framelist_object_append(of, ow);

   ow = e_widget_check_add(evas, _("Search Recent Files"),
			   &(cfdata->search_recent));
   e_widget_framelist_object_append(of, ow);

   e_widget_list_object_append(o, of, 1, 1, 0.5);
   return o;
}

static void *
_create_data(E_Config_Dialog *cfd)
  {
     E_Config_Dialog_Data *cfdata = NULL;

     cfdata = E_NEW(E_Config_Dialog_Data, 1);
     _fill_data(cfdata);
     return cfdata;
  }

static void
_free_data(E_Config_Dialog *cfd, E_Config_Dialog_Data *cfdata)
{
   _conf->cfd = NULL;
   E_FREE(cfdata);
}

static void
_fill_data(E_Config_Dialog_Data *cfdata)
{
#define C(_name) cfdata->_name = _conf->_name;
   C(show_homedir);
   C(show_recent);
   C(search_recent);
#undef C
}

static int
_basic_apply(E_Config_Dialog *cfd, E_Config_Dialog_Data *cfdata)
{
#define C(_name) _conf->_name = cfdata->_name;
   C(show_homedir);
   C(show_recent);
   C(search_recent);
#undef C
   printf("apply\n");

   e_config_domain_save("module.everything-files", conf_edd, _conf);
   e_config_save_queue();
   return 1;
}

static void
_conf_new(void)
{
   printf("new!!!!!\n");
   _conf = E_NEW(Module_Config, 1);
   _conf->version = (MOD_CONFIG_FILE_EPOCH << 16);

#define IFMODCFG(v) if ((_conf->version & 0xffff) < v) {
#define IFMODCFGEND }

   /* setup defaults */
   IFMODCFG(0x008d);
   _conf->show_recent = 0;
   _conf->show_homedir = 1;
   _conf->search_recent = 1;
   IFMODCFGEND;

   _conf->version = MOD_CONFIG_FILE_VERSION;

   e_config_domain_save("module.everything-files", conf_edd, _conf);
   e_config_save_queue();
}

static void
_conf_free(void)
{
   E_FREE(_conf);
}

static void
_conf_init(E_Module *m)
{
   char buf[4096];

   snprintf(buf, sizeof(buf), "%s/e-module.edj", m->dir);

   e_configure_registry_category_add("extensions", 80, _("Extensions"),
				     NULL, "preferences-extensions");

   e_configure_registry_item_add("extensions/everything-files", 110, _("Everything Files"),
				 NULL, buf, _conf_dialog);

   conf_edd = E_CONFIG_DD_NEW("Module_Config", Module_Config);
   
#undef T
#undef D
#define T Module_Config
#define D conf_edd
   E_CONFIG_VAL(D, T, version, INT);
   E_CONFIG_VAL(D, T, show_homedir, UCHAR);
   E_CONFIG_VAL(D, T, show_recent, UCHAR);
   E_CONFIG_VAL(D, T, search_recent, UCHAR);
#undef T
#undef D

   _conf = e_config_domain_load("module.everything-files", conf_edd);

   if (_conf && !evry_util_module_config_check(_("Everything Files"), _conf->version,
					       MOD_CONFIG_FILE_EPOCH, MOD_CONFIG_FILE_VERSION))
	  _conf_free();

   if (!_conf) _conf_new();

   _conf->module = m;
}

static void
_conf_shutdown(void)
{
   E_FREE(_conf);

   E_CONFIG_DD_FREE(conf_edd);
}

/***************************************************************************/

static Eina_Bool active = EINA_FALSE;

EAPI E_Module_Api e_modapi =
{
   E_MODULE_API_VERSION,
   "everything-files"
};

EAPI void *
e_modapi_init(E_Module *m)
{
   module = m;

   if (e_datastore_get("everything_loaded"))
     active = module_init();

   _conf_init(m);

   e_module_delayed_set(m, 1);

   return m;
}

EAPI int
e_modapi_shutdown(E_Module *m)
{
   if (active && e_datastore_get("everything_loaded"))
     module_shutdown();

   _conf_shutdown();

   return 1;
}

EAPI int
e_modapi_save(E_Module *m)
{
   e_config_domain_save("module.everything-files", conf_edd, _conf);
   return 1;
}