/*
 * 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.
 */
#include "zip_writer.h"

#include <cerrno>
#include <stdio.h>

#include "app_log_wrapper.h"
#include "contrib/minizip/zip.h"
#include "directory_ex.h"
#include "zip_internal.h"

using namespace OHOS::AppExecFwk;

namespace OHOS {
namespace AppExecFwk {
namespace LIBZIP {
namespace {
// Numbers of pending entries that trigger writting them to the ZIP file.
constexpr size_t g_MaxPendingEntriesCount = 50;
const std::string SEPARATOR = "/";
std::mutex g_mutex;;

bool AddFileContentToZip(zipFile zip_file, FilePath &file_path)
{
    char buf[kZipBufSize];
    if (!FilePathCheckValid(file_path.Value())) {
        APP_LOGI("filePath is invalid file_path=%{public}s", file_path.Value().c_str());
        return false;
    }
    if (!FilePath::PathIsValid(file_path)) {
        APP_LOGI("PathIsValid returns false");
        return false;
    }

    FILE *fp = fopen(file_path.Value().c_str(), "rb");
    if (fp == nullptr) {
        APP_LOGI("filePath to realPath failed filePath:%{private}s errno:%{public}s",
            file_path.Value().c_str(), strerror(errno));
        return false;
    }

    uint32_t num_bytes;
    while (!feof(fp)) {
        num_bytes = fread(buf, 1, kZipBufSize, fp);
        if (num_bytes > 0) {
            if (zipWriteInFileInZip(zip_file, buf, num_bytes) != ZIP_OK) {
                APP_LOGI("Could not write data to zip for path:%{private}s ", file_path.Value().c_str());
                fclose(fp);
                fp = nullptr;
                return false;
            }
        } else {
            break;
        }
    }
    fclose(fp);
    fp = nullptr;
    return true;
}

bool OpenNewFileEntry(
    zipFile zip_file, FilePath &path, bool isDirectory, struct tm *lastModified, const OPTIONS &options)
{
    std::string strPath = path.Value();

    if (isDirectory) {
        strPath += SEPARATOR;
    }

    return ZipOpenNewFileInZip(zip_file, strPath, options, lastModified);
}

bool CloseNewFileEntry(zipFile zip_file)
{
    return zipCloseFileInZip(zip_file) == ZIP_OK;
}

bool AddFileEntryToZip(zipFile zip_file, FilePath &relativePath, FilePath &absolutePath, const OPTIONS &options)
{
    struct tm *lastModified = GetCurrentSystemTime();
    if (lastModified == nullptr) {
        return false;
    }
    if (!OpenNewFileEntry(zip_file, relativePath, false, lastModified, options)) {
        return false;
    }
    bool success = AddFileContentToZip(zip_file, absolutePath);
    if (!CloseNewFileEntry(zip_file)) {
        APP_LOGI("CloseNewFileEntry returnValule is false");
        return false;
    }
    return success;
}

bool AddDirectoryEntryToZip(zipFile zip_file, FilePath &path, struct tm *lastModified, const OPTIONS &options)
{
    return OpenNewFileEntry(zip_file, path, true, lastModified, options) && CloseNewFileEntry(zip_file);
}

}  // namespace

zipFile ZipWriter::InitZipFileWithFd(PlatformFile zipFilefd)
{
    std::lock_guard<std::mutex> lock(g_mutex);
    if (zipFilefd == kInvalidPlatformFile) {
        return nullptr;
    }

    zipFile zip_file = OpenFdForZipping(zipFilefd, APPEND_STATUS_CREATE);
    if (!zip_file) {
        APP_LOGI("Couldn't create ZIP file for FD");
        return nullptr;
    }
    return zip_file;
}

zipFile ZipWriter::InitZipFileWithFile(const FilePath &zip_file_path)
{
    std::lock_guard<std::mutex> lock(g_mutex);
    FilePath zipFilePath = zip_file_path;
    if (zipFilePath.Value().empty()) {
        APP_LOGI("Path is empty");
        return nullptr;
    }

    zipFile zip_file = OpenForZipping(zipFilePath.Value(), APPEND_STATUS_CREATE);
    if (!zip_file) {
        APP_LOGI("Couldn't create ZIP file at path");
        return nullptr;
    }
    return zip_file;
}

ZipWriter::ZipWriter(zipFile zip_file) : zipFile_(zip_file)
{}

ZipWriter::~ZipWriter()
{
    pendingEntries_.clear();
}

bool ZipWriter::WriteEntries(const std::vector<std::pair<FilePath, FilePath>> &paths, const OPTIONS &options)
{
    return AddEntries(paths, options) && Close(options);
}

bool ZipWriter::AddEntries(const std::vector<std::pair<FilePath, FilePath>> &paths, const OPTIONS &options)
{
    if (!zipFile_) {
        return false;
    }
    pendingEntries_.insert(pendingEntries_.end(), paths.begin(), paths.end());
    return FlushEntriesIfNeeded(false, options);
}

bool ZipWriter::Close(const OPTIONS &options)
{
    bool success = FlushEntriesIfNeeded(true, options) && zipClose(zipFile_, nullptr) == ZIP_OK;
    zipFile_ = nullptr;
    return success;
}

bool ZipWriter::FlushEntriesIfNeeded(bool force, const OPTIONS &options)
{
    if (pendingEntries_.size() < g_MaxPendingEntriesCount && !force) {
        return true;
    }
    while (pendingEntries_.size() >= g_MaxPendingEntriesCount || (force && !pendingEntries_.empty())) {
        size_t entry_count = std::min(pendingEntries_.size(), g_MaxPendingEntriesCount);
        std::vector<std::pair<FilePath, FilePath>> relativePaths;
        relativePaths.insert(relativePaths.begin(), pendingEntries_.begin(), pendingEntries_.begin() + entry_count);
        pendingEntries_.erase(pendingEntries_.begin(), pendingEntries_.begin() + entry_count);
        for (size_t i = 0; i < relativePaths.size(); i++) {
            FilePath &relativePath = relativePaths[i].first;
            FilePath &absolutePath = relativePaths[i].second;
            bool isValid = FilePath::PathIsValid(absolutePath);
            bool isDir = FilePath::IsDir(absolutePath);
            if (isValid && !isDir) {
                if (!AddFileEntryToZip(zipFile_, relativePath, absolutePath, options)) {
                    APP_LOGI("Failed to write file");
                    return false;
                }
            } else {
                // Missing file or directory case.
                struct tm *last_modified = GetCurrentSystemTime();
                if (!AddDirectoryEntryToZip(zipFile_, relativePath, last_modified, options)) {
                    APP_LOGI("Failed to write directory");
                    return false;
                }
            }
        }
    }
    return true;
}

}  // namespace LIBZIP
}  // namespace AppExecFwk
}  // namespace OHOS