#!/usr/bin/python

import logging
import itertools
import datetime
import re
from . import register, Site, Section, Chapter

logger = logging.getLogger(__name__)


@register
class FictionLive(Site):
    """fiction.live: it's... mostly smut, I think? Terrible smut. But, hey, I had a rec to follow."""
    @staticmethod
    def matches(url):
        # e.g. https://fiction.live/stories/Descendant-of-a-Demon-Lord/SBBA49fQavNQMWxFT
        match = re.match(r'^(https?://fiction\.live/(?:stories|Sci-fi)/[^\/]+/[0-9a-zA-Z\-]+)/?.*', url)
        if match:
            return match.group(1)

    def extract(self, url):
        workid = re.match(r'^https?://fiction\.live/(?:stories|Sci-fi)/[^\/]+/([0-9a-zA-Z\-]+)/?.*', url).group(1)

        response = self.session.get(f'https://fiction.live/api/node/{workid}').json()

        story = Section(
            title=response['t'],
            author=response['u'][0]['n'],
            # Could normalize the URL here from the returns, but I'd have to
            # go look up how they handle special characters in titles...
            url=url
        )
        # There's a summary (or similar) in `d` and `b`, if I want to use that later.

        # TODO: extract these #special ones and send them off to an endnotes section?
        chapters = ({'ct': 0},) + tuple(c for c in response['bm'] if not c['title'].startswith('#special')) + ({'ct': 9999999999999999},)

        for prevc, currc, nextc in contextiterate(chapters):
            # `id`, `title`, `ct`, `isFirst`
            # https://fiction.live/api/anonkun/chapters/SBBA49fQavNQMWxFT/0/1448245168594
            # https://fiction.live/api/anonkun/chapters/SBBA49fQavNQMWxFT/1449266444062/1449615394752
            # https://fiction.live/api/anonkun/chapters/SBBA49fQavNQMWxFT/1502823848216/9999999999999998
            # i.e. format is [current timestamp] / [next timestamp - 1]
            chapter_url = f'https://fiction.live/api/anonkun/chapters/{workid}/{currc["ct"]}/{nextc["ct"] - 1}'
            logger.info("Extracting chapter \"%s\" @ %s", currc['title'], chapter_url)
            data = self.session.get(chapter_url).json()
            html = []

            updated = currc['ct']
            for segment in (d for d in data if not d.get('t', '').startswith('#special')):
                updated = max(updated, segment['ct'])
                # TODO: work out if this is actually enough types handled
                # There's at least also a reader post type, which mostly seems to be used for die rolls.
                try:
                    if segment['nt'] == 'chapter':
                        html.extend(('<div>', segment['b'].replace('<br>', '<br/>'), '</div>'))
                    elif segment['nt'] == 'choice':
                        if 'votes' not in segment:
                            # Somehow, sometime, we end up with a choice without votes (or choices)
                            continue
                        votes = {}
                        for vote in segment['votes']:
                            votechoices = segment['votes'][vote]
                            if isinstance(votechoices, str):
                                # This caused issue #30, where for some reason one
                                # choice on a story was a string rather than an
                                # index into the choices array.
                                continue
                            if isinstance(votechoices, int):
                                votechoices = (votechoices,)
                            for choice in votechoices:
                                if int(choice) < len(segment['choices']):
                                    # sometimes someone has voted for a presumably-deleted choice
                                    choice = segment['choices'][int(choice)]
                                    votes[choice] = votes.get(choice, 0) + 1
                        choices = [(votes[v], v) for v in votes]
                        choices.sort(reverse=True)
                        html.append('<hr/><ul>')
                        for votecount, choice in choices:
                            html.append(f'<li>{choice}: {votecount}</li>')
                        html.append('</ul><hr/>')
                    elif segment['nt'] == 'readerPost':
                        pass
                    else:
                        logger.info("Skipped chapter-segment of unhandled type: %s", segment['nt'])
                except Exception as e:
                    logger.error("Skipped chapter-segment due to parsing error", exc_info=e)

            story.add(Chapter(
                title=currc['title'],
                contents='\n'.join(html),
                date=datetime.datetime.fromtimestamp(updated / 1000.0)
            ))

        self._finalize(story)

        return story


# Stolen from the itertools docs
def contextiterate(iterable):
    "s -> (s0,s1), (s1,s2), (s2, s3), ..."
    a, b, c = itertools.tee(iterable, 3)
    next(b, None)
    next(c, None)
    next(c, None)
    return zip(a, b, c)