# -*- coding: utf-8 -*-

# Copyright 2017 Mike Fährmann
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License version 2 as
# published by the Free Software Foundation.

"""Direct link handling"""

from .common import Extractor, Message
from .. import text


class DirectlinkExtractor(Extractor):
    """Extractor for direct links to images"""
    category = "directlink"
    directory_fmt = []
    filename_fmt = "{filename}"
    pattern = [r"https?://[^?&#]+\.(?:jpe?g|png|gif|webm|mp4|ogg)"]
    test = [(("https://photos.smugmug.com/The-World/Hawaii/"
              "i-SWz2K6n/2/X3/IMG_0311-X3.jpg"), {
        "url": "32ee1045881e17ef3f13a9958595afa42421ec6c",
        "keyword": "5bd7871a6957e1693c43abf31d7c0a38e39a81de",
    })]

    def __init__(self, match):
        Extractor.__init__(self)
        self.url = match.string

    def items(self):
        data = text.nameext_from_url(self.url)
        yield Message.Version, 1
        yield Message.Directory, data
        yield Message.Url, self.url, data