from ntfs_utils.db_config import GetNTFSBootInfo, InsertInfoToDBConfig
from ntfs_utils.db_device import ScanSpecialVolumes, InsertVolumesToDB
from ntfs_utils.db_extend_name import InsertExtensionsToDB
from ntfs_utils.db_group import InsertGroupToDB
from ntfs_utils.db_node import InsertNodeDataToDB
from ntfs_utils.db_path import ScanVolume, InsertPathDataToDB
from ntfs_utils.db_user import InsertUserToDB

volume_letter = 'Y'


def main():
    #  初始化 db_config 表
    config_data = GetNTFSBootInfo(volume_letter)
    InsertInfoToDBConfig(config_data)

    # 初始化  db_device 表
    device_data = ScanSpecialVolumes(volume_letter)
    InsertVolumesToDB([device_data])

    # 初始化 db_user 表
    user_list = ["Copier"]
    InsertUserToDB(user_list)

    #  初始化 db_group 表
    group_name_list = ["Copier"]
    InsertGroupToDB(group_name_list)

    # 初始化 db_path 表
    scanned_data = ScanVolume(volume_letter)
    InsertPathDataToDB(scanned_data)

    # 初始化 db_extend_name 表
    common_extensions = [
        "txt", "log", "csv", "xls", "xlsx", "doc", "docx",
        "ppt", "pptx", "pdf", "jpg", "jpeg", "png", "gif",
        "bmp", "mp3", "wav", "mp4", "avi", "mkv", "mov",
        "exe", "dll", "bat", "ini", "reg", "zip", "rar", "7z",
        "json", "xml", "html", "css", "js", "py", "java", "cpp"
    ]
    count = InsertExtensionsToDB(common_extensions)
    print(f"共插入 {count} 个新扩展名。")

    # 初始化 db_node 表
    InsertNodeDataToDB(volume_letter)


if __name__ == '__main__':
    main()