from pathlib import Path
import shutil
import subprocess
from urllib.parse import quote

from unit.applications.proto import ApplicationProto
from unit.option import option


class ApplicationWasmComponent(ApplicationProto):
    @staticmethod
    def prepare_env(script):
        try:
            subprocess.check_output(['cargo', 'component', '--help'])
        except (subprocess.CalledProcessError, FileNotFoundError):
            return None

        temp_dir = Path(f'{option.temp_dir}/wasm_component/')

        if not temp_dir.exists():
            temp_dir.mkdir()

        app_path = f'{temp_dir}/{script}'

        shutil.copytree(f'{option.test_dir}/wasm_component/{script}', app_path)

        try:
            output = subprocess.check_output(
                ['cargo', 'component', 'build', '--release'],
                cwd=app_path,
                stderr=subprocess.STDOUT,
            )
        except KeyboardInterrupt:
            raise

        except subprocess.CalledProcessError:
            return None

        return output

    def load(self, script, **kwargs):
        self.prepare_env(script)

        component_path = f'{option.temp_dir}/wasm_component/{script}/target/wasm32-wasi/release/test_wasi_component.wasm'

        self._load_conf(
            {
                "listeners": {
                    "*:8080": {"pass": f"applications/{quote(script, '')}"}
                },
                "applications": {
                    script: {
                        "type": "wasm-wasi-component",
                        "processes": {"spare": 0},
                        "component": component_path,
                    }
                },
            },
            **kwargs,
        )