認証

{
  "type": "register",
  "payload": {
    "pilot": {
      "id": "my-pilot-001",
      "name": "My WiFi Controller",
      "version": "1.0.0",
      "vendor": "My Company"
    },
    "auth": {},  // 初回接続では空
    "capabilities": [...]
  }
}

┌─────────────────────────────────────────┐
│  新しい実行ノード接続リクエスト         │
├─────────────────────────────────────────┤
│                                         │
│  名前: My WiFi Controller               │
│  ID: my-pilot-001                       │
│  ベンダー: My Company                   │
│                                         │
│  機能:                                  │
│  • wifi.scan - WiFi ネットワークをスキャン
│  • wifi.connect - WiFi に接続          │
│  • wifi.status - WiFi ステータス取得   │
│                                         │
│  ⚠️  信頼できるソースからの実行ノード  │
│      のみ承認してください               │
│                                         │
│  120秒後に自動拒否                      │
│                                         │
│  ┌─────────┐  ┌──────────────────────┐ │
│  │ 拒否    │  │ 承認                  │ │
│  └─────────┘  └──────────────────────┘ │
└─────────────────────────────────────────┘

{
  "type": "register_response",
  "payload": {
    "success": true,
    "pilotId": "my-pilot-001",
    "assignedKey": "abc123-generated-key-xyz789",
    "message": "Pilot approved and registered",
    "heartbeat": {
      "interval": 30000,
      "timeout": 90000
    }
  }
}

{
  "type": "register_response",
  "payload": {
    "success": false,
    "error": {
      "code": "USER_REJECTED",
      "message": "User rejected this pilot"
    }
  }
}

# キーをコンフィグファイルに保存
config = {
    "pilot_id": "my-pilot-001",
    "auth_key": "abc123-generated-key-xyz789"
}
with open("pilot_config.json", "w") as f:
    json.dump(config, f)

{
  "type": "register",
  "payload": {
    "pilot": {
      "id": "my-pilot-001",
      "name": "My WiFi Controller",
      "version": "1.0.0"
    },
    "auth": {
      "key": "abc123-generated-key-xyz789"  // ← 保存されたキーを使用
    },
    "capabilities": [...]
  }
}

{
  "type": "register_response",
  "payload": {
    "success": true,
    "pilotId": "my-pilot-001",
    "message": "Pilot registered successfully",
    "heartbeat": {
      "interval": 30000,
      "timeout": 90000
    }
  }
}

{
  "authorizedPilots": {
    "production-pilot-001": {
      "name": "Production WiFi Controller",
      "key": "pre-shared-secret-key-12345",
      "authorizedAt": "2026-01-21T10:00:00.000Z"
    }
  }
}

{
  "pilot_id": "production-pilot-001",
  "auth_key": "pre-shared-secret-key-12345"
}

フォーマット: <vendor>-<type>-<instance>
例:
  - acme-wifi-controller-001
  - mycompany-sensor-hub-lab-01
  - test-pilot-dev-instance

{
  "pilot": {
    "id": "my-pilot-001",
    "version": "1.2.3",  // セマンティック バージョニング
    "name": "My WiFi Controller v1.2"
  }
}

async def handle_register_response(response):
    payload = response.get("payload", {})

    if not payload.get("success"):
        error = payload.get("error", {})

        if error.get("code") == "AUTH_FAILED":
            print("⚠️  キー検証に失敗")
            print("💡 キーなしで再接続...")

            # 保存されたキーを削除
            os.remove("pilot_config.json")

            # キーなしで再接続(初回フロー)
            await reconnect_without_key()

# キーなしで実行ノードを実行
python wifi_pilot.py

# 保存されたキーで実行ノードを実行
python wifi_pilot.py --config pilot_config.json

# コンフィグのキーを無効な値に変更
python wifi_pilot.py --config pilot_config.json

import json
import os

class AuthManager:
    def __init__(self, config_file="pilot_config.json"):
        self.config_file = config_file
        self.pilot_id = "my-pilot-001"
        self.key = None
        self.load_key()

    def load_key(self):
        """コンフィグファイルからキーを読み込む"""
        if os.path.exists(self.config_file):
            with open(self.config_file, "r") as f:
                config = json.load(f)
                self.key = config.get("auth_key")
                print(f"✅ {self.config_file} からキーを読み込み")
        else:
            print("ℹ️  キーが見つかりません。新しいキーをリクエストします")

    def save_key(self, key):
        """キーをコンフィグファイルに保存"""
        config = {
            "pilot_id": self.pilot_id,
            "auth_key": key
        }
        with open(self.config_file, "w") as f:
            json.dump(config, f, indent=2)

        # セキュアなファイルパーミッションを設定(Unix)
        os.chmod(self.config_file, 0o600)
        print(f"💾 {self.config_file} にキーを保存")

    def delete_key(self):
        """保存されたキーを削除"""
        if os.path.exists(self.config_file):
            os.remove(self.config_file)
            self.key = None
            print(f"🗑️  {self.config_file} を削除")

    def get_auth_payload(self):
        """登録用の認証ペイロードを取得"""
        if self.key:
            return {"key": self.key}
        else:
            return {}

    def handle_register_response(self, payload):
        """登録応答を処理"""
        if payload.get("success"):
            assigned_key = payload.get("assignedKey")
            if assigned_key:
                self.save_key(assigned_key)
                self.key = assigned_key
            return True
        else:
            error = payload.get("error", {})
            if error.get("code") == "AUTH_FAILED":
                print("❌ 認証に失敗。キーを削除")
                self.delete_key()
            return False

# 使用
auth = AuthManager()

# 登録ペイロードを構築
payload = {
    "pilot": {
        "id": auth.pilot_id,
        "name": "My WiFi Controller",
        "version": "1.0.0"
    },
    "auth": auth.get_auth_payload(),  # 空またはキー付き
    "capabilities": [...]
}