| | |
| | import json |
| | from pathlib import Path |
| |
|
| | ROOT = Path('/data/appdata/kiro-stack') |
| | GO_CONFIG = ROOT / 'kiro-go' / 'data' / 'config.json' |
| | GW_ENV = ROOT / 'kiro-gateway' / '.env' |
| |
|
| |
|
| | def load_env(path: Path): |
| | out = {} |
| | lines = [] |
| | if path.exists(): |
| | lines = path.read_text(encoding='utf-8').splitlines() |
| | for line in lines: |
| | s = line.strip() |
| | if not s or s.startswith('#') or '=' not in line: |
| | continue |
| | k, v = line.split('=', 1) |
| | out[k.strip()] = v.strip().strip('"') |
| | return out, lines |
| |
|
| |
|
| | def quote(v: str) -> str: |
| | return '"' + v.replace('"', '\\"') + '"' |
| |
|
| |
|
| | def main(): |
| | data = json.loads(GO_CONFIG.read_text(encoding='utf-8')) |
| | accounts = data.get('accounts', []) |
| |
|
| | env_map, lines = load_env(GW_ENV) |
| | primary = env_map.get('REFRESH_TOKEN', '') |
| |
|
| | tokens = [] |
| | seen = set() |
| | for a in accounts: |
| | if not a.get('enabled', False): |
| | continue |
| | token = (a.get('refreshToken') or '').strip() |
| | if not token: |
| | continue |
| | if token == primary: |
| | continue |
| | if token in seen: |
| | continue |
| | seen.add(token) |
| | tokens.append(token) |
| |
|
| | joined = ','.join(tokens) |
| | kv = f'KIRO_REFRESH_TOKENS={quote(joined)}' |
| |
|
| | replaced = False |
| | new_lines = [] |
| | for line in lines: |
| | if line.startswith('KIRO_REFRESH_TOKENS='): |
| | new_lines.append(kv) |
| | replaced = True |
| | else: |
| | new_lines.append(line) |
| |
|
| | if not replaced: |
| | if new_lines and new_lines[-1].strip() != '': |
| | new_lines.append('') |
| | new_lines.append(kv) |
| |
|
| | GW_ENV.write_text('\n'.join(new_lines) + '\n', encoding='utf-8') |
| | print(f'synced_tokens={len(tokens)}') |
| |
|
| |
|
| | if __name__ == '__main__': |
| | main() |
| |
|
