Merge pull request #18 from Abir-Tx/Abir-Tx/issue13 & issue6

Add DIR Tree view(#13) & goback globally in #6

include/dirTree.hpp

@@ -0,0 +1,30 @@
+#if !defined(DIRTREE_H)
+#define DIRTREE_H
+#include <iostream>
+#include <iomanip>
+#include <fstream>
+#include <iostream>
+
+void dirTree()
+{
+ using namespace std;
+ string driveLetterHolder;
+
+ ifstream driveLetterReader("C:\\SFO\\Data\\DriveLetterKeeper.dat");
+
+ while(getline(driveLetterReader, driveLetterHolder))
+ {
+ driveLetterHolder = driveLetterHolder;
+ }
+
+ //Starting the dir tree
+ system("cls");
+ system("color 03");
+ cout<<setw(75)<<setfill(' ')<<"Directory Tree Viewer"<<endl<<endl;
+
+ cout<<"Printing the folders structure view inside the selected drive: "<<endl;
+ std::cout<<std::endl;
+ string treeDir = "tree /f "+driveLetterHolder+":\\\"University Study\"";
+ system(treeDir.c_str());
+}
+#endif // DIRTREE_H

include/driveLetterTaker.hpp

@@ -2,6 +2,8 @@
 #define DRIVELETTERTAKER_H
 #include <iostream>
 #include <iomanip>
+#include <fstream>
+#include <direct.h>
 
 //Global Variables:
 static std::string driveLetter;
@@ -12,6 +14,16 @@ void driveTaker()
 
  std::cout << "What is your drive letter: ";
  std::cin >> driveLetter;
+
+ //Keeping the data in a file for using it in other funcs
+ //Creating the appData folder:
+ _mkdir("C:\\SFO");
+ _mkdir("C:\\SFO\\Data");
+
+ //File operation:
+ std::ofstream driveLetterKeeper("C:\\SFO\\Data\\DriveLetterKeeper.dat");
+ driveLetterKeeper<<driveLetter;
+ driveLetterKeeper.close();
 }
 
 #endif // DRIVELETTERTAKER_H

include/home.hpp

@@ -5,6 +5,7 @@
 #include <vector>
 #include <direct.h>
 #include <iomanip>
+#include "../include/dirTree.hpp"
 
 //User Defined Headers
 #include "../include/create.hpp"
@@ -28,7 +29,7 @@ void home()
 
  std::cout << "Your Options: " << std::endl;
 
- std::vector<std::string> homeOptions = {"Create Mode", "About"};
+ std::vector<std::string> homeOptions = {"Create Mode", "About", "Directory Tree", "Exit"};
 
  for (int i = 0; i < homeOptions.size(); i++)
  {
@@ -56,6 +57,15 @@ void home()
  case 2:
  about();
  break;
+ case 3:
+ dirTree();
+ break;
+ case 4:
+ system("cls");
+ system("color 04");
+ std::cout<<"Exitin the program.......";
+ exit(0);
+ break;
  default:
  std::cout << "Invalid";
  break;

src/StudyFolderOrgranizer.cpp

@@ -4,6 +4,21 @@ Anyone can modify this codes to meet their needs if they like the idea of the so
 #include <iostream>
 #include "../include/home.hpp"
 
-int main(){
- home();
+int main()
+{
+ int choice;
+ do
+ {
+ home();
+ std::cout<<std::endl;
+ std::cout << "Press 1 to go to homepage again & others to Exit: ";
+ std::cin >> choice;
+ } while (choice == 1);
+ if(choice != 1)
+ {
+ system("cls");
+ system("color 04");
+ std::cout<<"Exitin the program.......";
+ exit(0);
+ }
 }