diff --git a/components/security_apps/orchestration/k8s_policy_gen/k8s_policy_gen.cc b/components/security_apps/orchestration/k8s_policy_gen/k8s_policy_gen.cc index 9eef08a..7cf3195 100644 --- a/components/security_apps/orchestration/k8s_policy_gen/k8s_policy_gen.cc +++ b/components/security_apps/orchestration/k8s_policy_gen/k8s_policy_gen.cc @@ -13,9 +13,12 @@ #include "k8s_policy_gen.h" +#include +#include #include #include #include +#include #include #include #include @@ -854,9 +857,28 @@ private: string token; map practice_name_to_id_map; + bool + isPlaygroundEnv() + { + string playground_variable = "PLAYGROUND"; + const char* env_string = getenv(playground_variable.c_str()); + + if (env_string) + { + string env_value = env_string; + std::transform(env_value.begin(), env_value.end(), env_value.begin(), + [](unsigned char c){ return std::tolower(c); }); + return env_value == "true"; + } + + return false; + } + bool getClusterId() { + string playground_uid = isPlaygroundEnv() ? "playground-" : ""; + dbgTrace(D_K8S_POLICY) << "Getting cluster UID"; NamespaceData namespaces_data; bool res = messaging->sendObject( @@ -885,7 +907,7 @@ private: uid, EnvKeyAttr::LogSection::SOURCE ); - Singleton::Consume::by()->setClusterId(uid); + Singleton::Consume::by()->setClusterId(playground_uid + uid); return true; } }