/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#ifndef ArchiveReader_h_
#define ArchiveReader_h_

#include <stdio.h>
#include "mar.h"

#ifdef XP_WIN
#  include <windows.h>

typedef WCHAR NS_tchar;
#else
typedef char NS_tchar;
#endif

// This class provides an API to extract files from an update archive.
class ArchiveReader {
 public:
  ArchiveReader() = default;
  ~ArchiveReader() { Close(); }

  int Open(const NS_tchar* path);
  int VerifySignature();
  int VerifyProductInformation(const char* MARChannelID,
                               const char* appVersion);
  void Close();

  int ExtractFile(const char* item, const NS_tchar* destination);
  int ExtractFileToStream(const char* item, FILE* fp);

 private:
  int ExtractItemToStream(const MarItem* item, FILE* fp);

  MarFile* mArchive = nullptr;
  uint8_t* mInBuf = nullptr;
  uint8_t* mOutBuf = nullptr;
  size_t mInBufSize = 262144;
  size_t mOutBufSize = 262144;
};

#endif  // ArchiveReader_h_
