library/cpp/tvmauth/client/misc/tool/meta_info.h


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

#pragma once

#include "settings.h"

#include <library/cpp/tvmauth/client/misc/utils.h>

#include <library/cpp/tvmauth/client/logger.h>

#include <library/cpp/http/simple/http_client.h>

namespace NTvmAuth::NTvmTool {
    class TMetaInfo {
    public:
        using TDstAliases = THashMap<TClientSettings::TAlias, TTvmId>;

        struct TConfig {
            TTvmId SelfTvmId = 0;
            EBlackboxEnv BbEnv = EBlackboxEnv::Prod;
            TDstAliases DstAliases;

            bool AreTicketsRequired() const {
                return !DstAliases.empty();
            }

            TString ToString() const;

            bool operator==(const TConfig& c) const {
                return SelfTvmId == c.SelfTvmId &&
                       BbEnv == c.BbEnv &&
                       DstAliases == c.DstAliases;
            }
        };
        using TConfigPtr = std::shared_ptr<TConfig>;

    public:
        TMetaInfo(TLoggerPtr logger);

        TConfigPtr Init(TKeepAliveHttpClient& client,
                        const TClientSettings& settings);

        static TString GetRequestForTickets(const TMetaInfo::TConfig& config);

        const TKeepAliveHttpClient::THeaders& GetAuthHeader() const {
            return AuthHeader_;
        }

        TConfigPtr GetConfig() const {
            return Config_.Get();
        }

        bool TryUpdateConfig(TKeepAliveHttpClient& client);

    protected:
        void TryPing(TKeepAliveHttpClient& client);
        TString Fetch(TKeepAliveHttpClient& client) const;
        static TConfigPtr ParseMetaString(const TString& meta, const TString& self);
        void ApplySettings(const TClientSettings& settings);
        static EBlackboxEnv BbEnvFromString(const TString& env, const TString& meta);

    protected:
        NUtils::TProtectedValue<TConfigPtr> Config_;
        TKeepAliveHttpClient::THeaders AuthHeader_;

        TLoggerPtr Logger_;
        TString SelfAlias_;
    };
}