Merge pull request #17 from fga-eps-mds/fix-login-social

Fix login social

requirements.txt

@@ -73,3 +73,4 @@ websocket-client==1.6.4
 websockets==11.0.3
 wrapt==1.15.0
 zope.interface==6.1
+fastapi_filter

src/controller/authController.py

@@ -64,6 +64,21 @@ async def login(data: authSchema.UserLogin, db: Session = Depends(get_db)):
 
   return JSONResponse(status_code=200, content={ "access_token": access_token, "refresh_token": refresh_token, "token_type": "bearer" })
 
+@auth.post("/login/social")
+async def login_social(user: authSchema.UserSocial, db: Session = Depends(get_db)):
+  existing_user = userRepository.get_user_by_email(db, user.email)
+
+  if existing_user is None:
+    new_user = userRepository.create_user_social(db, user.name, user.email)
+    access_token = security.create_access_token(data={"id": new_user.id, "email": new_user.email, "role": new_user.role})
+
+    return JSONResponse(status_code=200, content={ "access_token": access_token, "token_type": "bearer", "is_new_user": True, "user_id": new_user.id })
+  else:
+    access_token = security.create_access_token(data={"id": existing_user.id, "email": existing_user.email, "role": existing_user.role})
+    refresh_token = security.create_refresh_token(data={ "id": existing_user.id })
+
+    return JSONResponse(status_code=200, content={ "access_token": access_token, "refresh_token": refresh_token, "token_type": "bearer", "is_new_user": False })
+
 @auth.post("/refresh", response_model=authSchema.RefreshTokenResponse)
 def refresh_token(token: dict = Depends(security.verify_token)):
   access_token=security.create_access_token(token)

src/domain/authSchema.py

@@ -11,6 +11,10 @@ class UserCreate(BaseModel):
   email: str
   password: str
 
+class UserSocial(BaseModel):
+  name: str
+  email:str
+
 class Token(BaseModel):
   access_token: str
   refresh_token: str

src/main.py

@@ -8,7 +8,7 @@
 load_dotenv()
 dotenv.validate_dotenv()
 
-from controller import userController, authController, googleController, facebookController
+from controller import userController, authController
 from database import engine 
 from model import userModel
 
@@ -30,8 +30,6 @@
 # Routers
 app.include_router(prefix="/api", router=authController.auth)
 app.include_router(prefix="/api", router=userController.user)
-app.include_router(router=googleController.google)
-app.include_router(router=facebookController.facebook)
 
 @app.get("/")
 def read_root():

src/model/userModel.py

@@ -14,9 +14,9 @@ class User(Base):
   connection = Column(String, nullable=False)
   role = Column(String, default="USER")
   email = Column(String, unique=True, index=True, nullable=False)
-  password = Column(String, nullable=False)
+  password = Column(String, nullable=True)
   is_active = Column(Boolean, default=False)
   activation_code = Column(Integer, nullable=True)
   password_reset_code = Column(Integer, nullable=True)
 
-  
+

src/repository/userRepository.py

@@ -44,26 +44,18 @@ def create_user(db: Session, name, connection, email, password, activation_code)
   db.refresh(db_user)
   return db_user
 
-async def get_or_create_user(email: str, name: str, db: Session):
-
-    user = userRepository.get_user_by_email(db, email)
-
-    if user is None:
-        user = userRepository.create_by_login(db, name=name, email=email)
-
-    return user
-
-def create_by_login(db: Session, name, email):
-
-    db_user = userModel.User(
-        name=name,
-        email=email
-    )
-    db.add(db_user)
-    db.commit()
-    db.refresh(db_user)
-    return db_user
+def create_user_social(db: Session, name, email):
+  db_user = userModel.User(
+  name=name,
+  connection="ESTUDANTE",
+  role="USER",
+  email=email,
+  is_active=True,)
 
+  db.add(db_user)
+  db.commit()
+  db.refresh(db_user)
+  return db_user
 
 def update_user(db: Session, db_user: userSchema.User, user: userSchema.UserUpdate):
   user_data = user.dict(exclude_unset=True)