import pyrebase
import firebase_admin

firebaseConfig = {
    'apiKey': "AIzaSyAdL0W5HscjEDFPK4BDi6Cnc7FLa30GPYY",
    'authDomain': "vehicleantitheftrecognition.firebaseapp.com",
    'databaseURL': "https://vehicleantitheftrecognition.firebaseio.com",
    'projectId': "vehicleantitheftrecognition",
    'storageBucket': "vehicleantitheftrecognition.appspot.com",
    'messagingSenderId': "163692530359",
    'appId': "1:163692530359:web:b6dc7ccfc56a79afb11b32",
    'measurementId': "G-EPWP2LK89Q",
    'serviceAccount': 'vehicleantitheftrecognition-firebase-adminsdk-krrgw-05da515de5.json'
  }

firebase = pyrebase.initialize_app(firebaseConfig)
db = firebase.database()
auth = firebase.auth()
storage = firebase.storage()

class DBHelper:

    # Create account function which creates a new authentication info.
    def createAccount(username, password, confirmpassword):
        email = username + "@hotmail.com"
        if password == confirmpassword:
            auth.create_user_with_email_and_password(email,password)
            print("Account sucessfully created.")
        else:
            print("Confirmed password doesn't match to other password.")

    # Login function which verifies the given authentication info.
    def login(username, password):
        email = username + "@hotmail.com"
        try:
            auth.sign_in_with_email_and_password(email, password)
            print("Successfully Logged in.")
        except:
            print("Invalid username or password.")

    # Uploads the data of specified user uploaded into firebase.
    def uploadData(userID, firstname, lastname, email, phone, address):
        data = {"First Name": firstname, "Last Name": lastname, "E-Mail": email, "Phone": phone, "Address": address}
        db.child("Users").child(userID).set(data)

    # Removes the data of specified user uploaded into firebase.
    def removeData(userID):
        db.child("Users").child(userID).remove()

    # Returns the first name or else an empty string.
    def getFirstName(userID):
        firstname = ""
        users = db.child("Users").get()
        for user in users.each():
            if user.key() == userID:
                firstname = user.val()["First Name"]
        return firstname

    # Returns the last name or else an empty string.
    def getLastName(userID):
        lastname = ""
        users = db.child("Users").get()
        for user in users.each():
            if user.key() == userID:
                lastname = user.val()["Last Name"]
        return lastname

    # Returns the e-mail or else an empty string.
    def getEmail(userID):
        email = ""
        users = db.child("Users").get()
        for user in users.each():
            if user.key() == userID:
                email = user.val()["E-Mail"]
        return email

    # Returns the phone or else an empty string.
    def getPhone(userID):
        phone = ""
        users = db.child("Users").get()
        for user in users.each():
            if user.key() == userID:
                phone = user.val()["Phone"]
        return phone

    # Returns the address or else an empty string.
    def getAddress(userID):
        address = ""
        users = db.child("Users").get()
        for user in users.each():
            if user.key() == userID:
                address = user.val()["Address"]
        return address

    # Uploads the photo of user, input should be something like "example.png"
    def uploadUserPhoto(userphoto):
        userphoto_str = str(userphoto)
        storage.child("Photos_of_Users/" + str(userphoto)).put("Photos_of_Users/" + str(userphoto))

    # Uploads the photo of thief, input should be something like "example.png"
    def uploadThiefPhoto(userphoto):
        userphoto_str = str(userphoto)
        storage.child("Photos_of_Thieves/" + str(userphoto)).put("Photos_of_Thieves/" + str(userphoto))

    # Downloads all the user photos.
    def downloadAllUserphotos(self):
        storage.child("Photos_of_Users").download("Storage_from_Database")

    # Downloads all the thief photos.
    def downloadAllThiefphotos(self):
        storage.child("Photos_of_Thieves").download("Storage_from_Thieves")

   # Deletes photo of the specified user.
    def deleteUserPhoto(userphoto):
       storage.delete('Photos_of_Users/' + userphoto)