#Original script by drdaxxy
# https://gist.github.com/drdaxxy/1e43b3aee3e08a5898f61a45b96e4cb4

# Thanks to ToshiroScan for fixing after a shonen update broke this.
# And QPDEH for helping fix the black bars issue with some manga.
# This script has been tested to work with python 3.10

# To install required libraries: 
#
# pip install requests
# pip install Pillow

import sys
import os
import requests
import errno
import json
import hashlib
from PIL import Image

login = False # Set this to True if you want to login
username = "your email here"
password = "your password"


loginheaders = {
    'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:130.0) Gecko/20100101 Firefox/130.0',
    'origin': 'https://pocket.shonenmagazine.com',
    'x-requested-with': 'XMLHttpRequest'
}

loginurl = "https://pocket.shonenmagazine.com/user_account/login"

sess = requests.Session()

if login:

    logindata = {"email_address": username, "password": password, "return_location_path" : "/"}

    r = sess.post(loginurl, headers=loginheaders, data = logindata)

    if r.ok:

        print('LOGIN SUCCESS')
        print(sess.cookies)
    else:
        print('LOGIN FAILED')
        print(r.headers)
        print(r.status_code)
        print(r.reason)
        print(r.text)



if len(sys.argv) != 3:
    print("usage: shonenripperjson.py <episode id> <destination folder>")
    sys.exit(1)


destination = sys.argv[2]

if not os.path.exists(destination):
    try:
        os.makedirs(destination)
    except OSError as exc:
        if exc.errno != errno.EEXIST:
            raise

episode_id = sys.argv[1]

def mangahash(episode_id):
    e = {"platform": "3", "episode_id": episode_id}
    t = sorted(list(e.keys()))
    def Ed(e, t): return dc(e) + "_" + _h(t)
    def dc(a): return hashlib.sha256(a.encode()).hexdigest()
    def _h(a): return hashlib.sha512(a.encode()).hexdigest()
    r = []
    for u in t: r.append(Ed(u, e[u]))
    n = dc(",".join(r))
    return _h(n + Ed("", ""))

headers = {
    'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:130.0) Gecko/20100101 Firefox/130.0'}
apiheaders = {
    'x-manga-is-crawler': "false",
    'x-manga-hash': mangahash(episode_id)  # w/o this header site won't return links to images
}

print("Getting from episode: " + episode_id)
r = sess.get(url="https://api.pocket.shonenmagazine.com/web/episode/viewer?platform=3&episode_id=" + episode_id, headers={**headers, **apiheaders})
data = r.json()

#with open("resp.json", "w+", encoding="utf-8") as file_:file_.write(r.text)

if not r.ok:
    print("Invalid request")
    print(r)
    sys.exit(1)


def dlImage(url, outFilename, drm, drmkey):
    r = sess.get(url, stream=True, headers=headers)

    if not r.ok:
        print(r)
        return

    content_type = r.headers.get('content-type')
    if content_type == "image/jpeg":
        outFilename = outFilename + ".jpg"
    elif content_type == "image/png":
        outFilename = outFilename + ".png"
    else:
        print("content type not recognized!")
        print(r)
        return

    with open(outFilename, 'wb') as file:
        for block in r.iter_content(1024):
            if not block:
                break
            file.write(block)

    if drm == True:
        source = Image.open(outFilename)
        dest = source.copy()
        
        def draw_subimage(sx, sy, sWidth, sHeight, dx, dy):
            rect = source.crop((sx, sy, sx+sWidth, sy+sHeight))
            dest.paste(rect, (dx, dy, dx+sWidth, dy+sHeight))

        def ve(num):
            def rshift(val, n): return (val % 0x100000000) >> n
            e = num
            while True:
                e ^= (e << 13)
                e ^= rshift(e, 17)
                e ^= (e << 5)
                e %= 2**32
                yield e

        DIVIDE_NUM = 4
        MULTIPLE = 8
        cell_width = (source.width // (DIVIDE_NUM * MULTIPLE)) * MULTIPLE
        cell_height = (source.height // (DIVIDE_NUM * MULTIPLE)) * MULTIPLE
        s = ve(drmkey)
        for e in enumerate(map(lambda x:x[1], sorted(map(lambda x:(next(s), x), range(DIVIDE_NUM*DIVIDE_NUM))))):
            t = e[1] // DIVIDE_NUM * cell_height
            n = e[1] % DIVIDE_NUM * cell_width
            s = e[0] % DIVIDE_NUM * cell_width
            c = (e[0] // DIVIDE_NUM) * cell_height
            draw_subimage(n, t, cell_width, cell_height, s, c)

        dest.save(outFilename)

nextEpisodeId = None

if 'next_episode' in data and 'episode_id' in data['next_episode']:
    nextEpisodeId = data['next_episode']['episode_id']

pages = data['page_list'] if 'page_list' in data else []

if len(pages) == 0:
    print("No pages found")
    sys.exit(1)

if data.get('scramble_seed') == None:
    print('Could not find scramble seed.')
    sys.exit(1)

#choJuGiga = data['choJuGiga'] if 'choJuGiga' in data else ''
#print('choJuGiga: ', choJuGiga)
#drm = choJuGiga != "usagi"
drm = True
drmkey = data['scramble_seed']

pageIndex = 0

for page in pages:
    src = page
    print(src)
    pageIndex += 1
    outFile = os.path.join(destination, f"{pageIndex:04d}")
    dlImage(src, outFile, drm, drmkey)

if nextEpisodeId != None:
    print("Next episode:", nextEpisodeId)