exiv2/app/exiv2app.hpp

// SPDX-License-Identifier: GPL-2.0-or-later
/*!
  @brief   Defines class Params, used for the command line handling of exiv2
  @author  Andreas Huggel (ahu)
           <a href="mailto:ahuggel@gmx.net">ahuggel@gmx.net</a>
  @date    08-Dec-03, ahu: created
 */
#ifndef EXIV2APP_HPP_
#define EXIV2APP_HPP_

// *****************************************************************************
// included header files
#include <exiv2/exiv2.hpp>

#include "getopt.hpp"
#include "types.hpp"

// + standard includes
#include <iostream>
#include <regex>
#include <set>

//! Command identifiers
enum class CmdId {
  invalid,
  add,
  set,
  del,
  reg,
};
//! Metadata identifiers
enum class MetadataId : uint32_t {
  invalid = Exiv2::mdNone,  // 0
  exif = Exiv2::mdExif,     // 1
  iptc = Exiv2::mdIptc,     // 2
  xmp = Exiv2::mdXmp,       // 8
};

inline MetadataId operator&(MetadataId x, MetadataId y) {
  return (MetadataId)(uint32_t(x) & uint32_t(y));
}

inline MetadataId operator|(MetadataId x, MetadataId y) {
  return (MetadataId)(uint32_t(x) | uint32_t(y));
}

inline MetadataId& operator|=(MetadataId& x, MetadataId y) {
  return x = x | y;
}

//! Structure for one parsed modification command
struct ModifyCmd {
  //! C'tor
  ModifyCmd() = default;
  CmdId cmdId_{CmdId::invalid};                 //!< Command identifier
  std::string key_;                             //!< Exiv2 key string
  MetadataId metadataId_{MetadataId::invalid};  //!< Metadata identifier
  Exiv2::TypeId typeId_{Exiv2::invalidTypeId};  //!< Exiv2 type identifier
  //! Flag to indicate if the type was explicitly specified (true)
  bool explicitType_{false};
  std::string value_;  //!< Data
};
//! Container for modification commands
using ModifyCmds = std::vector<ModifyCmd>;
//! Structure to link command identifiers to strings
using CmdIdAndString = std::pair<CmdId, std::string>;
/*!
  @brief Implements the command line handling for the program.

  Derives from Util::Getopt to use the command line argument parsing
  functionality provided there. This class is implemented as a singleton,
  i.e., there is only one global instance of it, which can be accessed
  from everywhere.

  <b>Usage example:</b> <br>
  @code
  #include "params.h"

  int main(int argc, char* const argv[])
  {
      Params& params = Params::instance();
      if (params.getopt(argc, argv)) {
          params.usage();
          return 1;
      }
      if (params.help_) {
          params.help();
          return 0;
      }
      if (params.version_) {
          params.version();
          return 0;
      }

      // do something useful here...

      return 0;
  }
  @endcode
 */

class Params : public Util::Getopt {
 private:
  std::string optstring_;

 public:
  //! Container for command files
  using CmdFiles = std::vector<std::string>;
  //! Container for commands from the command line
  using CmdLines = std::vector<std::string>;
  //! Container to store filenames.
  using Files = std::vector<std::string>;
  //! Container for preview image numbers
  using PreviewNumbers = std::set<int>;
  //! Container for keys
  using Keys = std::vector<std::string>;

  /*!
    @brief Controls all access to the global Params instance.
    @return Reference to the global Params instance.
  */
  static Params& instance();

  //! Prevent copy-construction: not implemented.
  ~Params() override = default;
  Params(const Params&) = delete;
  Params& operator=(const Params&) = delete;

  //! Enumerates print modes
  enum PrintMode {
    pmSummary,
    pmList,
    pmComment,
    pmPreview,
    pmStructure,
    pmXMP,
    pmIccProfile,
    pmRecursive,
  };

  //! Individual items to print, bitmap
  enum PrintItem : uint32_t {
    prTag = 1,
    prGroup = 2,
    prKey = 4,
    prName = 8,
    prLabel = 16,
    prType = 32,
    prCount = 64,
    prSize = 128,
    prValue = 256,
    prTrans = 512,
    prHex = 1024,
    prSet = 2048
  };

  //! Enumerates common targets, bitmap
  enum CommonTarget : uint32_t {
    ctExif = 1,
    ctIptc = 2,
    ctComment = 4,
    ctThumb = 8,
    ctXmp = 16,
    ctXmpSidecar = 32,
    ctPreview = 64,
    ctIccProfile = 128,
    ctXmpRaw = 256,
    ctStdInOut = 512,
    ctIptcRaw = 1024
  };

  //! Enumerates the policies to handle existing files in rename action
  enum FileExistsPolicy {
    overwritePolicy,
    renamePolicy,
    askPolicy,
  };

  //! Enumerates year, month and day adjustments.
  enum Yod {
    yodYear,
    yodMonth,
    yodDay,
  };

  //! Structure for year, month and day adjustment command line arguments.
  struct YodAdjust {
    bool flag_;           //!< Adjustment flag.
    const char* option_;  //!< Adjustment option string.
    int64_t adjustment_;  //!< Adjustment value.
  };

  bool help_{false};                              //!< Help option flag.
  bool version_{false};                           //!< Version option flag.
  bool verbose_{false};                           //!< Verbose (talkative) option flag.
  bool force_{false};                             //!< Force overwrites flag.
  bool binary_{false};                            //!< Suppress long binary values.
  bool unknown_{true};                            //!< Suppress unknown tags.
  bool preserve_{false};                          //!< Preserve timestamps flag.
  bool timestamp_{false};                         //!< Rename also sets the file timestamp.
  bool timestampOnly_{false};                     //!< Rename only sets the file timestamp.
  FileExistsPolicy fileExistsPolicy_{askPolicy};  //!< What to do if file to rename exists.
  bool adjust_{false};                            //!< Adjustment flag.
  PrintMode printMode_{pmSummary};                //!< Print mode.
  PrintItem printItems_{0};                       //!< Print items.
  MetadataId printTags_{Exiv2::mdNone};           //!< Print tags (bitmap of MetadataId flags).
  //! %Action (integer rather than TaskType to avoid dependency).
  int action_{0};
  CommonTarget target_;  //!< What common target to process.

  int64_t adjustment_{0};          //!< Adjustment in seconds.
  YodAdjust yodAdjust_[3];         //!< Year, month and day adjustment info.
  std::string format_;             //!< Filename format (-r option arg).
  bool formatSet_{false};          //!< Whether the format is set with -r
  CmdFiles cmdFiles_;              //!< Names of the modification command files
  CmdLines cmdLines_;              //!< Commands from the command line
  ModifyCmds modifyCmds_;          //!< Parsed modification commands
  std::string jpegComment_;        //!< Jpeg comment to set in the image
  std::string directory_;          //!< Location for files to extract/insert
  std::string suffix_;             //!< File extension of the file to insert
  Files files_;                    //!< List of non-option arguments.
  PreviewNumbers previewNumbers_;  //!< List of preview numbers
  std::vector<std::regex> greps_;  //!< List of keys to 'grep' from the metadata
  Keys keys_;                      //!< List of keys to match from the metadata
  std::string charset_;            //!< Charset to use for UNICODE Exif user comment

  Exiv2::DataBuf stdinBuf;  //!< DataBuf with the binary bytes from stdin

 private:
  bool first_{true};

  Params();

  //! @name Helpers
  //@{
  int setLogLevel(const std::string& optarg);
  int evalGrep(const std::string& optarg);
  int evalKey(const std::string& optarg);
  int evalRename(int opt, const std::string& optarg);
  int evalAdjust(const std::string& optarg);
  int evalYodAdjust(const Yod& yod, const std::string& optarg);
  int evalPrint(const std::string& optarg);
  int evalPrintFlags(const std::string& optarg);
  int evalDelete(const std::string& optarg);
  int evalExtract(const std::string& optarg);
  int evalInsert(const std::string& optarg);
  int evalModify(int opt, const std::string& optarg);
  //@}

 public:
  /*!
    @brief Call Getopt::getopt() with optstring, to initiate command line
           argument parsing, perform consistency checks after all command line
           arguments are parsed.

    @param argc Argument count as passed to main() on program invocation.
    @param argv Argument array as passed to main() on program invocation.

    @return 0 if successful, >0 in case of errors.
   */
  int getopt(int argc, char* const argv[]);

  //! Handle options and their arguments.
  int option(int opt, const std::string& optarg, int optopt) override;

  //! Handle non-option parameters.
  int nonoption(const std::string& argv) override;

  //! Print a minimal usage note to an output stream.
  void usage(std::ostream& os = std::cout) const;

  //! Print further usage explanations to an output stream.
  void help(std::ostream& os = std::cout) const;

  //! Print version information to an output stream.
  static void version(bool verbose = false, std::ostream& os = std::cout);

  //! getStdin binary data read from stdin to DataBuf
  /*
      stdin can be used by multiple images in the exiv2 command line:
      For example: $ cat foo.icc | exiv2 -iC- a.jpg b.jpg c.jpg will modify the ICC profile in several images.
  */
  void getStdin(Exiv2::DataBuf& buf);

};  // class Params

inline Params::CommonTarget operator|(Params::CommonTarget x, Params::CommonTarget y) {
  return (Params::CommonTarget)(uint32_t(x) | uint32_t(y));
}

inline Params::CommonTarget& operator|=(Params::CommonTarget& x, Params::CommonTarget y) {
  return x = x | y;
}

inline Params::PrintItem operator|(Params::PrintItem x, Params::PrintItem y) {
  return (Params::PrintItem)(uint32_t(x) | uint32_t(y));
}

inline Params::PrintItem& operator|=(Params::PrintItem& x, Params::PrintItem y) {
  return x = x | y;
}

#endif  // #ifndef EXIV2APP_HPP_