Add auth functions, register, login and logout

Author: chermdev

database/auth.py

@@ -1,6 +1,9 @@
-from db import DB
-from db import Client
+from database.db import DB
+from supabase import Client
 from gotrue.errors import AuthApiError
+from gotrue.types import AuthResponse
+from gotrue.types import OAuthResponse
+from postgrest import SyncPostgrestClient
 
 
 # TODO: not complete, initial commit.
@@ -9,24 +12,46 @@ class User():
     def __init__(self) -> None:
         pass
 
-    def register(self, email: str, password: str) -> DB:
-        db = DB()
-        db.supabase.auth.sign_up({"email": email, "password": password})
-        return db
+    def register(self, username: str, email: str, password: str) -> AuthResponse:
+        supabase = DB().supabase
+        response = supabase.auth.sign_up({
+            "email": email,
+            "password": password,
+            "options": {
+                "data": {
+                    "username": username
+                }
+            }
+        })
+        return response
 
-    def login(self, email: str, password: str) -> DB:
+    def login(self, username: str, password: str) -> AuthResponse:
         """ returns login token """
-        try:
-            session = DB().supabase.auth.sign_in_with_password(
-                {"email": email, "password": password})
-        # TODO: Improve auth error
-        except AuthApiError as err:
-            # temporal print
-            print(err)
-        else:
-            db = DB()
-            db.supabase.postgrest.auth(session.session.access_token)
-            return db
+        supabase = DB().supabase
+        response = supabase.auth.sign_in_with_password(
+            {"email": username, "password": password})
+        # response.session.access_token
+        return response
+    
+    
+    def login_github(self) -> OAuthResponse:
+        """ returns login token """
+        provider = 'github'
+        supabase = DB().supabase
+        response = supabase.auth.sign_in_with_oauth({'provider': provider})
+        # response.session.access_token
+        return response
+    
+    def login_google(self) -> OAuthResponse:
+        """ returns login token """
+        provider = 'google'
+        supabase = DB().supabase
+        response = supabase.auth.sign_in_with_oauth({'provider': provider})
+        # response.session.access_token
+        return response
 
-    def logout(self, supabase: Client) -> DB:
+    def logout(self, token: str) -> None:
+        supabase = DB().supabase
+        supabase.postgrest.auth(token=token)
         supabase.auth.sign_out()
+