zrb 0.18.0__py3-none-any.whl → 0.22.0__py3-none-any.whl

zrb/__init__.py

@@ -45,6 +45,7 @@ from zrb.task_input.bool_input import BoolInput
 from zrb.task_input.choice_input import ChoiceInput
 from zrb.task_input.float_input import FloatInput
 from zrb.task_input.int_input import IntInput
+from zrb.task_input.multiline_input import MultilineInput
 from zrb.task_input.password_input import PasswordInput
 from zrb.task_input.str_input import StrInput
 from zrb.task_input.task_input import Input
@@ -93,6 +94,7 @@ assert BoolInput
 assert ChoiceInput
 assert FloatInput
 assert IntInput
+assert MultilineInput
 assert PasswordInput
 assert StrInput
 assert Env

zrb/builtin/project/add/app/generator/generator.py

@@ -26,6 +26,7 @@ _CURRENT_DIR = os.path.dirname(__file__)
 @python_task(
     name="validate",
     inputs=[project_dir_input, package_name_input],
+    retry=0,
 )
 async def validate(*args: Any, **kwargs: Any):
     project_dir = kwargs.get("project_dir")

zrb/builtin/project/add/app/generator/template/src/kebab-zrb-package-name/src/snake_zrb_package_name/snake_zrb_generator_name/_input.py

@@ -1,5 +1,6 @@
 import os
 
+from zrb.helper.util import coalesce, to_kebab_case, to_snake_case
 from zrb.task_input.int_input import IntInput
 from zrb.task_input.str_input import StrInput
 
@@ -41,9 +42,9 @@ app_image_name_input = StrInput(
     name="app-image-name",
     description="App image name",
     prompt="App image name",
-    default=app_image_default_namespace
-    + "/"
-    + "{{util.to_kebab_case(input.app_name)}}",  # noqa
+    default=lambda m: "/".join(
+        [app_image_default_namespace, to_kebab_case(m.get("app_name"))]
+    ),
 )
 
 http_port_input = IntInput(
@@ -58,5 +59,7 @@ env_prefix_input = StrInput(
     name="env-prefix",
     description="OS environment prefix",
     prompt="OS environment prefix",
-    default='{{util.to_snake_case(util.coalesce(input.app_name, input.task_name, "MY")).upper()}}',  # noqa
+    default=lambda m: to_snake_case(
+        coalesce(m.get("app_name"), m.get("task_name"), "MY")
+    ).upper(),
 )

zrb/builtin/project/add/app/generator/template/src/kebab-zrb-package-name/src/snake_zrb_package_name/snake_zrb_generator_name/snake_zrb_generator_name.py

@@ -32,6 +32,7 @@ _snake_app_name_tpl = "{{util.to_snake_case(input.app_name)}}"
 @python_task(
     name="validate",
     inputs=[project_dir_input, app_name_input],
+    retry=0,
 )
 async def validate(*args: Any, **kwargs: Any):
     project_dir = kwargs.get("project_dir")

zrb/builtin/project/add/app/python/python.py

@@ -31,6 +31,7 @@ _snake_app_name_tpl = "{{util.to_snake_case(input.app_name)}}"
 @python_task(
     name="validate",
     inputs=[project_dir_input, app_name_input],
+    retry=0,
 )
 async def validate(*args: Any, **kwargs: Any):
     project_dir = kwargs.get("project_dir")

zrb/builtin/project/add/fastapp/app/_input.py

@@ -1,5 +1,6 @@
 import os
 
+from zrb.helper.util import coalesce, to_kebab_case, to_snake_case
 from zrb.task_input.int_input import IntInput
 from zrb.task_input.str_input import StrInput
 
@@ -37,13 +38,14 @@ app_description_input = StrInput(
 app_image_default_namespace = os.getenv(
     "PROJECT_IMAGE_DEFAULT_NAMESPACE", "docker.io/" + SYSTEM_USER
 )
+
 app_image_name_input = StrInput(
     name="app-image-name",
     description="App image name",
     prompt="App image name",
-    default=app_image_default_namespace
-    + "/"
-    + "{{util.to_kebab_case(input.app_name)}}",  # noqa
+    default=lambda m: "/".join(
+        [app_image_default_namespace, to_kebab_case(m.get("app_name"))]
+    ),
 )
 
 http_port_input = IntInput(
@@ -58,5 +60,7 @@ env_prefix_input = StrInput(
     name="env-prefix",
     description="OS environment prefix",
     prompt="OS environment prefix",
-    default='{{util.to_snake_case(util.coalesce(input.app_name, input.task_name, "MY")).upper()}}',  # noqa
+    default=lambda m: to_snake_case(
+        coalesce(m.get("app_name"), m.get("task_name"), "MY")
+    ).upper(),
 )

zrb/builtin/project/add/fastapp/app/app.py

@@ -35,6 +35,7 @@ _snake_app_name_tpl = "{{util.to_snake_case(input.app_name)}}"
 @python_task(
     name="validate",
     inputs=[project_dir_input, app_name_input],
+    retry=0,
 )
 async def validate(*args: Any, **kwargs: Any):
     project_dir = kwargs.get("project_dir")

zrb/builtin/project/add/fastapp/crud/_input.py

@@ -10,9 +10,10 @@ entity_name_input = StrInput(
 
 plural_entity_name_input = StrInput(
     name="plural-entity-name",
+    shortcut="p",
     description="Plural entity name",
     prompt="Plural entity name",
-    default="books",
+    default=lambda m: m.get("entity_name", "") + "s",
 )
 
 main_column_name_input = StrInput(

zrb/builtin/project/add/fastapp/crud/crud.py

@@ -35,6 +35,7 @@ _CODEMOD_DIR = os.path.join(_CURRENT_DIR, "nodejs", "codemod")
 @python_task(
     name="validate",
     inputs=[project_dir_input, app_name_input, module_name_input, entity_name_input],
+    retry=0,
 )
 async def validate(*args: Any, **kwargs: Any):
     project_dir = kwargs.get("project_dir")

zrb/builtin/project/add/fastapp/crud/nodejs/codemod/package-lock.json

@@ -60,10 +60,13 @@
       }
     },
     "node_modules/@types/node": {
-      "version": "20.6.0",
-      "resolved": "https://registry.npmjs.org/@types/node/-/node-20.6.0.tgz",
-      "integrity": "sha512-najjVq5KN2vsH2U/xyh2opaSEz6cZMR2SetLIlxlj08nOcmPOemJmUK2o4kUzfLqfrWE0PIrNeE16XhYDd3nqg==",
-      "dev": true
+      "version": "20.14.2",
+      "resolved": "https://registry.npmjs.org/@types/node/-/node-20.14.2.tgz",
+      "integrity": "sha512-xyu6WAMVwv6AKFLB+e/7ySZVr/0zLCzOa7rSpq6jNwpqOrUbcACDWC+53d4n2QHOnDou0fbIsg8wZu/sxrnI4Q==",
+      "dev": true,
+      "dependencies": {
+        "undici-types": "~5.26.4"
+      }
     },
     "node_modules/balanced-match": {
       "version": "1.0.2",
@@ -79,11 +82,11 @@
       }
     },
     "node_modules/braces": {
-      "version": "3.0.2",
-      "resolved": "https://registry.npmjs.org/braces/-/braces-3.0.2.tgz",
-      "integrity": "sha512-b8um+L1RzM3WDSzvhm6gIz1yfTbBt6YTlcEKAvsmqCZZFw46z626lVj9j1yEPW33H5H+lBQpZMP1k8l+78Ha0A==",
+      "version": "3.0.3",
+      "resolved": "https://registry.npmjs.org/braces/-/braces-3.0.3.tgz",
+      "integrity": "sha512-yQbXgO/OSZVD2IsiLlro+7Hf6Q18EJrKSEsdoMzKePKXct3gvD8oLcOQdIzGupr5Fj+EDe8gO/lxc1BzfMpxvA==",
       "dependencies": {
-        "fill-range": "^7.0.1"
+        "fill-range": "^7.1.1"
       },
       "engines": {
         "node": ">=8"
@@ -95,9 +98,9 @@
       "integrity": "sha512-q4dMFMlXtKR3XNBHyMHt/3pwYNA69EDk00lloMOaaUMKPUXBw6lpXtbu3MMVG6/uOihGnRDOlkyqsONEUj60+w=="
     },
     "node_modules/fast-glob": {
-      "version": "3.3.1",
-      "resolved": "https://registry.npmjs.org/fast-glob/-/fast-glob-3.3.1.tgz",
-      "integrity": "sha512-kNFPyjhh5cKjrUltxs+wFx+ZkbRaxxmZ+X0ZU31SOsxCEtP9VPgtq2teZw1DebupL5GmDaNQ6yKMMVcM41iqDg==",
+      "version": "3.3.2",
+      "resolved": "https://registry.npmjs.org/fast-glob/-/fast-glob-3.3.2.tgz",
+      "integrity": "sha512-oX2ruAFQwf/Orj8m737Y5adxDQO0LAB7/S5MnxCdTNDd4p6BsyIVsv9JQsATbTSq8KHRpLwIHbVlUNatxd+1Ow==",
       "dependencies": {
         "@nodelib/fs.stat": "^2.0.2",
         "@nodelib/fs.walk": "^1.2.3",
@@ -110,17 +113,17 @@
       }
     },
     "node_modules/fastq": {
-      "version": "1.15.0",
-      "resolved": "https://registry.npmjs.org/fastq/-/fastq-1.15.0.tgz",
-      "integrity": "sha512-wBrocU2LCXXa+lWBt8RoIRD89Fi8OdABODa/kEnyeyjS5aZO5/GNvI5sEINADqP/h8M29UHTHUb53sUu5Ihqdw==",
+      "version": "1.17.1",
+      "resolved": "https://registry.npmjs.org/fastq/-/fastq-1.17.1.tgz",
+      "integrity": "sha512-sRVD3lWVIXWg6By68ZN7vho9a1pQcN/WBFaAAsDDFzlJjvoGx0P8z7V1t72grFJfJhu3YPZBuu25f7Kaw2jN1w==",
       "dependencies": {
         "reusify": "^1.0.4"
       }
     },
     "node_modules/fill-range": {
-      "version": "7.0.1",
-      "resolved": "https://registry.npmjs.org/fill-range/-/fill-range-7.0.1.tgz",
-      "integrity": "sha512-qOo9F+dMUmC2Lcb4BbVvnKJxTPjCm+RRpe4gDuGrzkL7mEVl/djYSu2OdQ2Pa302N4oqkSg9ir6jaLWJ2USVpQ==",
+      "version": "7.1.1",
+      "resolved": "https://registry.npmjs.org/fill-range/-/fill-range-7.1.1.tgz",
+      "integrity": "sha512-YsGpe3WHLK8ZYi4tWDg2Jy3ebRz2rXowDxnld4bkQB00cc/1Zw9AWnC0i9ztDJitivtQvaI9KaLyKrc+hBW0yg==",
       "dependencies": {
         "to-regex-range": "^5.0.1"
       },
@@ -175,11 +178,11 @@
       }
     },
     "node_modules/micromatch": {
-      "version": "4.0.5",
-      "resolved": "https://registry.npmjs.org/micromatch/-/micromatch-4.0.5.tgz",
-      "integrity": "sha512-DMy+ERcEW2q8Z2Po+WNXuw3c5YaUSFjAO5GsJqfEl7UjvtIuFKO6ZrKvcItdy98dwFI2N1tg3zNIdKaQT+aNdA==",
+      "version": "4.0.7",
+      "resolved": "https://registry.npmjs.org/micromatch/-/micromatch-4.0.7.tgz",
+      "integrity": "sha512-LPP/3KorzCwBxfeUuZmaR6bG2kdeHSbe0P2tY3FLRU4vYrjYz5hI4QZwV0njUx3jeuKe67YukQ1LSPZBKDqO/Q==",
       "dependencies": {
-        "braces": "^3.0.2",
+        "braces": "^3.0.3",
         "picomatch": "^2.3.1"
       },
       "engines": {
@@ -301,9 +304,9 @@
       }
     },
     "node_modules/typescript": {
-      "version": "5.2.2",
-      "resolved": "https://registry.npmjs.org/typescript/-/typescript-5.2.2.tgz",
-      "integrity": "sha512-mI4WrpHsbCIcwT9cF4FZvr80QUeKvsUsUvKDoR+X/7XHQH98xYD8YHZg7ANtz2GtZt/CBq2QJ0thkGJMHfqc1w==",
+      "version": "5.4.5",
+      "resolved": "https://registry.npmjs.org/typescript/-/typescript-5.4.5.tgz",
+      "integrity": "sha512-vcI4UpRgg81oIRUFwR0WSIHKt11nJ7SAVlYNIu+QpqeyXP+gpQJy/Z4+F0aGxSE4MqwjyXvW/TzgkLAx2AGHwQ==",
       "dev": true,
       "bin": {
         "tsc": "bin/tsc",
@@ -312,6 +315,12 @@
       "engines": {
         "node": ">=14.17"
       }
+    },
+    "node_modules/undici-types": {
+      "version": "5.26.5",
+      "resolved": "https://registry.npmjs.org/undici-types/-/undici-types-5.26.5.tgz",
+      "integrity": "sha512-JlCMO+ehdEIKqlFxk6IfVoAUVmgz7cU7zD/h9XZ0qzeosSHmUJVOzSQvvYSYWXkFXC+IfLKSIffhv0sVZup6pA==",
+      "dev": true
     }
   }
 }

zrb/builtin/project/add/fastapp/field/field.py

@@ -32,6 +32,7 @@ from zrb.task.task import Task
 @python_task(
     name="validate",
     inputs=[project_dir_input, app_name_input, module_name_input, entity_name_input],
+    retry=0,
 )
 async def validate(*args: Any, **kwargs: Any):
     project_dir = kwargs.get("project_dir")

zrb/builtin/project/add/plugin/plugin.py

@@ -28,7 +28,11 @@ from zrb.task.task import Task
 _CURRENT_DIR = os.path.dirname(__file__)
 
 
-@python_task(name="validate", inputs=[project_dir_input, package_name_input])
+@python_task(
+    name="validate",
+    inputs=[project_dir_input, package_name_input],
+    retry=0,
+)
 async def validate(*args: Any, **kwargs: Any):
     project_dir = kwargs.get("project_dir")
     validate_existing_project_dir(project_dir)

zrb/builtin/project/add/task/cmd/add.py

@@ -23,6 +23,7 @@ _CURRENT_DIR = os.path.dirname(__file__)
         project_dir_input,
         task_name_input,
     ],
+    retry=0,
 )
 async def validate(*args: Any, **kwargs: Any):
     project_dir = kwargs.get("project_dir")

zrb/builtin/project/add/task/docker_compose/add.py

@@ -27,6 +27,7 @@ _CURRENT_DIR = os.path.dirname(__file__)
         project_dir_input,
         task_name_input,
     ],
+    retry=0,
 )
 async def validate(*args: Any, **kwargs: Any):
     project_dir = kwargs.get("project_dir")

zrb/builtin/project/add/task/python/add.py

@@ -23,6 +23,7 @@ _CURRENT_DIR = os.path.dirname(__file__)
         project_dir_input,
         task_name_input,
     ],
+    retry=0,
 )
 async def validate(*args: Any, **kwargs: Any):
     project_dir = kwargs.get("project_dir")

zrb/config/config.py

@@ -32,6 +32,7 @@ def _get_default_tmp_dir() -> str:
 
 tmp_dir = os.getenv("ZRB_TMP_DIR", _get_default_tmp_dir())
 default_shell = os.getenv("ZRB_SHELL", _get_current_shell())
+default_editor = os.getenv("ZRB_EDITOR", "nano")
 init_script_str = os.getenv("ZRB_INIT_SCRIPTS", "")
 init_scripts = init_script_str.split(":") if init_script_str != "" else []
 logging_level = untyped_to_logging_level(os.getenv("ZRB_LOGGING_LEVEL", "WARNING"))

zrb/helper/multiline.py

@@ -0,0 +1,13 @@
+import click
+
+from zrb.helper.typecheck import typechecked
+
+
+@typechecked
+def edit(editor: str, mark_comment: str, text: str, extension: str = "txt") -> str:
+    result = click.edit(
+        text="\n".join([mark_comment, text]), editor=editor, extension=f".{extension}"
+    )
+    if result is None:
+        result = text
+    return "\n".join(result.split(mark_comment)).strip()

zrb/task/any_task.py

@@ -568,6 +568,13 @@ class AnyTask(ABC):
         """
         pass
 
+    @abstractmethod
+    def _lock_checkers(self):
+        """
+        Lock checkers so that it cannot be altered anymore
+        """
+        pass
+
     @abstractmethod
     def _lock_upstreams(self):
         """

zrb/task/base_task/base_task.py

@@ -467,10 +467,9 @@ class BaseTask(FinishTracker, AttemptTracker, Renderer, BaseTaskModel, AnyTask):
                 )
             )
         # set checker keyval
+        self._lock_checkers()
         checker_coroutines = []
         for checker_task in self._get_checkers():
-            checker_task.add_input(*self._get_inputs())
-            checker_task.add_env(*self._get_envs())
             checker_coroutines.append(
                 asyncio.create_task(
                     checker_task._set_keyval(kwargs=new_kwargs, env_prefix=env_prefix)

zrb/task/base_task/component/common_task_model.py

@@ -65,6 +65,11 @@ class CommonTaskModel:
         self._group = group
         if group is not None:
             group._add_task(self)
+        checkers_cp: List[AnyTask] = [checker.copy() for checker in checkers]
+        for checker in checkers_cp:
+            checker.add_env(*envs)
+            checker.add_env_file(*env_files)
+            checker.add_input(*inputs)
         self._description = coalesce_str(description, name)
         self._inputs = inputs
         self._envs = envs
@@ -75,7 +80,7 @@ class CommonTaskModel:
         self._retry_interval = retry_interval
         self._upstreams = upstreams
         self._fallbacks = fallbacks
-        self._checkers = [checker.copy() for checker in checkers]
+        self._checkers = checkers_cp
         self._checking_interval = checking_interval
         self._run_function: Optional[Callable[..., Any]] = run
         self._on_triggered = on_triggered
@@ -101,6 +106,9 @@ class CommonTaskModel:
         self.__has_already_inject_fallbacks: bool = False
         self.__all_inputs: Optional[List[AnyInput]] = None
 
+    def _lock_checkers(self):
+        self.__allow_add_checkers = False
+
     def _lock_upstreams(self):
         self.__allow_add_upstreams = False
 
@@ -164,11 +172,15 @@ class CommonTaskModel:
         if not self.__allow_add_inputs:
             raise Exception(f"Cannot insert inputs for `{self.get_name()}`")
         self._inputs = list(inputs) + list(self._inputs)
+        for checker in self._get_checkers():
+            checker.insert_input(*inputs)
 
     def add_input(self, *inputs: AnyInput):
         if not self.__allow_add_inputs:
             raise Exception(f"Cannot add inputs for `{self.get_name()}`")
         self._inputs = list(self._inputs) + list(inputs)
+        for checker in self._get_checkers():
+            checker.add_input(*inputs)
 
     def inject_inputs(self):
         pass
@@ -219,11 +231,15 @@ class CommonTaskModel:
         if not self.__allow_add_envs:
             raise Exception(f"Cannot insert envs to `{self.get_name()}`")
         self._envs = list(envs) + list(self._envs)
+        for checker in self._get_checkers():
+            checker.insert_env(*envs)
 
     def add_env(self, *envs: Env):
         if not self.__allow_add_envs:
             raise Exception(f"Cannot add envs to `{self.get_name()}`")
         self._envs = list(self._envs) + list(envs)
+        for checker in self._get_checkers():
+            checker.add_env(*envs)
 
     def inject_envs(self):
         pass
@@ -255,11 +271,15 @@ class CommonTaskModel:
         if not self.__allow_add_env_files:
             raise Exception(f"Cannot insert env_files to `{self.get_name()}`")
         self._env_files = list(env_files) + list(self._env_files)
+        for checker in self._get_checkers():
+            checker.insert_env_file(*env_files)
 
     def add_env_file(self, *env_files: EnvFile):
         if not self.__allow_add_env_files:
             raise Exception(f"Cannot add env_files to `{self.get_name()}`")
         self._env_files = list(self._env_files) + list(env_files)
+        for checker in self._get_checkers():
+            checker.add_env_file(*env_files)
 
     def inject_env_files(self):
         pass
@@ -317,15 +337,26 @@ class CommonTaskModel:
     def insert_checker(self, *checkers: AnyTask):
         if not self.__allow_add_checkers:
             raise Exception(f"Cannot insert checkers to `{self.get_name()}`")
-        additional_checkers = [checker.copy() for checker in checkers]
+        additional_checkers = self.__complete_new_checkers(checkers)
         self._checkers = additional_checkers + self._checkers
 
     def add_checker(self, *checkers: AnyTask):
         if not self.__allow_add_checkers:
             raise Exception(f"Cannot add checkers to `{self.get_name()}`")
-        additional_checkers = [checker.copy() for checker in checkers]
+        additional_checkers = self.__complete_new_checkers(checkers)
         self._checkers = self._checkers + additional_checkers
 
+    def __complete_new_checkers(self, new_checkers: List[AnyTask]) -> List[AnyTask]:
+        """
+        For internal use: copy and completing new checkers
+        """
+        checkers: List[AnyTask] = [checker.copy() for checker in new_checkers]
+        for checker in checkers:
+            checker.add_input(*self._get_inputs())
+            checker.add_env(*self._get_envs())
+            checker.add_env_file(*self._get_env_files())
+        return checkers
+
     def inject_checkers(self):
         pass
 

zrb/task/cmd_task.py

@@ -1,5 +1,6 @@
 import asyncio
 import atexit
+import logging
 import os
 import pathlib
 import signal
@@ -7,7 +8,7 @@ import subprocess
 import sys
 import time
 
-from zrb.config.config import default_shell
+from zrb.config.config import default_shell, logging_level
 from zrb.helper.accessories.color import colored
 from zrb.helper.log import logger
 from zrb.helper.typecheck import typechecked
@@ -55,6 +56,18 @@ def _reset_stty():
             _has_stty = False
 
 
+def _log_error(message: Any):
+    if logging_level > logging.ERROR:
+        return
+    colored_message = colored(f"{message}", color="red", attrs=["bold"])
+    logger.error(colored_message, exc_info=True)
+
+
+def _print_out_dark(message: Any):
+    message_str = f"{message}"
+    print(colored(message_str, attrs=["dark"]), file=sys.stderr)
+
+
 CmdVal = Union[
     JinjaTemplate,
     Iterable[JinjaTemplate],
@@ -272,7 +285,7 @@ class CmdTask(BaseTask):
     def __on_kill(self, signum: Any, frame: Any):
         self._global_state.no_more_attempt = True
         self._global_state.is_killed_by_signal = True
-        self.print_out_dark(f"Getting signal {signum}")
+        _print_out_dark(f"Getting signal {signum}")
         for pid in self._pids:
             self.__kill_by_pid(pid)
         tasks = asyncio.all_tasks()
@@ -282,7 +295,7 @@ class CmdTask(BaseTask):
             except Exception as e:
                 self.print_err(e)
         time.sleep(0.3)
-        self.print_out_dark(f"Exiting with signal {signum}")
+        _print_out_dark(f"Exiting with signal {signum}")
         sys.exit(signum)
 
     def __on_exit(self):
@@ -297,20 +310,20 @@ class CmdTask(BaseTask):
             process_ever_exists = False
             if self.__is_process_exist(pid):
                 process_ever_exists = True
-                self.print_out_dark(f"Send SIGTERM to process {pid}")
+                _print_out_dark(f"Send SIGTERM to process {pid}")
                 os.killpg(os.getpgid(pid), signal.SIGTERM)
                 time.sleep(0.3)
             if self.__is_process_exist(pid):
-                self.print_out_dark(f"Send SIGINT to process {pid}")
+                _print_out_dark(f"Send SIGINT to process {pid}")
                 os.killpg(os.getpgid(pid), signal.SIGINT)
                 time.sleep(0.3)
             if self.__is_process_exist(pid):
-                self.print_out_dark(f"Send SIGKILL to process {pid}")
+                _print_out_dark(f"Send SIGKILL to process {pid}")
                 os.killpg(os.getpgid(pid), signal.SIGKILL)
             if process_ever_exists:
-                self.print_out_dark(f"Process {pid} is killed successfully")
+                _print_out_dark(f"Process {pid} is killed successfully")
         except Exception:
-            self.log_error(f"Cannot kill process {pid}")
+            _log_error(f"Cannot kill process {pid}")
 
     def __is_process_exist(self, pid: int) -> bool:
         try:

zrb/task_input/base_input.py

@@ -1,8 +1,10 @@
 from zrb.config.config import show_prompt
 from zrb.helper.accessories.color import colored
 from zrb.helper.log import logger
+from zrb.helper.string.conversion import to_variable_name
+from zrb.helper.string.jinja import is_probably_jinja
 from zrb.helper.typecheck import typechecked
-from zrb.helper.typing import Any, List, Mapping, Optional, Union
+from zrb.helper.typing import Any, Callable, List, Mapping, Optional, Union
 from zrb.task_input.any_input import AnyInput
 from zrb.task_input.constant import RESERVED_INPUT_NAMES
 
@@ -10,6 +12,9 @@ logger.debug(colored("Loading zrb.task_input.base_input", attrs=["dark"]))
 
 # flake8: noqa E501
 
+InputCallback = Callable[[Mapping[str, Any], Any], Any]
+InputDefault = Callable[[Mapping[str, Any]], Any]
+
 
 @typechecked
 class BaseInput(AnyInput):
@@ -22,8 +27,9 @@ class BaseInput(AnyInput):
         name (str): The name of the input, used as a unique identifier.
         shortcut (Optional[str]): An optional single-character shortcut for the input.
         default (Optional[Any]): The default value of the input.
+        callback (Optional[Any]): The default value of the input.
         description (Optional[str]): A brief description of what the input is for.
-        show_default (Union[bool, JinjaTemplate, None]): Determines whether the default value should be displayed.
+        show_default (Union[bool, JinjaTemplate, None]): Determines the default value to be shown.
         prompt (Union[bool, str]): The prompt text to be displayed when asking for the input.
         confirmation_prompt (Union[bool, str]): A prompt for confirmation if required.
         prompt_required (bool): Indicates whether a prompt is required.
@@ -50,13 +56,17 @@ class BaseInput(AnyInput):
         >>> )
     """
 
+    __value_cache: Mapping[str, Any] = {}
+    __default_cache: Mapping[str, Any] = {}
+
     def __init__(
         self,
         name: str,
         shortcut: Optional[str] = None,
-        default: Optional[Any] = None,
+        default: Optional[Union[Any, InputDefault]] = None,
+        callback: Optional[InputCallback] = None,
         description: Optional[str] = None,
-        show_default: Union[bool, str, None] = None,
+        show_default: Union[bool, str] = True,
         prompt: Union[bool, str] = True,
         confirmation_prompt: Union[bool, str] = False,
         prompt_required: bool = True,
@@ -79,6 +89,7 @@ class BaseInput(AnyInput):
         self._shortcut = shortcut
         self._prompt = prompt
         self._default = default
+        self._callback = callback
         self._help = description if description is not None else name
         self._type = type
         self._show_default = show_default
@@ -100,6 +111,8 @@ class BaseInput(AnyInput):
         return self._name
 
     def get_default(self) -> Any:
+        if callable(self._default):
+            return self._default(self.__value_cache)
         return self._default
 
     def get_param_decl(self) -> List[str]:
@@ -110,10 +123,9 @@ class BaseInput(AnyInput):
 
     def get_options(self) -> Mapping[str, Any]:
         options: Mapping[str, Any] = {
-            "default": self._default,
             "help": self._help,
             "type": self._type,
-            "show_default": self._show_default,
+            "show_default": self._get_calculated_show_default(),
             "confirmation_prompt": self._confirmation_prompt,
             "prompt_required": self._prompt_required,
             "hide_input": self._hide_input,
@@ -126,11 +138,44 @@ class BaseInput(AnyInput):
             "show_choices": self._show_choices,
             "show_envvar": self._show_envvar,
             "nargs": self._nargs,
+            "callback": self._wrapped_callback,
+            "default": self._wrapped_default,
         }
         if show_prompt:
             options["prompt"] = self._prompt
         return options
 
+    def _wrapped_callback(self, ctx, param, value) -> Any:
+        var_name = to_variable_name(self.get_name())
+        if var_name not in self.__value_cache:
+            if callable(self._callback):
+                result = self._callback(self.__value_cache, value)
+                self.__value_cache[var_name] = result
+                return result
+            self.__value_cache[var_name] = value
+            return value
+        return self.__value_cache[var_name]
+
+    def _wrapped_default(self) -> Any:
+        var_name = to_variable_name(self.get_name())
+        if var_name not in self.__default_cache:
+            if callable(self._default):
+                default = self._default(self.__value_cache)
+                self.__default_cache[var_name] = default
+                return default
+            self.__default_cache[var_name] = self._default
+            return self._default
+        return self.__default_cache[var_name]
+
+    def _get_calculated_show_default(self) -> str:
+        if self._show_default != True:
+            return self._show_default
+        if callable(self._default) or (
+            self.__should_render and is_probably_jinja(self._default)
+        ):
+            return colored("Autogenerated", color="yellow")
+        return f"{self._default}"
+
     def should_render(self) -> bool:
         return self.__should_render