Merge pull request #10502 from ydb-platform/mergelibs-241016-1210

Library import 241016-1210

1 files changed, 95 insertions, 0 deletions

diff --git a/contrib/libs/jinja2cpp/src/template_env.cpp b/contrib/libs/jinja2cpp/src/template_env.cpp
new file mode 100644
index 0000000000..239a4f02e1
--- /dev/null
+++ b/contrib/libs/jinja2cpp/src/template_env.cpp
@@ -0,0 +1,95 @@
+#include <jinja2cpp/template.h>
+#include <jinja2cpp/template_env.h>
+
+namespace jinja2
+{
+template<typename CharT>
+struct TemplateFunctions;
+
+template<>
+struct TemplateFunctions<char>
+{
+    using ResultType = nonstd::expected<Template, ErrorInfo>;
+    static Template CreateTemplate(TemplateEnv* env) { return Template(env); }
+    static auto LoadFile(const std::string& fileName, const IFilesystemHandler* fs) { return fs->OpenStream(fileName); }
+};
+
+template<>
+struct TemplateFunctions<wchar_t>
+{
+    using ResultType = nonstd::expected<TemplateW, ErrorInfoW>;
+    static TemplateW CreateTemplate(TemplateEnv* env) { return TemplateW(env); }
+    static auto LoadFile(const std::string& fileName, const IFilesystemHandler* fs) { return fs->OpenWStream(fileName); }
+};
+
+template<typename CharT, typename T, typename Cache>
+auto TemplateEnv::LoadTemplateImpl(TemplateEnv* env, std::string fileName, const T& filesystemHandlers, Cache& cache)
+{
+    using Functions = TemplateFunctions<CharT>;
+    using ResultType = typename Functions::ResultType;
+    using ErrorType = typename ResultType::error_type;
+    auto tpl = Functions::CreateTemplate(env);
+
+    {
+        std::shared_lock<std::shared_timed_mutex> l(m_guard);
+        auto p = cache.find(fileName);
+        if (p != cache.end())
+        {
+            if (m_settings.autoReload)
+            {
+                auto lastModified = p->second.handler->GetLastModificationDate(fileName);
+                if (!lastModified || (p->second.lastModification && lastModified.value() <= p->second.lastModification.value()))
+                    return ResultType(p->second.tpl);
+            }
+            else
+                return ResultType(p->second.tpl);
+        }
+    }
+
+    for (auto& fh : filesystemHandlers)
+    {
+        if (!fh.prefix.empty() && fileName.find(fh.prefix) != 0)
+            continue;
+
+        auto stream = Functions::LoadFile(fileName, fh.handler.get());
+        if (stream)
+        {
+            auto res = tpl.Load(*stream, fileName);
+            if (!res)
+                return ResultType(res.get_unexpected());
+
+            if (m_settings.cacheSize != 0)
+            {
+                auto lastModified = fh.handler->GetLastModificationDate(fileName);
+                std::unique_lock<std::shared_timed_mutex> l(m_guard);
+                auto& cacheEntry = cache[fileName];
+                cacheEntry.tpl = tpl;
+                cacheEntry.handler = fh.handler;
+                cacheEntry.lastModification = lastModified;
+            }
+
+            return ResultType(tpl);
+        }
+    }
+
+    typename ErrorType::Data errorData;
+    errorData.code = ErrorCode::FileNotFound;
+    errorData.srcLoc.col = 1;
+    errorData.srcLoc.line = 1;
+    errorData.srcLoc.fileName = "";
+    errorData.extraParams.push_back(Value(fileName));
+
+    return ResultType(nonstd::make_unexpected(ErrorType(errorData)));
+}
+
+nonstd::expected<Template, ErrorInfo> TemplateEnv::LoadTemplate(std::string fileName)
+{
+    return LoadTemplateImpl<char>(this, std::move(fileName), m_filesystemHandlers, m_templateCache);
+}
+
+nonstd::expected<TemplateW, ErrorInfoW> TemplateEnv::LoadTemplateW(std::string fileName)
+{
+    return LoadTemplateImpl<wchar_t>(this, std::move(fileName), m_filesystemHandlers, m_templateWCache);
+}
+
+} // namespace jinja2