#!/usr/bin/python3
#
# container.py
# Linux Container (e.g. Docker) plugin for Caterpillar Proxy
#
# Caterpillar Proxy - The simple and parasitic web proxy with SPAM filter
# Namyheon Go (Catswords Research) <gnh1201@gmail.com>
# https://github.com/gnh1201/caterpillar
# Created at: 2024-03-04
# Updated at: 2024-07-06
#

import docker

from base import Extension, Logger

logger = Logger("Container")


class Container(Extension):
    def __init__(self):
        self.type = "rpcmethod"
        self.method = "container_init"
        self.exported_methods = ["container_run", "container_stop"]

        # docker
        self.client = docker.from_env()

    def dispatch(self, type, id, params, conn):
        logger.info("[*] Greeting! dispatch")
        conn.send(b'Greeting! dispatch')

    def container_run(self, type, id, params, conn):
        devices = params['devices']
        image = params['image']
        devices = params['devices']
        name = params['name']
        environment = params['environment']
        volumes = params['volumes']

        container = self.client.containers.run(
            image,
            devices=devices,
            name=name,
            volumes=volumes,
            environment=environment,
            detach=True
        )
        container.logs()

        logger.info("[*] Running...")

    def container_stop(self, type, id, params, conn):
        name = params['name']

        container = self.client.containers.get(name)
        container.stop()

        logger.info("[*] Stopped")