git: port from mercurial repo.

Signed-off-by: Amlal El Mahrouss <amlal.elmahrouss@icloud.com>

1 files changed, 61 insertions, 0 deletions

diff --git a/Source/DriveManager.cxx b/Source/DriveManager.cxx
new file mode 100644
index 00000000..4c924b75
--- /dev/null
+++ b/Source/DriveManager.cxx
@@ -0,0 +1,61 @@
+/*
+ *	========================================================
+ *
+ *	hCore
+ * 	Copyright Mahrouss Logic, all rights reserved.
+ *
+ * 	========================================================
+ */
+
+#include <FSKit/NewFS.hxx>
+#include <NewKit/Utils.hpp>
+
+#include <KernelKit/DebugOutput.hpp>
+#include <KernelKit/DriveManager.hpp>
+
+namespace hCore
+{
+    DriveSelector::DriveSelector() : fDrive(nullptr) {}
+    
+    DriveSelector::~DriveSelector()
+    {
+    	if (fDrive)
+    	{
+            this->Unmount();
+    	}	
+    }
+	
+    DriveTraits& DriveSelector::GetMounted() { MUST_PASS(fDrive != nullptr); return *fDrive; }
+    
+    bool DriveSelector::Mount(DriveTraits* drive)
+    {
+    	if (drive &&
+            drive->fReady() &&
+            fDrive == nullptr)
+        {
+            fDrive = drive;
+            fDrive->fMount();
+
+            kcout << "[Mount] Mount drive: " << fDrive->fName << "\n";
+
+            return true;
+        }
+        
+        return false;
+    }
+
+    DriveTraits* DriveSelector::Unmount()
+    {
+        if (!fDrive)
+            return nullptr;
+
+        auto drivePointer = fDrive;
+
+        fDrive->fUnmount();
+        fDrive = nullptr;
+
+        kcout << "[Unmount] Mount drive: " << drivePointer->fName << "\n";
+
+        return drivePointer;
+    }
+} // namespace hCore