1 files changed, 133 insertions, 0 deletions
diff --git a/libs/archive/src/extractcallback.h b/libs/archive/src/extractcallback.h
new file mode 100644
index 0000000..f95fc86
--- /dev/null
+++ b/libs/archive/src/extractcallback.h
@@ -0,0 +1,133 @@
+/*
+Mod Organizer archive handling
+
+Copyright (C) 2012 Sebastian Herbord, 2020 MO2 Team. All rights reserved.
+
+This library is free software; you can redistribute it and/or
+modify it under the terms of the GNU Lesser General Public
+License as published by the Free Software Foundation; either
+version 3 of the License, or (at your option) any later version.
+
+This library is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+Lesser General Public License for more details.
+
+You should have received a copy of the GNU Lesser General Public
+License along with this library; if not, write to the Free Software
+Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
+*/
+
+#ifndef EXTRACTCALLBACK_H
+#define EXTRACTCALLBACK_H
+
+#include <atomic>
+#include <chrono>
+#include <filesystem>
+#include <format>
+
+#include <7zip/Archive/IArchive.h>
+#include <7zip/IPassword.h>
+
+#include "compat.h"
+
+#include "archive.h"
+#include "formatter.h"
+#include "instrument.h"
+#include "multioutputstream.h"
+#include "unknown_impl.h"
+
+class FileData;
+
+class CArchiveExtractCallback : public IArchiveExtractCallback,
+                                public ICryptoGetTextPassword
+{
+
+  // A note: It appears that the IArchiveExtractCallback interface includes the
+  // IProgress interface, swo we need to respond to it
+  UNKNOWN_3_INTERFACE(IArchiveExtractCallback, ICryptoGetTextPassword, IProgress);
+
+public:
+  CArchiveExtractCallback(Archive::ProgressCallback progressCallback,
+                          Archive::FileChangeCallback fileChangeCallback,
+                          Archive::ErrorCallback errorCallback,
+                          Archive::PasswordCallback passwordCallback,
+                          Archive::LogCallback logCallback, IInArchive* archiveHandler,
+                          std::wstring const& directoryPath, FileData* const* fileData,
+                          std::size_t nbFiles, UInt64 totalFileSize,
+                          std::wstring* password);
+
+  virtual ~CArchiveExtractCallback();
+
+  void SetCanceled(bool aCanceled);
+
+  Z7_IFACE_COM7_IMP(IProgress)
+  Z7_IFACE_COM7_IMP(IArchiveExtractCallback)
+
+  // ICryptoGetTextPassword
+  STDMETHOD(CryptoGetTextPassword)(BSTR* aPassword) throw();
+
+private:
+  void reportError(const std::wstring& message);
+
+  template <class... Args>
+  void reportError(std::wformat_string<Args...> format, Args&&... args)
+  {
+    reportError(std::format(format, std::forward<Args>(args)...));
+  }
+
+  template <typename T>
+  bool getOptionalProperty(UInt32 index, int property, T* result) const;
+  template <typename T>
+  bool getProperty(UInt32 index, int property, T* result) const;
+
+private:
+  CComPtr<IInArchive> m_ArchiveHandler;
+
+  UInt64 m_Total;
+
+  std::filesystem::path m_DirectoryPath;
+  bool m_Extracting;
+  std::atomic<bool> m_Canceled;
+
+  struct
+  {
+    ArchiveTimers::Timer GetStream;
+    struct
+    {
+      ArchiveTimers::Timer SetMTime;
+      ArchiveTimers::Timer Close;
+      ArchiveTimers::Timer Release;
+      ArchiveTimers::Timer SetFileAttributesW;
+    } SetOperationResult;
+  } m_Timers;
+
+  struct CProcessedFileInfo
+  {
+    FILETIME MTime;
+    UInt32 Attrib;
+    bool isDir;
+    bool AttribDefined;
+    bool MTimeDefined;
+  } m_ProcessedFileInfo;
+
+  MultiOutputStream* m_OutputFileStream;
+  CComPtr<MultiOutputStream> m_OutFileStreamCom;
+
+  std::vector<std::filesystem::path> m_FullProcessedPaths;
+
+  FileData* const* m_FileData;
+  std::size_t m_NbFiles;
+  UInt64 m_TotalFileSize;
+  UInt64 m_LastCallbackFileSize;
+  UInt64 m_ExtractedFileSize;
+
+  Archive::ProgressCallback m_ProgressCallback;
+  Archive::FileChangeCallback m_FileChangeCallback;
+  Archive::ErrorCallback m_ErrorCallback;
+  Archive::PasswordCallback m_PasswordCallback;
+  Archive::LogCallback m_LogCallback;
+  std::wstring* m_Password;
+};
+
+#endif  // EXTRACTCALLBACK_H