ssh-to-code 1.0.0__py3-none-any.whl

save_config.py

@@ -0,0 +1,340 @@
+#!/usr/bin/env python3
+"""
+Interactive SSH Configuration Manager
+Easily save and manage frequently used SSH connections
+"""
+
+import sys
+import os
+
+# Add current directory to path
+sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
+
+from config_manager import ConfigManager
+
+
+def print_header(text):
+    print("\n" + "=" * 60)
+    print(f"  {text}")
+    print("=" * 60 + "\n")
+
+
+def add_host_interactive():
+    """Interactive host addition"""
+    print_header("Add New SSH Host Configuration")
+    
+    config = ConfigManager()
+    
+    # Get host details
+    print("Enter the SSH connection details:\n")
+    
+    name = input("Host nickname (e.g., 'my-ec2', 'production'): ").strip()
+    if not name:
+        print("❌ Name is required")
+        return False
+    
+    # Check if already exists
+    existing = config.get_host(name)
+    if existing:
+        print(f"\n⚠️  Host '{name}' already exists!")
+        print(f"   Current config: {existing['username']}@{existing['hostname']}")
+        overwrite = input("   Overwrite? (yes/no): ").strip().lower()
+        if overwrite != 'yes':
+            print("Cancelled.")
+            return False
+        config.remove_host(name)
+    
+    hostname = input("Hostname or IP (e.g., 'example.com', '1.2.3.4'): ").strip()
+    if not hostname:
+        print("❌ Hostname is required")
+        return False
+    
+    username = input("Username (e.g., 'ubuntu', 'ec2-user', 'root'): ").strip()
+    if not username:
+        print("❌ Username is required")
+        return False
+    
+    port_input = input("Port (default: 22): ").strip()
+    port = int(port_input) if port_input else 22
+    
+    key_file = input("Path to SSH key file (optional, press Enter to skip): ").strip()
+    if key_file:
+        key_file = os.path.expanduser(key_file)
+        if not os.path.exists(key_file):
+            print(f"⚠️  Warning: Key file not found: {key_file}")
+            proceed = input("   Save anyway? (yes/no): ").strip().lower()
+            if proceed != 'yes':
+                return False
+    else:
+        key_file = None
+    
+    default_path = input("Default starting directory (optional, press Enter to skip): ").strip()
+    if not default_path:
+        default_path = None
+    
+    # Confirm
+    print("\n" + "-" * 60)
+    print("Configuration Summary:")
+    print("-" * 60)
+    print(f"  Name:         {name}")
+    print(f"  Connection:   {username}@{hostname}:{port}")
+    if key_file:
+        print(f"  Key file:     {key_file}")
+    if default_path:
+        print(f"  Start path:   {default_path}")
+    print("-" * 60)
+    
+    confirm = input("\nSave this configuration? (yes/no): ").strip().lower()
+    if confirm == 'yes':
+        success = config.add_host(
+            name=name,
+            hostname=hostname,
+            username=username,
+            port=port,
+            key_file=key_file,
+            default_path=default_path
+        )
+        
+        if success:
+            print(f"\n✅ Configuration '{name}' saved successfully!")
+            print(f"\nYou can now connect with:")
+            print(f"  ./ssh-browse {name}")
+            print(f"  # or")
+            print(f"  python ssh_dir_browser.py {username}@{hostname} -p {port}", end="")
+            if key_file:
+                print(f" -i {key_file}", end="")
+            print()
+            return True
+        else:
+            print("\n❌ Failed to save configuration")
+            return False
+    else:
+        print("Cancelled.")
+        return False
+
+
+def list_hosts():
+    """List all saved hosts"""
+    print_header("Saved SSH Host Configurations")
+    
+    config = ConfigManager()
+    hosts = config.get_hosts()
+    
+    if not hosts:
+        print("No saved configurations found.\n")
+        print("💡 Add one with: python save_config.py add")
+        return
+    
+    print(f"Found {len(hosts)} saved configuration(s):\n")
+    
+    for i, host in enumerate(hosts, 1):
+        print(f"{i}. {host['name']}")
+        print(f"   Connection: {host['username']}@{host['hostname']}:{host.get('port', 22)}")
+        if 'key_file' in host:
+            print(f"   Key file:   {host['key_file']}")
+        if 'default_path' in host:
+            print(f"   Start path: {host['default_path']}")
+        if 'last_path' in host:
+            print(f"   Last used:  {host['last_path']}")
+        print()
+    
+    print("Connect with: ./ssh-browse <name>")
+    print(f"Config file: {config.config_path}\n")
+
+
+def remove_host_interactive():
+    """Interactive host removal"""
+    print_header("Remove SSH Host Configuration")
+    
+    config = ConfigManager()
+    hosts = config.get_hosts()
+    
+    if not hosts:
+        print("No saved configurations found.\n")
+        return
+    
+    print("Saved configurations:\n")
+    for i, host in enumerate(hosts, 1):
+        print(f"{i}. {host['name']} ({host['username']}@{host['hostname']})")
+    
+    print()
+    choice = input("Enter name or number to remove (or 'cancel'): ").strip()
+    
+    if choice.lower() == 'cancel':
+        print("Cancelled.")
+        return
+    
+    # Try as number first
+    host_to_remove = None
+    try:
+        idx = int(choice) - 1
+        if 0 <= idx < len(hosts):
+            host_to_remove = hosts[idx]['name']
+    except ValueError:
+        host_to_remove = choice
+    
+    if host_to_remove:
+        confirm = input(f"Remove '{host_to_remove}'? (yes/no): ").strip().lower()
+        if confirm == 'yes':
+            if config.remove_host(host_to_remove):
+                print(f"✅ Removed '{host_to_remove}'")
+            else:
+                print(f"❌ Host '{host_to_remove}' not found")
+        else:
+            print("Cancelled.")
+
+
+def quick_add_from_command():
+    """Quick add from command line arguments"""
+    if len(sys.argv) < 4:
+        print("Usage: python save_config.py quick <name> <user@host> [options]")
+        print("\nOptions:")
+        print("  -i <key_file>        Path to SSH key")
+        print("  -p <port>            SSH port (default: 22)")
+        print("  --path <dir>         Default starting directory")
+        print("\nExample:")
+        print("  python save_config.py quick myserver ubuntu@example.com -i ~/key.pem --path /var/www")
+        return
+    
+    name = sys.argv[2]
+    user_host = sys.argv[3]
+    
+    if '@' not in user_host:
+        print("❌ Invalid format. Use: user@hostname")
+        return
+    
+    username, hostname = user_host.split('@', 1)
+    
+    # Parse options
+    port = 22
+    key_file = None
+    default_path = None
+    
+    i = 4
+    while i < len(sys.argv):
+        if sys.argv[i] == '-i' and i + 1 < len(sys.argv):
+            key_file = os.path.expanduser(sys.argv[i + 1])
+            i += 2
+        elif sys.argv[i] == '-p' and i + 1 < len(sys.argv):
+            port = int(sys.argv[i + 1])
+            i += 2
+        elif sys.argv[i] == '--path' and i + 1 < len(sys.argv):
+            default_path = sys.argv[i + 1]
+            i += 2
+        else:
+            i += 1
+    
+    config = ConfigManager()
+    
+    if config.add_host(name, hostname, username, port, key_file, default_path):
+        print(f"✅ Configuration '{name}' saved!")
+        print(f"\nConnect with: ./ssh-browse {name}")
+    else:
+        print(f"❌ Failed to save configuration")
+
+
+def show_help():
+    """Show help message"""
+    print("""
+SSH Directory Browser - Configuration Manager
+==============================================
+
+USAGE:
+  python save_config.py [command]
+
+COMMANDS:
+  add, a              Add new host configuration (interactive)
+  list, ls, l         List all saved configurations
+  remove, rm, r       Remove a host configuration
+  quick <name> ...    Quick add from command line
+  help, h             Show this help message
+
+EXAMPLES:
+
+1. Interactive mode (recommended for first time):
+   python save_config.py add
+
+2. Quick add from command line:
+   python save_config.py quick myec2 ubuntu@ec2-1-2-3-4.compute.amazonaws.com -i ~/aws-key.pem
+
+3. List all saved hosts:
+   python save_config.py list
+
+4. Remove a host:
+   python save_config.py remove
+
+5. After saving, connect with:
+   ./ssh-browse myec2
+
+CONFIGURATION FILE:
+  Location: ~/.ssh-dir-browser.json
+  Format:   JSON
+
+DIRECT EDITING:
+  You can also edit the config file directly:
+  nano ~/.ssh-dir-browser.json
+
+  Example format:
+  {
+    "version": "1.0",
+    "hosts": [
+      {
+        "name": "myserver",
+        "hostname": "example.com",
+        "username": "user",
+        "port": 22,
+        "key_file": "~/.ssh/id_rsa",
+        "default_path": "/var/www"
+      }
+    ]
+  }
+""")
+
+
+def main():
+    """Main entry point"""
+    if len(sys.argv) < 2:
+        # Interactive mode by default
+        while True:
+            print("\n" + "=" * 60)
+            print("  SSH Directory Browser - Configuration Manager")
+            print("=" * 60)
+            print("\n1. Add new host")
+            print("2. List saved hosts")
+            print("3. Remove host")
+            print("4. Exit")
+            
+            choice = input("\nSelect option (1-4): ").strip()
+            
+            if choice == '1':
+                add_host_interactive()
+            elif choice == '2':
+                list_hosts()
+            elif choice == '3':
+                remove_host_interactive()
+            elif choice == '4':
+                print("\nGoodbye!")
+                break
+            else:
+                print("Invalid option")
+        return
+    
+    command = sys.argv[1].lower()
+    
+    if command in ['add', 'a']:
+        add_host_interactive()
+    elif command in ['list', 'ls', 'l']:
+        list_hosts()
+    elif command in ['remove', 'rm', 'r']:
+        remove_host_interactive()
+    elif command == 'quick':
+        quick_add_from_command()
+    elif command in ['help', 'h', '--help', '-h']:
+        show_help()
+    else:
+        print(f"Unknown command: {command}")
+        print("Run 'python save_config.py help' for usage information")
+
+
+if __name__ == "__main__":
+    main()

ssh_dir_browser.py

@@ -0,0 +1,362 @@
+#!/usr/bin/env python3
+"""
+SSH Directory Browser with VS Code Integration
+Browse remote directories via SSH and open them in VS Code
+"""
+
+import curses
+import sys
+import subprocess
+import os
+from typing import List, Optional, Tuple
+import argparse
+
+
+class DirectoryBrowser:
+    """Interactive terminal-based directory browser"""
+    
+    def __init__(self, ssh_handler, start_path: str = "/"):
+        self.ssh_handler = ssh_handler
+        self.current_path = start_path
+        self.selected_index = 0
+        self.scroll_offset = 0
+        self.items: List[Tuple[str, str]] = []  # (name, type)
+        self.status_message = ""
+        
+    def get_directory_contents(self) -> List[Tuple[str, str]]:
+        """Get directory contents from remote server"""
+        try:
+            # List directory with details (-F adds indicators for file types)
+            cmd = f"ls -1F '{self.current_path}'"
+            output = self.ssh_handler.execute_command(cmd)
+            
+            if not output:
+                return []
+            
+            items = []
+            # Add parent directory option if not at root
+            if self.current_path != "/":
+                items.append(("..", "dir"))
+            
+            for line in output.strip().split('\n'):
+                if not line:
+                    continue
+                    
+                # Check file type indicator
+                if line.endswith('/'):
+                    items.append((line[:-1], "dir"))
+                elif line.endswith('*'):
+                    items.append((line[:-1], "exec"))
+                elif line.endswith('@'):
+                    items.append((line[:-1], "link"))
+                else:
+                    items.append((line, "file"))
+            
+            return items
+        except Exception as e:
+            self.status_message = f"Error: {str(e)}"
+            return []
+    
+    def navigate_to(self, item_name: str):
+        """Navigate to a directory"""
+        if item_name == "..":
+            # Go to parent directory
+            self.current_path = os.path.dirname(self.current_path)
+            if not self.current_path:
+                self.current_path = "/"
+        else:
+            # Go to subdirectory
+            if self.current_path == "/":
+                self.current_path = f"/{item_name}"
+            else:
+                self.current_path = f"{self.current_path}/{item_name}"
+        
+        self.selected_index = 0
+        self.scroll_offset = 0
+        self.items = self.get_directory_contents()
+    
+    def create_folder(self, stdscr) -> Optional[str]:
+        """Prompt user to create a new folder"""
+        height, width = stdscr.getmaxyx()
+        
+        # Create input window
+        input_win = curses.newwin(5, min(60, width - 4), height // 2 - 2, (width - min(60, width - 4)) // 2)
+        input_win.box()
+        input_win.addstr(1, 2, "Create New Folder:", curses.A_BOLD)
+        input_win.addstr(2, 2, "Name: ")
+        input_win.refresh()
+        
+        # Enable cursor for input
+        curses.curs_set(1)
+        curses.echo()
+        
+        # Get folder name
+        folder_name = ""
+        try:
+            folder_name = input_win.getstr(2, 8, 40).decode('utf-8').strip()
+        except:
+            pass
+        finally:
+            curses.noecho()
+            curses.curs_set(0)
+        
+        if not folder_name:
+            return None
+        
+        # Validate folder name
+        if '/' in folder_name or folder_name in ['.', '..']:
+            return f"Invalid folder name: {folder_name}"
+        
+        # Create folder on remote server
+        try:
+            new_path = os.path.join(self.current_path, folder_name)
+            cmd = f"mkdir -p '{new_path}'"
+            self.ssh_handler.execute_command(cmd)
+            return f"Created folder: {folder_name}"
+        except Exception as e:
+            return f"Error creating folder: {str(e)}"
+    
+    def open_in_vscode(self):
+        """Open current directory in VS Code"""
+        try:
+            host = self.ssh_handler.hostname
+            user = self.ssh_handler.username
+            port = self.ssh_handler.port
+            
+            # VS Code Remote-SSH URI format
+            # vscode://vscode-remote/ssh-remote+user@host:port/path
+            remote_uri = f"vscode://vscode-remote/ssh-remote+{user}@{host}"
+            if port != 22:
+                remote_uri += f":{port}"
+            remote_uri += self.current_path
+            
+            # Open VS Code
+            subprocess.Popen(['code', '--remote', f'ssh-remote+{user}@{host}', self.current_path],
+                           stdout=subprocess.DEVNULL, 
+                           stderr=subprocess.DEVNULL)
+            
+            return True, f"Opening {self.current_path} in VS Code..."
+        except Exception as e:
+            return False, f"Error opening VS Code: {str(e)}"
+    
+    def draw_ui(self, stdscr):
+        """Draw the terminal UI"""
+        stdscr.clear()
+        height, width = stdscr.getmaxyx()
+        
+        # Header
+        header = f"SSH Directory Browser - {self.ssh_handler.username}@{self.ssh_handler.hostname}"
+        stdscr.addstr(0, 0, header[:width-1], curses.A_REVERSE)
+        
+        # Current path
+        path_line = f"Path: {self.current_path}"
+        stdscr.addstr(1, 0, path_line[:width-1], curses.A_BOLD)
+        
+        # Help line
+        help_text = "↑/↓: Navigate | Enter: Open | 'o': VS Code | 'n': New Folder | 'r': Refresh | 'q': Quit"
+        stdscr.addstr(2, 0, help_text[:width-1], curses.A_DIM)
+        
+        # Separator
+        stdscr.addstr(3, 0, "─" * (width-1))
+        
+        # Directory contents
+        display_height = height - 6  # Reserve space for header, footer, and status
+        
+        # Calculate visible range
+        if self.selected_index < self.scroll_offset:
+            self.scroll_offset = self.selected_index
+        elif self.selected_index >= self.scroll_offset + display_height:
+            self.scroll_offset = self.selected_index - display_height + 1
+        
+        # Display items
+        for i in range(self.scroll_offset, min(len(self.items), self.scroll_offset + display_height)):
+            y = 4 + i - self.scroll_offset
+            name, item_type = self.items[i]
+            
+            # Format display
+            if item_type == "dir":
+                display = f"📁 {name}/"
+                attr = curses.A_BOLD
+            elif item_type == "exec":
+                display = f"⚙️  {name}*"
+                attr = curses.A_NORMAL
+            elif item_type == "link":
+                display = f"🔗 {name}@"
+                attr = curses.A_DIM
+            else:
+                display = f"📄 {name}"
+                attr = curses.A_NORMAL
+            
+            # Highlight selected item
+            if i == self.selected_index:
+                attr |= curses.A_REVERSE
+            
+            try:
+                stdscr.addstr(y, 2, display[:width-3], attr)
+            except curses.error:
+                pass  # Ignore errors from writing to last line
+        
+        # Status bar
+        status_y = height - 2
+        stdscr.addstr(status_y, 0, "─" * (width-1))
+        
+        if self.status_message:
+            status = self.status_message[:width-1]
+            stdscr.addstr(status_y + 1, 0, status, curses.A_BOLD)
+        else:
+            info = f"Items: {len(self.items)} | Selected: {self.selected_index + 1}/{len(self.items)}"
+            stdscr.addstr(status_y + 1, 0, info[:width-1])
+        
+        stdscr.refresh()
+    
+    def run(self, stdscr):
+        """Main event loop"""
+        curses.curs_set(0)  # Hide cursor
+        stdscr.keypad(True)
+        
+        # Initialize colors if available
+        if curses.has_colors():
+            curses.init_pair(1, curses.COLOR_CYAN, curses.COLOR_BLACK)
+            curses.init_pair(2, curses.COLOR_GREEN, curses.COLOR_BLACK)
+        
+        # Load initial directory
+        self.items = self.get_directory_contents()
+        
+        while True:
+            self.draw_ui(stdscr)
+            
+            # Get user input
+            key = stdscr.getch()
+            
+            if key == ord('q') or key == ord('Q'):
+                break
+            elif key == curses.KEY_UP:
+                if self.selected_index > 0:
+                    self.selected_index -= 1
+                    self.status_message = ""
+            elif key == curses.KEY_DOWN:
+                if self.selected_index < len(self.items) - 1:
+                    self.selected_index += 1
+                    self.status_message = ""
+            elif key == ord('\n') or key == curses.KEY_ENTER:
+                # Enter selected item
+                if self.items and self.selected_index < len(self.items):
+                    name, item_type = self.items[self.selected_index]
+                    if item_type == "dir" or name == "..":
+                        self.navigate_to(name)
+                        self.status_message = ""
+                    else:
+                        self.status_message = f"'{name}' is not a directory"
+            elif key == ord('o') or key == ord('O'):
+                # Open in VS Code
+                success, message = self.open_in_vscode()
+                self.status_message = message
+                if success:
+                    # Wait a moment to show the message, then exit
+                    stdscr.timeout(2000)
+                    stdscr.getch()
+                    break
+            elif key == ord('h') or key == ord('H'):
+                # Go to home directory
+                home_cmd = "echo $HOME"
+                home_path = self.ssh_handler.execute_command(home_cmd).strip()
+                if home_path:
+                    self.current_path = home_path
+                    self.selected_index = 0
+                    self.scroll_offset = 0
+                    self.items = self.get_directory_contents()
+                    self.status_message = f"Navigated to home: {home_path}"
+            elif key == ord('r') or key == ord('R'):
+                # Refresh
+                self.items = self.get_directory_contents()
+                self.status_message = "Refreshed"
+            elif key == ord('n') or key == ord('N'):
+                # Create new folder
+                result = self.create_folder(stdscr)
+                if result:
+                    self.status_message = result
+                    # Refresh directory listing
+                    self.items = self.get_directory_contents()
+                else:
+                    self.status_message = "Folder creation cancelled"
+
+
+def main():
+    """Main entry point"""
+    parser = argparse.ArgumentParser(
+        description="SSH Directory Browser - Browse remote directories and open in VS Code",
+        formatter_class=argparse.RawDescriptionHelpFormatter,
+        epilog="""
+Examples:
+  %(prog)s user@hostname
+  %(prog)s user@hostname -p 2222
+  %(prog)s user@hostname --start-path /var/www
+  %(prog)s user@hostname -i ~/.ssh/id_rsa
+        """
+    )
+    parser.add_argument('host', help='SSH host in format user@hostname')
+    parser.add_argument('-p', '--port', type=int, default=22, help='SSH port (default: 22)')
+    parser.add_argument('-i', '--identity', help='SSH private key file')
+    parser.add_argument('--start-path', default='~', help='Starting directory (default: ~)')
+    parser.add_argument('--password', action='store_true', 
+                       help='Prompt for password authentication')
+    
+    args = parser.parse_args()
+    
+    # Parse user@host
+    if '@' not in args.host:
+        print("Error: Host must be in format user@hostname")
+        sys.exit(1)
+    
+    username, hostname = args.host.split('@', 1)
+    
+    # Import SSH handler
+    try:
+        from ssh_handler import SSHHandler
+    except ImportError:
+        print("Error: ssh_handler module not found")
+        print("Make sure ssh_handler.py is in the same directory")
+        sys.exit(1)
+    
+    # Connect to SSH
+    print(f"Connecting to {username}@{hostname}:{args.port}...")
+    
+    try:
+        ssh = SSHHandler(
+            hostname=hostname,
+            username=username,
+            port=args.port,
+            key_filename=args.identity,
+            use_password=args.password
+        )
+        
+        if not ssh.connect():
+            print("Failed to connect to SSH server")
+            sys.exit(1)
+        
+        print("Connected successfully!")
+        
+        # Resolve start path
+        start_path = args.start_path
+        if start_path == '~':
+            home_cmd = "echo $HOME"
+            start_path = ssh.execute_command(home_cmd).strip() or "/"
+        
+        # Start the browser
+        browser = DirectoryBrowser(ssh, start_path)
+        curses.wrapper(browser.run)
+        
+        # Cleanup
+        ssh.disconnect()
+        print("\nDisconnected. Goodbye!")
+        
+    except KeyboardInterrupt:
+        print("\n\nInterrupted by user")
+        sys.exit(0)
+    except Exception as e:
+        print(f"\nError: {str(e)}")
+        sys.exit(1)
+
+
+if __name__ == "__main__":
+    main()