Initial commit

env/lib/python3.10/site-packages/wagtail/embeds/finders/facebook.py

@@ -0,0 +1,103 @@
+import json
+from urllib import request as urllib_request
+from urllib.error import HTTPError, URLError
+from urllib.parse import urlencode
+from urllib.request import Request
+
+from wagtail.embeds.exceptions import EmbedException, EmbedNotFoundException
+
+from .oembed import OEmbedFinder
+
+
+class AccessDeniedFacebookOEmbedException(EmbedException):
+    pass
+
+
+FACEBOOK_PROVIDERS = [
+    # Videos
+    {
+        "endpoint": "https://graph.facebook.com/v11.0/oembed_video",
+        "urls": [
+            r"^https://(?:www\.)?facebook\.com/.+?/videos/.+$",
+            r"^https://(?:www\.)?facebook\.com/video\.php\?(?:v|id)=.+$",
+            r"^https://fb.watch/.+$",
+        ],
+    },
+    # Posts
+    {
+        "endpoint": "https://graph.facebook.com/v11.0/oembed_post",
+        "urls": [
+            r"^https://(?:www\.)?facebook\.com/.+?/(?:posts|activity)/.+$",
+            r"^https://(?:www\.)?facebook\.com/photo\.php\?fbid=.+$",
+            r"^https://(?:www\.)?facebook\.com/(?:photos|questions)/.+$",
+            r"^https://(?:www\.)?facebook\.com/permalink\.php\?story_fbid=.+$",
+            r"^https://(?:www\.)?facebook\.com/media/set/?\?set=.+$",
+            r"^https://(?:www\.)?facebook\.com/notes/.+?/.+?/.+$",
+            # At the moment, not documented on https://developers.facebook.com/docs/plugins/oembed-endpoints
+            # Works for posts with a single photo
+            r"^https://(?:www\.)?facebook\.com/.+?/photos/.+$",
+        ],
+    },
+]
+
+
+class FacebookOEmbedFinder(OEmbedFinder):
+    """
+    An embed finder that supports the authenticated Facebook oEmbed Endpoint.
+    https://developers.facebook.com/docs/plugins/oembed
+    """
+
+    def __init__(self, omitscript=False, app_id=None, app_secret=None):
+        # {settings.facebook_APP_ID}|{settings.facebook_APP_SECRET}
+        self.app_id = app_id
+        self.app_secret = app_secret
+        self.omitscript = omitscript
+
+        super().__init__(providers=FACEBOOK_PROVIDERS)
+
+    def find_embed(self, url, max_width=None, max_height=None):
+        # Find provider
+        endpoint = self._get_endpoint(url)
+        if endpoint is None:
+            raise EmbedNotFoundException
+
+        params = {"url": url, "format": "json"}
+        if max_width:
+            params["maxwidth"] = max_width
+        if max_height:
+            params["maxheight"] = max_height
+        if self.omitscript:
+            params["omitscript"] = "true"
+
+        # Configure request
+        request = Request(endpoint + "?" + urlencode(params))
+        request.add_header("Authorization", f"Bearer {self.app_id}|{self.app_secret}")
+
+        # Perform request
+        try:
+            r = urllib_request.urlopen(request)
+        except (HTTPError, URLError) as e:
+            if isinstance(e, HTTPError) and e.code == 404:
+                raise EmbedNotFoundException
+            elif isinstance(e, HTTPError) and e.code in [400, 401, 403]:
+                raise AccessDeniedFacebookOEmbedException
+            else:
+                raise EmbedNotFoundException
+        oembed = json.loads(r.read().decode("utf-8"))
+
+        # Return embed as a dict
+        return {
+            "title": oembed["title"] if "title" in oembed else "",
+            "author_name": oembed["author_name"] if "author_name" in oembed else "",
+            "provider_name": oembed["provider_name"]
+            if "provider_name" in oembed
+            else "Facebook",
+            "type": oembed["type"],
+            "thumbnail_url": oembed.get("thumbnail_url"),
+            "width": oembed.get("width"),
+            "height": oembed.get("height"),
+            "html": oembed.get("html"),
+        }
+
+
+embed_finder_class = FacebookOEmbedFinder