File size: 2,281 Bytes
0731c95 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 |
#!/usr/bin/env python3
"""
自动读取Kiro token的脚本
在Docker容器启动时自动读取宿主机的token文件
"""
import os
import json
import sys
from pathlib import Path
def get_token_file_path():
"""获取token文件路径"""
home = Path.home()
return home / ".aws" / "sso" / "cache" / "kiro-auth-token.json"
def read_tokens():
"""读取token文件"""
token_file = get_token_file_path()
if not token_file.exists():
print(f"❌ Token文件不存在: {token_file}")
print("请确保已登录Kiro，或手动创建token文件")
return None, None
try:
with open(token_file, 'r', encoding='utf-8') as f:
data = json.load(f)
access_token = data.get('accessToken')
refresh_token = data.get('refreshToken')
if not access_token or not refresh_token:
print("❌ Token文件格式错误，缺少accessToken或refreshToken")
return None, None
return access_token, refresh_token
except json.JSONDecodeError:
print("❌ Token文件JSON格式错误")
return None, None
except Exception as e:
print(f"❌ 读取token文件失败: {e}")
return None, None
def create_env_file(access_token, refresh_token):
"""创建.env文件"""
env_content = f"""# Kiro Token配置
# 自动生成于 {os.path.basename(__file__)}
KIRO_ACCESS_TOKEN={access_token}
KIRO_REFRESH_TOKEN={refresh_token}
"""
with open('.env', 'w', encoding='utf-8') as f:
f.write(env_content)
print("✅ .env文件已创建/更新")
def main():
"""主函数"""
print("🔍 正在读取Kiro token...")
access_token, refresh_token = read_tokens()
if access_token and refresh_token:
create_env_file(access_token, refresh_token)
print("✅ Token读取成功，服务即将启动...")
return 0
else:
print("❌ 无法获取token，请检查：")
print("1. 是否已登录Kiro (https://kiro.dev)")
print("2. token文件是否存在: ~/.aws/sso/cache/kiro-auth-token.json")
print("3. 或手动创建.env文件并设置token")
return 1
if __name__ == "__main__":
sys.exit(main()) |