import cv2
import threading
import time
import face_processing as fp
from access_control_mqtt import access_control
from database import Database, Student, Parent
from display import Display

SERIAL_PORT_DISPLAY = "/dev/serial0"
CAMERA_INDEX = 0

cam = cv2.VideoCapture(CAMERA_INDEX)
actrl = access_control()
display = Display(SERIAL_PORT_DISPLAY)
db = Database()

global img
global frame_ready
frame_ready = False

def read_webcam():
    global img
    global frame_ready
    while True:
        ret, img = cam.read()
        frame_ready = True

'''
display.set_page("student")
display.set_string("msg.txt","Door Open")
'''

threading.Thread(target=read_webcam).start()

while True:
    while not frame_ready:
        time.sleep(1)

    #Try to identify face
    #if actrl.get_scan_state():
    if True:
        print("SCAN ACTIVE!")
        faces = fp.identify_face(img, target_condidence=0.6) ## Scan student face
        if(len(faces)==1):
            facefile = faces[0]['name'] ## File name
            student_info = db.get_student_info(facefile)## Check is it student?
            if(student_info!=None): #is student, request another scan for parent
                time_count = time.perf_counter()
                endtime = time_count + 10
                while (time_count != endtime):
                    #db.log_access_student_file()
                    time_count = time.perf_counter()
                    time.sleep(1)
                    display.set_page("student")
                    display.set_string("msg.txt","Scan parent face")
                    parent_face = fp.identify_face(img, target_condidence=0.6)   
                    parent_info = db.get_parent_info(parent_face)
                    #DONE TODO Fill in student info Resolve
                    display.set_page("student")
                    display.set_string("name_std.txt", student_info.name)
                    display.set_string("surname_std.txt", student_info.surname)
                    display.set_string("id_std.txt", student_info.id)
                    time_count = time.perf_counter()

                    if(parent_info != None): # Identified parent
                        if(db.check_relationship(student_info,parent_info)): # Check if the detected parent is right for the detected student.
                            #DONE TODO Log Access STDw/P
                            db.log_access_student_with_parent(student_info, student_info)
                            #DONE TODO write parrent info to display
                            display.set_page("student")
                            display.set_string("name_std.txt", parent_info.name)
                            display.set_string("surname_std.txt", parent_info.surname)
                            display.set_string("id_std.txt", parent_info.id)
                            access_control.unlock_door() # Door open
                            display.set_string("msg.txt","Get your kid")
                            #TODO Send timer to display
                            #display.set_page("student")
                            i = 10
                            while(i>=1):
                                display.set_string("msg.txt","Door will close in {}".format(i))
                                time.sleep(1)
                                i -= 1
                                actrl.lock_door()
                            
                            i = 10
                            #TODO Check if door close, lock
                            while(i>=0):
                                if (not actrl.get_door_state()): # False door is left open
                                    display.set_string("msg.txt","Door is left open, Alram in {} sec".format(i))
                                    time.sleep(1)
                                    i -= 1
                                    if i == 0:
                                        display.set_string("msg.txt", "Alarm Active!! Please close the door")
                                        actrl.activate_alarm()
                                else:
                                    actrl.lock_door()
                                    break
                                
                        else:
                            display.set_page("scan")
                            display.set_string("msg.txt","Wrong parent. \t Try Again")
                            #TODO Try again does not try again
                    else:
                        display.set_page("student")
                        display.set_string("msg.txt","Cannot find parent data. \t Try Again")
                should_start = True

            else:
                display.set_page("student")
                display.set_string("msg.txt","Cannot find student data. \t Try Again")
                pass
                    
        elif (len(faces)>1):
            #More than one people, error.
            print("more than one people. \t try again")
            display.set_string("msg.txt","More than one people. \t try again")
            time.sleep(2)
            pass
        print("DONE")

    else:
        print("Scan inactivate!!")


'''                
                    #is student, request another scan for parent
                    parent_face = fp.identify_face(img, target_condidence=0.6)
                    if(len(faces)==1):
                        parent_info = db.get_parent_info(faces[0]['name'])
                        if(parent_info!=None):
                            if(db.check_relationship(student_info,parent_info)):
                                #Student is under parent, open door.
                                pass
                            else:
                                #Wrong Parent, retry.
                                pass
                        else:
                            #Not a parent, retry
                            pass
'''