zrb 0.28.0__py3-none-any.whl → 1.0.0__py3-none-any.whl

zrb/__init__.py

@@ -1,103 +1,111 @@
-from zrb.helper.default_env import inject_default_env
-from zrb.runner import runner
-from zrb.task.any_task import AnyTask
-from zrb.task.any_task_event_handler import (
-    OnFailed,
-    OnReady,
-    OnRetry,
-    OnSkipped,
-    OnStarted,
-    OnTriggered,
-    OnWaiting,
+from zrb.attr.type import (
+    AnyAttr,
+    BoolAttr,
+    FloatAttr,
+    IntAttr,
+    StrAttr,
+    StrDictAttr,
+    fstring,
 )
-from zrb.task.checker import Checker
+from zrb.callback.any_callback import AnyCallback
+from zrb.callback.callback import Callback
+from zrb.cmd.cmd_result import CmdResult
+from zrb.cmd.cmd_val import Cmd, CmdPath
+from zrb.config import LOAD_BUILTIN
+from zrb.content_transformer.any_content_transformer import AnyContentTransformer
+from zrb.content_transformer.content_transformer import ContentTransformer
+from zrb.context.any_context import AnyContext
+from zrb.context.any_shared_context import AnySharedContext
+from zrb.context.context import Context
+from zrb.context.shared_context import SharedContext
+from zrb.env.any_env import AnyEnv
+from zrb.env.env import Env
+from zrb.env.env_file import EnvFile
+from zrb.env.env_map import EnvMap
+from zrb.group.any_group import AnyGroup
+from zrb.group.group import Group
+from zrb.input.any_input import AnyInput
+from zrb.input.base_input import BaseInput
+from zrb.input.bool_input import BoolInput
+from zrb.input.float_input import FloatInput
+from zrb.input.int_input import IntInput
+from zrb.input.option_input import OptionInput
+from zrb.input.password_input import PasswordInput
+from zrb.input.str_input import StrInput
+from zrb.input.text_input import TextInput
+from zrb.runner.cli import cli
+from zrb.runner.web_config.config_factory import web_config
+from zrb.runner.web_schema.user import User
+from zrb.session.session import Session
+from zrb.task.any_task import AnyTask
+from zrb.task.base_task import BaseTask
+from zrb.task.base_trigger import BaseTrigger
 from zrb.task.cmd_task import CmdTask
-from zrb.task.decorator import python_task
-from zrb.task.docker_compose_start_task import DockerComposeStartTask
-from zrb.task.docker_compose_task import DockerComposeTask, ServiceConfig
-from zrb.task.flow_task import FlowTask
-from zrb.task.http_checker import HTTPChecker
-from zrb.task.notifier import Notifier
-from zrb.task.parallel import AnyParallel, Parallel
-from zrb.task.path_checker import PathChecker
-from zrb.task.path_watcher import PathWatcher
-from zrb.task.port_checker import PortChecker
-from zrb.task.recurring_task import RecurringTask
-from zrb.task.remote_cmd_task import RemoteCmdTask
-from zrb.task.resource_maker import (
-    Replacement,
-    ReplacementMutator,
-    ResourceMaker,
-    get_default_resource_excludes,
-    get_default_resource_skip_parsing,
-)
+from zrb.task.http_check import HttpCheck
+from zrb.task.llm_task import LLMTask
+from zrb.task.make_task import make_task
 from zrb.task.rsync_task import RsyncTask
-from zrb.task.server import Controller, Server
+from zrb.task.scaffolder import Scaffolder
+from zrb.task.scheduler import Scheduler
 from zrb.task.task import Task
-from zrb.task.time_watcher import TimeWatcher
-from zrb.task.watcher import Watcher
-from zrb.task.wiki_task import create_wiki_tasks
-from zrb.task_env.env import Env
-from zrb.task_env.env_file import EnvFile
-from zrb.task_group.group import Group
-from zrb.task_input.any_input import AnyInput
-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
+from zrb.task.tcp_check import TcpCheck
+from zrb.util.load import load_file, load_module
+from zrb.xcom.xcom import Xcom
 
-assert create_wiki_tasks
-assert runner
-assert AnyTask
-assert OnTriggered
-assert OnWaiting
-assert OnSkipped
-assert OnStarted
-assert OnReady
-assert OnRetry
-assert OnFailed
-assert AnyParallel
-assert Parallel
-assert python_task
-assert Task
-assert CmdTask
-assert DockerComposeTask
-assert DockerComposeStartTask
-assert ServiceConfig
-assert RemoteCmdTask
-assert RsyncTask
-assert Notifier
-assert Checker
-assert HTTPChecker
-assert PortChecker
-assert PathChecker
-assert PathWatcher
-assert Controller
-assert Server
-assert Watcher
-assert TimeWatcher
-assert ResourceMaker
-assert FlowTask
-assert RecurringTask
-assert get_default_resource_excludes
-assert get_default_resource_skip_parsing
-assert Replacement
-assert ReplacementMutator
+assert load_file
+assert load_module
+assert fstring
+assert AnyAttr
+assert BoolAttr
+assert IntAttr
+assert FloatAttr
+assert StrAttr
+assert StrDictAttr
+assert AnyCallback
+assert Callback
+assert AnyEnv
+assert Env
+assert EnvFile
+assert EnvMap
 assert AnyInput
-assert Input
+assert BaseInput
 assert BoolInput
-assert ChoiceInput
+assert Cmd
+assert CmdPath
+assert CmdResult
+assert CmdTask
+assert HttpCheck
+assert TcpCheck
 assert FloatInput
 assert IntInput
-assert MultilineInput
+assert OptionInput
 assert PasswordInput
 assert StrInput
-assert Env
-assert EnvFile
+assert TextInput
+assert AnyGroup
 assert Group
+assert AnyTask
+assert BaseTask
+assert BaseTrigger
+assert RsyncTask
+assert Task
+assert LLMTask
+assert Session
+assert AnyContext
+assert Context
+assert AnySharedContext
+assert SharedContext
+assert make_task
+assert AnyContentTransformer
+assert ContentTransformer
+assert Scaffolder
+assert Scheduler
+assert cli
+assert Xcom
+assert web_config
+assert User
+
+if LOAD_BUILTIN:
+    from zrb import builtin
 
-inject_default_env()
+    assert builtin

zrb/__main__.py

@@ -1,8 +1,26 @@
-from zrb.helper.cli import create_cli
-from zrb.helper.default_env import inject_default_env
+import sys
 
-inject_default_env()
-cli = create_cli()
+from zrb.config import INIT_MODULES, INIT_SCRIPTS
+from zrb.runner.cli import cli
+from zrb.util.cli.style import stylize_error, stylize_warning
+from zrb.util.group import NodeNotFoundError
+from zrb.util.load import load_file, load_module
 
-if __name__ == "__main__":
-    cli()
+
+def serve_cli():
+    try:
+        for init_module in INIT_MODULES:
+            load_module(init_module)
+        for init_script in INIT_SCRIPTS:
+            load_file(init_script, -1)
+        cli.run(sys.argv[1:])
+    except KeyboardInterrupt:
+        print(stylize_warning("\nStopped"), file=sys.stderr)
+        sys.exit(1)
+    except RuntimeError as e:
+        if f"{e}".lower() != "event loop is closed":
+            raise e
+        sys.exit(1)
+    except NodeNotFoundError as e:
+        print(stylize_error(f"{e}"), file=sys.stderr)
+        sys.exit(1)

zrb/attr/type.py

@@ -0,0 +1,12 @@
+from typing import Any, Callable
+
+from zrb.context.any_shared_context import AnySharedContext
+
+fstring = str
+AnyAttr = Any | fstring | Callable[[AnySharedContext], Any]
+StrAttr = str | fstring | Callable[[AnySharedContext], str]
+BoolAttr = bool | fstring | Callable[[AnySharedContext], bool]
+IntAttr = int | fstring | Callable[[AnySharedContext], int]
+FloatAttr = float | fstring | Callable[[AnySharedContext], float]
+StrDictAttr = dict[str, StrAttr] | Callable[[AnySharedContext], dict[str, Any]]
+StrListAttr = list[StrAttr] | Callable[[AnySharedContext], list[str]]

zrb/builtin/__init__.py

@@ -1,135 +1,66 @@
-from zrb.builtin.base64 import base64_group, decode_base64, encode_base64
-from zrb.builtin.devtool import (
-    devtool_group,
-    devtool_install_group,
-    install_aws,
-    install_docker,
-    install_gcloud,
-    install_gvm,
-    install_helix,
-    install_helm,
-    install_kubectl,
-    install_nvm,
-    install_pulumi,
-    install_pyenv,
-    install_sdkman,
-    install_selenium,
-    install_terraform,
-    install_tmux,
-    install_zsh,
+from zrb.builtin.base64 import decode_base64, encode_base64
+from zrb.builtin.git import (
+    get_git_diff,
+    git_commit,
+    git_pull,
+    git_push,
+    prune_local_branches,
 )
-from zrb.builtin.docker import docker_group, prune_docker
-from zrb.builtin.env import env_group, get_env
-from zrb.builtin.eval import evaluate
-from zrb.builtin.explain import explain_group, explain_tasks
-from zrb.builtin.git import clear_git_branch, get_git_file_changes, git_group
-from zrb.builtin.md5 import hash_text_md5, md5_group, sum_file_md5
-from zrb.builtin.monorepo import add_to_monorepo, pull_from_monorepo, push_to_monorepo
-from zrb.builtin.process import (
-    get_process_pid_by_name,
-    get_process_pid_by_port,
-    process_group,
-    process_pid_group,
+from zrb.builtin.git_subtree import git_add_subtree, git_pull_subtree, git_push_subtree
+from zrb.builtin.llm.llm_chat import llm_chat
+from zrb.builtin.md5 import hash_md5, sum_md5
+from zrb.builtin.project.add.fastapp.fastapp_task import add_fastapp_to_project
+from zrb.builtin.project.create.project_task import create_project
+from zrb.builtin.python import format_python_code
+from zrb.builtin.random import shuffle_values, throw_dice
+from zrb.builtin.setup.asdf.asdf import setup_asdf
+from zrb.builtin.setup.latex.ubuntu import setup_latex_on_ubuntu
+from zrb.builtin.setup.tmux.tmux import setup_tmux
+from zrb.builtin.setup.ubuntu import setup_ubuntu
+from zrb.builtin.setup.zsh.zsh import setup_zsh
+from zrb.builtin.shell.autocomplete.bash import make_bash_autocomplete
+from zrb.builtin.shell.autocomplete.subcmd import get_shell_subcommands
+from zrb.builtin.shell.autocomplete.zsh import make_zsh_autocomplete
+from zrb.builtin.todo import (
+    add_todo,
+    archive_todo,
+    complete_todo,
+    edit_todo,
+    list_todo,
+    log_todo,
+    show_todo,
 )
-from zrb.builtin.project import (
-    add_cmd_task,
-    add_docker_compose_task,
-    add_fastapp_application,
-    add_fastapp_crud,
-    add_fastapp_field,
-    add_fastapp_module,
-    add_plugin,
-    add_project_tasks,
-    add_python_app,
-    add_python_task,
-    create_project,
-    project_add_app_group,
-    project_add_fastapp_group,
-    project_add_group,
-    project_add_task_group,
-    project_group,
-)
-from zrb.builtin.say import say
-from zrb.builtin.schedule import schedule
-from zrb.builtin.ubuntu import (
-    install_ubuntu_all,
-    install_ubuntu_essentials,
-    install_ubuntu_tex,
-    install_ubuntu_toys,
-    ubuntu_group,
-    ubuntu_install_group,
-    update_ubuntu,
-)
-from zrb.builtin.update import update_zrb
-from zrb.builtin.version import get_version
-from zrb.builtin.watch_changes import watch_changes
 
-assert base64_group
-assert decode_base64
-assert encode_base64
-assert devtool_group
-assert devtool_install_group
-assert install_aws
-assert install_docker
-assert install_gcloud
-assert install_gvm
-assert install_helix
-assert install_helm
-assert install_kubectl
-assert install_nvm
-assert install_pulumi
-assert install_pyenv
-assert install_sdkman
-assert install_selenium
-assert install_terraform
-assert install_tmux
-assert install_zsh
-assert docker_group
-assert prune_docker
-assert env_group
-assert get_env
-assert git_group
-assert clear_git_branch
-assert get_git_file_changes
-assert md5_group
-assert hash_text_md5
-assert sum_file_md5
-assert process_group
-assert process_pid_group
-assert get_process_pid_by_name
-assert get_process_pid_by_port
-assert project_group
-assert project_add_group
-assert project_add_app_group
-assert project_add_fastapp_group
-assert project_add_task_group
 assert create_project
-assert add_plugin
-assert add_fastapp_application
-assert add_fastapp_crud
-assert add_fastapp_module
-assert add_fastapp_field
-assert add_python_app
-assert add_project_tasks
-assert add_cmd_task
-assert add_docker_compose_task
-assert add_python_task
-assert add_to_monorepo
-assert pull_from_monorepo
-assert push_to_monorepo
-assert ubuntu_group
-assert ubuntu_install_group
-assert update_ubuntu
-assert ubuntu_install_group
-assert install_ubuntu_all
-assert install_ubuntu_essentials
-assert install_ubuntu_tex
-assert install_ubuntu_toys
-assert explain_group
-assert explain_tasks
-assert evaluate
-assert say
-assert schedule
-assert update_zrb
-assert get_version
-assert watch_changes
+assert add_fastapp_to_project
+assert get_shell_subcommands
+assert make_bash_autocomplete
+assert make_zsh_autocomplete
+assert encode_base64
+assert decode_base64
+assert llm_chat
+assert hash_md5
+assert sum_md5
+assert get_git_diff
+assert prune_local_branches
+assert format_python_code
+assert git_commit
+assert git_pull
+assert git_push
+assert git_add_subtree
+assert git_pull_subtree
+assert git_push_subtree
+assert list_todo
+assert add_todo
+assert archive_todo
+assert edit_todo
+assert complete_todo
+assert log_todo
+assert show_todo
+assert throw_dice
+assert shuffle_values
+assert setup_ubuntu
+assert setup_latex_on_ubuntu
+assert setup_asdf
+assert setup_tmux
+assert setup_zsh

zrb/builtin/base64.py

@@ -0,0 +1,34 @@
+from zrb.builtin.group import base64_group
+from zrb.context.any_context import AnyContext
+from zrb.input.str_input import StrInput
+from zrb.task.make_task import make_task
+
+
+@make_task(
+    name="encode-base64",
+    input=StrInput(name="text", description="Text", prompt="Text to encode"),
+    description="🔐 Encode text to base64",
+    group=base64_group,
+    alias="encode",
+)
+def encode_base64(ctx: AnyContext) -> str:
+    import base64
+
+    result = base64.b64encode(ctx.input.text.encode()).decode()
+    ctx.print(result)
+    return result
+
+
+@make_task(
+    name="decode-base64",
+    input=StrInput(name="text", description="Text", prompt="Text to encode"),
+    description="🔓 Decode base64 text",
+    group=base64_group,
+    alias="decode",
+)
+def decode_base64(ctx: AnyContext) -> str:
+    import base64
+
+    result = base64.b64decode(ctx.input.text.encode()).decode()
+    ctx.print(result)
+    return result

zrb/builtin/git.py

@@ -0,0 +1,169 @@
+from zrb.builtin.group import git_branch_group, git_group
+from zrb.context.any_context import AnyContext
+from zrb.input.bool_input import BoolInput
+from zrb.input.str_input import StrInput
+from zrb.task.make_task import make_task
+from zrb.util.cli.style import stylize_faint, stylize_green, stylize_red, stylize_yellow
+from zrb.util.git import (
+    add,
+    commit,
+    delete_branch,
+    get_branches,
+    get_current_branch,
+    get_diff,
+    get_repo_dir,
+    pull,
+    push,
+)
+
+
+@make_task(
+    name="get-git-diff",
+    input=[
+        StrInput(
+            name="source",
+            description="Source branch/tag/commit",
+            prompt="Source branch/tag/commit",
+            default_str="main",
+        ),
+        StrInput(
+            name="current",
+            description="Current branch/tag/commit",
+            prompt="Current branch/tag/commit",
+            default_str="HEAD",
+        ),
+        BoolInput(
+            name="created",
+            description="Include created files",
+            prompt="Include created files",
+            default_str="True",
+        ),
+        BoolInput(
+            name="removed",
+            description="Include removed files",
+            prompt="Include removed files",
+            default_str="True",
+        ),
+        BoolInput(
+            name="updated",
+            description="Include updated files",
+            prompt="Include updated files",
+            default_str="True",
+        ),
+    ],
+    description="🔍 Get modified files",
+    group=git_group,
+    alias="diff",
+)
+async def get_git_diff(ctx: AnyContext):
+    ctx.print(stylize_faint("Get directory"))
+    repo_dir = await get_repo_dir(print_method=ctx.print)
+    diff = await get_diff(
+        repo_dir, ctx.input.source, ctx.input.current, print_method=ctx.print
+    )
+    result = []
+    decorated = []
+    if ctx.input.created and diff.created:
+        decorated += [stylize_green(f"++ {name}") for name in diff.created]
+        result += diff.created
+    if ctx.input.updated and diff.updated:
+        decorated += [stylize_yellow(f"+- {name}") for name in diff.updated]
+        result += diff.updated
+    if ctx.input.removed and diff.removed:
+        decorated += [stylize_red(f"-- {name}") for name in diff.removed]
+        result += diff.removed
+    if len(decorated) > 0:
+        decorated = [""] + decorated + [""]
+    ctx.print(
+        "\n" + "\n".join((f"    {decorated_line}" for decorated_line in decorated))
+    )
+    return "\n".join(result)
+
+
+@make_task(
+    name="prune-local-git-branches",
+    description="🧹 Prune local branches",
+    group=git_branch_group,
+    alias="prune",
+)
+async def prune_local_branches(ctx: AnyContext):
+    ctx.print(stylize_faint("Get directory"))
+    repo_dir = await get_repo_dir(print_method=ctx.print)
+    ctx.print(stylize_faint("Get existing branches"))
+    branches = await get_branches(repo_dir, print_method=ctx.print)
+    ctx.print(stylize_faint("Get current branch"))
+    current_branch = await get_current_branch(repo_dir, print_method=ctx.print)
+    for branch in branches:
+        if branch == current_branch or branch == "main" or branch == "master":
+            continue
+        ctx.print(stylize_faint(f"Removing local branch: {branch}"))
+        try:
+            await delete_branch(repo_dir, branch, print_method=ctx.print)
+        except Exception as e:
+            ctx.log_error(e)
+
+
+@make_task(
+    name="git-commit",
+    input=StrInput(
+        name="message",
+        description="Commit message",
+        prompt="Commit message",
+        default_str="Add feature/fix bug",
+    ),
+    description="📝 Commit changes",
+    group=git_group,
+    alias="commit",
+)
+async def git_commit(ctx: AnyContext):
+    ctx.print(stylize_faint("Get directory"))
+    repo_dir = await get_repo_dir(print_method=ctx.print)
+    ctx.print(stylize_faint("Add changes to staging"))
+    await add(repo_dir, print_method=ctx.print)
+    ctx.print(stylize_faint("Commit changes"))
+    await commit(repo_dir, ctx.input.message, print_method=ctx.print)
+
+
+@make_task(
+    name="git-pull",
+    description="📥 Pull changes from remote repository",
+    input=StrInput(
+        name="remote",
+        description="Remote name",
+        prompt="Remote name",
+        default_str="origin",
+    ),
+    upstream=git_commit,
+    group=git_group,
+    alias="pull",
+)
+async def git_pull(ctx: AnyContext):
+    ctx.print(stylize_faint("Get directory"))
+    repo_dir = await get_repo_dir(print_method=ctx.print)
+    ctx.print(stylize_faint("Get current branch"))
+    current_branch = await get_current_branch(repo_dir, print_method=ctx.print)
+    remote = ctx.input.remote
+    ctx.print(stylize_faint(f"Pulling from {remote}/{current_branch}"))
+    await pull(repo_dir, remote, current_branch, print_method=ctx.print)
+
+
+@make_task(
+    name="git-push",
+    description="📤 Push changes to remote repository",
+    input=StrInput(
+        name="remote",
+        description="Remote name",
+        prompt="Remote name",
+        default_str="origin",
+    ),
+    upstream=git_commit,
+    group=git_group,
+    alias="push",
+)
+async def git_push(ctx: AnyContext):
+    repo_dir = await get_repo_dir(print_method=ctx.print)
+    ctx.print(stylize_faint("Get current branch"))
+    current_branch = await get_current_branch(repo_dir, print_method=ctx.print)
+    remote = ctx.input.remote
+    ctx.print(stylize_faint(f"Pushing to {remote}/{current_branch}"))
+    await push(repo_dir, remote, current_branch, print_method=ctx.print)