Azalea
61 lines
1.6 KiB
Python
Executable File
61 lines
1.6 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
from __future__ import annotations
|
|
import importlib
|
|
import subprocess
|
|
import sys
|
|
from pathlib import Path
|
|
from urllib.request import Request, urlopen
|
|
|
|
from color_utils import printc
|
|
|
|
|
|
def import_or_install(module: str, package: str | None = None):
|
|
try:
|
|
return importlib.import_module(module)
|
|
except ModuleNotFoundError:
|
|
subprocess.check_call([sys.executable, "-m", "pip", "install", package or module])
|
|
return importlib.import_module(module)
|
|
|
|
|
|
GITHUB_USERS = ['hykilpikonna']
|
|
KEYS_PATH = Path.home() / '.ssh' / 'authorized_keys'
|
|
|
|
|
|
def fetch_keys(user: str) -> set[str]:
|
|
req = Request(f"https://github.com/{user}.keys")
|
|
with urlopen(req) as response:
|
|
assert response.status == 200
|
|
resp = response.read().decode()
|
|
return {f"{l} {user}@github" for l in resp.strip().splitlines()}
|
|
|
|
|
|
def fetch_all_keys() -> set[str]:
|
|
return {l for u in GITHUB_USERS for l in fetch_keys(u)}
|
|
|
|
|
|
def normalize_key(key: str) -> str:
|
|
"""
|
|
Remove comment from keys
|
|
"""
|
|
return ' '.join(key.split(' ')[:2])
|
|
|
|
|
|
def update_ssh_keys():
|
|
all_keys = list(fetch_all_keys())
|
|
normalized_keys = {normalize_key(k) for k in all_keys}
|
|
|
|
existing_keys = set(KEYS_PATH.read_text('utf-8').strip().splitlines())
|
|
for k in existing_keys:
|
|
if normalize_key(k) not in normalized_keys:
|
|
all_keys.append(k)
|
|
|
|
len_diff = len(all_keys) - len(existing_keys)
|
|
if len_diff > 0:
|
|
printc(f"&aSSH Keys Updated! Added {len_diff} keys")
|
|
|
|
KEYS_PATH.write_text('\n'.join(all_keys))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
update_ssh_keys()
|