machineconfig 1.92__py3-none-any.whl → 1.94__py3-none-any.whl

machineconfig/scripts/python/devops.py

@@ -1,118 +1,219 @@
-
-
-"""devops
+"""devops with emojis
 """
 
-from machineconfig.utils.utils import display_options, PROGRAM_PATH, write_shell_script
+from machineconfig.utils.utils import display_options, PROGRAM_PATH, write_shell_script_to_default_program_path
 from platform import system
 from enum import Enum
 from typing import Optional
 
 
 class Options(Enum):
-    update         = 'UPDATE essential repos'
-    cli_install    = 'DEVAPPS install'
-    ve             = 'VE install'
-    sym_path_shell = 'SYMLINKS, PATH & SHELL PROFILE'
-    sym_new        = 'SYMLINKS new'
-    ssh_add_pubkey = 'SSH add pub key to this machine'
-    ssh_add_id     = 'SSH add identity (private key) to this machine'
-    ssh_use_pair   = 'SSH use key pair to connect two machines'
-    ssh_setup      = 'SSH setup'
-    ssh_setup_wsl  = 'SSH setup wsl'
-    backup         = 'BACKUP'
-    retreive       = 'RETRIEVE'
-    scheduler      = 'SCHEDULER'
+    update         = '🔄 UPDATE essential repos'
+    cli_install    = '⚙️ DEVAPPS install'
+    ve             = '🐍 VE install'
+    sym_path_shell = '🔗 SYMLINKS, PATH & SHELL PROFILE'
+    sym_new        = '🆕 SYMLINKS new'
+    ssh_add_pubkey = '🔑 SSH add pub key to this machine'
+    ssh_add_id     = '🗝️ SSH add identity (private key) to this machine'
+    ssh_use_pair   = '🔐 SSH use key pair to connect two machines'
+    ssh_setup      = '📡 SSH setup'
+    ssh_setup_wsl  = '🐧 SSH setup wsl'
+    dot_files_sync = '🔗 DOTFILES sync'
+    backup         = '💾 BACKUP'
+    retreive       = '📥 RETRIEVE'
+    scheduler      = '⏰ SCHEDULER'
 
 
 def args_parser():
+    print(f"""
+╔{'═' * 70}╗
+║ 🛠️  DevOps Tool Suite                                                    ║
+╚{'═' * 70}╝
+""")
+    
     import argparse
     parser = argparse.ArgumentParser()
-    parser.add_argument("-w", "--which", help="which option to run", type=str, default=None)  # , choices=[op.value for op in Options]
+    new_line = "\n\n"
+    parser.add_argument("-w", "--which", help=f"""which option to run\nChoose one of those:\n{new_line.join([f"{item.name}: {item.value}" for item in list(Options)])}""", type=str, default=None)  # , choices=[op.value for op in Options]
     args = parser.parse_args()
     main(which=args.which)
 
 
 def main(which: Optional[str] = None):
     PROGRAM_PATH.delete(sure=True, verbose=False)
+    print(f"""
+╭{'─' * 70}╮
+│ 🚀 Initializing DevOps operation...                                      │
+╰{'─' * 70}╯
+""")
+    
     options = [op.value for op in Options]
     if which is None:
-        # import questionary
-        # choice_key = questionary.autocomplete(message="Which option to run?", choices=options, default=options[0]).ask()
-        choice_key = display_options(msg="", options=options, header="DEVOPS", default=options[0])
+        try:
+            choice_key = display_options(msg="", options=options, header="🛠️ DEVOPS", default=options[0])
+        except KeyboardInterrupt:
+            print(f"""
+╔{'═' * 70}╗
+║ ❌ Operation cancelled by user                                           ║
+╚{'═' * 70}╝
+""")
+            return
     else: choice_key = Options[which].value
 
+    print(f"""
+╔{'═' * 70}╗
+║ 🔧 SELECTED OPERATION                                                    ║
+╠{'═' * 70}╣
+║ {choice_key.center(68)} ║
+╚{'═' * 70}╝
+""")
+
     if choice_key == Options.update.value:
+        print(f"""
+╭{'─' * 70}╮
+│ 🔄 Updating essential repositories...                                    │
+╰{'─' * 70}╯
+""")
         import machineconfig.scripts.python.devops_update_repos as helper
         program = helper.main()
 
     elif choice_key == Options.ve.value:
-        program = ""
-        reply: bool=False
-        # if P.cwd().joinpath(".ve.ini").exists():
-        #     reply = input("Detected .ve.ini file. Do you want to use it to build ve? (y/[n]): ") == "y"
-        #     if reply:
-        #         from machineconfig.utils.ve import get_ve_install_script_from_specs
-        #         program_win = get_ve_install_script_from_specs(repo_root=P.cwd().to_str(), system="Windows")
-        #         program_lin = get_ve_install_script_from_specs(repo_root=P.cwd().to_str(), system="Linux")
-        #         install_reply = input("Proceed with installation? (y/[n]): ") == "y"
-        #         if not install_reply: program = ""
-        #         else:
-        #             if system() == "Windows": program = program_win
-        #             elif system() == "Linux": program = program_lin
-        #             else: raise ValueError(f"Unknown system: {system()}")
-
-        if not reply:
-            from machineconfig.utils.ve import get_ve_install_script
-            program = get_ve_install_script()
+        print(f"""
+╭{'─' * 70}╮
+│ 🐍 Setting up virtual environment...                                     │
+╰{'─' * 70}╯
+""")
+        from machineconfig.utils.ve import get_ve_install_script
+        program = get_ve_install_script()
 
     elif choice_key == Options.cli_install.value:
+        print(f"""
+╭{'─' * 70}╮
+│ ⚙️  Installing development applications...                                │
+╰{'─' * 70}╯
+""")
         import machineconfig.scripts.python.devops_devapps_install as helper
         program = helper.main()
 
     elif choice_key == Options.sym_new.value:
+        print(f"""
+╭{'─' * 70}╮
+│ 🔄 Creating new symlinks...                                              │
+╰{'─' * 70}╯
+""")
         import machineconfig.jobs.python.python_ve_symlink as helper
         program = helper.main()
 
     elif choice_key == Options.sym_path_shell.value:
+        print(f"""
+╭{'─' * 70}╮
+│ 🔗 Setting up symlinks, PATH, and shell profile...                       │
+╰{'─' * 70}╯
+""")
         import machineconfig.profile.create as helper
         helper.main()
         program = "echo '✅ done with symlinks'"
 
     elif choice_key == Options.ssh_add_pubkey.value:
+        print(f"""
+╭{'─' * 70}╮
+│ 🔑 Adding public SSH key to this machine...                              │
+╰{'─' * 70}╯
+""")
         import machineconfig.scripts.python.devops_add_ssh_key as helper
         program = helper.main()
 
     elif choice_key == Options.ssh_use_pair.value:
+        print(f"""
+╔{'═' * 70}╗
+║ ❌ ERROR: Not Implemented                                                ║
+║ SSH key pair connection feature is not yet implemented                   ║
+╚{'═' * 70}╝
+""")
         raise NotImplementedError
 
     elif choice_key == Options.ssh_add_id.value:  # so that you can SSH directly withuot pointing to identity key.
+        print(f"""
+╭{'─' * 70}╮
+│ 🗝️  Adding SSH identity (private key) to this machine...                  │
+╰{'─' * 70}╯
+""")
         import machineconfig.scripts.python.devops_add_identity as helper
         program = helper.main()
 
     elif choice_key == Options.ssh_setup.value:
+        print(f"""
+╭{'─' * 70}╮
+│ 📡 Setting up SSH...                                                     │
+╰{'─' * 70}╯
+""")
         program_windows = """Invoke-WebRequest https://raw.githubusercontent.com/thisismygitrepo/machineconfig/main/src/machineconfig/setup_windows/openssh_all.ps1 | Invoke-Expression  # https://github.com/thisismygitrepo.keys"""
         program_linux = """curl https://raw.githubusercontent.com/thisismygitrepo/machineconfig/main/src/machineconfig/setup_linux/openssh_all.sh | sudo bash  # https://github.com/thisismygitrepo.keys"""
         program = program_linux if system() == "Linux" else program_windows
 
     elif choice_key == Options.ssh_setup_wsl.value:
+        print(f"""
+╭{'─' * 70}╮
+│ 🐧 Setting up SSH for WSL...                                             │
+╰{'─' * 70}╯
+""")
         program = """curl https://raw.githubusercontent.com/thisismygitrepo/machineconfig/main/src/machineconfig/setup_linux/openssh_wsl.sh | sudo bash"""
 
     elif choice_key == Options.backup.value:
+        print(f"""
+╭{'─' * 70}╮
+│ 💾 Creating backup...                                                    │
+╰{'─' * 70}╯
+""")
         from machineconfig.scripts.python.devops_backup_retrieve import main_backup_retrieve as helper
         program = helper(direction="BACKUP")
+        
     elif choice_key == Options.retreive.value:
+        print(f"""
+╭{'─' * 70}╮
+│ 📥 Retrieving backup...                                                  │
+╰{'─' * 70}╯
+""")
         from machineconfig.scripts.python.devops_backup_retrieve import main_backup_retrieve as helper
         program = helper(direction="RETRIEVE")
 
     elif choice_key == Options.scheduler.value:
+        print(f"""
+╭{'─' * 70}╮
+│ ⏰ Setting up scheduler...                                               │
+╰{'─' * 70}╯
+""")
         from machineconfig.scripts.python.scheduler import main as helper
         program = helper()
 
-    else: raise ValueError(f"Unimplemented choice: {choice_key}")
+    elif choice_key == Options.dot_files_sync.value:
+        print(f"""
+╭{'─' * 70}╮
+│ 🔗 Synchronizing dotfiles...                                             │
+╰{'─' * 70}╯
+""")
+        from machineconfig.scripts.python.cloud_repo_sync import main as helper, P
+        program = helper(cloud=None, path=str(P.home() / "dotfiles"), pwd=None, action="ask")
+
+    else: 
+        print(f"""
+╔{'═' * 70}╗
+║ ❌ ERROR: Invalid choice                                                 ║
+║ The selected operation is not implemented: {choice_key}                  
+╚{'═' * 70}╝
+""")
+        raise ValueError(f"Unimplemented choice: {choice_key}")
+        
     if program:
-        write_shell_script(program=program, display=True, preserve_cwd=True, desc="Shell script prepared by Python.", execute=True if which is not None else False)
-    else: write_shell_script(program="echo 'Done.'", display=False, desc="Shell script prepared by Python.", preserve_cwd=True, execute=False)  # Python did not return any script to run.
+        print(f"""
+╭{'─' * 70}╮
+│ 📜 Preparing shell script...                                             │
+╰{'─' * 70}╯
+""")
+        write_shell_script_to_default_program_path(program=program, display=True, preserve_cwd=True, desc="🔧 Shell script prepared by Python.", execute=True if which is not None else False)
+    else: 
+        write_shell_script_to_default_program_path(program="echo '✨ Done.'", display=False, desc="🔧 Shell script prepared by Python.", preserve_cwd=True, execute=False)
+
 
 
 if __name__ == "__main__":

machineconfig/scripts/python/devops_add_identity.py

@@ -1,25 +1,128 @@
-
 """ID
 """
 
 
 # from platform import system
 from crocodile.file_management import P
-from machineconfig.utils.utils import display_options
+from machineconfig.utils.utils_options import display_options
 
 
 def main():
+    print(f"""
+╔{'═' * 70}╗
+║ 🔑 SSH IDENTITY MANAGEMENT                                               ║
+╚{'═' * 70}╝
+""")
+    
+    print(f"""
+╭{'─' * 70}╮
+│ 🔍 Searching for existing SSH keys...                                    │
+╰{'─' * 70}╯
+""")
+    
     private_keys = P.home().joinpath(".ssh").search("*.pub").apply(lambda x: x.with_name(x.stem)).filter(lambda x: x.exists())
+    
+    if private_keys:
+        print(f"""
+╭{'─' * 70}╮
+│ ✅ Found {len(private_keys)} SSH private key(s)                                    │
+╰{'─' * 70}╯
+""")
+    else:
+        print(f"""
+╭{'─' * 70}╮
+│ ⚠️  No SSH private keys found                                             │
+╰{'─' * 70}╯
+""")
+        
     choice = display_options(msg="Path to private key to be used when ssh'ing: ", options=private_keys.apply(str).list + ["I have the path to the key file", "I want to paste the key itself"])
-    if choice == "I have the path to the key file": path_to_key = P(input("Input path here: ")).expanduser().absolute()
-    elif choice == "I want to paste the key itself": path_to_key = P.home().joinpath(f".ssh/{input('file name (default: my_pasted_key): ') or 'my_pasted_key'}").write_text(input("Paste the private key here: "))
-    elif isinstance(choice, str): path_to_key = P(choice)
-    else: raise NotImplementedError(f"Choice {choice} not supported")
+    
+    if choice == "I have the path to the key file":
+        print(f"""
+╭{'─' * 70}╮
+│ 📄 Please enter the path to your private key file                         │
+╰{'─' * 70}╯
+""")
+        path_to_key = P(input("📋 Input path here: ")).expanduser().absolute()
+        print(f"""
+╭{'─' * 70}╮
+│ 📂 Using key from custom path: {path_to_key}              │
+╰{'─' * 70}╯
+""")
+        
+    elif choice == "I want to paste the key itself":
+        print(f"""
+╭{'─' * 70}╮
+│ 📋 Please provide a filename and paste the private key content            │
+╰{'─' * 70}╯
+""")
+        key_filename = input("📝 File name (default: my_pasted_key): ") or "my_pasted_key"
+        path_to_key = P.home().joinpath(f".ssh/{key_filename}").write_text(input("🔑 Paste the private key here: "))
+        print(f"""
+╭{'─' * 70}╮
+│ 💾 Key saved to: {path_to_key}                           │
+╰{'─' * 70}╯
+""")
+        
+    elif isinstance(choice, str): 
+        path_to_key = P(choice)
+        print(f"""
+╭{'─' * 70}╮
+│ 🔑 Using selected key: {path_to_key.name}                                 │
+╰{'─' * 70}╯
+""")
+        
+    else: 
+        print(f"""
+╔{'═' * 70}╗
+║ ❌ ERROR: Invalid choice                                                 ║
+║ The selected option is not supported: {choice}                           ║
+╚{'═' * 70}╝
+""")
+        raise NotImplementedError(f"Choice {choice} not supported")
+    
     txt = f"IdentityFile {path_to_key.collapseuser().as_posix()}"  # adds this id for all connections, no host specified.
     config_path = P.home().joinpath(".ssh/config")
-    if config_path.exists(): config_path.modify_text(txt_search=txt, txt_alt=txt, replace_line=True, notfound_append=True, prepend=True)  # note that Identity line must come on top of config file otherwise it won't work, hence `prepend=True`
-    else: config_path.write_text(txt)
-    program = f"echo 'Finished adding identity to ssh config file. {'*'*50} Consider reloading config file.'"
+    
+    print(f"""
+╭{'─' * 70}╮
+│ 📝 Updating SSH configuration...                                          │
+╰{'─' * 70}╯
+""")
+    
+    if config_path.exists(): 
+        config_path.modify_text(txt_search=txt, txt_alt=txt, replace_line=True, notfound_append=True, prepend=True)  # note that Identity line must come on top of config file otherwise it won't work, hence `prepend=True`
+        print(f"""
+╭{'─' * 70}╮
+│ ✏️  Updated existing SSH config file                                       │
+╰{'─' * 70}╯
+""")
+    else: 
+        config_path.write_text(txt)
+        print(f"""
+╭{'─' * 70}╮
+│ 📄 Created new SSH config file                                            │
+╰{'─' * 70}╯
+""")
+    
+    program = f"""echo '
+╔{'═' * 70}╗
+║ ✅ SSH IDENTITY CONFIGURATION COMPLETE                                   ║
+╠{'═' * 70}╣
+║ Identity added to SSH config file                                        ║
+║ Consider reloading the SSH config to apply changes                       ║
+╚{'═' * 70}╝
+'"""
+    
+    print(f"""
+╔{'═' * 70}╗
+║ 🎉 CONFIGURATION SUCCESSFUL                                              ║
+╠{'═' * 70}╣
+║ Identity added: {path_to_key.name}                                       
+║ Config file: {config_path}                                
+╚{'═' * 70}╝
+""")
+    
     return program
 
 

machineconfig/scripts/python/devops_add_ssh_key.py

@@ -1,4 +1,3 @@
-
 """SSH
 """
 
@@ -9,18 +8,68 @@ from crocodile.file_management import P
 
 
 def get_add_ssh_key_script(path_to_key: P):
-    if system() == "Linux": authorized_keys = P.home().joinpath(".ssh/authorized_keys")
-    elif system() == "Windows": authorized_keys = P("C:/ProgramData/ssh/administrators_authorized_keys")
-    else: raise NotImplementedError
+    print(f"""
+╔{'═' * 70}╗
+║ 🔑 SSH KEY CONFIGURATION                                                 ║
+╚{'═' * 70}╝
+""")
+    
+    if system() == "Linux": 
+        authorized_keys = P.home().joinpath(".ssh/authorized_keys")
+        print(f"""
+╭{'─' * 70}╮
+│ 🐧 Linux SSH configuration                                               │
+│ 📄 Authorized keys file: {authorized_keys}                      │
+╰{'─' * 70}╯
+""")
+    elif system() == "Windows": 
+        authorized_keys = P("C:/ProgramData/ssh/administrators_authorized_keys")
+        print(f"""
+╭{'─' * 70}╮
+│ 🪟 Windows SSH configuration                                             │
+│ 📄 Authorized keys file: {authorized_keys}                │
+╰{'─' * 70}╯
+""")
+    else: 
+        print(f"""
+╔{'═' * 70}╗
+║ ❌ ERROR: Unsupported operating system                                   ║
+║ Only Linux and Windows are supported                                     ║
+╚{'═' * 70}╝
+""")
+        raise NotImplementedError
 
     if authorized_keys.exists():
         split = "\n"
-        print(f'ℹ️ Users that can access this machine have private keys of those pub keys:\n{authorized_keys.read_text().split(split)}')
-
+        keys_text = authorized_keys.read_text().split(split)
+        key_count = len([k for k in keys_text if k.strip()])
+        
+        print(f"""
+╭{'─' * 70}╮
+│ 🔍 Current SSH authorization status                                      │
+│ ✅ Found {key_count} authorized key(s)                                        │
+╰{'─' * 70}╯
+""")
+        
         if path_to_key.read_text() in authorized_keys.read_text():
-            print(f"⚠️ Key {path_to_key} already added to {authorized_keys}. No action is taken.")
+            print(f"""
+╔{'═' * 70}╗
+║ ⚠️  Key already authorized                                                ║
+╠{'═' * 70}╣
+║ Key: {path_to_key.name}
+║ Status: Already present in authorized_keys file
+║ No action required
+╚{'═' * 70}╝
+""")
             program = ""
         else:
+            print(f"""
+╭{'─' * 70}╮
+│ ➕ Adding new SSH key to authorized keys                                 │
+│ 🔑 Key file: {path_to_key.name}                                          │
+╰{'─' * 70}╯
+""")
+            
             if system() == "Linux":
                 program = f"cat {path_to_key} >> ~/.ssh/authorized_keys"
             elif system() == "Windows":
@@ -29,16 +78,35 @@ def get_add_ssh_key_script(path_to_key: P):
                 place_holder = r'$sshfile = "$env:USERPROFILE\.ssh\pubkey.pub"'
                 assert place_holder in program, f"This section performs string manipulation on the script {program_path} to add the key to the authorized_keys file. The script has changed and the string {place_holder} is not found."
                 program = program.replace(place_holder, f'$sshfile = "{path_to_key}"')
-                print(f"Replaced {place_holder} with {path_to_key} in {program_path}.")
+                print(f"""
+╭{'─' * 70}╮
+│ 🔧 Configured PowerShell script for Windows                              │
+│ 📝 Replaced placeholder with actual key path                             │
+╰{'─' * 70}╯
+""")
             else: raise NotImplementedError
     else:
+        print(f"""
+╭{'─' * 70}╮
+│ 📝 Creating new authorized_keys file                                     │
+│ 🔑 Using key: {path_to_key.name}                                         │
+╰{'─' * 70}╯
+""")
+        
         if system() == "Linux":
             program = f"cat {path_to_key} > ~/.ssh/authorized_keys"
         else:
-            program = LIBRARY_ROOT.joinpath("setup_windows/openssh-server_add-sshkey.ps1")
-            program = P(program).expanduser().read_text().replace('$sshfile=""', f'$sshfile="{path_to_key}"')
+            program_path = LIBRARY_ROOT.joinpath("setup_windows/openssh-server_add-sshkey.ps1")
+            program = P(program_path).expanduser().read_text().replace('$sshfile=""', f'$sshfile="{path_to_key}"')
+            print(f"""
+╭{'─' * 70}╮
+│ 🔧 Configured PowerShell script for Windows                              │
+│ 📝 Set key path in script                                                │
+╰{'─' * 70}╯
+""")
 
-    if system() == "Linux": program += """
+    if system() == "Linux": 
+        program += """
 
 sudo chmod 700 ~/.ssh
 sudo chmod 644 ~/.ssh/authorized_keys
@@ -47,22 +115,109 @@ sudo service ssh --full-restart
 # from superuser.com/questions/215504/permissions-on-private-key-in-ssh-folder
 
 """
+        print(f"""
+╭{'─' * 70}╮
+│ 🔒 Setting proper SSH permissions and restarting service                 │
+╰{'─' * 70}╯
+""")
+        
+    print(f"""
+╔{'═' * 70}╗
+║ ✅ SSH KEY CONFIGURATION PREPARED                                        ║
+╚{'═' * 70}╝
+""")
+        
     return program
 
 
 def main():
+    print(f"""
+╔{'═' * 70}╗
+║ 🔐 SSH PUBLIC KEY AUTHORIZATION TOOL                                     ║
+╚{'═' * 70}╝
+""")
+    
+    print(f"""
+╭{'─' * 70}╮
+│ 🔍 Searching for public keys...                                          │
+╰{'─' * 70}╯
+""")
+    
     pub_keys = P.home().joinpath(".ssh").search("*.pub")
+    
+    if pub_keys:
+        print(f"""
+╭{'─' * 70}╮
+│ ✅ Found {len(pub_keys)} public key(s)                                           │
+╰{'─' * 70}╯
+""")
+    else:
+        print(f"""
+╭{'─' * 70}╮
+│ ⚠️  No public keys found                                                  │
+╰{'─' * 70}╯
+""")
+    
     all_keys_option = f"all pub keys available ({len(pub_keys)})"
     i_have_path_option = "I have the path to the key file"
     i_paste_option = "I want to paste the key itself"
+    
     res = display_options("Which public key to add? ", options=pub_keys.apply(str).list + [all_keys_option, i_have_path_option, i_paste_option])
     assert isinstance(res, str), f"Got {res} of type {type(res)} instead of str."
-    if res == all_keys_option: program = "\n\n\n".join(pub_keys.apply(get_add_ssh_key_script))
-    elif res == i_have_path_option: program = get_add_ssh_key_script(P(input("Path: ")).expanduser().absolute())
-    elif res == i_paste_option: program = get_add_ssh_key_script(P.home().joinpath(f".ssh/{input('file name (default: my_pasted_key.pub): ') or 'my_pasted_key.pub'}").write_text(input("Paste the pub key here: ")))
+    
+    if res == all_keys_option:
+        print(f"""
+╭{'─' * 70}╮
+│ 🔄 Processing all {len(pub_keys)} public keys...                                  │
+╰{'─' * 70}╯
+""")
+        program = "\n\n\n".join(pub_keys.apply(get_add_ssh_key_script))
+    
+    elif res == i_have_path_option:
+        print(f"""
+╭{'─' * 70}╮
+│ 📂 Please provide the path to your public key                             │
+╰{'─' * 70}╯
+""")
+        key_path = P(input("📋 Path: ")).expanduser().absolute()
+        print(f"""
+╭{'─' * 70}╮
+│ 📄 Using key from path: {key_path}                        │
+╰{'─' * 70}╯
+""")
+        program = get_add_ssh_key_script(key_path)
+    
+    elif res == i_paste_option:
+        print(f"""
+╭{'─' * 70}╮
+│ 📋 Please provide a filename and paste the public key content             │
+╰{'─' * 70}╯
+""")
+        key_filename = input("📝 File name (default: my_pasted_key.pub): ") or "my_pasted_key.pub"
+        key_path = P.home().joinpath(f".ssh/{key_filename}")
+        key_path.write_text(input("🔑 Paste the public key here: "))
+        print(f"""
+╭{'─' * 70}╮
+│ 💾 Key saved to: {key_path}                           │
+╰{'─' * 70}╯
+""")
+        program = get_add_ssh_key_script(key_path)
+    
     else:
+        print(f"""
+╭{'─' * 70}╮
+│ 🔑 Using selected key: {P(res).name}                                     │
+╰{'─' * 70}╯
+""")
         program = get_add_ssh_key_script(P(res))
-        print(program)
+    
+    print(f"""
+╔{'═' * 70}╗
+║ 🚀 SSH KEY AUTHORIZATION READY                                           ║
+║ Run the generated script to apply changes                                ║
+╚{'═' * 70}╝
+""")
+    
     return program