import importlib.util
import sys
import socket
import os

# 加载 默认配置文件 envar.py 和 本机配置文件 (envar.本机编码.py 例如 envar.lin237.py)
def load_envar():
    # Get current OS type
    if sys.platform.startswith('win'):
        ostype = 'win'
    elif sys.platform.startswith('darwin'):
        ostype = 'mac'
    elif sys.platform.startswith('linux'):
        ostype = 'lin'
    else:
        ostype = 'other'
    # Get current LAN IP address 获取当前主机的局域网 IP 例如 192.168.168.237
    s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    s.connect(("8.8.8.8", 80))
    lanip = s.getsockname()[0]
    s.close()
    # Load environment variables
    this_script_dir = os.path.dirname(os.path.abspath(__file__))
    calling_script_dir = os.path.dirname(os.path.abspath(sys.argv[0]))  # Get the directory of the calling script
    envFileLocal = os.path.join(calling_script_dir, "envar." + ostype + lanip.split(".")[-1] + ".py")  # 本主机配置文件
    envFileDefault = os.path.join(calling_script_dir, "envar.py")  # 默认配置文件
    # 总是加载默认配置文件，作为基础配置
    print(f"*** Loading default envar file: {envFileDefault}")
    if not os.path.exists(envFileDefault):
        raise FileNotFoundError("Default envar file not found")
    spec = importlib.util.spec_from_file_location("envar", envFileDefault)
    envar = importlib.util.module_from_spec(spec)
    sys.modules[spec.name] = envar
    spec.loader.exec_module(envar)
    # 如果存在本机配置文件，加载并覆盖默认配置
    if os.path.exists(envFileLocal):
        print(f"*** Loading and merging local envar file: {envFileLocal}")
        spec_local = importlib.util.spec_from_file_location("env_local", envFileLocal)
        env_local = importlib.util.module_from_spec(spec_local)
        spec_local.loader.exec_module(env_local)
        # 合并配置到env中
        for attr in dir(env_local):
            if not attr.startswith('__'):
                setattr(envar, attr, getattr(env_local, attr))
    return envar