import subprocess
import psutil
from pycaw.pycaw import AudioUtilities, IAudioEndpointVolume
from comtypes import CLSCTX_ALL
from ctypes import POINTER, cast

# 설정된 WiFi 네트워크 목록
SILENCE_WIFI_LIST = ["NETWORK_1_5G", "OFFICE_WIFI"]       # 해당 네트워크 연결 시 음소거
UNMUTE_WIFI_LIST = ["HOME_WIFI", "DESKTOP_NETWORK"]      # 해당 네트워크 연결 시 볼륨 복원

# 오디오 상태 조절 함수
def adjust_volume(mute_state):
    devices = AudioUtilities.GetSpeakers()
    interface = devices.Activate(IAudioEndpointVolume._iid_, CLSCTX_ALL, None)
    volume = cast(interface, POINTER(IAudioEndpointVolume))
    current_status = volume.GetMute()
    if mute_state:
        if not current_status:
            print("시스템 음소거 중...")
            volume.SetMute(1, None)
    else:
        if current_status:
            print("시스템 볼륨 복원 중...")
            volume.SetMute(0, None)

# 현재 연결된 WiFi 정보 추출
def get_current_network():
    try:
        result = subprocess.check_output(["netsh", "wlan", "show", "interfaces"], encoding="mbcs")
        for line in result.splitlines():
            if "SSID" in line and "BSSID" not in line:
                ssid = line.split(":")[1].strip()
                print(f"현재 연결된 WiFi: {ssid}")
                return ssid
    except Exception as e:
        print(f"네트워크 정보 조회 실패: {e}")
    return None

# 유선 연결 여부 확인
def check_ethernet():
    for iface, addrs in psutil.net_if_addrs().items():
        if "Ethernet" in iface or "以太网" in iface:
            for addr in addrs:
                if addr.family == 2:  # IPv4 주소
                    print(f"유선 연결 감지: {iface}")
                    return True
    return False

# 메인 실행 로직
def main():
    current_network = get_current_network()
    if current_network:
        print(f"연결된 네트워크: {current_network}")
        if current_network in SILENCE_WIFI_LIST:
            adjust_volume(True)
        elif current_network in UNMUTE_WIFI_LIST:
            adjust_volume(False)
        else:
            print("기본 설정으로 유지됩니다.")
    elif check_ethernet():
        print("유선 연결 감지됨, 볼륨 복원 중...")
        adjust_volume(False)
    else:
        print("모든 네트워크 연결 해제됨.")

if __name__ == "__main__":
    main()