From 92f8ded7cc4710794bed0eb59ca76232b907e516 Mon Sep 17 00:00:00 2001
From: SpyCheese <mikle98@yandex.ru>
Date: Thu, 29 Jun 2023 11:57:33 +0300
Subject: [PATCH] Use keyStoreTypeDirectory in rldp-http-proxy and
 storage-daemon

Mainly for caching synced block in tonlib.
---
 rldp-http-proxy/rldp-http-proxy.cpp       | 16 +++++++++++++++-
 storage/storage-daemon/storage-daemon.cpp |  4 +++-
 2 files changed, 18 insertions(+), 2 deletions(-)

diff --git a/rldp-http-proxy/rldp-http-proxy.cpp b/rldp-http-proxy/rldp-http-proxy.cpp
index e91868228..0d518d6d1 100644
--- a/rldp-http-proxy/rldp-http-proxy.cpp
+++ b/rldp-http-proxy/rldp-http-proxy.cpp
@@ -54,6 +54,7 @@
 #include "git.h"
 #include "td/utils/BufferedFd.h"
 #include "common/delay.h"
+#include "td/utils/port/path.h"
 
 #include "tonlib/tonlib/TonlibClientWrapper.h"
 #include "DNSResolver.h"
@@ -920,6 +921,12 @@ class RldpHttpProxy : public td::actor::Actor {
   }
 
   void run() {
+    if (!db_root_.empty()) {
+      td::mkpath(db_root_ + "/").ensure();
+    } else if (!is_client_) {
+      LOG(ERROR) << "DB root is required for server proxy";
+      std::_Exit(2);
+    }
     keyring_ = ton::keyring::Keyring::create(is_client_ ? std::string("") : (db_root_ + "/keyring"));
     {
       auto S = load_global_config();
@@ -955,9 +962,16 @@ class RldpHttpProxy : public td::actor::Actor {
     auto conf_dataR = td::read_file(global_config_);
     conf_dataR.ensure();
 
+    ton::tl_object_ptr<tonlib_api::KeyStoreType> key_store;
+    if (db_root_.empty()) {
+      key_store = tonlib_api::make_object<tonlib_api::keyStoreTypeInMemory>();
+    } else {
+      td::mkpath(db_root_ + "/tonlib-cache/").ensure();
+      key_store = tonlib_api::make_object<tonlib_api::keyStoreTypeDirectory>(db_root_ + "/tonlib-cache/");
+    }
     auto tonlib_options = tonlib_api::make_object<tonlib_api::options>(
         tonlib_api::make_object<tonlib_api::config>(conf_dataR.move_as_ok().as_slice().str(), "", false, false),
-        tonlib_api::make_object<tonlib_api::keyStoreTypeInMemory>());
+        std::move(key_store));
     tonlib_client_ = td::actor::create_actor<tonlib::TonlibClientWrapper>("tonlibclient", std::move(tonlib_options));
     dns_resolver_ = td::actor::create_actor<DNSResolver>("dnsresolver", tonlib_client_.get());
   }
diff --git a/storage/storage-daemon/storage-daemon.cpp b/storage/storage-daemon/storage-daemon.cpp
index 372829f4a..98388428d 100644
--- a/storage/storage-daemon/storage-daemon.cpp
+++ b/storage/storage-daemon/storage-daemon.cpp
@@ -927,9 +927,11 @@ class StorageDaemon : public td::actor::Actor {
     }
     auto r_conf_data = td::read_file(global_config_);
     r_conf_data.ensure();
+    std::string key_store = db_root_ + "/tonlib/";
+    td::mkpath(key_store).ensure();
     auto tonlib_options = tonlib_api::make_object<tonlib_api::options>(
         tonlib_api::make_object<tonlib_api::config>(r_conf_data.move_as_ok().as_slice().str(), "", false, false),
-        tonlib_api::make_object<tonlib_api::keyStoreTypeInMemory>());
+        tonlib_api::make_object<tonlib_api::keyStoreTypeDirectory>(key_store));
     tonlib_client_ = td::actor::create_actor<tonlib::TonlibClientWrapper>("tonlibclient", std::move(tonlib_options));
   }