rage/src/bin/albumart.c

315 lines
8.2 KiB
C

#include <Elementary.h>
#include "video.h"
#include "win.h"
#include "rage_config.h"
#include "config.h"
#include "albumart.h"
#include "sha1.h"
#define Q_START "http://www.google.com/search?as_st=y&tbm=isch&hl=en&as_q="
//#define Q_END "&as_epq=&as_oq=&as_eq=&cr=&as_sitesearch=&safe=images&tbs=iar:s,ift:jpg"
#define Q_END "&as_epq=&as_oq=&as_eq=&cr=&as_sitesearch=&safe=images&tbs=ift:jpg"
static Ecore_Con_Url *fetch = NULL;
static Ecore_Event_Handler *handle_data = NULL;
static Ecore_Event_Handler *handle_complete = NULL;
static Eina_Strbuf *sb_result = NULL;
static Eina_Bool fetch_image = EINA_FALSE;
static char *fetchfile = NULL;
static FILE *fout = NULL;
static Evas_Object *fetchwin = NULL;
static char *
_inpath(const char *file)
{
char *realpath = NULL;
if (!strncasecmp(file, "file:/", 6))
{
Efreet_Uri *uri = efreet_uri_decode(file);
if (uri)
{
realpath = ecore_file_realpath(uri->path);
efreet_uri_free(uri);
}
}
else if ((!strncasecmp(file, "http:/", 6)) ||
(!strncasecmp(file, "https:/", 7)))
realpath = strdup(file);
else
realpath = ecore_file_realpath(file);
if (realpath && (!realpath[0]))
{
free(realpath);
return NULL;
}
return realpath;
}
static char *
_thumbpath(const char *file)
{
char buf_base[PATH_MAX];
char buf_file[PATH_MAX];
unsigned char sum[20];
if (!sha1((unsigned char *)file, strlen(file), sum)) return NULL;
snprintf(buf_base, sizeof(buf_base), "%s/rage/albumart/%02x",
efreet_cache_home_get(), sum[0]);
if (!ecore_file_mkpath(buf_base)) return NULL;
snprintf(buf_file, sizeof(buf_base),
"%s/%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x"
"%02x%02x%02x%02x%02x%02x%02x%02x.jpg",
buf_base,
sum[1], sum[2], sum[3],
sum[4], sum[5], sum[6], sum[7],
sum[8], sum[9], sum[10], sum[11],
sum[12], sum[13], sum[14], sum[15],
sum[16], sum[17], sum[18], sum[19]);
return strdup(buf_file);
}
static Ecore_Con_Url *
_fetch(Eina_Strbuf *sb)
{
Ecore_Con_Url *f;
const char *qs;
qs = eina_strbuf_string_get(sb);
if (!qs) return NULL;
f = ecore_con_url_new(qs);
if (!f) return NULL;
ecore_con_url_additional_header_add
(f, "user-agent",
"Mozilla/5.0 (Windows NT 5.1; rv:31.0) Gecko/20100101 Firefox/31.0");
ecore_con_url_get(f);
return f;
}
static Eina_Bool
_cb_http_data(void *data EINA_UNUSED, int type EINA_UNUSED, void *event)
{
Ecore_Con_Event_Url_Data *ev = event;
if (ev->url_con != fetch) return EINA_TRUE;
if (fetch_image) fwrite(ev->data, ev->size, 1, fout);
else if (sb_result)
eina_strbuf_append_length(sb_result, (char *)ev->data, (size_t)ev->size);
return EINA_FALSE;
}
static Eina_Bool
_cb_http_complete(void *data EINA_UNUSED, int type EINA_UNUSED, void *event)
{
Ecore_Con_Event_Url_Complete *ev = event;
Eina_Bool ok = EINA_FALSE;
if (ev->url_con != fetch) return EINA_TRUE;
if (fetch_image)
{
char *path;
fetch_image = EINA_FALSE;
fclose(fout);
fout = NULL;
path = _thumbpath(fetchfile);
if (path)
{
win_art(fetchwin, path);
free(path);
}
}
else
{
if (sb_result)
{
const char *res = eina_strbuf_string_get(sb_result);
if (res)
{
const char *p, *pe;
Eina_Strbuf *sb;
sb = eina_strbuf_new();
p = strstr(res, "<img data-sz=\"f\" name=\"");
if (p)
{
p = strstr(p, "src=\"http");
p += 5;
pe = strchr(p, '"');
if (pe)
{
eina_strbuf_append_length(sb, p, pe - p);
ok = EINA_TRUE;
}
}
if (!ok)
{
p = strstr(res, "imgurl=");
if (p)
{
p += 7;
pe = strstr(p, "&amp;");
if (pe)
{
eina_strbuf_append_length(sb, p, pe - p);
ok = EINA_TRUE;
}
}
}
if (ok)
{
char *path;
ecore_con_url_free(fetch);
fetch = NULL;
path = _thumbpath(fetchfile);
if (path)
{
fout = fopen(path, "wb");
if (fout)
{
fetch_image = EINA_TRUE;
fetch = _fetch(sb);
}
free(path);
}
}
eina_strbuf_free(sb);
}
eina_strbuf_free(sb_result);
sb_result = NULL;
}
}
if (!ok)
{
ecore_con_url_free(fetch);
fetch = NULL;
if (fetchfile)
{
free(fetchfile);
fetchfile = NULL;
}
}
return EINA_FALSE;
}
static Eina_Bool
_search_append(Eina_Strbuf *sb, const char *str, Eina_Bool hadword)
{
const char *s;
Eina_Bool word = EINA_FALSE;
for (s = str; *s; s++)
{
if (((*s >= 'a') && (*s <= 'z')) ||
((*s >= 'A') && (*s <= 'Z')) ||
((*s >= '0') && (*s <= '9')))
{
if (!word)
{
if (hadword)
{
eina_strbuf_append_char(sb, '+');
word = EINA_FALSE;
}
}
eina_strbuf_append_char(sb, *s);
word = EINA_TRUE;
hadword = EINA_TRUE;
}
else word = EINA_FALSE;
if (*s == '.') break;
}
return hadword;
}
void
albumart_find(Evas_Object *win, Evas_Object *vid)
{
const char *file, *album, *artist, *title;
Eina_Strbuf *sb;
char *path;
if (fetchfile)
{
free(fetchfile);
fetchfile = NULL;
}
if (fetch)
{
ecore_con_url_free(fetch);
fetch = NULL;
}
if (fout)
{
fclose(fout);
fout = NULL;
}
if (sb_result)
{
eina_strbuf_free(sb_result);
sb_result = NULL;
}
fetch_image = EINA_FALSE;
if (!vid) return;
if (!handle_data)
handle_data = ecore_event_handler_add(ECORE_CON_EVENT_URL_DATA,
_cb_http_data, NULL);
if (!handle_complete)
handle_complete = ecore_event_handler_add(ECORE_CON_EVENT_URL_COMPLETE,
_cb_http_complete, NULL);
file = video_file_get(vid);
if (!file) return;
fetchfile = _inpath(file);
path = _thumbpath(fetchfile);
if (path)
{
if (ecore_file_exists(path))
{
win_art(win, path);
free(path);
free(fetchfile);
fetchfile = NULL;
return;
}
else free(path);
}
fetchwin = win;
sb = eina_strbuf_new();
eina_strbuf_append(sb, Q_START);
title = video_meta_title_get(vid);
artist = video_meta_artist_get(vid);
album = video_meta_album_get(vid);
if ((title) || (album) || (artist))
{
Eina_Bool had = EINA_FALSE;
if (artist) had = _search_append(sb, artist, had);
if (album) had = _search_append(sb, album, had);
if (title) had = _search_append(sb, title, had);
}
else
_search_append(sb, ecore_file_file_get(fetchfile), EINA_FALSE);
eina_strbuf_append(sb, Q_END);
if (sb_result) eina_strbuf_free(sb_result);
sb_result = eina_strbuf_new();
fetch = _fetch(sb);
eina_strbuf_free(sb);
}
char *
albumart_file_get(const char *file)
{
return _thumbpath(file);
}