int main(int ac, char* av[])
{
mega_session* s;
gc_error_free GError *local_err = NULL;
GSList *l = NULL, *i;
gint j;
tool_init(&ac, &av, "- list files stored at mega.nz", entries);
s = tool_start_session();
if (!s)
return 1;
// gather nodes
if (ac == 1)
{
l = mega_session_ls_all(s);
opt_names = FALSE;
}
else
{
if (ac > 2 || opt_recursive)
opt_names = FALSE;
for (j = 1; j < ac; j++)
{
gc_free gchar* path = tool_convert_filename(av[j], FALSE);
mega_node* n = mega_session_stat(s, path);
if (n && (n->type == MEGA_NODE_FILE || !opt_names))
l = g_slist_append(l, n);
l = g_slist_concat(l, mega_session_ls(s, path, opt_recursive));
}
}
l = g_slist_sort(l, (GCompareFunc)compare_node);
// export if requested
if (opt_export && !mega_session_addlinks(s, l, &local_err))
{
g_printerr("ERROR: Can't read links info from mega.nz: %s\n", local_err->message);
g_slist_free(l);
tool_fini(s);
return 1;
}
if (l && opt_long && opt_header && !opt_export)
{
g_print("===================================================================================\n");
g_print("%-11s %-11s %-1s %13s %-19s %s\n", "Handle", "Owner", "T", "Size", "Mod. Date", opt_names ? "Filename" : "Path");
g_print("===================================================================================\n");
}
for (i = l; i; i = i->next)
{
mega_node* n = i->data;
gc_free gchar* node_path = mega_node_get_path_dup(n);
if (opt_export)
g_print("%73s ", n->link ? mega_node_get_link(n, TRUE) : "");
if (opt_long)
{
GDateTime* dt = g_date_time_new_from_unix_local(n->timestamp);
gc_free gchar* time_str = g_date_time_format(dt, "%Y-%m-%d %H:%M:%S");
g_date_time_unref(dt);
gc_free gchar* size_str = NULL;
if (opt_human)
size_str = n->size > 0 ? g_format_size_full(n->size, G_FORMAT_SIZE_IEC_UNITS) : g_strdup("-");
else
size_str = n->size > 0 ? g_strdup_printf("%" G_GUINT64_FORMAT, n->size) : g_strdup("-");
g_print("%-11s %-11s %d %13s %19s %s\n",
n->handle,
n->user_handle ? n->user_handle : "",
n->type,
size_str,
n->timestamp > 0 ? time_str : "",
opt_names ? n->name : node_path
);
}
else
g_print("%s\n", opt_names ? n->name : node_path);
}
g_slist_free(l);
tool_fini(s);
return 0;
}
int main(int ac, char* av[])
{
GError *local_err = NULL;
GRegex *file_regex, *folder_regex;
gchar* key;
gchar* handle;
gint i;
tool_init_bare(&ac, &av, "- download exported files from mega.co.nz", entries);
if (!strcmp(opt_path, "-"))
opt_noprogress = opt_stream = TRUE;
if (ac < 2)
{
g_printerr("ERROR: No links specified for download!\n");
tool_fini(NULL);
return 1;
}
if (opt_stream && ac != 2)
{
g_printerr("ERROR: Can't stream from multiple files!\n");
tool_fini(NULL);
return 1;
}
// prepare link parsers
file_regex = g_regex_new("^https?://mega.co.nz/#!([a-z0-9_-]{8})!([a-z0-9_-]{43})$", G_REGEX_CASELESS, 0, NULL);
g_assert(file_regex != NULL);
folder_regex = g_regex_new("^https?://mega.co.nz/#F!([a-z0-9_-]{8})!([a-z0-9_-]{22})$", G_REGEX_CASELESS, 0, NULL);
g_assert(folder_regex != NULL);
// create session
s = mega_session_new();
mega_session_watch_status(s, status_callback, NULL);
// process links
for (i = 1; i < ac; i++)
{
GMatchInfo* m1 = NULL;
GMatchInfo* m2 = NULL;
key = NULL;
handle = NULL;
if (g_regex_match(file_regex, av[i], 0, &m1))
{
handle = g_match_info_fetch(m1, 1);
key = g_match_info_fetch(m1, 2);
// perform download
if (!mega_session_dl(s, handle, key, opt_stream ? NULL : opt_path, &local_err))
{
if (!opt_noprogress)
g_print("\r" ESC_CLREOL "\n");
g_printerr("ERROR: Download failed for '%s': %s\n", av[i], local_err->message);
g_clear_error(&local_err);
}
else
{
if (!opt_noprogress)
g_print("\r" ESC_CLREOL "Downloaded %s\n", cur_file);
}
}
else if (g_regex_match(folder_regex, av[i], 0, &m2))
{
if (opt_stream)
{
g_printerr("ERROR: Can't stream from a directory!\n");
if (m1)
g_match_info_unref(m1);
if (m2)
g_match_info_unref(m2);
tool_fini(s);
return 1;
}
handle = g_match_info_fetch(m2, 1);
key = g_match_info_fetch(m2, 2);
// perform download
if (!mega_session_open_exp_folder(s, handle, key, &local_err))
{
g_printerr("ERROR: Can't open folder '%s': %s\n", av[i], local_err->message);
g_clear_error(&local_err);
}
else
{
GSList* l = mega_session_ls(s, "/", FALSE);
if (g_slist_length(l) == 1)
{
mega_node* root_node = l->data;
GFile* local_dir = g_file_new_for_path(opt_path);
if (g_file_query_file_type(local_dir, G_FILE_QUERY_INFO_NOFOLLOW_SYMLINKS, NULL) == G_FILE_TYPE_DIRECTORY)
{
dl_sync_dir(root_node, local_dir, root_node->path);
}
else
{
g_printerr("ERROR: %s must be a directory\n", opt_path);
}
}
else
{
g_printerr("ERROR: EXP folder fs has multiple toplevel nodes? Weird!\n");
}
}
}
else
{
g_printerr("WARNING: Skipping invalid Mega download link: %s\n", av[i]);
}
if (m1)
g_match_info_unref(m1);
if (m2)
g_match_info_unref(m2);
g_free(handle);
g_free(key);
}
tool_fini(s);
return 0;
}
