websoft9/appmanage/api/service/manage.py

import os
import io
import sys
import platform
import shutil
import time
import subprocess
import json
import datetime
import socket
import re
from threading import Thread
from api.utils import shell_execute, docker, const
from api.model.app import App
from api.model.response import Response
from api.utils import lock
from api.utils.common_log import myLogger


# 获取所有app的信息
def get_my_app():
    ret = Response(code=const.RETURN_FAIL, message="App query failed!")

    # get all info
    cmd = "docker compose ls -a --format json"
    output = shell_execute.execute_command_output_all(cmd)
    if int(output["code"]) == 0:
        output_list = json.loads(output["result"])
        app_list, has_add = get_apps_from_compose(output_list)
        list = get_apps_from_queue(app_list, has_add)
        ret = Response(code=const.RETURN_SUCCESS, message="The app query is successful.", data=list)
    ret = ret.dict()
    return ret


# 获取具体某个app的信息
def get_app_detail(app_id):
    ret = {}
    ret['code'] = const.RETURN_FAIL
    ret['message'] = 'App query failed!'
    ret['data'] = None

    if docker.check_app_id(app_id):
        # get all info
        cmd = "docker compose ls -a --format json"
        output = shell_execute.execute_command_output_all(cmd)
        if int(output["code"]) == 0:
            output_list = json.loads(output["result"])
            app_list, has_add = get_apps_from_compose(output_list)
            list = get_apps_from_queue(app_list, has_add)
            flag = 0
            app_info = None
            for app in list:
                if app["app_id"] == app_id:
                    list.clear()
                    list.append(app)
                    app_info = app
                    flag = 1
                    break
            if flag == 1:
                ret['code'] = const.RETURN_SUCCESS
                ret['message'] = "The app query is successful."
                ret['data'] = app_info
    else:
        ret['message'] = "AppID is not legal!"
    return ret


# 查询某个正在安装的app的 具体状态：waiting（等待安装）pulling（拉取镜像）initializing（初始化）running（正常运行）
def install_app_process(app_id):
    app_name = split_app_id(app_id)
    var_path = "/data/apps/" + app_name + "/variables.json"
    real_name = docker.read_var(var_path, 'name')
    if docker.check_app_directory(real_name):
        percentage = docker.get_process_perc(app_name, real_name)
        ret = Response(code=const.RETURN_SUCCESS, message=percentage)
        ret = ret.dict()
    else:
        ret = Response(code=const.RETURN_FAIL, message="This app is not currently installed.")
        ret = ret.dict()
    return ret


def install_app(app_name, customer_app_name, app_version):
    ret = Response(code=const.RETURN_FAIL, message=" ")
    ret.code, ret.message = check_app(app_name, customer_app_name, app_version)
    if ret.code == const.RETURN_SUCCESS:
        ret.code, ret.message = prepare_app(app_name, customer_app_name)
        if ret.code == const.RETURN_SUCCESS:
            t1 = Thread(target=install_app_job, args=(customer_app_name, app_version,))
            t1.start()
            ret.message = "The app is starting, please check again in a few minutes."
    ret = ret.dict()
    return ret


def start_app(app_id):
    ret = Response(code=const.RETURN_FAIL, message="")
    if docker.check_app_id(app_id):
        app_name = split_app_id(app_id)
        info, code = if_app_exits(app_name)
        if code:
            app_path = info.split()[-1].rsplit('/', 1)[0]
            docker.check_app_compose(app_path + '/.env')
            cmd = "docker compose -f " + app_path + "/docker-compose.yml start"
            output = shell_execute.execute_command_output_all(cmd)
            if int(output["code"]) == 0:
                ret.code = const.RETURN_SUCCESS
                ret.message = "The app starts successfully."
            else:
                ret.message = "The app failed to start!"
        else:
            ret.message = "The app is not installed!"
    else:
        ret.message = "AppID is not legal!"
    ret = ret.dict()
    return ret


def stop_app(app_id):
    ret = Response(code=const.RETURN_FAIL, message="")
    if docker.check_app_id(app_id):
        app_name = split_app_id(app_id)
        info, code = if_app_exits(app_name)
        if code:
            app_path = info.split()[-1].rsplit('/', 1)[0]
            cmd = "docker compose -f " + app_path + "/docker-compose.yml stop"
            output = shell_execute.execute_command_output_all(cmd)
            if int(output["code"]) == 0:
                ret.code = const.RETURN_SUCCESS
                ret.message = "The app stopped successfully."
            else:
                ret.message = "App stop failed!"
        else:
            ret.message = "The app is not installed!"
    else:
        ret.message = 'AppID is not legal!'
    ret = ret.dict()
    return ret


def restart_app(app_id):
    ret = Response(code=const.RETURN_FAIL, message="")
    if docker.check_app_id(app_id):
        app_name = split_app_id(app_id)
        info, code = if_app_exits(app_name)
        if code:
            app_path = info.split()[-1].rsplit('/', 1)[0]
            cmd = "docker compose -f " + app_path + "/docker-compose.yml restart"
            output = shell_execute.execute_command_output_all(cmd)
            if int(output["code"]) == 0:
                ret.code = const.RETURN_SUCCESS
                ret.message = "The app restarts successfully."
            else:
                ret.message = "App restart failed!"
        else:
            ret.message = "The app is not installed!"
    else:
        ret.message = 'AppID is not legal!'
    ret = ret.dict()
    return ret


def uninstall_app(app_id):
    ret = Response(code=const.RETURN_FAIL, message="")
    if docker.check_app_id(app_id):
        info, code = if_app_exits(app_name)
        if code:
            app_path = info.split()[-1].rsplit('/', 1)[0]
            if_stopped = stop_app(app_id)  # stop_app
            app_name = split_app_id(app_id)
            if if_stopped["code"] == 0:
                cmd = "docker compose -f " + app_path + "/docker-compose.yml down -v"
                lib_path = '/data/library/apps/' + app_name
                if app_path != lib_path:
                    cmd = cmd + " && sudo rm -rf " + app_path
                output = shell_execute.execute_command_output_all(cmd)
                if int(output["code"]) == 0:
                    ret.code = 0
                    ret.message = "The app is deleted successfully"
                else:
                    ret.message = "App deletion failed!"
            else:
                ret.message = if_stopped["message"]
        else:
            ret.message = 'AppID is not legal!'
    ret = ret.dict()
    return ret


def check_app(app_name, customer_app_name, app_version):
    message = " "
    code = const.RETURN_FAIL
    if app_name == None or customer_app_name == None or app_version == None:
        message = "Please fill in the APP information completely!"
    elif not docker.check_app_directory(app_name):
        message = "Installing the app is not supported!"
    elif re.match('^[a-z0-9]+$', customer_app_name) == None:
        message = "App names must be lowercase letters and numbers!"
    elif docker.check_directory("/data/apps/" + customer_app_name):
        message = "The APP name is already in use, please specify a different name to reinstall."
    elif not docker.check_vm_resource(app_name):
        message = "System resources (memory, CPU, disk) are insufficient, and continuing to install may cause the app to not run or the server to be abnormal!"
    else:
        code = const.RETURN_SUCCESS
    return code, message


def prepare_app(app_name, customer_app_name):
    library_path = "/data/library/apps/" + app_name
    install_path = "/data/apps/" + customer_app_name
    message = " "
    code = const.RETURN_SUCCESS
    output = shell_execute.execute_command_output_all("cp -r " + library_path + " " + install_path)
    if int(output["code"]) != 0:
        message = "creating" + customer_app_name + "directory failed!"
        code = const.RETURN_FAIL
    return code, message


def install_app_job(customer_app_name, app_version):
    # write running_apps.txt
    file_path = "/data/apps/running_apps.txt"
    shell_execute.execute_command_output_all("echo " + customer_app_name + " >> " + file_path)
    # modify env
    env_path = "/data/apps/" + customer_app_name + "/.env"
    docker.modify_env(env_path, 'APP_NAME', customer_app_name)
    docker.modify_env(env_path, "APP_VERSION", app_version)
    # check port
    docker.check_app_compose('/data/apps/' + customer_app_name)
    # modify running_apps.txt
    cmd = "cd /data/apps/" + customer_app_name + " && sudo docker compose up --pull always -d"
    shell_execute.execute_command_output_all(cmd)
    # delete
    output = shell_execute.execute_command_output_all("sed -n \'/^" + customer_app_name + "/=\' " + file_path)
    if int(output["code"]) == 0 and output["result"] != "":
        line_num = output["result"].split("\n")[0]
        shell_execute.execute_command_output_all("sed -i \'" + line_num + "d\' " + file_path)


def if_app_exits(app_name):
    info = ""
    cmd = "docker compose ls -a | grep \'/" + app_name + "/\'"
    output = shell_execute.execute_command_output_all(cmd)
    if int(output["code"]) == -1:
        return info, False
    else:
        info = output["result"]
        myLogger.info_logger("APP info: " + info)
        return info, True


def split_app_id(app_id):
    return app_id.split("_")[1]


def get_apps_from_compose(output_list):
    ip_result = shell_execute.execute_command_output_all("curl ifconfig.me")
    ip = ip_result["result"]
    app_list = []
    has_add = []
    for app_info in output_list:
        volume = app_info["ConfigFiles"]  # volume
        app_path = volume.rsplit('/', 1)[0]
        app_name = volume.split('/')[-2]
        app_id = app_name + "_" + app_name  # app_id
        real_name = ""
        trade_mark = ""
        port = 0
        url = ""
        admin_url = ""
        image_url = ""
        user_name = ""
        password = ""
        official_app = False

        if app_name in ['appmanage', 'nginxproxymanager']:
            continue
        # get code
        case = app_info["Status"].split("(")[0]  # case
        if case == "running":
            case_code = const.RETURN_RUNNING  # case_code
        elif case == "exited":
            case = "stop"
            case_code = const.RETURN_STOP
        elif case == "created":
            case_code = const.RETURN_READY
            case = "installing"
        else:
            case_code = const.RETURN_ERROR

        var_path = app_path + "/variables.json"
        official_app = check_if_official_app(var_path)
        if official_app:
            real_name = docker.read_var(var_path, 'name')
            app_id = real_name + "_" + app_name  # app_id
            # get trade_mark
            trade_mark = docker.read_var(var_path, 'trademark')
            image_url = get_Image_url(real_name)
            # get env info
            path = app_path + "/.env"
            # get port and url
            try:
                http_port = list(docker.read_env(
                    path, "APP_HTTP_PORT").values())[0]
                port = int(http_port)
                easy_url = "http://" + ip + ":" + str(port)
                url = get_url(real_name, easy_url)
                admin_url = get_admin_url(real_name, url)
            except IndexError:
                try:
                    db_port = list(docker.read_env(
                        path, "APP_DB.*_PORT").values())[0]
                    port = int(db_port)
                except IndexError:
                    pass
            # get user_name
            try:
                user_name = list(docker.read_env(path, "APP_USER").values())[0]
            except IndexError:
                pass
            # get password
            try:
                password = list(docker.read_env(
                    path, "POWER_PASSWORD").values())[0]
            except IndexError:
                pass

        has_add.append(app_name)
        app = App(app_id=app_id, name=real_name, customer_name=app_name, status_code=case_code, status=case, port=port,
                  volume=volume, url=url,
                  image_url=image_url, admin_url=admin_url, trade_mark=trade_mark, user_name=user_name,
                  password=password, official_app=official_app)
        app_list.append(app.dict())
    return app_list, has_add


def check_if_official_app(var_path):
    if docker.check_directory(var_path):
        if docker.read_var(var_path, 'name') != "" and docker.read_var(var_path, 'trademark') != "" and docker.read_var(
                var_path, 'requirements') != "":
            requirements = docker.read_var(var_path, 'requirements')
            try:
                cpu = requirements['cpu']
                mem = requirements['memory']
                return True
            except:
                return False
    else:
        return False


def get_apps_from_queue(app_list, has_add):
    file_path = "/data/apps/running_apps.txt"
    if docker.check_directory(file_path):
        output = shell_execute.execute_command_output_all("cat " + file_path)
        apps = output["result"].split("\n")
        for running_app_name in apps:
            running_app_name = re.sub("\n", "", running_app_name)
            if running_app_name not in has_add and running_app_name != "":
                var_path = "/data/apps/" + running_app_name + "/variables.json"
                trade_mark = docker.read_var(var_path, 'trademark')
                real_name = docker.read_var(var_path, 'name')
                image_url = get_Image_url(real_name)
                app = App(app_id=real_name + "_" + running_app_name, name=real_name, customer_name=running_app_name,
                          status_code=const.RETURN_READY, status="installing", port=0, volume="-",
                          url="-", image_url=image_url, admin_url="-", trade_mark=trade_mark, user_name="-",
                          password="-", official_app=True)
                app_list.append(app.dict())
    return app_list


def get_Image_url(app_name):
    image_url = "static/images/" + app_name + "-websoft9.png"
    if not os.path.exists(image_url):
        return ""
    return image_url


def get_url(app_name, easy_url):
    url = easy_url
    if app_name == "joomla":
        url = easy_url + "/administrator"
    elif app_name == "other":
        url = easy_url + "/administrator"
    else:
        url = easy_url
    return url


def get_admin_url(app_name, url):
    admin_url = ""
    if app_name == "wordpress":
        admin_url = url + "/wp-admin"
    elif app_name == "other":
        admin_url = url + "/admin"
    else:
        admin_url = ""
    return admin_url