machineconfig 8.14__py3-none-any.whl → 8.50__py3-none-any.whl

machineconfig/scripts/python/helpers_network/ssh_add_identity.py

@@ -1,116 +0,0 @@
-"""ID
-
-On windows:
-
-# on the side of the local machine, you need to sort out the following:
-# if the $sshfile doesn't have a standard name, you will need to explicitly specify the identity while SSHing (e.g. ssh u@s -i ~/.ssh/my_id)
-# However, this must be done every time. For permanent solutions, use .ssh/config
-$sshfile = "$env:USERPROFILE/.ssh/id_rsa"
-Set-Service ssh-agent -StartupType Manual  # allow the service to be started manually
-ssh-agent  # start the service
-ssh-add.exe $sshfile # add the key to the agent
-
-# add key:
-$sshfile=""
-$ErrorActionPreference = "Stop"
-$sshd_dir = "$env:ProgramData/ssh"
-cp "$sshd_dir/administrators_authorized_keys" "$sshd_dir/administrators_authorized_keys.orig"
-Get-Content $sshfile >> "$sshd_dir/administrators_authorized_keys"
-Restart-Service sshd -Force
-
-# copy ssh key:
-# This is the Windows equivalent of copy-ssh-id on Linux.
-# Just like the original function, it is a convenient way of doing two things in one go:
-# 1- copy a certain public key to the remote machine.
-#    scp ~/.ssh/id_rsa.pub $remote_user@$remote_host:~/.ssh/authorized_keys
-# 2- Store the value on the remote in a file called .ssh/authorized_keys
-#    ssh $remote_user@$remote_host "echo $public_key >> ~/.ssh/authorized_keys"
-# Idea from: https://www.chrisjhart.com/Windows-10-ssh-copy-id/
-
-$key_value = cat ($env:USERPROFILE + "/.ssh/id_rsa.pub")
-ssh $args[0] "powershell.exe -Command type $key_value >> .ssh/authorized_keys"
-$my_keys='https://github.com/thisismygitrepo.keys'
-(Invoke-WebRequest $my_keys).Content >> .ssh/authorized_keys
-
-"""
-
-# from platform import system
-from machineconfig.utils.path_extended import PathExtended
-from machineconfig.utils.options import choose_from_options
-from rich.panel import Panel
-from rich.text import Text
-
-BOX_WIDTH = 150  # width for box drawing
-
-
-def main() -> None:
-    title = "🔑 SSH IDENTITY MANAGEMENT"
-    print(Panel(Text(title, justify="center"), expand=False))
-    print(Panel("🔍 Searching for existing SSH keys...", expand=False))
-    private_keys = [x.with_name(x.stem) for x in PathExtended.home().joinpath(".ssh").search("*.pub")]
-    private_keys = [x for x in private_keys if x.exists()]
-    if private_keys:
-        print(Panel(f"✅ Found {len(private_keys)} SSH private key(s)", expand=False))
-    else:
-        print(Panel("⚠️  No SSH private keys found", expand=False))
-
-    choice = choose_from_options(msg="Path to private key to be used when ssh'ing: ", options=[str(x) for x in private_keys] + ["I have the path to the key file", "I want to paste the key itself"], multi=False)
-
-    if choice == "I have the path to the key file":
-        print(Panel("📄 Please enter the path to your private key file", expand=False))
-        path_to_key = PathExtended(input("📋 Input path here: ")).expanduser().absolute()
-        print(Panel(f"📂 Using key from custom path: {path_to_key}", expand=False))
-
-    elif choice == "I want to paste the key itself":
-        print(Panel("📋 Please provide a filename and paste the private key content", expand=False))
-        key_filename = input("📝 File name (default: my_pasted_key): ") or "my_pasted_key"
-        path_to_key = PathExtended.home().joinpath(f".ssh/{key_filename}")
-        path_to_key.parent.mkdir(parents=True, exist_ok=True)
-        path_to_key.write_text(input("🔑 Paste the private key here: "), encoding="utf-8")
-        print(Panel(f"💾 Key saved to: {path_to_key}", expand=False))
-
-    else:
-        path_to_key = PathExtended(choice)
-        print(Panel(f"🔑 Using selected key: {path_to_key.name}", expand=False))
-
-    txt = f"IdentityFile {path_to_key.collapseuser().as_posix()}"  # adds this id for all connections, no host specified.
-    config_path = PathExtended.home().joinpath(".ssh/config")
-
-    print(Panel("📝 Updating SSH configuration...", expand=False))
-
-    # Inline the previous modify_text behavior (now deprecated):
-    # - If file doesn't exist, seed content with txt_search
-    # - Otherwise, replace a matching line or append if not found
-    if config_path.exists():
-        current = config_path.read_text(encoding="utf-8")
-        print(Panel("✏️  Updated existing SSH config file", expand=False))
-    else:
-        current = txt
-        print(Panel("📄 Created new SSH config file", expand=False))
-    lines = current.split("\n")
-    found = False
-    for i, line in enumerate(lines):
-        if txt in line:
-            lines[i] = txt
-            found = True
-    if not found:
-        lines.insert(0, txt)
-    new_content = "\n".join(lines)
-    config_path.write_text(new_content, encoding="utf-8")
-
-    panel_complete = Panel(Text("✅ SSH IDENTITY CONFIGURATION COMPLETE\nIdentity added to SSH config file\nConsider reloading the SSH config to apply changes", justify="center"), expand=False, border_style="green")
-    program = f"echo '{panel_complete}'"
-
-    success_message = f"🎉 CONFIGURATION SUCCESSFUL\nIdentity added: {path_to_key.name}\nConfig file: {config_path}"
-    print(Panel(Text(success_message, justify="center"), expand=False, border_style="green"))
-
-    import subprocess
-
-    # run program
-    subprocess.run(program, shell=True, check=True, text=True)
-    print(Panel("🔐 Identity added to SSH agent", expand=False, border_style="green"))
-    return None
-
-
-if __name__ == "__main__":
-    pass

machineconfig/scripts/python/helpers_network/ssh_add_ssh_key.py

@@ -1,153 +0,0 @@
-"""SSH"""
-
-from platform import system
-from machineconfig.utils.source_of_truth import LIBRARY_ROOT
-from machineconfig.utils.path_extended import PathExtended
-from rich.console import Console
-from rich.panel import Panel
-from rich import box
-from typing import Optional, Annotated
-import typer
-
-
-console = Console()
-
-
-def get_add_ssh_key_script(path_to_key: PathExtended) -> str:
-    console.print(Panel("🔑 SSH KEY CONFIGURATION", title="[bold blue]SSH Setup[/bold blue]"))
-    if system() == "Linux" or system() == "Darwin":
-        authorized_keys = PathExtended.home().joinpath(".ssh/authorized_keys")
-        console.print(Panel(f"🐧 Linux SSH configuration\n📄 Authorized keys file: {authorized_keys}", title="[bold blue]System Info[/bold blue]"))
-    elif system() == "Windows":
-        authorized_keys = PathExtended("C:/ProgramData/ssh/administrators_authorized_keys")
-        console.print(Panel(f"🪟 Windows SSH configuration\n📄 Authorized keys file: {authorized_keys}", title="[bold blue]System Info[/bold blue]"))
-    else:
-        console.print(Panel("❌ ERROR: Unsupported operating system\nOnly Linux and Windows are supported", title="[bold red]Error[/bold red]"))
-        raise NotImplementedError
-
-    if authorized_keys.exists():
-        split = "\n"
-        keys_text = authorized_keys.read_text(encoding="utf-8").split(split)
-        key_count = len([k for k in keys_text if k.strip()])
-        console.print(Panel(f"🔍 Current SSH authorization status\n✅ Found {key_count} authorized key(s)", title="[bold blue]Status[/bold blue]"))
-        if path_to_key.read_text(encoding="utf-8") in authorized_keys.read_text(encoding="utf-8"):
-            console.print(Panel(f"⚠️  Key already authorized\nKey: {path_to_key.name}\nStatus: Already present in authorized_keys file\nNo action required", title="[bold yellow]Warning[/bold yellow]"))
-            program = ""
-        else:
-            console.print(Panel(f"➕ Adding new SSH key to authorized keys\n🔑 Key file: {path_to_key.name}", title="[bold blue]Action[/bold blue]"))
-            if system() == "Linux" or system() == "Darwin":
-                program = f"cat {path_to_key} >> ~/.ssh/authorized_keys"
-            elif system() == "Windows":
-                program_path = LIBRARY_ROOT.joinpath("setup_windows/ssh/add-sshkey.ps1")
-                program = program_path.expanduser().read_text(encoding="utf-8")
-                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}"')
-                console.print(Panel("🔧 Configured PowerShell script for Windows\n📝 Replaced placeholder with actual key path", title="[bold blue]Configuration[/bold blue]"))
-            else:
-                raise NotImplementedError
-    else:
-        console.print(Panel(f"📝 Creating new authorized_keys file\n🔑 Using key: {path_to_key.name}", title="[bold blue]Action[/bold blue]"))
-        if system() == "Linux" or system() == "Darwin":
-            program = f"cat {path_to_key} > ~/.ssh/authorized_keys"
-        else:
-            program_path = LIBRARY_ROOT.joinpath("setup_windows/ssh/openssh-server_add-sshkey.ps1")
-            program = PathExtended(program_path).expanduser().read_text(encoding="utf-8").replace('$sshfile=""', f'$sshfile="{path_to_key}"')
-            console.print(Panel("🔧 Configured PowerShell script for Windows\n📝 Set key path in script", title="[bold blue]Configuration[/bold blue]"))
-
-    if system() == "Linux" or system() == "Darwin":
-        program += """
-sudo chmod 700 ~/.ssh
-sudo chmod 644 ~/.ssh/authorized_keys
-sudo chmod 644 ~/.ssh/*.pub
-sudo service ssh --full-restart
-# from superuser.com/questions/215504/permissions-on-private-key-in-ssh-folder
-"""
-    return program
-
-
-"""
-Common pitfalls: 
-🚫 Wrong line endings (LF/CRLF) in config files
-🌐 Network port conflicts (try 2222 -> 2223) between WSL and Windows
-sudo service ssh restart
-sudo service ssh status
-sudo nano /etc/ssh/sshd_config
-"""
-
-
-def main(pub_path: Annotated[Optional[str], typer.Argument(..., help="Path to the public key file")] = None,
-         pub_choose: Annotated[bool, typer.Option(..., "--choose", "-c", help="Choose from available public keys in ~/.ssh")] = False,
-         pub_val: Annotated[bool, typer.Option(..., "--paste", "-p", help="Paste the public key content manually")] = False,
-         from_github: Annotated[Optional[str], typer.Option(..., "--from-github", "-g", help="Fetch public keys from a GitHub username")] = None
-         ) -> None:
-    
-    if pub_path:
-        key_path = PathExtended(pub_path).expanduser().absolute()
-        key_path.parent.mkdir(parents=True, exist_ok=True)
-        if not key_path.exists():
-            console.print(Panel(f"❌ ERROR: Provided key path does not exist\nPath: {key_path}", title="[bold red]Error[/bold red]"))
-            raise FileNotFoundError(f"Provided key path does not exist: {key_path}")
-        console.print(Panel(f"📄 Using provided public key file: {key_path}", title="[bold blue]Info[/bold blue]"))
-        program = get_add_ssh_key_script(key_path)
-        from machineconfig.utils.code import run_shell_script
-        run_shell_script(script=program)
-        console.print(Panel("✅ SSH KEY AUTHORIZATION COMPLETED", box=box.DOUBLE_EDGE, title_align="left"))
-        return
-    elif pub_choose:
-        console.print(Panel("🔐 SSH PUBLIC KEY AUTHORIZATION TOOL", box=box.DOUBLE_EDGE, title_align="left"))
-        console.print(Panel("🔍 Searching for public keys...", title="[bold blue]SSH Setup[/bold blue]", border_style="blue"))
-        pub_keys = PathExtended.home().joinpath(".ssh").search("*.pub")
-        if pub_keys:
-            console.print(Panel(f"✅ Found {len(pub_keys)} public key(s)", title="[bold green]Status[/bold green]", border_style="green"))
-        else:
-            console.print(Panel("⚠️  No public keys found", title="[bold yellow]Warning[/bold yellow]", border_style="yellow"))
-            return
-        console.print(Panel(f"🔄 Processing all {len(pub_keys)} public keys...", title="[bold blue]Processing[/bold blue]", border_style="blue"))
-        program = "\n\n\n".join([get_add_ssh_key_script(key) for key in pub_keys])
-
-    elif pub_val:
-        console.print(Panel("📋 Please provide a filename and paste the public key content", title="[bold blue]Input Required[/bold blue]", border_style="blue"))
-        key_filename = input("📝 File name (default: my_pasted_key.pub): ") or "my_pasted_key.pub"
-        key_path = PathExtended.home().joinpath(f".ssh/{key_filename}")
-        key_path.parent.mkdir(parents=True, exist_ok=True)
-        key_path.write_text(input("🔑 Paste the public key here: "), encoding="utf-8")
-        console.print(Panel(f"💾 Key saved to: {key_path}", title="[bold green]Success[/bold green]", border_style="green"))
-        program = get_add_ssh_key_script(key_path)
-    elif from_github:
-        console.print(Panel(f"🌐 Fetching public keys from GitHub user: {from_github}", title="[bold blue]GitHub Fetch[/bold blue]", border_style="blue"))
-        import requests
-        response = requests.get(f"https://api.github.com/users/{from_github}/keys")
-        if response.status_code != 200:
-            console.print(Panel(f"❌ ERROR: Failed to fetch keys from GitHub user {from_github}\nStatus Code: {response.status_code}", title="[bold red]Error[/bold red]", border_style="red"))
-            raise RuntimeError(f"Failed to fetch keys from GitHub user {from_github}: Status Code {response.status_code}")
-        keys = response.json()
-        if not keys:
-            console.print(Panel(f"⚠️  No public keys found for GitHub user: {from_github}", title="[bold yellow]Warning[/bold yellow]", border_style="yellow"))
-            return
-        console.print(Panel(f"✅ Found {len(keys)} public key(s) for user: {from_github}", title="[bold green]Success[/bold green]", border_style="green"))
-        key_path = PathExtended.home().joinpath(f".ssh/{from_github}_github_keys.pub")
-        key_path.parent.mkdir(parents=True, exist_ok=True)
-        key_path.write_text("\n".join([key["key"] for key in keys]), encoding="utf-8")
-        console.print(Panel(f"💾 Keys saved to: {key_path}", title="[bold green]Success[/bold green]", border_style="green"))
-        program = get_add_ssh_key_script(key_path)
-    else:
-        console.print(Panel("❌ ERROR: No method provided to add SSH key\nUse --help for options", title="[bold red]Error[/bold red]", border_style="red"))
-        raise ValueError("No method provided to add SSH key. Use --help for options.")
-    console.print(Panel("🚀 SSH KEY AUTHORIZATION READY\nRun the generated script to apply changes", box=box.DOUBLE_EDGE, title_align="left"))
-    from machineconfig.utils.code import run_shell_script
-    run_shell_script(script=program)
-
-    import machineconfig.scripts.python.helpers_network.address as helper
-    res = helper.select_lan_ipv4(prefer_vpn=False)
-    if res is None:
-        console.print(Panel("❌ ERROR: Could not determine local LAN IPv4 address", title="[bold red]Error[/bold red]", border_style="red"))
-        raise typer.Exit(code=1)
-    local_ip_v4 = res
-
-    console.print(Panel(f"🌐 This computer is accessible at: {local_ip_v4}", title="[bold green]Network Info[/bold green]", border_style="green"))
-    console.print(Panel("✅ SSH KEY AUTHORIZATION COMPLETED", box=box.DOUBLE_EDGE, title_align="left"))
-
-
-if __name__ == "__main__":
-    pass