#!/usr/bin/env python3
from tqdm import tqdm
import requests
import shutil
import simplejson.errors
import sys
import os

if len(sys.argv) < 2:
    raise Exception('sys.argv[1]')

def parse_chapter_api(data):
    if not data['server'].startswith('http'):
        data['server'] = 'https://mangadex.org/data/'
    base_url = data['server'] + data['hash'] + '/'
    return list(map(lambda page: base_url + page, data['page_array']))

for chapter in tqdm(sys.argv[1:]):
    chapter_api_response = requests.get(
        f'https://mangadex.org/api/?id={chapter}&type=chapter'
    )

    chapter_number = chapter_api_response.json()['chapter']

    os.makedirs(chapter_number, exist_ok=True)

    try:
        for image in tqdm(parse_chapter_api(chapter_api_response.json())):
            r = requests.get(image, stream=True)
            with open(chapter_number + '/' + image.split('/')[-1], 'wb') as f:
                r.raw.decode_content = True
                shutil.copyfileobj(r.raw, f)
    except simplejson.errors.JSONDecodeError as e:
        if chapter_api_response.status_code != 200:
            raise Exception(f'API request failed with HTTP status code {chapter_api_response.status_code}') from None
        else:
            raise e