update archive ttl question

Author: yungwine

scripts/install.py

@@ -1,6 +1,53 @@
+import datetime
 import os
 import subprocess
+import time
 import inquirer
+import requests
+
+
+def get_archive_ttl_message(answers: dict):
+    global default_archive_ttl
+    archive_blocks = answers.get('archive-blocks')
+    if not archive_blocks and answers['archive-blocks'] != 0:
+        return 'Send the number of seconds to keep the block data in the node database. Default is 2592000 (30 days)'
+    block_from = archive_blocks.split()[0]
+    # or sent -1 to store downloaded blocks always
+    if block_from.isdigit():
+        seqno = int(block_from)
+        url = f'https://toncenter.com/api/v2/getBlockHeader?workchain=-1&shard={-2**63}&seqno={seqno}'
+        if answers['network'] == 'Testnet':
+            url = url.replace('toncenter.com', 'testnet.toncenter.com')
+        data = requests.get(url).json()
+        utime = int(data['result']['gen_utime'])
+    else:
+        utime = int(datetime.datetime.strptime(block_from, '%Y-%m-%d').timestamp())
+    default_archive_ttl = int(time.time() - (utime - datetime.timedelta(days=30).total_seconds()))
+    answers['archive-ttl-default'] = default_archive_ttl
+    return f'Send the number of seconds to keep the block data in the node database. Default is {default_archive_ttl} to keep archive blocks from {datetime.datetime.fromtimestamp(utime - datetime.timedelta(days=30).total_seconds())}\nOr send -1 to keep downloaded blocks always (recommended).'
+
+
+def is_valid_date_format(date_str):
+    try:
+        datetime.datetime.strptime(date_str, '%Y-%m-%d')
+        return True
+    except ValueError:
+        return False
+
+
+def validate_archive_blocks(_, value):
+    if not value:
+        return True
+    parts = value.split()
+    if len(parts) > 2:
+        return False
+
+    for part in parts:
+        if part.isdigit() and int(part) < 0:
+            return False
+        elif not is_valid_date_format(part):
+            return False
+    return True
 
 
 def run_cli():
@@ -21,13 +68,6 @@ def run_cli():
             ignore=lambda x: x["network"] != "Other",  # do not ask this question if network is not 'Other'
             validate=lambda _, x: x.startswith("http"),
         ),
-        inquirer.Text(
-            "archive-ttl",
-            message="Send the number of seconds to keep the block data in the node database. Default is 2592000 (30 days)",
-            ignore=lambda x: x["mode"] != "liteserver",  # do not ask this question if mode is not liteserver
-            validate=lambda _, x: not x or x.isdigit(),  # must be empty string or a number
-            # default=2592000
-        ),
         inquirer.List(
             "validator-mode",
             message="Select mode for validator usage. You can skip and set up this later",
@@ -36,9 +76,18 @@ def run_cli():
         ),
         inquirer.Text(
             "archive-blocks",
-            message="Do you want to download archive blocks via TON Storage? If yes, provide block seqno to start from or press Enter to skip.",
+            message="Do you want to download archive blocks via TON Storage? Press Enter to skip.\n"
+                    "If yes, provide block seqno or date to start from and (optionally) block seqno or date to end with (send 0 to download all blocks and setup full archive node).\n"
+                    "Examples: `30850000`, `10000000 10200000`, `2025-01-01`, `2025-01-01 2025-01-30`",
             ignore=lambda x: x["mode"] == "validator",
-            validate=lambda _, x: not x or (x.isdigit() and int(x) >= 0),
+            validate=validate_archive_blocks,
+        ),
+        inquirer.Text(
+            "archive-ttl",
+            message=get_archive_ttl_message,
+            ignore=lambda x: x["mode"] != "liteserver",  # do not ask this question if mode is not liteserver
+            validate=lambda _, x: not x or x.isdigit(),  # must be empty string or a number
+            # default=get_default_archive_ttl
         ),
         inquirer.Confirm(
             "dump",
@@ -64,7 +113,7 @@ def parse_args(answers: dict):
     mode = answers["mode"]
     network = answers["network"].lower()
     config = answers["config"]
-    archive_ttl = answers["archive-ttl"]
+    archive_ttl = answers["archive-ttl"] or answers.get("archive-ttl-default")
     add_shard = answers["add-shard"]
     validator_mode = answers["validator-mode"]
     archive_blocks = answers["archive-blocks"]