From eccc7dcfb22fb9c6c42bbcd5566cd044acd1a2f3 Mon Sep 17 00:00:00 2001
From: Maksim Denisov <denisov.maksim@huawei.com>
Date: Tue, 25 Feb 2025 12:20:44 +0100
Subject: ArtifactBlob: Support construction in memory

---
 src/buildtool/common/artifact_blob.cpp | 27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+)

(limited to 'src/buildtool/common/artifact_blob.cpp')

diff --git a/src/buildtool/common/artifact_blob.cpp b/src/buildtool/common/artifact_blob.cpp
index 3b199e5e..c25e1db1 100644
--- a/src/buildtool/common/artifact_blob.cpp
+++ b/src/buildtool/common/artifact_blob.cpp
@@ -14,8 +14,35 @@
 
 #include "src/buildtool/common/artifact_blob.hpp"
 
+#include <exception>
+
+#include "fmt/core.h"
+#include "src/buildtool/common/artifact_digest_factory.hpp"
 #include "src/utils/cpp/hash_combine.hpp"
 
+auto ArtifactBlob::FromMemory(HashFunction hash_function,
+                              ObjectType type,
+                              std::string content) noexcept
+    -> expected<ArtifactBlob, std::string> {
+    try {
+        auto digest = IsTreeObject(type)
+                          ? ArtifactDigestFactory::HashDataAs<ObjectType::Tree>(
+                                hash_function, content)
+                          : ArtifactDigestFactory::HashDataAs<ObjectType::File>(
+                                hash_function, content);
+        return ArtifactBlob{
+            std::move(digest),
+            std::make_shared<std::string const>(std::move(content)),
+            IsExecutableObject(type)};
+    } catch (const std::exception& e) {
+        return unexpected{fmt::format(
+            "ArtifactBlob::FromMemory: Got an exception:\n{}", e.what())};
+    } catch (...) {
+        return unexpected<std::string>{
+            "ArtifactBlob::FromMemory: Got an unknown exception"};
+    }
+}
+
 auto ArtifactBlob::ReadContent() const noexcept
     -> std::shared_ptr<std::string const> {
     return content_;
-- 
cgit v1.2.3