mirror of
https://github.com/hubHarmony/servii-backend.git
synced 2024-11-17 21:40:31 +00:00
76 lines
2.5 KiB
Python
76 lines
2.5 KiB
Python
from datetime import datetime
|
|
|
|
import jwt
|
|
import firebase_admin
|
|
from firebase_admin import credentials, auth, firestore
|
|
from google.api_core.exceptions import NotFound, PermissionDenied, Aborted, ResourceExhausted, OutOfRange, DataLoss
|
|
|
|
cred = credentials.Certificate('servii.json')
|
|
firebase_admin.initialize_app(cred)
|
|
firestore_database = firestore.client()
|
|
|
|
'''
|
|
TODO
|
|
Write a function that launches upon app's startup, it does check in the firestore for any already running servers.
|
|
Fetches the PID's.
|
|
Stops all the current processes,
|
|
Tell the database they have now stopped running.
|
|
Also ensure the program can add an additional argument to avoid this checking for scalability.
|
|
'''
|
|
|
|
|
|
def get_user_from_id(user_id):
|
|
return auth.get_user(user_id)
|
|
|
|
|
|
def verify_jwt_token(token):
|
|
try:
|
|
decoded_token = jwt.decode(token, options={"verify_signature": False})
|
|
user_id = decoded_token.get('sub')
|
|
return True, user_id
|
|
except jwt.ExpiredSignatureError:
|
|
return False, None
|
|
except jwt.InvalidTokenError:
|
|
return False, None
|
|
|
|
|
|
def create_firestore(user_id: str, data: dict) -> bool:
|
|
doc_ref = firestore_database.collection('users').document(user_id)
|
|
try:
|
|
doc_ref.create(data)
|
|
return True
|
|
except (NotFound, PermissionDenied, Aborted, ResourceExhausted,
|
|
OutOfRange, DataLoss, TypeError, Exception, ValueError) as e:
|
|
log_exception_to_firestore(e, user_id, data)
|
|
return False
|
|
|
|
|
|
def update_firestore(user_id: str, data: dict) -> bool:
|
|
doc_ref = firestore_database.collection('users').document(user_id)
|
|
try:
|
|
doc_ref.update(data)
|
|
return True
|
|
except (NotFound, PermissionDenied, Aborted, ResourceExhausted,
|
|
OutOfRange, DataLoss, TypeError, Exception, ValueError) as e:
|
|
log_exception_to_firestore(e, user_id, data)
|
|
return False
|
|
|
|
|
|
def log_exception_to_firestore(exception: Exception = None, user_id: str = None, data: dict = None):
|
|
new_id: str = datetime.now().strftime('%Y-%m-%d %H:%M:%S %Z%z')
|
|
log_entry = {
|
|
'exception_name': str(type(exception).__name__),
|
|
'exception': str(exception) if exception else 'No exception',
|
|
'user_id': str(user_id) if user_id else 'No user_id',
|
|
'data': str(data) if data else 'No data provided',
|
|
}
|
|
try:
|
|
firestore_database.collection('firebase.logs').document(new_id).create(log_entry)
|
|
print("Log entry added successfully.")
|
|
except Exception as e:
|
|
print(f"Failed to add log entry: {e}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
pass
|