ふつうのITエンジニアの独り言

import subprocess
import objc
import os
from pathlib import Path

from CoreFoundation import CFRunLoopRun
from SystemConfiguration import (
    SCDynamicStoreCreate,
    SCDynamicStoreSetNotificationKeys,
    SCDynamicStoreCreateRunLoopSource
)
from CoreFoundation import (
    CFRunLoopAddSource, 
    CFRunLoopGetCurrent,
    kCFRunLoopDefaultMode
)

from logging import getLogger, basicConfig, DEBUG
basicConfig(filename='/tmp/Monitoring_WiFi.log', 
            encoding='utf-8', 
            level=DEBUG,
            format='%(asctime)s - %(levelname)s - %(message)s'
            )
logger = getLogger(__name__)

def getScriptFolderPath(name : str):
    """
    カレントディレクトリから指定した名前のフォルダパスを取得する。

    :param ssid: SSID名
    :type ssid: str
    """
    for path in Path(__file__).parent.iterdir():
        if path.is_dir() and path.name == f"{name}":
            logger.debug(f"getScriptFolderPath return {name}")
            return path
        
    logger.debug(f"getScriptFolderPath return None")
    return None

def execute_all_commands_in_folder(folder_path):
    """
    指定したフォルダ内のスクリプトをすべて実行する。
    順不定
    """
    for path in Path(folder_path).iterdir():
        if path.is_file() and path.name.endswith(".command"):
            logger.info(f"Executing: {path}")
            subprocess.Popen(["/bin/bash", path])

def get_current_ssid():
    """
    現在のssidを取得する
    """
    try:
        result = subprocess.run(
            ["bash", "-c", "system_profiler SPAirPortDataType | grep 'Current Network Information' -A1 | sed -n '2p'"],
            capture_output=True,
            text=True
        )
        # result例 = "CompletedProcess(args=['bash', '-c', 'system_profiler SPAirPortDataType | awk \'/Current Network Information:/ {getline; gsub(/^ +/, ""); print}\' | head -n 1'], returncode=0, stdout='name_of_ssid:\n', stderr='')"
        logger.debug(f"subprocess.run result: {result}")
        ssid = result.stdout.strip("\n: ") #改行とカンマとスペースを除去
        logger.debug(f"get_current_ssid return : {repr(ssid)}")
        return ssid
    except Exception as e:
        logger.debug(f"get_current_ssid err : {e}")
        logger.debug(f"get_current_ssid return : {e}")
        return None

last_ssid = "" #現在のssid名を保持
def ssid_changed(store, changed_keys, info):
    """
    SSIDの変化を検出した時に呼ばれる
    """
    logger.info(f"start ssid_changed")
    
    # 現在のssidを取得
    ssid = get_current_ssid()

    if not ssid:
        logger.debug(f"ssid is empty")
        return
    else:   
        # 前回から変化がない場合は何もしない。  
        global last_ssid
        if last_ssid == ssid:
            logger.debug(f"not change ssid: {ssid}")
            return
        
        logger.info(f"SSID change {last_ssid} to {ssid}")
        last_ssid = ssid

        # SSIDと同じ名前のフォルダを検索する。なければ"default"フォルダを検索する
        path = getScriptFolderPath(ssid) or getScriptFolderPath("default")
        
        if not path:
            #実行するフォルダがない場合は何もしない
            return
        else:
            execute_all_commands_in_folder(path)

def startWatchSSI():
    """
    SSID監視を開始
    """
    logger.debug(f"start startWatchSSI")

    # イベント監視のセットアップ
    # SCDynamicStoreCreate は、macOS の SystemConfiguration フレームワークに含まれる関数で、ネットワーク設定や状態の監視・取得を行うための「ダイナミックストア」オブジェクトを作成する関数です。
    store = SCDynamicStoreCreate(None, "SSIDMonitor", ssid_changed, None)
    if store is None:
        raise RuntimeError("SCDynamicStoreCreate failed")
    # SCDynamicStoreSetNotificationKeys は、macOS の SystemConfiguration フレームワークにおいて、どのネットワーク設定の変更を監視するかを指定する関数です
    SCDynamicStoreSetNotificationKeys(store, None, ["State:/Network/Interface/en0/AirPort"]) # en0 の Wi-Fi 状態の変化を監視

    # RunLoop 開始
    # SCDynamicStoreCreateRunLoopSource と CFRunLoopRun を使ってイベントループを開始することで、リアルタイムに変化を検出できます。
    rls = SCDynamicStoreCreateRunLoopSource(None, store, 0)
    if rls is None:
        raise RuntimeError("SCDynamicStoreCreateRunLoopSource failed")
    CFRunLoopAddSource(CFRunLoopGetCurrent(), rls, kCFRunLoopDefaultMode)
    CFRunLoopRun()

if __name__ == "__main__":
    startWatchSSI()