bgneal@285: """
bgneal@285: Views for the oembed application.
bgneal@285: """
bgneal@285: import re
bgneal@285: import urllib2
bgneal@285: 
bgneal@285: from django.http import HttpResponse
bgneal@285: from django.http import HttpResponseBadRequest
bgneal@285: from django.http import HttpResponseForbidden
bgneal@285: import django.utils.simplejson as json
bgneal@285: 
bgneal@285: from oembed.models import Provider
bgneal@285: from oembed.models import Oembed
bgneal@285: from oembed.core import get_oembed
bgneal@285: 
bgneal@285: 
bgneal@285: def fetch_media(request):
bgneal@285:     """
bgneal@285:     This view returns the HTML media of an embeddable resource.
bgneal@285:     This view is the target of an AJAX request. 
bgneal@285:     """
bgneal@285:     if not request.user.is_authenticated():
bgneal@285:         return HttpResponseForbidden('Please login or register.')
bgneal@285: 
bgneal@285:     url = request.POST.get('q')
bgneal@285: 
bgneal@285:     if not url:
bgneal@285:         return HttpResponseBadRequest('Please provide a valid URL.')
bgneal@285: 
bgneal@285:     # Is this already in our database?
bgneal@285:     try:
bgneal@285:         oembed = Oembed.objects.get(url=url)
bgneal@285:     except Oembed.DoesNotExist:
bgneal@285:         pass
bgneal@285:     else:
bgneal@285:         data = dict(id=oembed.id, embed=oembed.html)
bgneal@285:         return HttpResponse(json.dumps(data), content_type='application/json')
bgneal@285: 
bgneal@285:     # It isn't in the database, try to find it from our providers
bgneal@285:     providers = Provider.objects.all()
bgneal@285:     for provider in providers:
bgneal@285:         if re.match(provider.url_regex, url):
bgneal@285:             try:
bgneal@285:                 data = get_oembed(provider.api_endpoint, url)
bgneal@285:             except IOError, e:
bgneal@285:                 return HttpResponseBadRequest(
bgneal@285:                     "Sorry, we could not retrieve your video (%s)" % e)
bgneal@285: 
bgneal@285:             if 'type' not in data or data['type'] != 'video':
bgneal@285:                 return HttpResponseBadRequest(
bgneal@285:                     "Hey, this doesn't look like a video..??")
bgneal@285: 
bgneal@285:             oembed = Oembed(url=url,
bgneal@285:                     type=Oembed.VIDEO,
bgneal@285:                     title=data.get('title', ''),
bgneal@285:                     width=int(data.get('width', 0)),
bgneal@285:                     height=int(data.get('height', 0)),
bgneal@285:                     html=data.get('html', ''))
bgneal@285:             oembed.save()
bgneal@285: 
bgneal@285:             data = dict(id=oembed.id, embed=oembed.html)
bgneal@285:             return HttpResponse(json.dumps(data),
bgneal@285:                     content_type='application/json')
bgneal@285: 
bgneal@285:     return HttpBadRequest("Sorry, we couldn't find that video.")