/*
   Copyright 2025 Yağız Zengin

   Licensed under the Apache License, Version 2.0 (the "License");
   you may not use this file except in compliance with the License.
   You may obtain a copy of the License at

       http://www.apache.org/licenses/LICENSE-2.0

   Unless required by applicable law or agreed to in writing, software
   distributed under the License is distributed on an "AS IS" BASIS,
   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   See the License for the specific language governing permissions and
   limitations under the License.
*/

#ifndef LIBPMT_LIB_HPP
#define LIBPMT_LIB_HPP

#include <CLI/CLI11.hpp>
#include <libhelper/lib.hpp>
#include <libpartition_map/lib.hpp>
#include <memory>
#include <string>
#include <vector>

#define PMT "libpmt"
#define PMTE "pmt"
#define PMTF "libpmt-function-manager"

// Quick access to variables.
#define VARS (*Variables)
// Quick access to partition map.
#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
__attribute__((format(printf, 1, 2))) void print(const char *format, ...);
__attribute__((format(printf, 1, 2))) void println(const char *format, ...);

// If there is a delimiter in the string, CLI::detail::split returns; if not, an
// empty vector is returned. And checks duplicate arguments.
std::vector<std::string> splitIfHasDelim(const std::string &s, char delim,
                                         bool checkForBadUsage = false);

// Process vectors with input strings. Use for [flag(s)]-[other flag(s)]
// situations
void processCommandLine(std::vector<std::string> &vec1,
                        std::vector<std::string> &vec2, const std::string &s1,
                        const std::string &s2, char delim,
                        bool checkForBadUsage = false);

// Setting ups buffer size
void setupBufferSize(uint64_t &size, const std::string &entry);

std::string getLibVersion();

std::string getAppVersion(); // Not Android app version (an Android app is
                             // planned!), tells pmt version.
} // namespace PartitionManager

#endif // #ifndef LIBPMT_LIB_HPP