LinkIPFSPlugin.py

import os
import re
import gevent
import html
import socket

from Plugin import PluginManager
from Config import config
from Debug import Debug

from urllib.request import urlopen, Request
from urllib.parse import quote
from urllib.error import URLError, HTTPError

@PluginManager.registerTo("UiRequest")
class UiRequestPlugin(object):
    def actionWrapper(self, path, extra_headers=None):
        if path.startswith("/ipfs/") or path.startswith("/ipns/"):
        	local_url = "http://127.0.0.1:8080%s" % (quote(path))
        	local_timeout = 3
        	gateway_url = "https://ipfs.io%s" % (quote(path))
        	try:
        		request = Request(local_url)
        		response = urlopen(request, timeout=local_timeout)
        	except (HTTPError) as err:
        		return self.actionRedirect302(local_url)
        	except (URLError) as err:
        		return self.actionRedirect302(gateway_url)
        	except socket.timeout:
        		return self.actionRedirect302(gateway_url)
        	else:
        		return self.actionRedirect302(local_url)
        return super(UiRequestPlugin, self).actionWrapper(path, extra_headers)
    def actionRedirect302(self, url):
        self.start_response('302 Redirect', [('Location', str(url))])
        yield self.formatRedirect302(url)
    def formatRedirect302(self, url):
        return """
            <html>
            <body>
            Redirecting to <a href="{0}" target="_top">{0}</a>
            <script>
            window.top.location = "{0}"
            </script>
            </body>
            </html>
        """.format(html.escape(url))
        