import os
from flask import Flask, render_template, request, Response
from app import app
import time
import subprocess
import re
import json
from threading import Thread
netprofilepath = r'/prog/smartshelf/app/netprofile/'

#app = Flask(__name__)

@app.route('/ipconfig')
def ipconfig():
    print("begin ip flask")
    return render_template('ipconfig.html')

def Config_DHCP():

    netconfigfile = ''
    if os.path.exists('/etc/dhcpcd.conf'):
        netprofile = netprofilepath+'dhcpcd.conf.bak'
        netconfigfile = '/etc/dhcpcd.conf'

    else:
        netprofile = netprofilepath+'/NEOTEL.nmconnection.bak'
        netconfigfile = '/etc/NetworkManager/system-connections/NEOTEL.nmconnection'
    
    with open(netprofile, 'r') as f:
        file_lines = f.readlines()
        f.close()
    with open(netconfigfile, 'w') as f:
        # for line in file_lines:
        f.writelines(file_lines)
        f.flush()
        f.close()    


def Config_StaticIP(IPAddress, Router_IP):

    netconfigfile = ''
    if os.path.exists('/etc/dhcpcd.conf'):
        netprofile = netprofilepath+'dhcpcd.conf.bak'
        netconfigfile = '/etc/dhcpcd.conf'
        with open(netprofile, 'r') as f:
            file_lines = f.readlines()
            f.close()
        with open(netconfigfile, 'w') as f:
            # for line in file_lines:
            f.writelines(file_lines)
            # f.write("static ip_address=" + str(request.args.get('new_ip')))
            # interface eth0
            #IPAddress = str(request.args.get('new_ip')
            #Router_IP = str(request.args.get('new_ip')
            f.write("interface eth0" + "\n")
            if len(IPAddress) > 0:
                f.write("static ip_address=" + IPAddress + "\n")
            if len(Router_IP) > 0:
                f.write("static routers=" + Router_IP + "\n")
            # f.write("")
            f.flush()
            f.close()

    else:
        netprofile = netprofilepath+'/NEOTEL.nmconnection.bak'
        netconfigfile = '/etc/NetworkManager/system-connections/NEOTEL.nmconnection'

        with open(netprofile, 'r') as f:
            file_lines = f.readlines()
            f.close()
        with open(netconfigfile, 'w') as f:
            # for line in file_lines:
            for line in file_lines:
                f.write(line)
                if line.startswith('[ipv4]') and len(IPAddress) > 0:
                    if len(Router_IP) > 0:
                        f.write('address1='+ IPAddress + ','+Router_IP+"\n")
                    else:
                        f.write('address1='+ IPAddress + "\n")
                
            # f.write("static ip_address=" + str(request.args.get('new_ip')))
            # interface eth0
            #IPAddress = str(request.args.get('new_ip')
            #Router_IP = str(request.args.get('new_ip')
            
            f.flush()
            f.close()

def reboot_after_delay():
    # Wait for 5 seconds before rebooting
    time.sleep(5)
    os.system('sudo reboot')

def Reenable_NetworkService():
    #os.system('sudo systemctl stop networking')
    #os.system('sudo systemctl start networking')
    #os.system('sudo systemctl stop dhcpcd.service')
    #print("services stop")
    #os.system('sudo ip addr flush dev eth0')
    #os.system('sudo systemctl start dhcpcd.service')
    # Wait for 3 seconds before rebooting
    reboot_thread = Thread(target=reboot_after_delay)
    reboot_thread.start()
    return "Apply Successfully, System will reboot in 5 seconds"

    #return "Raspberry Pi will reboot in 3 seconds"
    #os.system('sudo reboot')

def netmask_to_length(netmask):
    # Convert the netmask to a list of integers
    netmask = [int(x) for x in netmask.split(".")]

    # Count the number of set bits in the netmask
    length = 0
    for octet in netmask:
        while octet > 0:
            length += octet & 1
            octet >>= 1

    return length

@app.route('/ip_config',methods=['POST'])
def IP_Config():
    dhcp_enabled = request.form.get('dhcp')
    if dhcp_enabled:
        Config_DHCP()
        print("DHCP ENABLED")
    else:
        IP_ADDR = str(request.form.get('new_ip'))
        print("input:",str(request.form.get('new_ip')))
        Router_ADDR = str(request.form.get('router_ip'))
        NET_MASK = str(request.form.get('new_mask'))
        IP_ADDR +="/"+str(netmask_to_length(NET_MASK))
        print(IP_ADDR)
        print(Router_ADDR)
        Config_StaticIP(IP_ADDR,Router_ADDR)
        print(IP_ADDR,Router_ADDR)
        print("IP CHANGED")
        #Reenable_NetworkService()
    #Response('Apply Successfully', status=200, mimetype='text/plain')
    return Reenable_NetworkService()
    #return #'Apply Successfully ' #{} with DHCP enabled: {}'.format(IP_ADDR, 'True' if dhcp_enabled else 'False')



#Config_StaticIP("192.168.1.59/24", "192.168.1.1")
#Reenable_NetworkService()

@app.route('/get_networkinfo',methods=['POST'])
def Get_NetworkInfo():


    """
    获取eth0接口信息，返回JSON格式字符串
    """
    result = {
        "interface": "eth0",
        "ip_address": None,
        "netmask": None,
        "gateway": None
    }
    
    # 获取eth0的IP和掩码
    try:
        ifconfig_output = subprocess.check_output(["ifconfig", "eth0"], stderr=subprocess.STDOUT)
        
        # 提取IP地址
        print("1")
        print(ifconfig_output)
        ifconfig_str = ifconfig_output.decode('utf-8')
        ip_match = re.search(r'inet (?:addr:)?([\d.]+)', ifconfig_str)
        print("c111")
        if ip_match:
            result["ip_address"] = ip_match.group(1)
        
        # 提取子网掩码
        print("a")
        mask_match = re.search(r'(?:Mask:|netmask )([\d.]+)', ifconfig_str)
        print("b")
        if mask_match:
            result["netmask"] = mask_match.group(1)
            
    except subprocess.CalledProcessError:
        # eth0接口不存在，直接返回null值
        print("2")
        return json.dumps(result, indent=2)
    except Exception:
        print("3")
        # 其他异常，也返回null值
        return json.dumps(result, indent=2)
    
    # 获取网关信息
    try:
        print("4")
        route_output = subprocess.check_output(["route", "-n"])
        route_str = route_output.decode('utf-8')
        print(route_output)
        print("5")
        for line in route_str.split('\n'):
            if line.startswith('0.0.0.0'):
                print("8")
                parts = line.split()
                if len(parts) >= 8 and parts[7] == 'eth0':
                    print("9")
                    result["gateway"] = parts[1]
                    break
    except Exception:
        # 获取网关失败，保持为None
        pass
    
    return json.dumps(result, indent=2)