from flask import Flask, Blueprint, request , current_app
from flask import jsonify
import os
import json
import re
import subprocess

# 创建bp
bp = Blueprint('home', __name__, url_prefix='')


def load_nginx_config():
 # 获取当前脚本的目录
 # current_dir = os.path.dirname(os.path.abspath(__file__))
 # 构建文件的相对路径
 # file_path = os.path.join(current_dir, 'config.json')
 file_path = "F:\\ziliao\\nginx-1.20.1\\conf\\nginx.conf"
 # 打开 JSON 文件并加载数据
 with open(file=file_path, encoding='utf-8', mode='r') as f:
 return f.read()


def update_nginx_config(string_to_write):
 # 文件名
 file_path = "F:\\ziliao\\nginx-1.20.1\\conf\\nginx.conf"

 try:
 with open(file_path, 'w', encoding='utf-8') as file:
 file.write(string_to_write)

 current_app.logger.critical(f"new str is write ok to {file_path} ")
 except Exception as e:
 current_app.logger.critical(f"write file error : {e}")


@bp.get('/')
def index():
 return jsonify(success=True, code=200, msg="hello world")


@bp.post('/register')
def register():
 ip_list = request.get_json(force=True)
 filtered_ips = [ip for ip in ip_list if ip.startswith('172.16.10.')]
 new_ip = filtered_ips[0]
 current_app.logger.critical('program send new ip address is : ' + new_ip)

 nginx_config_str = load_nginx_config()
 # 定义正则表达式模式,用于匹配IP地址中最后一部分可变的情况
 pattern = re.compile(r'172\.16\.10\.\d{1,3}')
 # 原始字符串

 # 使用findall方法找出所有符合正则表达式的内容,并形成列表
 matches = re.findall(pattern, nginx_config_str)
 old_ip = matches[0]
 current_app.logger.critical(f'old ip address in config is : {old_ip}')

 if old_ip != new_ip:
 current_app.logger.critical(f'ip address is change , start update config file ... ')

 # 使用sub方法进行替换,通过lambda函数来构建替换内容
 new_config_str = re.sub(pattern, new_ip, nginx_config_str)
 # print(f'------------------------------------------------新配置文件内容------------------------------------------------')
 # print(new_config_str)
 update_nginx_config(new_config_str)
 current_app.logger.critical(f'config file is update ok , restart nginx')
 exec_command([
 "net stop nginx",
 "net start nginx"
 # "F:",
 # "cd F:\\ziliao\\nginx-1.20.1",
 # "F:\\ziliao\\nginx-1.20.1\\nginx.exe -s reload"
 ])
 else:
 current_app.logger.critical(f'ip address has no change , keep watching...')

 return jsonify(success=True, code=200, msg="register success")


def exec_command(cmd_command_lines):
 # # 多行CMD命令
 # cmd_command_lines = [
 # "cd C:\\Program Files",
 # "dir"
 # ]
 # 将多行命令拼接成一个字符串
 cmd_command = " & ".join(cmd_command_lines)

 # 使用subprocess.run()方法执行CMD命令
 result = subprocess.run(cmd_command, shell=True, capture_output=True, text=True)

 # 输出命令执行结果
 if result.returncode == 0:
 current_app.logger.critical("command run success")
 current_app.logger.critical(result.stdout)
 else:
 current_app.logger.critical("command run fail . error message:")
 current_app.logger.critical(result.stderr)