pmt: improve PureTuple class and improve function manager.

include/PartitionManager/PartitionManager.hpp

@@ -34,63 +34,6 @@
 #define PART_MAP (*VARS.PartMap)
 
 namespace PartitionManager {
-enum basic_function_flags {
-  NO_SU = 1,
-  NO_MAP_CHECK = 2,
-  ADB_SUFFICIENT = 3,
-};
-
-// All function classes must inherit from this class.
-class basic_function {
-public:
-  CLI::App *cmd = nullptr;
-  std::vector<int> flags = {};
-
-  virtual bool init(CLI::App &_app) = 0;
-  virtual bool run() = 0;
-
-  [[nodiscard]] virtual bool isUsed() const = 0;
-  [[nodiscard]] virtual const char *name() const = 0;
-
-  virtual ~basic_function() = default;
-};
-
-// A class for function management.
-class basic_function_manager final {
-private:
-  std::vector<std::unique_ptr<basic_function>> _functions;
-
-public:
-  void registerFunction(std::unique_ptr<basic_function> _func, CLI::App &_app);
-
-  [[nodiscard]] bool hasFlagOnUsedFunction(int flag) const;
-  [[nodiscard]] bool isUsed(const std::string &name) const;
-  [[nodiscard]] bool handleAll() const;
-};
-
-class basic_variables final {
-public:
-  basic_variables();
-
-  std::unique_ptr<PartitionMap::BuildMap> PartMap;
-
-  std::string searchPath, logFile;
-  bool onLogical;
-  bool quietProcess;
-  bool verboseMode;
-  bool viewVersion;
-  bool forceProcess;
-};
-
-using FunctionBase = basic_function;
-using FunctionManager = basic_function_manager;
-using FunctionFlags = basic_function_flags;
-using VariableTable = basic_variables;
-using Error = Helper::Error;
-
-extern std::unique_ptr<VariableTable> Variables;
-extern FILE *pstdout, *pstderr;
-
 int Main(int argc, char **argv);
 
 // Print messages if not using quiet mode
@@ -116,6 +59,114 @@ std::string getLibVersion();
 
 std::string getAppVersion(); // Not Android app version (an Android app is
                              // planned!), tells pmt version.
+
+enum basic_function_flags {
+  NO_SU = 1,
+  NO_MAP_CHECK = 2,
+  ADB_SUFFICIENT = 3,
+};
+
+// All function classes must inherit from this class.
+class basic_function {
+public:
+  CLI::App *cmd = nullptr;
+  std::vector<int> flags = {};
+
+  virtual bool init(CLI::App &_app) = 0;
+  virtual bool run() = 0;
+
+  [[nodiscard]] virtual bool isUsed() const = 0;
+  [[nodiscard]] virtual const char *name() const = 0;
+
+  virtual ~basic_function() = default;
+};
+
+// A class for function management.
+template <class _Type>
+class basic_manager {
+private:
+  std::vector<std::unique_ptr<_Type>> _functions;
+
+public:
+  void registerFunction(std::unique_ptr<_Type> _func, CLI::App &_app) {
+    LOGN(PMTF, INFO) << "registering: " << _func->name() << std::endl;
+    for (const auto &f : _functions) {
+      if (std::string(_func->name()) == std::string(f->name())) {
+        LOGN(PMTF, INFO) << "Is already registered: " << _func->name()
+                         << ". Skipping." << std::endl;
+        return;
+      }
+    }
+    if (!_func->init(_app))
+      throw Helper::Error("Cannot init: %s", _func->name());
+    _functions.push_back(std::move(_func));
+    LOGN(PMTF, INFO) << _functions.back()->name() << " successfully registered."
+                     << std::endl;
+  }
+
+  [[nodiscard]] bool hasFlagOnUsedFunction(int flag) const {
+    for (const auto &func : _functions) {
+      if (func->isUsed()) {
+        std::for_each(func->flags.begin(), func->flags.end(), [&](const int x) {
+          LOGN(PMTF, INFO) << "Used flag " << x << " on " << func->name()
+                           << std::endl;
+        });
+        return std::find(func->flags.begin(), func->flags.end(), flag) !=
+               func->flags.end();
+      }
+    }
+    return false;
+  }
+
+  [[nodiscard]] bool isUsed(const std::string &name) const {
+    if (_functions.empty()) return false;
+    for (const auto &func : _functions) {
+      if (func->name() == name) return func->isUsed();
+    }
+    return false;
+  }
+
+  [[nodiscard]] bool handleAll() const {
+    LOGN(PMTF, INFO) << "running caught commands in command-line."
+                   << std::endl;
+    for (const auto &func : _functions) {
+      if (func->isUsed()) {
+        LOGN(PMTF, INFO) << func->name()
+                         << " is calling because used in command-line."
+                         << std::endl;
+        return func->run();
+      }
+    }
+
+    LOGN(PMTF, INFO) << "not found any used function from command-line."
+                     << std::endl;
+    println("Target progress is not specified. Specify a progress.");
+    return false;
+  }
+};
+
+class basic_variables final {
+public:
+  basic_variables();
+
+  std::unique_ptr<PartitionMap::BuildMap> PartMap;
+
+  std::string searchPath, logFile;
+  bool onLogical;
+  bool quietProcess;
+  bool verboseMode;
+  bool viewVersion;
+  bool forceProcess;
+};
+
+using FunctionBase = basic_function;
+using FunctionManager = basic_manager<FunctionBase>;
+using FunctionFlags = basic_function_flags;
+using VariableTable = basic_variables;
+using Error = Helper::Error;
+
+extern std::unique_ptr<VariableTable> Variables;
+extern FILE *pstdout, *pstderr;
 } // namespace PartitionManager
 
-#endif // #ifndef LIBPMT_LIB_HPP
+#endif // #ifndef LIBPMT_LIB_HPP