AWS SSO × Arc：複数アカウントの安全な切り替えを自動化する

#!/bin/bash

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" &> /dev/null && pwd)"
PYTHON_SCRIPT="${SCRIPT_DIR}/aws_sso_login.py"

if [ "$#" -eq 0 ]; then
    profiles=$("$PYTHON_SCRIPT")
    echo "利用可能なプロファイル:"
    echo "$profiles"
    echo "使用方法: aws_sso_login "
    exit 1
fi

"$PYTHON_SCRIPT" "$@"

#!/bin/bash

_aws_sso_login_completion() {
    local cur prev opts
    COMPREPLY=()
    cur="${COMP_WORDS[COMP_CWORD]}"
    prev="${COMP_WORDS[COMP_CWORD-1]}"
    opts=$(~/.aws/bin/aws_sso_login.py)

    COMPREPLY=( $(compgen -W "${opts}" -- ${cur}) )
    return 0
}

complete -F _aws_sso_login_completion aws_sso_login

#!/usr/bin/env python3

import os
import sys
import subprocess
import configparser
import webbrowser
import time
import re

def get_profiles_with_sso_session():
    config = configparser.ConfigParser()
    config.read(os.path.expanduser("~/.aws/config"))
    profiles = {}
    for section in config.sections():
        if section.startswith("profile "):
            profile_name = section.split()[1]
            sso_session = config[section].get("sso_session", "")
            if sso_session:  # sso_sessionが設定されている場合のみ追加
                profiles[profile_name] = sso_session
    return profiles

def run_aws_sso_login(profile_name, sso_session):
    command = f"aws sso login --profile {profile_name} --no-browser"

    print(f"プロファイル {profile_name} の AWS SSO ログインを開始します...")
    process = subprocess.Popen(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)

    url_with_code = None
    for line in process.stdout:
        print(line, end='')
        # user_code付きのURLを探す
        match = re.search(r'(https://\\S+user_code=\\S+)', line)
        if match:
            url_with_code = match.group(1)
            break

    if url_with_code:
        # URLにsso_sessionパラメータを追加
        modified_url = f"{url_with_code}&sso_session={sso_session}"

        print(f"\\n認証 URL をブラウザで開きます: {modified_url}")
        try:
            webbrowser.open(modified_url)
        except webbrowser.Error:
            print("ブラウザを自動で開けませんでした。以下のURLを手動でブラウザにコピー&ペーストしてください：")
            print(modified_url)

        print("\\nブラウザで認証を完了してください。")
        print("認証が完了したら、このターミナルに戻ってきてEnterキーを押してください。")
        input()

        # 残りの出力を表示
        for line in process.stdout:
            print(line, end='')
    else:
        print("user_code付きの認証 URL が見つかりませんでした。")

    process.wait()
    if process.returncode != 0:
        print("エラー: AWS SSO ログインに失敗しました。")
        print(process.stderr.read())
    else:
        print("AWS SSO ログインが正常に完了しました。")

def main():
    profiles = get_profiles_with_sso_session()

    if len(sys.argv) < 2:
        for profile in profiles:
            print(profile)
        sys.exit(0)

    profile_name = sys.argv[1]
    if profile_name not in profiles:
        print(f"エラー: プロファイル '{profile_name}' が見つかないか、sso_sessionが設定されていません。", file=sys.stderr)
        print("利用可能なプロファイル (sso_sessionが設定されているもの):", file=sys.stderr)
        for profile in profiles:
            print(profile, file=sys.stderr)
        sys.exit(1)

    sso_session = profiles[profile_name]
    run_aws_sso_login(profile_name, sso_session)

if __name__ == "__main__":
    main()