import os
import subprocess
import tempfile
from io import BytesIO
from uuid import uuid4

import requests


def get_vk_audio(m3u8):
    with tempfile.TemporaryDirectory() as d:
        fname = '{}.mp3'.format(uuid4())
        path = os.path.join(d, fname)
        subprocess.check_output(['ffmpeg', '-i', m3u8, path])
        with open(path, 'rb') as f:
            tf = BytesIO(f.read())
            tf.name = fname
            tf.seek(0)
            return tf


def get_vk_photo(attachment):
    for size in (2560, 1280, 807, 604, 130, 75):
        if f'photo_{size}' in attachment:
            return attachment[f'photo_{size}']
    return None


def get_file(url):
    fname = '?'.join(url.split('?')[:-1])
    extension = os.path.basename(fname).split('.')[-1]
    f = tempfile.NamedTemporaryFile(suffix=f'.{extension}' if extension else None)
    r = requests.get(url, stream=True)
    for chunk in r.iter_content(1024 * 1024):
        if chunk:
            f.write(chunk)
    f.seek(0)
    return f