From 088675785c2672abbb44bd4d3475158c85d959af Mon Sep 17 00:00:00 2001
From: JustIceO7 <oscarcao777@gmail.com>
Date: Thu, 30 Jan 2025 22:12:35 +0000
Subject: [PATCH] FEAT: Added blueprints for Google OAuth

---
 web_server/blueprints/oauth.py | 58 ++++++++++++++++++++++++++++++++++
 web_server/requirements.txt    |  3 +-
 2 files changed, 60 insertions(+), 1 deletion(-)
 create mode 100644 web_server/blueprints/oauth.py

diff --git a/web_server/blueprints/oauth.py b/web_server/blueprints/oauth.py
new file mode 100644
index 0000000..f294fe9
--- /dev/null
+++ b/web_server/blueprints/oauth.py
@@ -0,0 +1,58 @@
+from authlib.integrations.flask_client import OAuth, OAuthError
+from flask import Blueprint, url_for, jsonify, session
+
+oauth_bp = Blueprint("oauth", __name__)
+def init_oauth(app):
+    oauth = OAuth(app)
+
+    google = oauth.register(
+        'google',
+        client_id=app.config['GOOGLE_CLIENT_ID'],
+        client_secret=app.config['GOOGLE_CLIENT_SECRET'],
+        authorize_url='https://accounts.google.com/o/oauth2/auth',
+        authorize_params=None,
+        access_token_url='https://accounts.google.com/o/oauth2/token',
+        access_token_params=None,
+        refresh_token_url=None,
+        redirect_uri=url_for('google.google_auth', _external=True),
+        scope='openid profile email',
+    )
+
+
+    @oauth_bp.route('/login/google')
+    def login_google():
+            """
+            Redirects to Google's OAuth authorization page
+            """
+            return  google.authorize_redirect(url_for('google.google_auth', _external=True))
+
+    @oauth_bp.route('/google_auth')
+    def google_auth():
+        try:
+            token = google.authorize_access_token()
+            user = google.parse_id_token(token)
+            
+            # check if email exists else create a database entry
+            user_email = user.get("email")
+
+            session.clear()
+            session["username"] = "a"
+            session["user_id"] = 1
+
+            return jsonify({
+                'message': 'User authenticated successfully',
+            })
+        
+        except OAuthError as e:
+            # Handle OAuth errors like failed authentication or invalid token
+            return jsonify({
+                'message': 'Authentication failed',
+                'error': str(e)
+            }), 400 
+
+        except Exception as e:
+            # Handle other unexpected errors
+            return jsonify({
+                'message': 'An unexpected error occurred',
+                'error': str(e)
+            }), 500 
\ No newline at end of file
diff --git a/web_server/requirements.txt b/web_server/requirements.txt
index 14a4a9e..71a61e1 100644
--- a/web_server/requirements.txt
+++ b/web_server/requirements.txt
@@ -23,4 +23,5 @@ Werkzeug==3.1.3
 WTForms==3.2.1
 Gunicorn==20.1.0
 gevent>=22.10.2
-gevent-websocket
\ No newline at end of file
+gevent-websocket
+flask-oauthlib==0.9.6
\ No newline at end of file