trustchain-oc1-orchestral/authentik
Archived
10
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

cleanup

Browse source 
Signed-off-by: Jens Langhammer <jens@goauthentik.io>
This commit is contained in:
Jens Langhammer 2023-04-24 15:47:11 +03:00
parent cf9d8f64a2
commit 71ac1282f9
No known key found for this signature in database
1 changed files with 5 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
authentik/outposts/docker_ssh.py
View file

@ -7,8 +7,6 @@ from docker.errors import DockerException
from authentik.crypto.models import CertificateKeyPair from authentik.crypto.models import CertificateKeyPair
HEADER = "### Managed by authentik"
FOOTER = "### End Managed by authentik"
SSH_CONFIG_DIR = Path("/etc/ssh/ssh_config.d/") SSH_CONFIG_DIR = Path("/etc/ssh/ssh_config.d/")
def opener(path, flags): def opener(path, flags):
@ -28,13 +26,12 @@ class DockerInlineSSH:
key_path: str key_path: str
config_path: Path config_path: Path
header: str
def __init__(self, host: str, keypair: CertificateKeyPair) -> None: def __init__(self, host: str, keypair: CertificateKeyPair) -> None:
self.host = host self.host = host
self.keypair = keypair self.keypair = keypair
self.config_path = SSH_CONFIG_DIR / Path(self.host + ".conf") self.config_path = SSH_CONFIG_DIR / Path(self.host + ".conf")
if self.config_path.exists(): if not open(self.config_path, "w").writable():
# SSH Config file already exists and there's no header from us, meaning that it's # SSH Config file already exists and there's no header from us, meaning that it's
# been externally mapped into the container for more complex configs # been externally mapped into the container for more complex configs
raise SSHManagedExternallyException( raise SSHManagedExternallyException(
@ -43,26 +40,25 @@ class DockerInlineSSH:
if not self.keypair: if not self.keypair:
raise DockerException("keypair must be set for SSH connections") raise DockerException("keypair must be set for SSH connections")
def write_config(self, key_path: str) -> bool: def write_config(self, key_path: str):
"""Update the local user's ssh config file""" """Update the local user's ssh config file"""
with open(self.config_path, "w", encoding="utf-8") as ssh_config: with open(self.config_path, "w", encoding="utf-8") as ssh_config:
ssh_config.writelines( ssh_config.writelines(
[ [
f"Host {self.host}\n", f"Host {self.host}\n",
f" IdentityFile {key_path}\n", f" IdentityFile {str(key_path)}\n",
" StrictHostKeyChecking No\n", " StrictHostKeyChecking No\n",
" UserKnownHostsFile /dev/null\n", " UserKnownHostsFile /dev/null\n",
"\n", "\n",
] ]
) )
return True
def write_key(self): def write_key(self) -> Path:
"""Write keypair's private key to a temporary file""" """Write keypair's private key to a temporary file"""
path = Path(gettempdir(), f"{self.keypair.pk}_private.pem") path = Path(gettempdir(), f"{self.keypair.pk}_private.pem")
with open(path, "w", encoding="utf8", opener=opener) as _file: with open(path, "w", encoding="utf8", opener=opener) as _file:
_file.write(self.keypair.key_data) _file.write(self.keypair.key_data)
return str(path) return path
def write(self): def write(self):
"""Write keyfile and update ssh config""" """Write keyfile and update ssh config"""