/*
 * 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 "wm_trace.h"

#include <hitrace_meter.h>

#ifndef WINDOWS_PLATFORM
#include <securec.h>
#endif

namespace OHOS {
namespace Rosen {
namespace {
const size_t MAX_STRING_SIZE = 128;
}

bool WmTraceEnabled()
{
    return true;
}

void WmTraceBegin(const char* name)
{
    if (name == nullptr) {
        return;
    }
    std::string nameStr(name);
    StartTrace(HITRACE_TAG_WINDOW_MANAGER, nameStr);
}

void WmTraceEnd()
{
    if (WmTraceEnabled()) {
        FinishTrace(HITRACE_TAG_WINDOW_MANAGER);
    }
}

void WmAsyncTraceBegin(int32_t taskId, const char* name)
{
    if (name == nullptr) {
        return;
    }
    std::string nameStr(name);
    StartAsyncTrace(HITRACE_TAG_WINDOW_MANAGER, nameStr, taskId);
}

void WmAsyncTraceEnd(int32_t taskId, const char* name)
{
    if (name == nullptr) {
        return;
    }
    std::string nameStr(name);
    FinishAsyncTrace(HITRACE_TAG_WINDOW_MANAGER, nameStr, taskId);
}

void WmAsyncTraceWithArgs(bool isBegin, int32_t taskId, const char* format, ...)
{
    if (WmTraceEnabled()) {
        va_list args;
        va_start(args, format);
        WmAsyncTraceWithArgv(isBegin, taskId, format, args);
        va_end(args);
    }
}

void WmAsyncTraceWithArgv(bool isBegin, int32_t taskId, const char* format, va_list args)
{
    char name[MAX_STRING_SIZE] = { 0 };
    if (vsnprintf_s(name, sizeof(name), sizeof(name) - 1, format, args) < 0) {
        return;
    }
    if (isBegin) {
        WmAsyncTraceBegin(taskId, name);
    } else {
        WmAsyncTraceEnd(taskId, name);
    }
}

bool WmTraceBeginWithArgv(const char* format, va_list args)
{
    char name[MAX_STRING_SIZE] = { 0 };
    if (vsnprintf_s(name, sizeof(name), sizeof(name) - 1, format, args) < 0) {
        return false;
    }
    WmTraceBegin(name);
    return true;
}

bool WmTraceBeginWithArgs(const char* format, ...)
{
    if (WmTraceEnabled()) {
        va_list args;
        va_start(args, format);
        bool retVal = WmTraceBeginWithArgv(format, args);
        va_end(args);
        return retVal;
    }
    return false;
}

WmScopedTrace::WmScopedTrace(const char* format, ...) : traceEnabled_(WmTraceEnabled())
{
    if (traceEnabled_) {
        va_list args;
        va_start(args, format);
        traceEnabled_ = WmTraceBeginWithArgv(format, args);
        va_end(args);
    }
}

WmScopedTrace::~WmScopedTrace()
{
    if (traceEnabled_) {
        WmTraceEnd();
    }
}
} // namespace Rosen
} // namespace OHOS