enlightenment/src/modules/everything/evry_plug_tracker.c

423 lines
8.9 KiB
C

#include "Evry.h"
/* TODO check if trackerd is running and version */
typedef struct _Inst Inst;
struct _Inst
{
int active;
char *condition;
char *service;
int max_hits;
const char *input;
const char *matched;
Eina_List *items;
};
static E_DBus_Connection *conn = NULL;
static Eina_List *plugins = NULL;
static int
_begin(Evry_Plugin *p, const Evry_Item *it)
{
Inst *inst = p->private;
inst->active = 0;
if (!strcmp(it->plugin->type_out, "APPLICATION"))
{
Eina_List *l;
const char *mime;
Evry_App *app = it->data[0];
char mime_entry[256];
char rdf_query[32768];
int len = 0;
if (inst->condition[0]) free (inst->condition);
inst->condition = "";
if (!app->desktop || !app->desktop->mime_types)
return 1;
rdf_query[0] = '\0';
strcat(rdf_query, "<rdfq:Condition><rdfq:or>");
EINA_LIST_FOREACH(app->desktop->mime_types, l, mime)
{
if (!strcmp(mime, "x-directory/normal"))
return 0;
snprintf(mime_entry, 256,
"<rdfq:contains>"
"<rdfq:Property name=\"File:Mime\" />"
"<rdf:String>%s</rdf:String> "
"</rdfq:contains>",
mime);
strcat(rdf_query, mime_entry);
len += 256;
if (len > 32000) break;
}
strcat(rdf_query, "</rdfq:or></rdfq:Condition>");
inst->condition = strdup(rdf_query);
}
return 1;
}
static Evry_Item *
_item_add(Evry_Plugin *p, char *file, char *mime, int prio)
{
Evry_Item *it;
const char *filename;
int folder = (!strcmp(mime, "Folder"));
/* folders are specifically searched */
if (folder && p->begin)
return NULL;
filename = ecore_file_file_get(file);
if (!filename)
return NULL;
it = evry_item_new(p, filename, NULL);
if (!it)
return NULL;
it->priority = prio;
it->uri = eina_stringshare_add(file);
if (folder)
{
it->browseable = EINA_TRUE;
it->mime = eina_stringshare_add("x-directory/normal");
it->priority = 1;
}
else
it->mime = eina_stringshare_add(mime);
return it;
}
static void
_cleanup(Evry_Plugin *p)
{
Evry_Item *it;
Inst *inst = p->private;
inst->active = 0;
if (inst->input)
eina_stringshare_del(inst->input);
inst->input = NULL;
if (inst->matched)
eina_stringshare_del(inst->matched);
inst->matched = NULL;
EINA_LIST_FREE(inst->items, it)
evry_item_free(it);
if (p->items)
eina_list_free(p->items);
p->items = NULL;
}
static int
_cb_sort(const void *data1, const void *data2)
{
const Evry_Item *it1, *it2;
it1 = data1;
it2 = data2;
return (it2->priority - it1->priority);
}
static void
_dbus_cb_reply(void *data, DBusMessage *msg, DBusError *error)
{
DBusMessageIter array, iter, item;
char *uri, *mime, *date;
Evry_Item *it;
Eina_List *items = NULL;
Evry_Plugin *p = data;
Inst *inst = p->private;
if (inst->active) inst->active--;
if (dbus_error_is_set(error))
{
_cleanup(p);
printf("Error: %s - %s\n", error->name, error->message);
return;
}
if (inst->active) return;
dbus_message_iter_init(msg, &array);
if(dbus_message_iter_get_arg_type(&array) == DBUS_TYPE_ARRAY)
{
dbus_message_iter_recurse(&array, &item);
while(dbus_message_iter_get_arg_type(&item) == DBUS_TYPE_ARRAY)
{
dbus_message_iter_recurse(&item, &iter);
if (dbus_message_iter_get_arg_type(&iter) == DBUS_TYPE_STRING)
{
dbus_message_iter_get_basic(&iter, &uri);
dbus_message_iter_next(&iter);
/* dbus_message_iter_get_basic(&iter, &service); */
dbus_message_iter_next(&iter);
dbus_message_iter_get_basic(&iter, &mime);
dbus_message_iter_next(&iter);
dbus_message_iter_get_basic(&iter, &date);
if (uri && mime && date)
{
it = _item_add(p, uri, mime, atoi(date));
if (it)
items = eina_list_append(items, it);
}
}
dbus_message_iter_next(&item);
}
}
if (p->items)
eina_list_free(p->items);
p->items = NULL;
if (items)
{
Eina_List *l;
EINA_LIST_FREE(inst->items, it)
evry_item_free(it);
items = eina_list_sort(items, eina_list_count(items), _cb_sort);
inst->items = items;
EINA_LIST_FOREACH(inst->items, l, it)
p->items = eina_list_append(p->items, it);
if (inst->matched)
eina_stringshare_del(inst->matched);
if (inst->input)
inst->matched = eina_stringshare_add(inst->input);
else
inst->matched = NULL;
}
else if (inst->items && inst->input && inst->matched &&
(strlen(inst->input) > strlen(inst->matched)))
{
Eina_List *l;
EINA_LIST_FOREACH(inst->items, l, it)
if (evry_fuzzy_match(it->label, (inst->input + strlen(inst->matched))))
p->items = eina_list_append(p->items, it);
}
else
{
EINA_LIST_FREE(inst->items, it)
evry_item_free(it);
if (inst->input)
eina_stringshare_del(inst->input);
inst->input = NULL;
if (inst->matched)
eina_stringshare_del(inst->matched);
inst->matched = NULL;
}
evry_plugin_async_update(p, EVRY_ASYNC_UPDATE_ADD);
}
static int
_fetch(Evry_Plugin *p, const char *input)
{
Inst *inst = p->private;
DBusMessage *msg;
int live_query_id = 0;
int max_hits = inst->max_hits;
int offset = 0;
int sort_descending = 1;
int sort_by_service = 0;
int sort_by_access = 0;
char *search_text;
char *fields[2];
char *keywords[1];
char *sort_fields[1];
fields[0] = "File:Mime";
fields[1] = "File:Accessed";
keywords[0] = "";
sort_fields[0] = "File:Accessed";
char **_fields = fields;
char **_keywords = keywords;
char **_sort_fields = sort_fields;
if (inst->input)
eina_stringshare_del(inst->input);
inst->input = NULL;
if (!conn)
{
_cleanup(p);
return 0;
}
if (input && (strlen(input) > 2))
{
inst->input = eina_stringshare_add(input);
search_text = malloc(sizeof(char) * strlen(input) + 1);
sprintf(search_text, "%s", input);
max_hits = 50;
}
else if (!input && !p->begin && p->type == type_object)
{
sort_by_access = 1;
search_text = "";
}
else
{
_cleanup(p);
return 0;
}
inst->active++;
msg = dbus_message_new_method_call("org.freedesktop.Tracker",
"/org/freedesktop/Tracker/Search",
"org.freedesktop.Tracker.Search",
"Query");
dbus_message_append_args(msg,
DBUS_TYPE_INT32, &live_query_id,
DBUS_TYPE_STRING, &inst->service,
DBUS_TYPE_ARRAY, DBUS_TYPE_STRING,
&_fields, 2,
DBUS_TYPE_STRING, &search_text,
DBUS_TYPE_ARRAY, DBUS_TYPE_STRING,
&_keywords, 0,
DBUS_TYPE_STRING, &inst->condition,
DBUS_TYPE_BOOLEAN, &sort_by_service,
DBUS_TYPE_ARRAY, DBUS_TYPE_STRING,
&_sort_fields, sort_by_access,
DBUS_TYPE_BOOLEAN, &sort_descending,
DBUS_TYPE_INT32, &offset,
DBUS_TYPE_INT32, &max_hits,
DBUS_TYPE_INVALID);
e_dbus_message_send(conn, msg, _dbus_cb_reply, -1, p);
dbus_message_unref(msg);
if (input && (strlen(input) > 2))
free(search_text);
if (p->items) return 1;
return 0;
}
static Evas_Object *
_item_icon_get(Evry_Plugin *p __UNUSED__, const Evry_Item *it, Evas *e)
{
char *icon_path;
Evas_Object *o = NULL;
if (it->browseable)
{
o = e_icon_add(e);
evry_icon_theme_set(o, "folder");
}
else
{
icon_path = efreet_mime_type_icon_get(it->mime, e_config->icon_theme, 64);
if (icon_path)
{
o = e_util_icon_add(icon_path, e);
free(icon_path);
}
else
{
o = e_icon_add(e);
evry_icon_theme_set(o, "none");
}
}
return o;
}
static void
_plugin_new(const char *name, int type, char *service, int max_hits, int begin)
{
Evry_Plugin *p;
Inst *inst;
p = E_NEW(Evry_Plugin, 1);
p->name = name;
p->type = type;
p->type_in = "NONE";
p->type_out = "FILE";
p->async_query = 1;
if (begin)
p->begin = &_begin;
p->fetch = &_fetch;
p->cleanup = &_cleanup;
p->icon_get = &_item_icon_get;
inst = E_NEW(Inst, 1);
inst->condition = "";
inst->service = service;
inst->max_hits = max_hits;
inst->active = 0;
p->private = inst;
evry_plugin_register(p);
plugins = eina_list_append(plugins, p);
}
static Eina_Bool
_init(void)
{
conn = e_dbus_bus_get(DBUS_BUS_SESSION);
if (!conn) return EINA_FALSE;
_plugin_new("Folders", type_subject, "Folders", 20, 0);
_plugin_new("Images", type_subject, "Images", 20, 0);
_plugin_new("Music", type_subject, "Music", 20, 0);
_plugin_new("Videos", type_subject, "Videos", 20, 0);
_plugin_new("Documents", type_subject, "Documents", 20, 0);
_plugin_new("Find Files", type_object, "Files", 20, 1);
_plugin_new("Folders", type_object, "Folders", 20, 0);
return EINA_TRUE;
}
static void
_shutdown(void)
{
Inst *inst;
Evry_Plugin *p;
if (conn) e_dbus_connection_close(conn);
EINA_LIST_FREE(plugins, p)
{
evry_plugin_unregister(p);
inst = p->private;
if (inst->condition[0]) free(inst->condition);
E_FREE(inst);
E_FREE(p);
}
}
EINA_MODULE_INIT(_init);
EINA_MODULE_SHUTDOWN(_shutdown);