import os
import requests


def backup_book(api, env, selected_book, backup_extension, backup_extension_fileType):
    base_url = env['base_url']
    backup_dir = env['backup_dir']
    headers = {
        "Authorization": f"Token {env['token_id']}:{env['token_secret']}"
    }

    pages_list = api.get_pages_list()['data']
    chapter_list = api.get_chapters_list()['data']

    print(f"Backup type: {backup_extension}")

    for chapter in chapter_list:
        for page in pages_list:
            if page['book_id'] == selected_book['id']:
                if chapter['id'] == page['chapter_id']:
                    folder = os.path.join(backup_dir, chapter['name'])
                    filename = os.path.join(folder, page['name'] + backup_extension_fileType)
                    print(f"Backing up: {filename}")
                    if not os.path.exists(folder):
                        os.makedirs(folder, exist_ok=True)
                    response = requests.get(
                        f'{base_url}/api/pages/{page["id"]}/export/{backup_extension}',
                        headers=headers
                    )
                    with open(filename, 'wb') as file:
                        file.write(response.content)