from app import app, db
from app.sound_models import YouTubeAudio, SoundcloudAudio
from flask import Response, send_file, redirect, url_for, render_template, request
from multiprocessing import Value
from mutagen.mp3 import MP3
import os
import re
import json
import subprocess

# Retrieves the YouTubeAudio object relevant to the mediaid if available. If not, it facilitates the creation and writing of one.
# Also helps with access times.
def get_youtube(mediaid):
    audio = YouTubeAudio.query.get(mediaid)
    if audio is not None:
        audio.access()
        return audio # sets the access time to now
    audio = YouTubeAudio(id=mediaid)
    audio.fill_metadata()
    audio.download()
    # Commit and save new audio object into the database
    db.session.add(audio)
    db.session.commit()
    return audio

def downloadLimiter():
    service = request.args.get('service')
    mediaid = request.args.get('mediaid')
    if service == 'youtube':
        if YouTubeAudio.query.get(mediaid) is not None:
            return '5 per minute'
        else:
            return '1 per 30 seconds'
    else:
        return '10 per minute'

# Streams back the specified media back to the client 
@app.route('/stream/<service>/<mediaid>')
@limiter.limit(downloadLimiter)
def stream(service, mediaid):
    if service == 'youtube':
        audio = get_youtube(mediaid)
        return send_file(audio.getPath(alt=True), attachment_filename=audio.filename)
    elif service == 'soundcloud':
        return Response('Not implemented', status=501, mimetype='text/plain')
    elif service == 'spotify':
        return Response('Not implemented', status=501, mimetype='text/plain')
    else:
        return Response('Bad request', status=400, mimetype='text/plain')

# Returns the duration of a specific media
@app.route('/duration/<service>/<mediaid>')
def duration(service, mediaid):
    if service == 'youtube':
        duration = get_youtube(mediaid).duration
        return Response(str(duration), status=200, mimetype='text/plain')
    elif service == 'soundcloud':
        return Response('Not implemented', status=501, mimetype='text/plain')
    elif service == 'spotify':
        return Response('Not implemented', status=501, mimetype='text/plain')
    else:
        return Response('Bad request', status=400, mimetype='text/plain')

# Returns a detailed JSON export of a specific database entry.
# Will not create a new database entry where one didn't exist before.
@app.route('/status/<service>/<mediaid>')
def status(service, mediaid):
    if service == 'youtube':
        audio = YouTubeAudio.query.get(mediaid)
        if audio is None:
            if YouTubeAudio.isValid(mediaid):
                return Response('Media not yet downloaded', status=400, mimetype='text/plain')
            else:
                return Response('Invalid ID', status=400, mimetype='text/plain')
        else:
            return Response(audio.toJSON(), status=200, mimetype='application/json')
    elif service == 'soundcloud':
        return Response('Not implemented', status=501, mimetype='text/plain')
    elif service == 'spotify':
        return Response('Not implemented', status=501, mimetype='text/plain')
    else:
        return Response('Bad request', status=400, mimetype='text/plain')