/*
 * Copyright (c) 2022 Huawei Device Co., Ltd.
 * 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 OHOS_ABILITY_BASE_EXTRACTOR_H
#define OHOS_ABILITY_BASE_EXTRACTOR_H

#include <memory>
#include <mutex>
#include <optional>
#include <set>
#include <string>
#include <unordered_map>
#include <vector>

#include "file_mapper.h"
#include "zip_file.h"

namespace OHOS {
namespace AbilityBase {
struct FileInfo {
    std::string fileName;
    uint32_t offset = 0;
    uint32_t length = 0;
    uint16_t lastModTime = 0;
    uint16_t lastModDate = 0;
};

class Extractor {
public:
    explicit Extractor(const std::string &source);
    virtual ~Extractor();

    /**
     * @brief Open compressed file.
     * @return Returns true if the file is successfully opened; returns false otherwise.
     */
    virtual bool Init();

    /**
     * @brief Extract to dest stream by file name.
     * @param fileName Indicates the file name.
     * @param dest Indicates the obtained std::ostream object.
     * @return Returns true if the file extracted successfully; returns false otherwise.
     */
    bool ExtractByName(const std::string &fileName, std::ostream &dest) const;
    /**
     * @brief Get specified type names in a zip file.
     * @param fileNames Indicates the obtained file names in zip.
     * @param suffix Indicates the suffix of file.
     */
    void GetSpecifiedTypeFiles(std::vector<std::string> &fileNames, const std::string &suffix);
    /**
     * @brief Has entry by name.
     * @param entryName Indicates the entry name.
     * @return Returns true if the ZipEntry is successfully finded; returns false otherwise.
     */
    bool HasEntry(const std::string &fileName) const;
    bool IsDirExist(const std::string &dir);
    bool IsStageBasedModel(std::string abilityName);
    bool GetFileBuffer(const std::string& srcPath, std::ostringstream& dest);
    bool GetFileList(const std::string& srcPath, std::vector<std::string>& assetList);
    bool GetFileList(const std::string &srcPath, std::set<std::string> &fileSet);
    bool IsSameHap(const std::string& hapPath) const;

    std::unique_ptr<FileMapper> GetData(const std::string &fileName, bool safeRegion = false) const;
    /**
     * Do not use this method unless you exactly know what you are doing.
     * For file item that user will handle errors, to mmap to safe region.
     * User should make sure the extractor's release goes after the data's.
     */
    std::unique_ptr<FileMapper> GetMmapData(const std::string &fileName);

    bool UnzipData(std::unique_ptr<FileMapper> fileMapper, std::unique_ptr<uint8_t[]> &dataPtr, size_t &len) const;
    bool IsStageModel();

    bool GetFileInfo(const std::string &fileName, FileInfo &fileInfo) const;

    bool IsHapCompress(const std::string &fileName) const;

    bool ExtractToBufByName(const std::string &fileName, std::unique_ptr<uint8_t[]> &dataPtr, size_t &len);
    /**
     * For abc file only, to mmap to safe region.
     */
    std::shared_ptr<FileMapper> GetSafeData(const std::string &fileName);
    void SetCacheMode(CacheMode cacheMode)
    {
        zipFile_.SetCacheMode(cacheMode);
    }
private:
    ZipFile zipFile_;
    bool initial_ = false;
    std::string hapPath_;
    std::optional<bool> isStageModel_;
};

class ExtractorUtil {
public:
    static std::string GetLoadFilePath(const std::string &hapPath);
    static std::shared_ptr<Extractor> GetExtractor(const std::string &hapPath, bool &newCreate, bool cache = false);
    static void DeleteExtractor(const std::string &hapPath);

private:
    static std::mutex mapMutex_;
    static std::unordered_map<std::string, std::shared_ptr<Extractor>> extractorMap_;
};
}  // namespace AbilityBase
}  // namespace OHOS
#endif  // OHOS_ABILITY_BASE_EXTRACTOR_H