# Copyright (c) 2018, George Tokmaji

# Permission to use, copy, modify, and/or distribute this software for any
# purpose with or without fee is hereby granted, provided that the above
# copyright notice and this permission notice appear in all copies.

# THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
# WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
# MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
# ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
# WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
# ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
# OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.

from ..helpers import *
from datetime import datetime

@route("/api/media")
def get_media():
    notAllowed()


@route("/api/media/<id>")
def get_media_id(id):
    session = DBSession()
    try:
        file = session.query(File).filter_by(id=id).one()
    except db.orm.exc.NoResultFound:
        raise HTTPResponse(status=404)

    response.set_header("Content-Type", file.content_type)
    response.set_header("Content-Length", file.length)
    response.set_header("Date", file.date.strftime("%a, %d %b %Y %H:%M:%S GMT"))
    response.set_header("Content-Disposition", f"attachment; filename=\"{file.name}\"")
    if request.method == "GET":
        if file.download_url:
            #return requests.request(request.method, file.download_url, allow_redirects=True)
            return HTTPResponse(status=302, headers={"Location" : file.download_url})
        else:
            return static_file(file.id, os.path.join(os.getcwd(), "media"), file.content_type, file.name)
    
    raise HTTPResponse(status=404)