From 0e6ca31ed2eb75c82fc30e5c236c8c4682c571f3 Mon Sep 17 00:00:00 2001
From: amazing-username <kundeng94@gmail.com>
Date: Sat, 8 Jun 2019 00:09:26 -0400
Subject: [PATCH] Downloading functionality added #4 and #7

---
 src/Managers/ActionManager.cpp |  2 +-
 src/Managers/CommitManager.cpp | 34 +++++++++++++++++--
 src/Models/Song.h              |  3 +-
 src/Syncers/Download.cpp       | 61 ++++++++++++++++++++++++++++++++++
 src/Syncers/Download.h         |  5 +++
 src/Syncers/SyncerBase.h       |  3 ++
 6 files changed, 104 insertions(+), 4 deletions(-)

diff --git a/src/Managers/ActionManager.cpp b/src/Managers/ActionManager.cpp
index d1ba899..1be285d 100644
--- a/src/Managers/ActionManager.cpp
+++ b/src/Managers/ActionManager.cpp
@@ -73,7 +73,7 @@ namespace Managers
 	{
 		supportedFlags = vector<string>{
 			"-u", "-p", "-t", "-h", "-s",
-			"-d", "-D"
+			"-d", "-D", "-b"
 		};
 	}
 	void ActionManager::validateAction()
diff --git a/src/Managers/CommitManager.cpp b/src/Managers/CommitManager.cpp
index 4709ed5..16fdf0a 100644
--- a/src/Managers/CommitManager.cpp
+++ b/src/Managers/CommitManager.cpp
@@ -6,6 +6,7 @@
 #include"Models/Song.h"
 #include"Parsers/APIParser.h"
 #include"Syncers/Delete.h"
+#include"Syncers/Download.h"
 #include"Syncers/Upload.h"
 
 #include"TokenManager.h"
@@ -23,6 +24,7 @@ using Models::Song;
 using Parsers::APIParser;
 using Models::IcarusAction;
 using Syncers::Delete;
+using Syncers::Download;
 using Syncers::Upload;
 
 namespace Managers
@@ -49,7 +51,7 @@ namespace Managers
 			case downloadAct:
 				downloadSong();
 				break;
-			case retrieveAct:
+			case retrieveAct: // No plans to imeplement
 				break;
 			case uploadAct:
 				uploadSong();
@@ -96,7 +98,35 @@ namespace Managers
 	}
 	void CommitManager::downloadSong()
 	{
-		// TODO: Implement this functionality
+		cout<<"Starting downloading process..."<<endl;
+		UserManager usrMgr{icaAction};
+		auto user = usrMgr.retrieveUser();
+
+		APIParser apiPrs{icaAction};
+		auto api = apiPrs.retrieveAPI();
+
+		TokenManager tk{user, api};
+		auto token = tk.requestToken();
+
+		Song song{};
+
+		for (auto arg : icaAction.flags)
+		{
+			auto flag = arg.flag;
+			auto value = arg.value;
+
+			if (flag.compare("-d") == 0)
+			{
+				song.songPath.assign(arg.value);
+			}
+			if (flag.compare("-b") == 0)
+			{
+				song.id = atoi(value.c_str());
+			}
+		}
+
+		Download dnld{api};
+		dnld.downloadSong(token, song);
 	}
 	void CommitManager::uploadSong()
 	{
diff --git a/src/Models/Song.h b/src/Models/Song.h
index 3571157..085037c 100644
--- a/src/Models/Song.h
+++ b/src/Models/Song.h
@@ -15,7 +15,8 @@ namespace Models
 		std::string genre;
 		int year;
 		int duration;
-		char* songData;
+		char *songData;
+		std::string  data;
 		std::string songPath;
 	};
 } 
diff --git a/src/Syncers/Download.cpp b/src/Syncers/Download.cpp
index 2b367f4..4c680bc 100644
--- a/src/Syncers/Download.cpp
+++ b/src/Syncers/Download.cpp
@@ -1,5 +1,6 @@
 #include"Download.h"
 
+#include<exception>
 #include<iostream>
 #include<fstream>
 
@@ -7,24 +8,31 @@
 
 using std::cout;
 using std::endl;
+using std::exception;
 using std::ofstream;
 using std::string;
 
 using Models::API;
+using Models::Song;
+using Models::Token;
 
 namespace Syncers
 {
+	#pragma
 	Download::Download() { }
 	Download::Download(API api)
 	{
 		this->api = api;
+		this->api.endpoint = "song/data";
 	}
 	Download::Download(string filePath)
 	{
 		downloadFilePath = filePath;
 	}
+	#pragma Constructors
 
 
+	#pragma
 	void Download::downloadSong(const int id)
 	{
 		string urlRoot = "http://192.168.1.5";
@@ -45,4 +53,57 @@ namespace Syncers
 		cout<<r.header["content-type"]<<endl;
 		cout<<r.header["content-disposition"]<<endl;
 	}
+	void Download::downloadSong(const Token token, Song song)
+	{
+		try
+		{
+			string url = retrieveUrl(song);
+			song.songPath.append("track.mp3");
+			cout<<"song path "<<song.songPath<<endl;
+			string auth{token.tokenType};
+			auth.append(" " + token.accessToken);
+			auto r = cpr::Get(cpr::Url(url), 
+				cpr::Header{{"authorization", auth}});
+
+			int statusCode = r.status_code;
+
+			if (statusCode == OK)
+			{
+				song.data = r.text;
+				saveSong(&song);
+			}
+
+
+			cout<<"finsihed with status code "<<statusCode<<endl;
+		}
+		catch (exception e)
+		{
+			auto msg = e.what();
+			cout<<msg<<endl;
+		}
+	}
+
+	string Download::retrieveUrl(Song song)
+	{
+		string url{api.url + "api/" + api.version + "/" + 
+			api.endpoint + "/"};
+
+		url.append(std::to_string(song.id));
+		cout<<"url "<<url<<endl;
+
+		return url;
+	}
+
+	void Download::saveSong(Song *song)
+	{
+		cout<<"\nSaving song to: "<<song->songPath<<endl;
+		int bufferLength = song->data.length();
+		const char *data = song->data.c_str();
+		cout<<"buff length  "<<bufferLength<<endl;
+
+		ofstream saveSong{song->songPath, std::ios::binary};
+		saveSong.write(data, bufferLength);
+		saveSong.close();
+	}
+	#pragma Functions
 }
diff --git a/src/Syncers/Download.h b/src/Syncers/Download.h
index 1a097ca..aae0bfd 100644
--- a/src/Syncers/Download.h
+++ b/src/Syncers/Download.h
@@ -6,6 +6,7 @@
 
 #include"Models/API.h"
 #include"Models/Song.h"
+#include"Models/Token.h"
 
 #include"SyncerBase.h"
 
@@ -19,8 +20,12 @@ namespace Syncers
 		Download(std::string);
 
 		void downloadSong(int);
+		void downloadSong(const Models::Token token, Models::Song);
 	private:
+		std::string retrieveUrl(Models::Song);
+
 		std::string downloadFilePath;
+		void saveSong(Models::Song*);
 	};
 }
 
diff --git a/src/Syncers/SyncerBase.h b/src/Syncers/SyncerBase.h
index 4d0d983..78ae5f1 100644
--- a/src/Syncers/SyncerBase.h
+++ b/src/Syncers/SyncerBase.h
@@ -11,6 +11,9 @@ namespace Syncers
 	{
 	protected:
 		Models::API api;
+		const int OK = 200;
+		const int UNAUTHORIZED = 401;
+		const int NOTFOUND = 404;
 	};
 }