/* Distributed under the OSI-approved BSD 3-Clause License.  See accompanying
   file LICENSE.rst or https://cmake.org/licensing for details.  */
#pragma once

#include "cmConfigure.h" // IWYU pragma: keep

#include <iosfwd>
#include <string>

#include "cmArchiveWrite.h"
#include "cmCPackGenerator.h"

class cmCPackComponent;

/** \class cmCPackArchiveGenerator
 * \brief A generator base for libarchive generation.
 * The generator itself uses the libarchive wrapper
 * \ref cmArchiveWrite.
 *
 */
class cmCPackArchiveGenerator : public cmCPackGenerator
{
public:
  using Superclass = cmCPackGenerator;

  static cmCPackGenerator* Create7ZGenerator();
  static cmCPackGenerator* Create7ZStoreGenerator();
  static cmCPackGenerator* Create7ZDeflateGenerator();
  static cmCPackGenerator* Create7ZBzip2Generator();
  static cmCPackGenerator* Create7ZLzmaGenerator();
  static cmCPackGenerator* Create7ZLzma2Generator();
  static cmCPackGenerator* Create7ZZstdGenerator();
  static cmCPackGenerator* Create7ZPPMdGenerator();
  static cmCPackGenerator* CreateTBZ2Generator();
  static cmCPackGenerator* CreateTGZGenerator();
  static cmCPackGenerator* CreateTXZGenerator();
  static cmCPackGenerator* CreateTZGenerator();
  static cmCPackGenerator* CreateTZSTGenerator();
  static cmCPackGenerator* CreateTarGenerator();
  static cmCPackGenerator* CreateZIPGenerator();
  static cmCPackGenerator* CreateZipStoreGenerator();
  static cmCPackGenerator* CreateZipDeflateGenerator();
  static cmCPackGenerator* CreateZipBzip2Generator();
  static cmCPackGenerator* CreateZipLzmaGenerator();
  static cmCPackGenerator* CreateZipLzma2Generator();
  static cmCPackGenerator* CreateZipZstdGenerator();

  /**
   * Construct generator
   */
  cmCPackArchiveGenerator(cmArchiveWrite::Compress t, std::string format,
                          std::string extension);
  ~cmCPackArchiveGenerator() override;
  // Used to add a header to the archive
  virtual int GenerateHeader(std::ostream* os);
  // component support
  bool SupportsComponentInstallation() const override;

private:
  std::string GetArchiveFileName();
  // get archive component filename
  std::string GetArchiveComponentFileName(std::string const& component,
                                          bool isGroupName);

  class Deduplicator;

protected:
  int InitializeInternal() override;
  /**
   * Add the files belonging to the specified component
   * to the provided (already opened) archive.
   * @param[in,out] archive the archive object
   * @param[in] component the component whose file will be added to archive
   * @param[in] deduplicator file deduplicator utility.
   */
  int addOneComponentToArchive(cmArchiveWrite& archive,
                               cmCPackComponent* component,
                               Deduplicator* deduplicator);

  /**
   * The main package file method.
   * If component install was required this
   * method will call either PackageComponents or
   * PackageComponentsAllInOne.
   */
  int PackageFiles() override;
  /**
   * The method used to package files when component
   * install is used. This will create one
   * archive for each component group.
   */
  int PackageComponents(bool ignoreGroup);
  /**
   * Special case of component install where all
   * components will be put in a single installer.
   */
  int PackageComponentsAllInOne();

private:
  char const* GetNameOfClass() override { return "cmCPackArchiveGenerator"; }

  char const* GetOutputExtension() override
  {
    return this->OutputExtension.c_str();
  }

  int GetThreadCount() const;
  int GetCompressionLevel() const;
  std::string GetEncoding() const;

private:
  cmArchiveWrite::Compress Compress;
  std::string ArchiveFormat;
  std::string OutputExtension;
  int UID = 0;
  int GID = 0;
};
