1 files changed, 79 insertions, 0 deletions
diff --git a/special-mode-mgr/include/file.hpp b/special-mode-mgr/include/file.hpp
new file mode 100644
index 0000000..3ea0d50
--- /dev/null
+++ b/special-mode-mgr/include/file.hpp
@@ -0,0 +1,79 @@
+#pragma once
+
+#include <stdio.h>
+
+#include <filesystem>
+
+/** @class File
+ *  @brief Responsible for handling file pointer
+ *  Needed by putspent(3)
+ */
+class File
+{
+  private:
+    /** @brief handler for operating on file */
+    FILE* fp = NULL;
+
+    /** @brief File name. Needed in the case where the temp
+     *         needs to be removed
+     */
+    const std::string& name;
+
+    /** @brief Should the file be removed at exit */
+    bool removeOnExit = false;
+
+  public:
+    File() = delete;
+    File(const File&) = delete;
+    File& operator=(const File&) = delete;
+    File(File&&) = delete;
+    File& operator=(File&&) = delete;
+
+    /** @brief Opens file and uses it to do file operation
+     *
+     *  @param[in] name         - File name
+     *  @param[in] mode         - File open mode
+     *  @param[in] removeOnExit - File to be removed at exit or no
+     */
+    File(const std::string& filename, const std::string& mode,
+         bool removeExit = false) :
+        name(filename),
+        removeOnExit(removeExit)
+    {
+        fp = fopen(name.c_str(), mode.c_str());
+    }
+
+    /** @brief Opens file using provided file descriptor
+     *
+     *  @param[in] fd           - File descriptor
+     *  @param[in] name         - File name
+     *  @param[in] mode         - File open mode
+     *  @param[in] removeOnExit - File to be removed at exit or no
+     */
+    File(int fd, const std::string& filename, const std::string& mode,
+         bool removeExit = false) :
+        name(filename),
+        removeOnExit(removeExit)
+    {
+        fp = fdopen(fd, mode.c_str());
+    }
+
+    ~File()
+    {
+        if (fp)
+        {
+            fclose(fp);
+        }
+
+        // Needed for exception safety
+        if (removeOnExit && std::filesystem::exists(name))
+        {
+            std::filesystem::remove(name);
+        }
+    }
+
+    auto operator()()
+    {
+        return fp;
+    }
+};