import os
# Example usage:
import json
from datetime import datetime
import copy
LOCAL_USER_DATA_DIR = "data/user/" # only use locally, not for other .py files

class User:
    def __init__(self, user_id, select_course, head_icon, course_process_dict, save_path):
        self.user_id = user_id  # User ID
        self.select_course = select_course  # List of selected courses
        self.head_icon = head_icon  # URL of the user's head icon
        self.course_process_dict = course_process_dict  # Dictionary of courses with their progress
        self.save_path = save_path

    def save_chapter_memory(self, course_id, lesson_id, subchapter_title, mem_list, score, is_rebuttal):
        '''
        Save the memory of a chapter
        '''
        try:
            if(course_id not in self.course_process_dict):
                self.course_process_dict[course_id] = {'lesson_processs':[], 'course_put_in_date':'', 'course_last_study_date':'', 'course_total_study_time':0}
            self.course_process_dict[course_id]['course_last_study_date'] = datetime.now().strftime('%Y-%m-%d')
            for _ in range(len(self.course_process_dict[course_id]['lesson_processs'])):
                lesson_process, lid = self.course_process_dict[course_id]['lesson_processs'][_]
                if lid == lesson_id:
                    saveed = False
                    for i in range(len(lesson_process)):
                        chapter_info = lesson_process[i]
                        if chapter_info['title'] == subchapter_title:
                            chapter_info['dialog'] = mem_list
                            if is_rebuttal:
                                chapter_info['rebuttal_score'] = score
                            else:
                                chapter_info['score'] = score
                            chapter_info['is_rebuttal'] = is_rebuttal
                            saveed = True
                            break
                    if not saveed:
                        lesson_process.append({
                            'title': subchapter_title,
                            'dialog': mem_list,
                            'score': score,
                            'is_rebuttal': is_rebuttal,
                            'rebuttal_score': score if is_rebuttal else 0,
                        })
                    break
            self.save_to_file()
        except Exception as e:
            print("=-=-=-=-=-=")
            print(e)
    def get_course_progress(self):
        for course_name, course_data in self.course_process_dict.items():
            print(f"Course: {course_name}")
            for lessons,lesson_id  in course_data["lesson_processs"]:
                print(f"  Lesson ID: {lesson_id}")
                for chapter in lessons:
                    chapter_info = chapter
                    print(f"    Chapter Dialog: {chapter_info['dialog']}")
                    print(f"    Chapter Score: {chapter_info['score']}")
                    print(f"    Rebuttal: {'Yes' if chapter_info['is_rebuttal'] else 'No'}")
                    if chapter_info['is_rebuttal']:
                        print(f"    Rebuttal Score: {chapter_info['rebuttal_score']}")

    def select_new_course(self, course_id, course):
        self.select_course.append(course_id)

        self.course_process_dict[course_id] = {'lesson_processs':[], 'course_put_in_date':'', 'course_last_study_date':'','course_total_study_time':0}
        self.course_process_dict[course_id]['lesson_processs']
        for lesson in course.lessons:
            self.course_process_dict[course_id]['lesson_processs'].append(([], lesson['lesson_id']))


        with open(self.save_path, 'w') as f:
            json.dump(self.__json__(), f, indent=4)

    def save_to_file(self):
        print(f"saved at {self.save_path}")
        self.get_course_progress()
        with open(self.save_path, 'w') as f:
            json.dump(self.__json__(), f, indent=4)

    def __json__(self):
        return {
            'user_id': self.user_id,
            'select_course': self.select_course,
            'head_icon': self.head_icon,
            'course_process_dict': self.course_process_dict
        }

    def to_json_without_dialog(self):
        course_dict = copy.deepcopy(self.course_process_dict)
        for course_id in course_dict.keys():
            for lep,lid in course_dict[course_id]['lesson_processs']:
                for chapter_info in lep:
                    chapter_info['dialog']=[]
        return json.dumps({
            'user_id': self.user_id,
            'select_course': self.select_course,
            'head_icon': self.head_icon,
            'course_process_dict': course_dict
        }, indent=1)

    def to_json(self):
        return json.dumps(self.__json__(), indent=4)

def load_user_from_json(user_id, user_data_dir = None)-> User:
    USER_DATA_DIR = user_data_dir if user_data_dir else LOCAL_USER_DATA_DIR
    with open(os.path.join(USER_DATA_DIR, f'{user_id}.json'), "r") as f:
        raw_json = f.read()
        user_data = json.loads(raw_json)
        return User(save_path = os.path.join(USER_DATA_DIR, f'{user_id}.json'), **user_data)

def create_user_json(user_id, user_data_dir = None):
    USER_DATA_DIR = user_data_dir if user_data_dir else LOCAL_USER_DATA_DIR
    with open(os.path.join(USER_DATA_DIR, f'{user_id}.json'), "w") as f:
        f.write(json.dumps({
            'user_id': user_id,
            'select_course': [],
            'head_icon': '',
            'course_process_dict': {}
        }))

if(__name__ == "__main__"):

    user = load_user_from_json('cake')


    user.get_course_progress()