import os
import subprocess
from flask import Flask, request, Response
import paramiko
import socket

app = Flask(__name__)


class SSHRemote:
    def __init__(self, config_path: str, host: str) -> None:
        """
        config_path : ssh配置文件路径

        host : 配置文件中配置项的名字
        """

        self.host = host
        self.ssh_config = paramiko.SSHConfig().from_path(config_path).lookup(host)
        self.ssh_connect = paramiko.SSHClient()
        self.ssh_connect.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        self.ssh_connect.connect(hostname=self.ssh_config["hostname"], port=self.ssh_config["port"],
                                 username=self.ssh_config["user"], password=self.ssh_config["passwd"])

    def __enter__(self):
        return self

    def __exit__(self, exc_type, exc_value, traceback):
        self.ssh_connect.close()

    def compileCmd(self, cmd: str):
        ssh_stdin, ssh_stdout, ssh_stderr = self.ssh_connect.exec_command(
            " && ".join(cmd))
        for line in ssh_stdout:
            yield line.encode('utf-8') + b"<br>"  # 将每行结果转换为字节类型，并加上 HTML 换行

    def __del__(self):
        self.ssh_connect.close()


@app.route('/build-and-compile')
def build_and_compile():
    package = request.args.get('package')
    target_platform = request.args.get('platform')

    if package and target_platform:
        project, package_name = package.split('/')
        target_dir = f"{project}/{package_name}"
        print(target_dir)

        def generate_output():

            if target_platform == 'x86':
                commands = [
                    f"rm -rf ~/devel/{project}/{package_name}*",
                    f"cd ~/devel && osc co {project}/{package_name}",
                    f"cd ~/devel/{project}/{package_name} && osc up -S",
                    f"cd ~/devel/{project}/{package_name} && rm -f _service; for file in `ls | grep -v .osc`; do new_file=${{file##*:}}; mv $file $new_file; done",
                    f"cd ~/devel/{project}/{package_name} && osc build standard_x86_64_gcc x86_64"
                ]
                with SSHRemote("./ssh.config", "x86") as x86_remote:
                    for line in x86_remote.compileCmd(commands):
                        yield line
            elif target_platform == 'riscv':
                commands = [
                    f"rm -rf ~/devel/{project}/{package_name}*",
                    f"cd ~/devel && osc co {project}/{package_name}",
                    f"cd ~/devel/{project}/{package_name} && osc up -S",
                    f"cd ~/devel/{project}/{package_name} && rm -f _service; for file in `ls | grep -v .osc`; do new_file=${{file##*:}}; mv $file $new_file; done",
                    f"cd ~/devel/{project}/{package_name} && osc build standard_riscv64_gcc  riscv64"
                ]
                with SSHRemote("./ssh.config", "riscv") as riscv_remote:
                    for line in riscv_remote.compileCmd(commands):
                        yield line

        return Response(generate_output(), content_type='text/html')


if __name__ == "__main__":
    app.run()