/*
 * Copyright (c) 2021-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 <cstdio>
#include <image_type.h>
#include <iosfwd>
#include <iostream>
#include <memory>
#include <ostream>
#include <refbase.h>

#include "display_manager.h"
#include "parameters.h"
#include "snapshot_utils.h"

using namespace OHOS;
using namespace OHOS::Media;
using namespace OHOS::Rosen;
using OHOS::system::GetParameter;

// developer mode
static const std::string DEVELOPER_MODE_STATE_ON_DEFAULT = "false";
static const std::string DEVELOPER_MODE_PARAMETER = "const.security.developermode.state";
static const std::string IS_DEVELOPER_MODE = GetParameter(DEVELOPER_MODE_PARAMETER, DEVELOPER_MODE_STATE_ON_DEFAULT);

static bool GetScreenshotByCmdArgments(CmdArgments& cmdArgments, sptr<Display> display,
    std::shared_ptr<OHOS::Media::PixelMap>& pixelMap);

int main(int argc, char *argv[])
{
    CmdArgments cmdArgments;
    cmdArgments.fileName = "";

    if (!SnapShotUtils::ProcessArgs(argc, argv, cmdArgments)) {
        _exit(-1);
    }

    if (DEVELOPER_MODE_STATE_ON_DEFAULT == IS_DEVELOPER_MODE) {
        std::cout << "current mode is not developer mode, just return." << std::endl;
        _exit(-1);
    }

    auto display = DisplayManager::GetInstance().GetDisplayById(cmdArgments.displayId);
    if (display == nullptr) {
        std::cout << "error: GetDisplayById " << cmdArgments.displayId << " error!" << std::endl;
        _exit(-1);
    }

    std::cout << "process: display " << cmdArgments.displayId <<
        ": width " << display->GetWidth() << ", height " << display->GetHeight() << std::endl;

    // get PixelMap from DisplayManager API
    std::shared_ptr<OHOS::Media::PixelMap> pixelMap = nullptr;
    if (!GetScreenshotByCmdArgments(cmdArgments, display, pixelMap)) {
        _exit(-1);
    }

    bool ret = false;
    if (pixelMap != nullptr) {
        ret = SnapShotUtils::WriteToJpegWithPixelMap(cmdArgments.fileName, *pixelMap);
    }
    if (!ret) {
        std::cout << "\nerror: snapshot display " << cmdArgments.displayId <<
            ", write to " << cmdArgments.fileName << " as jpeg failed!" << std::endl;
        _exit(-1);
    }

    std::cout << "\nsuccess: snapshot display " << cmdArgments.displayId << " , write to " <<
        cmdArgments.fileName << " as jpeg, width " << pixelMap->GetWidth() <<
        ", height " << pixelMap->GetHeight() << std::endl;
    _exit(0);
}

static bool GetScreenshotByCmdArgments(CmdArgments& cmdArgments, sptr<Display> display,
    std::shared_ptr<OHOS::Media::PixelMap>& pixelMap)
{
    DmErrorCode errorCode;
    if (!cmdArgments.isWidthSet && !cmdArgments.isHeightSet) {
        // default width & height
        pixelMap = DisplayManager::GetInstance().GetScreenshot(cmdArgments.displayId, &errorCode, false);
    } else {
        if (!cmdArgments.isWidthSet) {
            cmdArgments.width = display->GetWidth();
            std::cout << "process: reset to display's width " << cmdArgments.width << std::endl;
        }
        if (!cmdArgments.isHeightSet) {
            cmdArgments.height = display->GetHeight();
            std::cout << "process: reset to display's height " << cmdArgments.height << std::endl;
        }
        if (!SnapShotUtils::CheckWidthAndHeightValid(cmdArgments.width, cmdArgments.height)) {
            std::cout << "error: width " << cmdArgments.width << " height " <<
            cmdArgments.height << " invalid!" << std::endl;
            return false;
        }
        SnapShotConfig snapConfig;
        snapConfig.displayId_ = cmdArgments.displayId;
        snapConfig.imageRect_ = { 0, 0, display->GetWidth(), display->GetHeight() };
        snapConfig.imageSize_ = { cmdArgments.width, cmdArgments.height };
        snapConfig.rotation_ = 0;
        pixelMap = DisplayManager::GetInstance().GetScreenshotwithConfig(snapConfig, &errorCode, false);
    }
    return true;
}