fix readlink in download_async_gui

1 files changed, 6 insertions, 5 deletions

diff --git a/src/DownloadUtils.cpp b/src/DownloadUtils.cpp
index 9834390..774fba9 100644
--- a/src/DownloadUtils.cpp
+++ b/src/DownloadUtils.cpp
@@ -214,11 +214,12 @@ namespace QuickMedia {
     }
 
     bool download_async_gui(const std::string &url, const std::string &file_manager_start_dir, bool use_youtube_dl, bool no_video) {
-        // TODO: Fix this not working when installed to /usr/bin/quickmedia for some reason
-        //char quickmedia_path[PATH_MAX];
-        //if(readlink("/proc/self/exe", quickmedia_path, sizeof(quickmedia_path)) == -1)
-        //    strcpy(quickmedia_path, "quickmedia");
-        const char *quickmedia_path = "quickmedia";
+        char quickmedia_path[PATH_MAX];
+        ssize_t bytes_written = readlink("/proc/self/exe", quickmedia_path, sizeof(quickmedia_path));
+        if(bytes_written == -1 || bytes_written == sizeof(quickmedia_path))
+            strcpy(quickmedia_path, "quickmedia");
+        else
+            quickmedia_path[bytes_written] = '\0';
 
         std::vector<const char*> args = { quickmedia_path, "download", "-u", url.c_str(), "--dir", file_manager_start_dir.c_str() };
         if(use_youtube_dl) args.push_back("--youtube-dl");