summaryrefslogtreecommitdiff
path: root/source/slang-record-replay/util/record-utility.cpp
blob: 8e3486ca5f69a7f90c74718c5474ad8e338857d8 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83

#include <string.h>
#include <stdlib.h>
#include <stdarg.h>
#include <mutex>

#include "record-utility.h"
#include "../../core/slang-string.h"
#include "../../core/slang-string-util.h"

constexpr const char* kRecordLayerEnvVar = "SLANG_RECORD_LAYER";
constexpr const char* kRecordLayerLogLevel = "SLANG_RECORD_LOG_LEVEL";

namespace SlangRecord
{
    static thread_local unsigned int g_logLevel = LogLevel::Silent;

    static bool getEnvironmentVariable(const char* name, Slang::String& out)
    {
#ifdef _WIN32
        char* envVar = nullptr;
        size_t sz = 0;
        if (_dupenv_s(&envVar, &sz, name) == 0 && envVar != nullptr)
        {
            out = envVar;
            free(envVar);
        }
#else
        if (const char* envVar = std::getenv(name))
        {
            out = envVar;
        }
#endif
        return out.getLength() > 0;
    }

    bool isRecordLayerEnabled()
    {
        Slang::String envVarStr;
        if(getEnvironmentVariable(kRecordLayerEnvVar, envVarStr))
        {
            if (envVarStr == "1")
            {
                return true;
            }
        }
        return false;
    }

    void setLogLevel()
    {
        // We only want to set the log level once
        if (g_logLevel != LogLevel::Silent)
        {
            return;
        }

        Slang::String envVarStr;
        if (getEnvironmentVariable(kRecordLayerLogLevel, envVarStr))
        {
            unsigned int logLevel = Slang::stringToUInt(envVarStr);
            g_logLevel = std::min((unsigned int)(LogLevel::Verbose), logLevel);
            return;
        }
    }

    void slangRecordLog(LogLevel logLevel, const char* fmt, ...)
    {
        if (logLevel > g_logLevel)
        {
            return;
        }

        Slang::StringBuilder builder;

        va_list args;
        va_start(args, fmt);
        Slang::StringUtil::append(fmt, args, builder);
        va_end(args);

        fprintf(stdout, "[slang-record-replay]: %s", builder.begin());
    }
}