import os
import time
from src.services.image_capture_service.capture_main import FrameSaver


class ImageCaptureService:
    def __init__(self, flag_path, source, image_path_to_save, image_hash_threshold):
        self.flag_path = flag_path
        self.source = source
        self.image_path_to_save = image_path_to_save
        self.image_hash_threshold = image_hash_threshold
        self.running = False

    def create_stop_flag(self):
        try:
            with open(self.flag_path, 'w') as flag_file:
                flag_file.write('False')  # Set initial content as 'False'
            print("Flag file created and set to 'False'.")
        except Exception as e:
            print(f"Error creating flag file: {e}")

    def check_stop_flag(self):
        try:
            if os.path.exists(self.flag_path):
                with open(self.flag_path, 'r') as flag_file:
                    content = flag_file.read().strip()
                    return content.lower() == "true"
            else:
                return False
        except Exception as e:
            print(f"Error checking stop flag: {e}")
            return False

    def update_stop_flag(self, value):
        try:
            with open(self.flag_path, 'w') as flag_file:
                flag_file.write(str(value))  # Write the provided value to the flag file
        except Exception as e:
            print(f"Error updating stop flag: {e}")

    def start_service(self):
        try:
            print("Image Capturing service starting.")
            self.running = True
            start_capture = FrameSaver(self.source, self.image_path_to_save, self.image_hash_threshold)
            start_capture.start_stream()

            while self.running:
                time.sleep(1)
                stop_flag = self.check_stop_flag()

                if stop_flag:
                    print("Stop flag detected. Stopping all frame capturing services.")
                    self.running = False
                    stop_successful = start_capture.stop_stream()
                    if stop_successful:
                        print("Frame capturing services successfully stopped.")
                    else:
                        print("Issue encountered while stopping frame capturing services.")

        except Exception as e:
            print(f"Error in setting up FrameSaver: {e}")
            print(f"Error in setting up FrameSaver: {e}")
        finally:
            if 'start_capture' in locals():
                stop_successful = start_capture.stop_stream()
                if stop_successful:
                    print("Frame capturing services successfully stopped in finally block.")
                else:
                    print("Issue encountered while stopping frame capturing services in finally block.")

    def stop_service(self):
        self.update_stop_flag("True")


if __name__ == "__main__":
    FLAG_PATH = "resources/image_capturing"
    # SOURCE = 'rtsp://ishwor:subedi@192.168.1.106:5555/h264_opus.sdp'
    SOURCE = 0
    IMAGE_PATH_TO_SAVE = "images/cam_images"
    IMAGE_HASH_THRESHOLD = 5

    image_capture_service = ImageCaptureService(FLAG_PATH, SOURCE, IMAGE_PATH_TO_SAVE, IMAGE_HASH_THRESHOLD)
    image_capture_service.create_stop_flag()
    image_capture_service.start_service()
    time.sleep(1)