add raw token injection

Author: ks6088ts

docs/index.md

@@ -125,7 +125,9 @@ uv run python scripts/microsoft_graphs.py --help
 uv run python scripts/microsoft_graphs.py get-access-token
 
 # Get my profile
-uv run python scripts/microsoft_graphs.py get-my-profile
+uv run python scripts/microsoft_graphs.py get-my-profile \
+  --access-token $ACCESS_TOKEN \
+  --expires-on $EXPIRES_ON
 ```
 
 ## MCP

scripts/microsoft_graphs.py

@@ -3,6 +3,7 @@
 import asyncio
 
 import typer
+from azure.core.credentials import AccessToken
 from azure.identity import DeviceCodeCredential
 from msgraph import GraphServiceClient
 from msgraph.generated.users.item.user_item_request_builder import UserItemRequestBuilder
@@ -15,27 +16,44 @@
 console = Console()
 
 
-def get_graph_client() -> tuple[GraphServiceClient, list[str]]:
+class RawAccessTokenProvider:
+    def __init__(self, access_token: str, expires_on: int):
+        self._access_token = access_token
+        self._expires_on = expires_on
+
+    def get_token(self, *scopes, **kwargs) -> AccessToken:
+        return AccessToken(self._access_token, self._expires_on)
+
+
+def get_graph_client(
+    access_token: str | None = None,
+    expires_on: int | None = None,
+) -> GraphServiceClient:
     """Microsoft Graph クライアントを取得する"""
     settings = get_microsoft_graph_settings()
     scopes = settings.microsoft_graph_user_scopes.split(" ")
 
+    if access_token and expires_on:
+        # アクセストークンが指定されている場合はそれを使用
+        access_token_provider = RawAccessTokenProvider(access_token, expires_on)
+        return GraphServiceClient(credentials=access_token_provider, scopes=scopes)
     device_code_credential = DeviceCodeCredential(
         client_id=settings.microsoft_graph_client_id,
         tenant_id=settings.microsoft_graph_tenant_id,
     )
-
-    client = GraphServiceClient(
+    return GraphServiceClient(
         credentials=device_code_credential,
         scopes=scopes,
     )
 
-    return client, scopes
-
 
 @app.command()
 def get_my_profile(
     fields: list[str] | None = typer.Option(None, "--field", "-f", help="取得するフィールドを指定（複数指定可能）"),
+    access_token: str | None = typer.Option(None, "--access-token", "-a", help="アクセストークンを指定して認証する"),
+    expires_on: int | None = typer.Option(
+        None, "--expires-on", "-e", help="アクセストークンの有効期限を指定（Unix時間）"
+    ),
 ):
     """自分のプロファイル情報を取得する"""
     console.print("[bold green]ユーザープロファイル[/bold green]を取得します")
@@ -46,7 +64,7 @@ def get_my_profile(
 
     async def _get_profile():
         try:
-            client, _ = get_graph_client()
+            client = get_graph_client(access_token=access_token, expires_on=expires_on)
 
             query_params = UserItemRequestBuilder.UserItemRequestBuilderGetQueryParameters(select=select_fields)
             request_config = UserItemRequestBuilder.UserItemRequestBuilderGetRequestConfiguration(
@@ -90,9 +108,8 @@ def get_access_token():
 
         access_token = device_code_credential.get_token(*scopes)
 
-        console.print("[bold green]トークン取得成功[/bold green]")
-        console.print(f"  トークン: {access_token.token}")
-        console.print(f"  期限: {access_token.expires_on}")
+        print(f"access_token: {access_token.token}")
+        print(f"expires_on: {access_token.expires_on}")
 
     except Exception as e:
         console.print(f"[bold red]エラー[/bold red]: {str(e)}")