# Copyright 2009-2022 Davide Alberani <da@erlug.linux.it>
#                2018 H. Turgut Uyar <uyar@tekir.org>
#                2022 Kostya Farber
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

"""
This module provides the classes (and the instances) that are used to parse
the pages for the showtimes of movies near you.

Pages:

https://www.imdb.com/showtimes/
"""

import re

from .piculet import Path, Rule, Rules
from .utils import DOMParserBase, analyze_imdbid, build_movie

re_space = re.compile(r'\s+')


class DOMHTMLMoviesNearYou(DOMParserBase):
    rules = [
        Rule(
            key='showtimes',
            extractor=Rules(
                foreach='//div[@itemtype="http://schema.org/MovieTheater"]',
                rules=[
                    Rule(
                        key='cinema',
                        extractor=Path('.//h3/a/text()')
                    ),
                    Rule(
                        key='address and contacts',
                        extractor=Path(
                            './/div[@itemtype="http://schema.org/PostalAddress"]//text()',
                            transform=lambda x: re_space.sub(' ', x).strip()
                        )
                    ),
                    Rule(
                        key='movies',
                        extractor=Rules(
                            foreach='.//div[@class="info"]',
                            rules=[
                                Rule(
                                    key='movie',
                                    extractor=Rules(
                                        rules=[
                                            Rule(
                                                key='link',
                                                extractor=Path(".//h4//a/@href")
                                            ),
                                            Rule(
                                                key='title',
                                                extractor=Path(".//h4//a/text()")
                                            ),
                                        ],
                                        transform=lambda x: build_movie(x.get('title'), analyze_imdbid(x.get('link')))
                                    )
                                ),
                                Rule(
                                    key='showtimes',
                                    extractor=Path(
                                        './/div[@class="showtimes"]//text()',
                                        transform=lambda x: re_space.sub(' ', x).strip()
                                    )
                                )
                            ])
                    )
                ]
            )
        )
    ]

    def postprocess_data(self, data):
        return data.get('showtimes', [])


_OBJECTS = {
    'showtime_parser': ((DOMHTMLMoviesNearYou,), None),
}