import uuid

import requests

from django.views.generic import RedirectView, View
from django.shortcuts import redirect
from django.conf import settings

from quickbooks import QuickBooks

 

# ボタンがクリックされたときにOAuthフローを開始するビュー
class QuickBooksAuthView(RedirectView):
    permanent = False
    
    def get_redirect_url(self, *args, **kwargs):
        # CSRF保護のために一意の状態トークンを生成する
        state_token = str(uuid.uuid4())
        self.request.session['state_token'] = state_token
        
        # QuickBooks OAuth 2.0承認URL
        auth_url = (
            "https://appcenter.intuit.com/connect/oauth2"
            "?client_id={}&response_type=code"
            "&scope=com.intuit.quickbooks.accounting"
            "&redirect_uri={}&state={}"
        ).format(
            settings.QB_CLIENT_ID,       # settings.pyからのクライアントID
            settings.QB_REDIRECT_URI,   # settings.pyからのリダイレクトURI
            state_token                 # セキュリティのための状態トークン
        )
        return auth_url

# QuickBooksからのリダイレクトURI（コールバック）を処理するビュー
class QuickBooksCallbackView(View):
    def get(self, request, *args, **kwargs):
        # コールバックURLからコードと状態を抽出する
        auth_code = request.GET.get('code')
        state = request.GET.get('state')
        realm_id = request.GET.get('realmId')
        
        # CSRFを防ぐために状態トークンの一致を確認する
        if state != request.session.get('state_token'):
            return redirect('error_page')  # エラーURLに置き換える
        
        # アクセスと更新トークンに対して承認コードを交換する
        token_url = "https://oauth.platform.intuit.com/oauth2/v1/tokens/bearer"
        payload = {
            'grant_type': 'authorization_code',
            'code': auth_code,
            'redirect_uri': settings.QB_REDIRECT_URI,
            'client_id': settings.QB_CLIENT_ID,
            'client_secret': settings.QB_CLIENT_SECRET,
        }
        headers = {'Content-Type': 'application/x-www-form-urlencoded'}
        response = requests.post(token_url, data=payload, headers=headers)
        
        if response.status_code == 200:
            tokens = response.json()
            # トークンを安全に保存する（セッションまたはデータベースなど）
            request.session['access_token'] = tokens['access_token']
            request.session['refresh_token'] = tokens['refresh_token']
            request.session['realm_id'] = realm_id
            
            # QuickBooksクライアントを初期化する
            client = QuickBooks(
                sandbox=True,
                consumer_key=settings.QB_CLIENT_ID,
                consumer_secret=settings.QB_CLIENT_SECRET,
                access_token=tokens['access_token'],
                access_token_secret=tokens.get('access_token_secret', ''),
                company_id=realm_id
            )
            # クライアントを保存するか、API呼び出しを続行する
            request.session['qb_client'] = client  # オプション：シリアル化されたクライアントを保存する
        
        return redirect('success_page')  # 成功URLに置き換える

from django.urls import path
from .views import QuickBooksAuthView, QuickBooksCallbackView

urlpatterns = [
    path('auth/', QuickBooksAuthView.as_view(), name='qb_auth'),
    path('callback/', QuickBooksCallbackView.as_view(), name='qb_callback'),
]