77 lines
2.7 KiB
Python
77 lines
2.7 KiB
Python
"""Docker controller"""
|
|
from docker import DockerClient, from_env
|
|
from docker.errors import NotFound
|
|
from docker.models.containers import Container
|
|
from yaml import safe_dump
|
|
|
|
from passbook import __version__
|
|
from passbook.outposts.controllers.base import BaseController
|
|
|
|
|
|
class DockerController(BaseController):
|
|
"""Docker controller"""
|
|
|
|
client: DockerClient
|
|
|
|
container: Container
|
|
|
|
image_base = "beryju/passbook"
|
|
|
|
def __init__(self, outpost_pk: str) -> None:
|
|
super().__init__(outpost_pk)
|
|
self.client = from_env()
|
|
|
|
def _get_container(self) -> Container:
|
|
container_name = f"passbook-proxy-{self.outpost.uuid.hex}"
|
|
try:
|
|
return self.client.containers.get(container_name)
|
|
except NotFound:
|
|
return self.client.containers.create(
|
|
image=f"{self.image_base}-{self.outpost.type}:{__version__}",
|
|
name=f"passbook-proxy-{self.outpost.uuid.hex}",
|
|
detach=True,
|
|
ports={x: x for _, x in self.deployment_ports.items()},
|
|
environment={
|
|
"PASSBOOK_HOST": self.outpost.config.passbook_host,
|
|
"PASSBOOK_INSECURE": str(
|
|
self.outpost.config.passbook_host_insecure
|
|
),
|
|
"PASSBOOK_TOKEN": self.outpost.token.token_uuid.hex,
|
|
},
|
|
)
|
|
|
|
def run(self):
|
|
container = self._get_container()
|
|
# Check if the container is out of date, delete it and retry
|
|
if len(container.image.tags) > 0:
|
|
tag: str = container.iamge.tags[0]
|
|
_, _, version = tag.partition(":")
|
|
if version != __version__:
|
|
container.kill()
|
|
container.remove(force=True)
|
|
return self.run()
|
|
if container.status != "running":
|
|
container.start()
|
|
return None
|
|
|
|
def get_static_deployment(self) -> str:
|
|
"""Generate docker-compose yaml for proxy, version 3.5"""
|
|
ports = [f"{x}:{x}" for _, x in self.deployment_ports.items()]
|
|
compose = {
|
|
"version": "3.5",
|
|
"services": {
|
|
f"passbook_{self.outpost.type}": {
|
|
"image": f"{self.image_base}-{self.outpost.type}:{__version__}",
|
|
"ports": ports,
|
|
"environment": {
|
|
"PASSBOOK_HOST": self.outpost.config.passbook_host,
|
|
"PASSBOOK_INSECURE": str(
|
|
self.outpost.config.passbook_host_insecure
|
|
),
|
|
"PASSBOOK_TOKEN": self.outpost.token.token_uuid.hex,
|
|
},
|
|
}
|
|
},
|
|
}
|
|
return safe_dump(compose, default_flow_style=False)
|