fixing shell identifier function

Author: mramkumar

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "sshkey-switch"
-version = "0.2.0"
+version = "0.4.0"
 description = "A command-line tool for switching SSH keys"
 readme = "README.md"
 authors = [

sshkey_switch/main.py

@@ -11,6 +11,7 @@
 import sys
 import re
 import shutil
+import platform
 import subprocess
 from pathlib import Path
 from collections import OrderedDict
@@ -28,16 +29,9 @@
 ACTIVATION_MESSAGE = False 
 EXCLUDE_FILES = {"authorized_keys", "known_hosts"}
 
-def get_os_shell_profile() -> Path | None:
-    """Returns the appropriate shell profile file path based on the user's current shell."""
-    shell = os.getenv("SHELL", "").split('/')[-1].lower()
-
-    shell_profiles = {
-        "bash": BASHRC_FILE,
-        "zsh": ZSHRC_FILE
-    }
-
-    return shell_profiles.get(shell, None)  # Return None for unsupported shells
+def get_os_shell_profile():
+     """Return the correct shell profile file based on OS."""
+     return ZSHRC_FILE if platform.system() == "Darwin" else BASHRC_FILE
 
 def check_ssh_environment():
     """Check if /root/.ssh directory and ssh-agent binary exist."""