1 files changed, 112 insertions, 78 deletions

diff --git a/src/plugins/fs/download.c b/src/plugins/fs/download.c
index f627dfb1..ab03525a 100644
--- a/src/plugins/fs/download.c
+++ b/src/plugins/fs/download.c
@@ -71,6 +71,7 @@ static int addFilesToDirectory
     gtk_tree_model_get_iter(GTK_TREE_MODEL(pos->model),
                             &iter,
                             path);
+    gtk_tree_path_free(path);
     for (i=gtk_tree_model_iter_n_children(pos->model,
                                           &iter)-1;i>=0;i--) {
       if (TRUE == gtk_tree_model_iter_nth_child(pos->model,
@@ -108,11 +109,12 @@ static void initiateDownload(GtkTreeModel * model,
   struct ECRS_MetaData * meta;
   char * filename;
   char * pfx;
+  char * lnk;
   char * fn;
   DownloadList * list;
   GtkTreeIter iiter;
   GtkWidget * spin;
-  char * name;
+  const char * name;
   struct stat sbuf;
   unsigned int anon;
 
@@ -127,7 +129,6 @@ static void initiateDownload(GtkTreeModel * model,
                      -1);
   if (uri == NULL) {
     BREAK();
-    FREENONNULL(name);
     return;
   }
   filename = ECRS_uriToString(uri);  
@@ -136,17 +137,16 @@ static void initiateDownload(GtkTreeModel * model,
         strlen(ECRS_URI_PREFIX) +
         strlen(ECRS_FILE_INFIX)) ) {
     BREAK();
-    FREENONNULL(name);
     FREENONNULL(filename);
     return;
   }
   if (name == NULL)
-    name = STRDUP(_("unnamed"));
+    name = filename;
   list = MALLOC(sizeof(DownloadList));
   list->next = head;
   list->rr = NULL;
   list->model = NULL;
-  if (ECRS_isDirectory(meta)) {
+  if (YES == ECRS_isDirectory(meta)) {
     list->rr = gtk_tree_row_reference_new(model, path);
     list->model = model;
   }
@@ -157,53 +157,41 @@ static void initiateDownload(GtkTreeModel * model,
                      " in section '%s' under '%s'."));
   mkdirp(fn);
   pfx = MALLOC(strlen(fn) + 2 +
-               strlen(filename) + strlen(name));
+               strlen(name));
+  lnk = MALLOC(strlen(fn) + 2 +
+               strlen(filename));
   strcpy(pfx, fn);
-  strcat(pfx, 
-         DIR_SEPARATOR_STR);
+  if (pfx[strlen(pfx)-1] != DIR_SEPARATOR)
+    strcat(pfx, 
+           DIR_SEPARATOR_STR);
+  strcpy(lnk, fn);
+  FREE(fn);
+  if (lnk[strlen(lnk)-1] != DIR_SEPARATOR)
+    strcat(lnk, 
+           DIR_SEPARATOR_STR);
   strcat(pfx, 
          name);
-  while ( (strlen(pfx) > 0) &&
-          (pfx[strlen(pfx)-1] == DIR_SEPARATOR) )
-    pfx[strlen(pfx)-1] = '\0';
-  if (0 == STAT(pfx,
+  strcat(lnk, 
+         &filename[strlen(ECRS_URI_PREFIX) +
+                   strlen(ECRS_FILE_INFIX)]);
+  if (0 != STAT(pfx,
                 &sbuf)) {
-    GtkWidget * dialog;
-    
-    dialog = gtk_message_dialog_new 
-      (NULL,
-       GTK_DIALOG_MODAL,
-       GTK_MESSAGE_WARNING,
-       GTK_BUTTONS_CLOSE,
-       _("File '%s' exists in '%s',\n"
-         "will store download under its"
-         " GNUnet URI '%s' instead.\n"),
-       name,
-       fn,
-       &filename[strlen(ECRS_URI_PREFIX) +
-                 strlen(ECRS_FILE_INFIX)]);
-    gtk_dialog_run(GTK_DIALOG(dialog));
-    gtk_widget_destroy(dialog);  
-    strcpy(pfx, fn);
-    strcat(pfx, 
-           DIR_SEPARATOR_STR);
-    strcat(pfx,
-           &filename[strlen(ECRS_URI_PREFIX) +
-                     strlen(ECRS_FILE_INFIX)]);
+    if (0 != SYMLINK(lnk, pfx))
+      LOG_FILE_STRERROR(LOG_ERROR, "symlink", pfx);
+  } else {
+    FREE(pfx);
+    pfx = STRDUP(lnk);
   }
-  FREE(fn);
-  
-
   list->filename = pfx;
   head = list;
-
   gtk_tree_store_insert(summary,
                         &iiter,
                         NULL,
                         0);
   gtk_tree_store_set(summary,
                      &iiter,
-                     DOWNLOAD_FILENAME, name,
+                     DOWNLOAD_FILENAME, lnk,
+                     DOWNLOAD_LINKNAME, pfx,
                      DOWNLOAD_SIZE, ECRS_fileSize(uri),
                      DOWNLOAD_PROGRESS, 0, /* progress */
                      DOWNLOAD_URISTRING, filename,
@@ -212,7 +200,6 @@ static void initiateDownload(GtkTreeModel * model,
                      /* internal: row reference! */
                      -1);
   FREE(filename);
-  FREE(name);
   spin = getAnonymityButtonFromTM(model);
   if (spin == NULL) {
     BREAK();
@@ -220,12 +207,12 @@ static void initiateDownload(GtkTreeModel * model,
   } else {
     anon = gtk_spin_button_get_value_as_int
       (GTK_SPIN_BUTTON(spin));
-  }
-  
+  }  
   FSUI_startDownload(ctx,
                      anon,
                      uri,
-                     pfx);
+                     lnk);
+  FREE(lnk);
 }
 
 void on_downloadButton_clicked(GtkWidget * treeview,
@@ -290,12 +277,17 @@ void displayDownloadUpdate(const struct ECRS_URI * uri,
 void displayDownloadComplete(const struct ECRS_URI * uri,
                              const char * filename) {
   char * ren;
+  const char * fn;
+  const char * ln;
   unsigned long long size;
   char * data;
   int fd;
   struct ECRS_MetaData * meta;
   DownloadList * pos;
-
+  GtkTreeIter iter;
+  GtkTreePath * path;
+  struct ECRS_URI * u;
+   
   pos = head;
   while (pos != NULL) {
     if (ECRS_equalsUri(uri,
@@ -303,40 +295,80 @@ void displayDownloadComplete(const struct ECRS_URI * uri,
       break;    
     pos = pos->next;
   }
-
-  ren = ECRS_suggestFilename(filename);
-  if (ren == NULL)
+  if (pos == NULL) 
     return;
-  if (0 != strcmp(ren,
-                  filename)) {
-    /* fixme: ask user for confirmation? */
-    /* rename(filename, ren); */
-  }
-  FREE(ren);
-  /* fixme: also update download model? */
-  /* update directory view (if applicable!) */
-  if ( (pos != NULL) &&
-       (pos->rr != NULL) &&
-       (OK == getFileSize(filename, &size)) ) {
-    fd = fileopen(filename, O_RDONLY);
-    data = MMAP(NULL,
-                size,
-                PROT_READ,
-                MAP_SHARED,
-                fd,
-                0);
-    meta = NULL;
-    if (data != NULL) {
-      ECRS_listDirectory(data,
-                         size,
-                         &meta,
-                         &addFilesToDirectory,
-                         (void*)uri);
-      MUNMAP(data, size);
+  if ( (pos->rr != NULL) &&
+       (gtk_tree_row_reference_valid(pos->rr)) ) {
+
+    /* update directory view (if applicable!) */
+    if (OK == getFileSize(filename, &size)) {
+      fd = fileopen(filename, O_RDONLY);
+      data = MMAP(NULL,
+                  size,
+                  PROT_READ,
+                  MAP_SHARED,
+                  fd,
+                  0);
+      meta = NULL;
+      if (data != NULL) {
+        ECRS_listDirectory(data,
+                           size,
+                           &meta,
+                           &addFilesToDirectory,
+                           (void*)uri);
+        MUNMAP(data, size);
+      }
+      CLOSE(fd);
+      if (meta != NULL)
+        ECRS_freeMetaData(meta);
     }
-    CLOSE(fd);
-    if (meta != NULL)
-      ECRS_freeMetaData(meta);
+
+    path = gtk_tree_row_reference_get_path(pos->rr);
+    if (gtk_tree_path_get_depth(path) > 1) {
+      gtk_tree_path_free(path);
+      return;
+    }
+    gtk_tree_path_free(path);  
+  }
+  
+  /* only rename top-level files, not files inside of directories! */
+  if (gtk_tree_model_get_iter_first(GTK_TREE_MODEL(summary),
+                                    &iter)) {
+    do {    
+      ln = NULL;
+      fn = NULL;
+      gtk_tree_model_get(GTK_TREE_MODEL(summary),
+                         &iter,
+                         DOWNLOAD_URI, &u,
+                         DOWNLOAD_FILENAME, &fn,
+                         DOWNLOAD_LINKNAME, &ln,
+                         -1);
+      if (ECRS_equalsUri(u, uri)) {
+        ren = ECRS_suggestFilename(ln);
+        if (ren != NULL) {
+          gtk_tree_store_set(summary,
+                             &iter,
+                             DOWNLOAD_LINKNAME, ren,
+                             -1);
+          if (0 == strcmp(fn, ln)) {
+            /* keep the actual data under the URI name
+               and create a symlink */
+            if (0 != RENAME(ren, fn))
+              LOG(LOG_ERROR,
+                  _("Renaming of file '%s' to '%s' failed: %s\n"),
+                  ren,
+                  fn,
+                  STRERROR(errno));
+            else
+              if (0 != SYMLINK(fn, ren)) 
+                LOG_FILE_STRERROR(LOG_ERROR, "symlink", ren);
+          }
+          FREE(ren);
+        }
+        break;
+      }
+    } while (gtk_tree_model_iter_next(GTK_TREE_MODEL(summary),
+                                      &iter));
   }
 
 }
@@ -440,6 +472,7 @@ static int addDownloadView(void * cls,
   gtk_tree_store_set(summary,
                      &iiter,
                      DOWNLOAD_FILENAME, filename,
+                     DOWNLOAD_LINKNAME, filename,
                      DOWNLOAD_SIZE, filesize,
                      DOWNLOAD_PROGRESS, progress,
                      DOWNLOAD_URISTRING, uriname,
@@ -460,7 +493,8 @@ void fs_download_start() {
                                       "activeDownloadsList");
   summary =
     gtk_tree_store_new(DOWNLOAD_NUM,
-                       G_TYPE_STRING, /* name */
+                       G_TYPE_STRING, /* name (URI as string) */
+                       G_TYPE_STRING, /* name (user-friendly name) */
                        G_TYPE_UINT64,  /* size */
                        G_TYPE_INT,  /* progress */
                        G_TYPE_STRING, /* uri */                      
@@ -474,7 +508,7 @@ void fs_download_start() {
                                               _("Name"),
                                               renderer,
                                               "value", DOWNLOAD_PROGRESS,
-                                              "text", DOWNLOAD_FILENAME,
+                                              "text", DOWNLOAD_LINKNAME,
                                               NULL);
   renderer = gtk_cell_renderer_text_new();
   gtk_tree_view_insert_column_with_attributes(GTK_TREE_VIEW(downloadList),