#!/usr/bin/env python3
from PIL import Image
from bs4 import BeautifulSoup
from tqdm import tqdm
import re
import requests
import sys


def parse_filename(filename):
    split = re.split(":|,|\+|>", filename)
    return map(int, split[1:])


def get_image_urls():
    url = sys.argv[1]
    soup = BeautifulSoup(requests.get(url).text, "html.parser")
    for div in soup.find(id="content").find_all("div"):
        yield url + div.get("data-ptimg")


pages = list(get_image_urls())

for metadata_url in tqdm(pages):
    image_url = re.sub("\.ptimg\.json$", ".jpg", metadata_url)

    ptimg_data = requests.get(metadata_url).json()
    image_data = requests.get(image_url, stream=True).raw

    scrambled_image = Image.open(image_data)
    combined_image = Image.new(
        "RGB", (ptimg_data["views"][0]["width"], ptimg_data["views"][0]["height"])
    )

    for from_x, from_y, width, height, to_x, to_y in map(
        parse_filename, ptimg_data["views"][0]["coords"]
    ):
        chunk_data = scrambled_image.crop(
            (from_x, from_y, from_x + width, from_y + height)
        )
        combined_image.paste(chunk_data, (to_x, to_y))

    combined_image.save(image_url.split("/")[-1])