machineconfig 1.5__py3-none-any.whl → 1.8__py3-none-any.whl

machineconfig/scripts/python/cloud_mount.py

@@ -1,52 +1,115 @@
-
-import platform
-from machineconfig.utils.utils import PROGRAM_PATH, display_options
-import crocodile.toolbox as tb
-
-
-def get_mprocs_mount_txt(cloud, rclone_cmd, localpath):
-    config = tb.Read.ini(tb.P.home().joinpath(".config/rclone/rclone.conf"))
-    cloud_brand = config[cloud]["type"]
-    header = f"{' ' + cloud + ' | ' + cloud_brand + ' '}".center(50, "=")
-    if platform.system() == "Windows":
-        sub_text_path = tb.P.tmpfile(suffix=".ps1").write_text(f"""
-echo "{header}"
-iex 'rclone about {cloud}:'
-echo 'See ~/mounts/{cloud} for the mounted cloud'
-
-echo ''
-""")
-        txt = f"""
-cd ~
-mkdir mounts -ErrorAction SilentlyContinue
-# mkdir mounts/{cloud}  # this is not needed on windows
-
-mprocs "powershell {sub_text_path}" "{rclone_cmd}" "btm" "timeout 2 & cd mounts & lf" "timeout 2 & cd mounts & pwsh" "pwsh" --names "info,service,monitor,explorer,main,terminal"
-"""
-    else:
-        txt = f"""
-cd ~
-mkdir mounts -p
-mkdir {localpath} -p
-mprocs "echo 'see ~/mounts/{cloud} for the mounted cloud'; rclone about {cloud}:" "{rclone_cmd}" "btm" "cd ~/mounts; lf" "bash" "bash" --names "about,service,monitor,explorer,main,shell"
-"""
-    return txt
-
-def main(cloud=None, network=None):
-    config = tb.Read.ini(tb.P.home().joinpath(".config/rclone/rclone.conf"))
-    # default = tb.P.home().joinpath(".machineconfig/
-    if cloud is None:
-        cloud = display_options(msg="which cloud", options=config.sections(), header="CLOUD MOUNT", default=None)
-
-    if network is None: mount_loc = tb.P.home().joinpath(f"mounts/{cloud}")
-    elif network and platform.system() == "Windows": mount_loc = "X: --network-mode"
-    else: raise ValueError("network mount only supported on windows")
-
-    mount_cmd = f"rclone mount {cloud}: {mount_loc} --vfs-cache-mode full --file-perms=0777"
-
-    txt = get_mprocs_mount_txt(cloud, mount_cmd, mount_loc)
-    PROGRAM_PATH.write_text(txt)
-
-
-if __name__ == '__main__':
-    main()
+
+"""Cloud mount script
+"""
+
+
+from machineconfig.utils.utils import PROGRAM_PATH, choose_one_option
+from crocodile.file_management import P, Read
+
+import platform
+import argparse
+from typing import Optional
+
+
+DEFAULT_MOUNT = "~/data/rclone"
+
+
+def get_rclone_config():
+    if platform.system() == "Windows": config = Read.ini(P.home().joinpath("AppData/Roaming/rclone/rclone.conf"))
+    elif platform.system() == "Linux": config = Read.ini(P.home().joinpath(".config/rclone/rclone.conf"))
+    else: raise ValueError("unsupported platform")
+    return config
+
+
+def get_mprocs_mount_txt(cloud: str, rclone_cmd: str, cloud_brand: str):  # cloud_brand = config[cloud]["type"]
+    # config = Read.ini(P.home().joinpath(".config/rclone/rclone.conf"))
+    header = f"{' ' + cloud + ' | ' + cloud_brand + ' '}".center(50, "=")
+    if platform.system() == "Windows":
+        sub_text_path = P.tmpfile(suffix=".ps1").write_text(f"""
+echo "{header}"
+iex 'rclone about {cloud}:'
+echo 'See {DEFAULT_MOUNT}/{cloud} for the mounted cloud'
+
+echo ''
+""")
+        txt = f"""
+cd ~
+mprocs "powershell {sub_text_path}" "{rclone_cmd}" "btm" "timeout 2 & cd {DEFAULT_MOUNT} & lf" "timeout 2 & cd {DEFAULT_MOUNT} & pwsh" "pwsh" --names "info,service,monitor,explorer,main,terminal"
+"""
+    else: txt = f"""
+mprocs "echo 'see {DEFAULT_MOUNT}/{cloud} for the mounted cloud'; rclone about {cloud}:" "{rclone_cmd}" "btm" "cd {DEFAULT_MOUNT}; lf" "bash" "bash" --names "about,service,monitor,explorer,main,shell"
+"""
+    return txt
+
+
+def mount(cloud: Optional[str], network: Optional[str], destination: Optional[str]) -> None:
+
+    config = get_rclone_config()
+    if cloud is None:
+        res = choose_one_option(msg="which cloud", options=config.sections(), header="CLOUD MOUNT", default=None)
+        if type(res) is str: cloud = res
+        else: raise ValueError("no cloud selected")
+
+
+    if network is None:
+        if destination is None: mount_loc = P(DEFAULT_MOUNT).expanduser().joinpath(cloud)
+        else: mount_loc = P(destination)
+
+        if platform.system() == "Windows": mount_loc.parent.create()
+        elif platform.system() == "Linux": mount_loc.create()
+        else: raise ValueError("unsupported platform")
+
+    elif network and platform.system() == "Windows": mount_loc = "X: --network-mode"
+    else: raise ValueError("network mount only supported on windows")
+
+    mount_cmd = f"rclone mount {cloud}: {mount_loc} --vfs-cache-mode full --file-perms=0777"
+
+    # txt = get_mprocs_mount_txt(cloud, mount_cmd)
+    if platform.system() == "Windows": txt = f"""
+wt --window 0 --profile "Windows PowerShell" --startingDirectory "$HOME/data/rclone" `; split-pane --horizontal  --profile "Command Prompt" --size 0.2 powershell -Command "{mount_cmd}" `; split-pane --vertical --profile "Windows PowerShell" --size 0.2 powershell -NoExit -Command "rclone about {cloud}:"  `; move-focus up
+"""
+    elif platform.system() == "Linux": txt = f"""
+
+ZJ_SESSIONS=$(zellij list-sessions)
+
+if [[ "${{ZJ_SESSIONS}}" != *"(current)"* ]]; then
+    echo "Not inside a zellij session ..."
+    echo '{mount_cmd}'
+    exit 1
+fi
+
+zellij run --direction down --name rclone -- {mount_cmd}
+sleep 1; zellij action resize decrease down
+sleep 0.2; zellij action resize decrease up
+sleep 0.2; zellij action resize decrease up
+sleep 0.1; zellij action resize decrease up
+sleep 0.1; zellij action resize decrease up
+sleep 0.1; zellij action resize decrease up
+sleep 0.1; zellij action resize decrease up
+sleep 0.1; zellij action resize decrease up
+sleep 0.1; zellij action resize decrease up
+sleep 0.1; zellij action resize decrease up
+zellij run --direction right --name about -- rclone about {cloud}:
+zellij action move-focus up
+# zellij action write-chars "cd $HOME/data/rclone/{cloud}; sleep 0.1; ls"
+zellij run --direction left --cwd $HOME/data/rclone/{cloud} -- lf
+zellij run --direction up -- btm --default_widget_type net --expanded
+zellij run --in-place --cwd $HOME/data/rclone/{cloud} -- bash
+zellij action move-focus up
+"""
+    else: raise ValueError("unsupported platform")
+    # print(f"running command: \n{txt}")
+    PROGRAM_PATH.write_text(txt)
+
+
+def main():
+    parser = argparse.ArgumentParser(description='mount cloud')
+    parser.add_argument('cloud', nargs='?', type=str, default=None, help='cloud to mount')
+    parser.add_argument('destination', nargs='?', type=str, default=None, help='destination to mount')
+    parser.add_argument('--network', type=str, default=None, help='mount network drive')
+    args = parser.parse_args()
+    mount(cloud=args.cloud, network=args.network, destination=args.destination)
+
+
+if __name__ == '__main__':
+    main()

machineconfig/scripts/python/cloud_repo_sync.py

@@ -1,114 +1,154 @@
-
-import crocodile.toolbox as tb
-import argparse
-import platform
-from machineconfig.utils.utils import print_programming_script, CONFIG_PATH, write_shell_script
-# import sys
-# import subprocess
-
-
-def get_wt_cmd(wd1: tb.P, wd2: tb.P) -> str: return f"""wt --window 0 new-tab --profile pwsh --title "gitdiff" --tabColor `#3b04d1 --startingDirectory {wd1} `  --colorScheme "Solarized Dark" `; split-pane --horizontal --profile pwsh --startingDirectory {wd2} --size 0.5 --colorScheme "Tango Dark" -- pwsh -Interactive"""
-
-
-def get_zellij_cmd(wd1: tb.P, wd2: tb.P) -> str: return f""" zellij action new-tab --name gitdiff; zellij action new-pane --direction down --name local --cwd ./data; zellij action write-chars "cd '{wd1}'; git status";  zellij action move-focus up; zellij action close-pane; zellij action new-pane --direction down --name remote --cwd code; zellij action write-chars "cd '{wd2}'; git status" """
-
-
-def args_parser():
-    parser = argparse.ArgumentParser(description="Secure Repo CLI.")
-
-    # parser.add_argument("cmd", help="command to run", choices=["pull", "push"])
-
-    parser.add_argument("--cloud", "-c", help="rclone cloud profile name.", default=None)
-    parser.add_argument("--message", "-m", help="Commit Message", default=f"new message {tb.randstr()}")
-    parser.add_argument("--key", "-k", help="Key for encryption", default=None)
-    parser.add_argument("--pwd", "-p", help="Password for encryption", default=None)
-    parser.add_argument("--push", "-u", help="Zip before sending.", action="store_true")  # default is False
-    args = parser.parse_args()
-    if args.cloud is None:
-        _path = tb.P.home().joinpath("dotfiles/config/setup/rclone_remote")
-        try: cloud = _path.read_text().replace("\n", "")
-        except FileNotFoundError:
-            print(f"No cloud profile found @ {_path}, please set one up or provide one via the --cloud flag.")
-            return ""
-    else: cloud = args.cloud
-    # repo_root = tb.P(args.repo).expanduser().absolute()
-    repo_root = tb.P.cwd()
-    repo_obj = tb.install_n_import("git", "gitpython").Repo(repo_root, search_parent_directories=True)
-    CONFIG_PATH.joinpath("remote").create()
-    repo_sync_root = CONFIG_PATH.joinpath("remote", repo_root.rel2home())  # .delete(sure=True)
-    try:
-        print("\n", "=============================== Downloading Remote Repo ====================================")
-        repo_root.from_cloud(cloud=cloud, localpath=repo_sync_root, unzip=True, decrypt=True, rel2home=True, key=args.key, pwd=args.pwd, os_specific=False)
-    except AssertionError:
-        print("Remote does not exist, creating it and exiting ... ")
-        repo_root.to_cloud(cloud=cloud, zip=True, encrypt=True, rel2home=True, key=args.key, pwd=args.pwd, os_specific=False)
-        return ""
-    repo_sync_obj = tb.install_n_import("git", "gitpython").Repo(repo_sync_root)
-    if repo_sync_obj.is_dirty():
-        print("=" * 50, '\n', f"WRANING: the remote `{repo_sync_root}` is dirty, please commit or stash changes before proceeding.", '\n', "=" * 50)
-
-    script = f"""
-echo ""
-echo "=============================== Committing Local Changes ==================================="
-cd {repo_root}
-git status
-git add .
-git commit -am "{args.message}"
-echo ""
-echo ""
-echo "=============================== Pulling Latest From Remote ================================"
-cd {repo_root}
-echo '-> Trying to removing originEnc remote from local repo if it exists.'
-git remote remove originEnc
-echo '-> Adding originEnc remote to local repo'
-git remote add originEnc {repo_sync_root}
-echo '-> Fetching originEnc remote.'
-git pull originEnc master
-
-"""
-    suffix = '.ps1' if platform.system() == 'Windows' else '.sh'
-    res = tb.Terminal().run(f". {tb.P.tmpfile(suffix=suffix).write_text(script)}", shell="powershell").capture().print()
-    if res.is_successful(strict_err=True, strict_returcode=True):
-        print("\n", "Pull succeeded, removing originEnc, the local copy of remote & pushing merged repo_root to remote ... ")
-        repo_sync_root.delete(sure=True)
-        from git.remote import Remote
-        Remote.remove(repo_obj, "originEnc")
-        if args.push:
-            repo_root.to_cloud(cloud=cloud, zip=True, encrypt=True, rel2home=True, key=args.key, pwd=args.pwd, os_specific=False)
-    else:
-        print(f"Failed to pull, keeping local copy of remote at {repo_sync_root} ... ")
-        program = f"""
-# Finalizing syncing of `{repo_root}` to `{cloud}` by pushing local changes to remote and deleting local copy of remote.
-repo_sync = tb.P(r'{repo_sync_root}')
-repo_root = tb.P(r'{repo_root}')
-repo_sync.delete(sure=True)
-from git.remote import Remote
-from git import Repo
-try: Remote.remove(Repo(repo_root), "originEnc")
-except: pass
-repo_root.to_cloud(cloud='{cloud}', zip=True, encrypt=True, rel2home=True, os_specific=False)
-"""
-        print_programming_script(program, lexer="py", desc="Abstaining from running the following autmomatically:")
-        resp = input("Would you like to run the above commands? y/[n] ") or "n"
-        if resp.lower() == "y":
-            repo_sync_root.delete(sure=True)
-            from git.remote import Remote
-            from git import Repo
-            try: Remote.remove(Repo(repo_root), "originEnc")
-            except: pass
-            repo_root.to_cloud(cloud=cloud, zip=True, encrypt=True, rel2home=True, os_specific=False)
-        else:
-            print(f"When ready, use this snippet: \n{program}")
-            if platform.system() == "Windows":
-                program = get_wt_cmd(wd1=repo_root, wd2=repo_sync_root)
-                write_shell_script(program=program, execute=True)
-                return None
-            elif platform.system() == "Linux":
-                program = get_zellij_cmd(wd1=repo_root, wd2=repo_sync_root)
-                write_shell_script(program=program, execute=True)
-                return None
-            else: raise NotImplementedError(f"Platform {platform.system()} not implemented.")
-
-
-if __name__ == "__main__":
-    args_parser()
+
+"""utils"""
+
+
+from machineconfig.utils.utils import CONFIG_PATH, DEFAULTS_PATH, write_shell_script, get_shell_file_executing_python_script
+from crocodile.file_management import P, Read, install_n_import
+from crocodile.core import randstr
+from crocodile.meta import Terminal
+import argparse
+import platform
+from typing import Optional
+# import sys
+# import subprocess
+
+
+def get_wt_cmd(wd1: P, wd2: P) -> str:
+    lines = [
+        f"""wt --window 0 new-tab --profile pwsh --title "gitdiff" --tabColor `#3b04d1 --startingDirectory {wd1} ` --colorScheme "Solarized Dark" """,
+        f"""split-pane --horizontal --profile pwsh --startingDirectory {wd2} --size 0.5 --colorScheme "Tango Dark" -- pwsh -Interactive """
+    ]
+    return " `; ".join(lines)
+
+def get_zellij_cmd(wd1: P, wd2: P) -> str:
+    lines = [f""" zellij action new-tab --name gitdiff""",
+             f"""zellij action new-pane --direction down --name local --cwd ./data """,
+             f"""zellij action write-chars "cd '{wd1}'; git status" """,
+             f"""zellij action move-focus up; zellij action close-pane """,
+             f"""zellij action new-pane --direction down --name remote --cwd code """,
+             f"""zellij action write-chars "cd '{wd2}' """,
+             f"""git status" """
+    ]
+    return "; ".join(lines)
+
+
+def args_parser():
+    parser = argparse.ArgumentParser(description="Secure Repo CLI.")
+
+    # parser.add_argument("cmd", help="command to run", choices=["pull", "push"])
+    parser.add_argument("path", nargs='?', type=str, help="Repository path, defaults to cwd.", default=None)
+
+    # parser.add_argument("--share", help="Repository path, defaults to cwd.", action="store_true", default=False)
+
+    parser.add_argument("--cloud", "-c", help="rclone cloud profile name.", default=None)
+    parser.add_argument("--message", "-m", help="Commit Message", default=f"new message {randstr()}")
+    parser.add_argument("--skip_confirmation", "-s", help="Skip confirmation.", action="store_true", default=False)
+    # parser.add_argument("--key", "-k", help="Key for encryption", default=None)
+    parser.add_argument("--pwd", "-p", help="Password for encryption", default=None)
+    parser.add_argument("--no_push", "-u", help="push to reomte.", action="store_true")  # default is False
+    args = parser.parse_args()
+
+    # if args.share:
+    #     from machineconfig.scripts.cloud.dotfiles import put
+    #     put()
+    #     return None
+    main(cloud=args.cloud, path=args.path, message=args.message, skip_confirmation=args.skip_confirmation, pwd=args.pwd, push=not args.no_push)
+
+
+def main(cloud: Optional[str] = None, path: Optional[str] = None, message: Optional[str] = None, skip_confirmation: bool = False, pwd: Optional[str] = None, push: bool = True):
+    if cloud is None:
+        try:
+            cloud_resolved = Read.ini(DEFAULTS_PATH)['general']['rclone_config_name']
+            print(f"⚠️ Using default cloud: `{cloud_resolved}` from {DEFAULTS_PATH} ⚠️")
+        except FileNotFoundError:
+            print(f"No cloud profile found @ {DEFAULTS_PATH}, please set one up or provide one via the --cloud flag.")
+            return ""
+    else: cloud_resolved = cloud
+    # repo_root = P(args.repo).expanduser().absolute()
+    repo_root = P.cwd() if path is None else P(path).expanduser().absolute()
+    repo_obj = install_n_import("git", "gitpython").Repo(repo_root, search_parent_directories=True)
+    repo_root = P(repo_obj.working_dir)  # cwd might have been in a sub directory of repo_root, so its better to redefine it.
+    CONFIG_PATH.joinpath("remote").create()
+    repo_sync_root = CONFIG_PATH.joinpath("remote", repo_root.rel2home())  # .delete(sure=True)
+    try:
+        print("\n", "=============================== Downloading Remote Repo ====================================")
+        remote_path = repo_root.get_remote_path(rel2home=True, os_specific=False, root="myhome") + ".zip.enc"
+        repo_sync_root.from_cloud(remotepath=remote_path, cloud=cloud_resolved, unzip=True, decrypt=True, rel2home=True, os_specific=False, pwd=pwd)
+    except AssertionError:
+        print("Remote does not exist, creating it and exiting ... ")
+        repo_root.to_cloud(cloud=cloud_resolved, zip=True, encrypt=True, rel2home=True, pwd=pwd, os_specific=False)
+        return ""
+    repo_sync_obj = install_n_import("git", "gitpython").Repo(repo_sync_root)
+    if repo_sync_obj.is_dirty():
+        print("=" * 50, '\n', f"WRANING: the remote `{repo_sync_root}` is dirty, please commit or stash changes before proceeding.", '\n', "=" * 50)
+
+    script = f"""
+echo ""
+echo "=============================== Committing Local Changes ==================================="
+cd {repo_root}
+git status
+git add .
+git commit -am "{message}"
+echo ""
+echo ""
+echo "=============================== Pulling Latest From Remote ================================"
+cd {repo_root}
+echo '-> Trying to removing originEnc remote from local repo if it exists.'
+git remote remove originEnc
+echo '-> Adding originEnc remote to local repo'
+git remote add originEnc {repo_sync_root}
+echo '-> Fetching originEnc remote.'
+git pull originEnc master
+
+"""
+    suffix = '.ps1' if platform.system() == 'Windows' else '.sh'
+    res = Terminal().run(f". {P.tmpfile(suffix=suffix).write_text(script)}", shell="powershell").capture().print()
+
+    if res.is_successful(strict_err=True, strict_returcode=True):
+        print("\n", "Pull succeeded, removing originEnc, the local copy of remote & pushing merged repo_root to remote ... ")
+        repo_sync_root.delete(sure=True)
+        from git.remote import Remote
+        Remote.remove(repo_obj, "originEnc")
+        if push:
+            repo_root.to_cloud(cloud=cloud_resolved, zip=True, encrypt=True, rel2home=True, pwd=pwd, os_specific=False)
+    else:
+        print(f"Failed to pull, keeping local copy of remote at {repo_sync_root} ... ")
+
+        if push:
+            if skip_confirmation: resp = "y"
+            else: resp = input(f"Would you like to proceed syncing `{repo_root}` to `{cloud_resolved}` by pushing local changes to remote and deleting local copy of remote? y/[n] ") or "n"
+        else: resp = "n"
+
+        if resp.lower() == "y":
+            delete_remote_repo_copy_and_push_local(remote_repo=repo_sync_root.str, local_repo=repo_root.str, cloud=cloud_resolved)
+        else:
+            program = f"""
+from machineconfig.scripts.python.cloud_repo_sync import delete_remote_repo_copy_and_push_local as func
+func(remote_repo=r'{repo_sync_root.str}', local_repo=r'{repo_root.str}', cloud=r'{cloud_resolved}')
+"""
+            shell_file = get_shell_file_executing_python_script(python_script=program)
+            print(f"When ready, use this snippet: \n. {shell_file}")
+            if platform.system() == "Windows":
+                program = get_wt_cmd(wd1=repo_root, wd2=repo_sync_root)
+                write_shell_script(program=program, execute=True)
+                return None
+            elif platform.system() == "Linux":
+                program = get_zellij_cmd(wd1=repo_root, wd2=repo_sync_root)
+                write_shell_script(program=program, execute=True)
+                return None
+            else: raise NotImplementedError(f"Platform {platform.system()} not implemented.")
+
+
+def delete_remote_repo_copy_and_push_local(remote_repo: str, local_repo: str, cloud: str):
+    repo_sync_root = P(remote_repo).expanduser().absolute()
+    repo_root_path = P(local_repo).expanduser().absolute()
+    repo_sync_root.delete(sure=True)
+    from git.remote import Remote
+    from git.repo import Repo
+    try: Remote.remove(Repo(repo_root_path), "originEnc")
+    except Exception: pass  # type: ignore
+    repo_root_path.to_cloud(cloud=cloud, zip=True, encrypt=True, rel2home=True, os_specific=False)
+
+
+if __name__ == "__main__":
+    args_parser()