import json
from rauth import OAuth2Service

from . import user


def get_oauth_token(code, state):
    service = OAuth2Service(
        name="Casdoor",
        client_id=user.authConfig.client_id,
        client_secret=user.authConfig.client_secret,
        authorize_url="%s/api/login/oauth/authorize" % user.authConfig.endpoint,
        access_token_url="%s/api/login/oauth/access_token" % user.authConfig.endpoint,
        base_url="%s/" % user.authConfig.endpoint,
    )

    data = {
        'code': code,
        'grant_type': 'authorization_code',
    }

    session = service.get_auth_session(data=data, decoder=json.loads)
    res = session.access_token
    return res