1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
|
import enum
import os
from collections import namedtuple
CLIENT_CERT_NAME = "client.crt"
CLIENT_KEY_NAME = "client.key"
SERVER_CERT_NAME = "redis.crt"
SERVER_KEY_NAME = "redis.key"
CA_CERT_NAME = "ca.crt"
class CertificateType(str, enum.Enum):
client = "client"
server = "server"
TLSFiles = namedtuple("TLSFiles", ["certfile", "keyfile", "ca_certfile"])
def get_tls_certificates(
subdir: str = "standalone",
cert_type: CertificateType = CertificateType.client,
):
root = os.path.join(os.path.dirname(__file__), "..")
cert_subdir = ("dockers", subdir, "tls")
cert_dir = os.path.abspath(os.path.join(root, *cert_subdir))
if not os.path.isdir(cert_dir): # github actions package validation case
cert_dir = os.path.abspath(os.path.join(root, "..", *cert_subdir))
if not os.path.isdir(cert_dir):
raise OSError(f"No SSL certificates found. They should be in {cert_dir}")
if cert_type == CertificateType.client:
return TLSFiles(
os.path.join(cert_dir, CLIENT_CERT_NAME),
os.path.join(cert_dir, CLIENT_KEY_NAME),
os.path.join(cert_dir, CA_CERT_NAME),
)
elif cert_type == CertificateType.server:
return TLSFiles(
os.path.join(cert_dir, SERVER_CERT_NAME),
os.path.join(cert_dir, SERVER_KEY_NAME),
os.path.join(cert_dir, CA_CERT_NAME),
)
|
