BeamMeUp

src/beam-me-up-get.c

@@ -0,0 +1,173 @@
+/* beam-me-up-get.c
+ *
+ * Copyright 2017 Julian Sparber <julian@sparber.com>
+ *
+ * Teleport is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#ifdef HAVE_CONFIG_H
+#include <config.h>
+#endif
+
+#include <libsoup/soup.h>
+#include "beam-me-up-app.h"
+#include "beam-me-up-window.h"
+#include "beam-me-up-get.h"
+
+static int saveFile (SoupMessage *, const gchar *, const gchar *);
+static gchar * getFilePath (const gchar *, const gchar *);
+static int get (const gchar *, const gchar *, const gchar *, const gchar *);
+
+static gboolean debug;
+
+static void
+finished (SoupSession *session,
+          SoupMessage *msg,
+          gpointer target)
+{
+  //GVariant *target array: {originDevice, url, filename, downloadDirectory}
+  if ((char *) g_variant_get_string (
+                                     g_variant_get_child_value ((GVariant *) target, 2), NULL) != NULL) {
+    saveFile(msg,
+             (char *) g_variant_get_string (
+                                            g_variant_get_child_value ((GVariant *) target, 3), NULL),
+             (char *) g_variant_get_string (
+                                            g_variant_get_child_value ((GVariant *) target, 2), NULL));
+
+    create_finished_notification ((char *) g_variant_get_string (
+                                                                 g_variant_get_child_value ((GVariant *) target, 0), NULL),
+                                  0,
+                                  g_variant_get_string (
+                                                        g_variant_get_child_value ((GVariant *) target, 2), NULL),
+                                  target);
+  }
+}
+
+static int
+get (const gchar *url,
+     const gchar *originDevice,
+     const gchar *downloadDirectory,
+     const gchar *outputFilename) {
+  SoupSession *session;
+  SoupLogger *logger = NULL;
+  SoupMessage *msg;
+
+  if (!soup_uri_new (url)) {
+    g_printerr ("Could not parse '%s' as a URL\n", url);
+    return (1);
+  }
+
+  session = g_object_new (SOUP_TYPE_SESSION,
+                          SOUP_SESSION_ADD_FEATURE_BY_TYPE,
+                          SOUP_TYPE_CONTENT_DECODER,
+                          SOUP_SESSION_USER_AGENT,
+                          "teleport ",
+                          SOUP_SESSION_ACCEPT_LANGUAGE_AUTO,
+                          TRUE,
+                          NULL);
+
+  if (debug) {
+    logger = soup_logger_new (SOUP_LOGGER_LOG_BODY, -1);
+    soup_session_add_feature (session, SOUP_SESSION_FEATURE (logger));
+    g_object_unref (logger);
+  }
+
+  msg = soup_message_new ("GET", url);
+  soup_message_set_flags (msg, SOUP_MESSAGE_NO_REDIRECT);
+
+  g_object_ref (msg);
+  //soup_session_queue_message (session, msg, finished, loop);
+  if (outputFilename == NULL) {
+    soup_session_queue_message (session, msg, NULL, NULL);
+  }
+  else {
+    GVariantBuilder *builder;
+    GVariant *target;
+
+    builder = g_variant_builder_new (G_VARIANT_TYPE ("as"));
+    g_variant_builder_add (builder, "s", originDevice);
+    g_variant_builder_add (builder, "s", url);
+    g_variant_builder_add (builder, "s", outputFilename);
+    g_variant_builder_add (builder, "s", downloadDirectory);
+    target = g_variant_new ("as", builder);
+    g_variant_builder_unref (builder);
+
+    soup_session_queue_message (session, msg, finished, target);
+  }
+
+  return 0;
+}
+
+static int 
+saveFile (SoupMessage *msg,
+          const gchar *outputDirectory,
+          const gchar *outputFilename) {
+  const char *name;
+  FILE *outputFile = NULL;
+
+  name = soup_message_get_uri (msg)->path;
+
+  if (SOUP_STATUS_IS_TRANSPORT_ERROR (msg->status_code))
+    g_print ("%s: %d %s\n", name, msg->status_code, msg->reason_phrase);
+
+  if (SOUP_STATUS_IS_REDIRECTION (msg->status_code)) {
+    g_print ("%s: %d %s\n", name, msg->status_code, msg->reason_phrase);
+  } else if (SOUP_STATUS_IS_SUCCESSFUL (msg->status_code)) {
+    //if there is no file name and path the page will not get saved
+    if (outputFilename == NULL) {
+      //g_print ("%s: Got a file offered form a other peer. Will not save anything.\n", name);
+    }
+    else {
+      outputFile = fopen (getFilePath(outputDirectory, outputFilename), "w");
+
+      if (!outputFile)
+        g_printerr ("Error trying to create file %s.\n", outputFilename);
+
+      if (outputFile) {
+        fwrite (msg->response_body->data,
+                1,
+                msg->response_body->length,
+                outputFile);
+
+        fclose (outputFile);
+      }
+    }
+  }
+  return 0;
+}
+
+
+static gchar *
+getFilePath (const gchar *outputDirectory,
+             const gchar *outputFilename) {
+  return g_strdup_printf("%s/%s", outputDirectory,
+                         g_uri_escape_string(outputFilename, NULL, TRUE));
+}
+
+int
+teleport_get_do_client_notify (const gchar *url) {
+  get (g_strdup(url), NULL, NULL, NULL);
+  g_print("Offering selected file to other machine.\n");
+  return 0;
+}
+
+int 
+teleport_get_do_downloading (const char *originDevice,
+                             const char *url,
+                             const char *filename) {
+  const gchar *outputDirectory = teleport_get_download_directory();
+  g_print("Downloading %s to %s\n", url, g_uri_escape_string(filename, NULL, TRUE));
+  get (g_strdup(url), originDevice, outputDirectory, filename);
+  return 0;
+}