mirror of
				https://github.com/ethereum/solidity
				synced 2023-10-03 13:03:40 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			84 lines
		
	
	
		
			3.1 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			84 lines
		
	
	
		
			3.1 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
import base64
 | 
						|
import docutils  # pragma pylint: disable=import-error
 | 
						|
 | 
						|
from sphinx.util import logging  # pragma pylint: disable=import-error
 | 
						|
 | 
						|
# NOTE: 2000 should generally be safe for all browsers, while 8000 for most of them.
 | 
						|
MAX_SAFE_URL_LENGTH = 10000
 | 
						|
 | 
						|
logger = logging.getLogger(__name__)
 | 
						|
 | 
						|
 | 
						|
def insert_node_before(child, new_sibling):
 | 
						|
    assert child in child.parent.children
 | 
						|
 | 
						|
    for position, node in enumerate(child.parent.children):
 | 
						|
        if node == child:
 | 
						|
            child.parent.insert(position, new_sibling)
 | 
						|
            break
 | 
						|
 | 
						|
 | 
						|
def remix_code_url(source_code, language, solidity_version):
 | 
						|
    # NOTE: base64 encoded data may contain +, = and / characters. Remix seems to handle them just
 | 
						|
    # fine without any escaping.
 | 
						|
    base64_encoded_source = base64.b64encode(source_code.encode('utf-8')).decode('ascii')
 | 
						|
    return f"https://remix.ethereum.org/?language={language}&version={solidity_version}&code={base64_encoded_source}"
 | 
						|
 | 
						|
 | 
						|
def build_remix_link_node(url):
 | 
						|
    link_icon_node = docutils.nodes.inline()
 | 
						|
    link_icon_node.set_class('link-icon')
 | 
						|
 | 
						|
    link_text_node = docutils.nodes.inline(text="open in Remix")
 | 
						|
    link_text_node.set_class('link-text')
 | 
						|
 | 
						|
    reference_node = docutils.nodes.reference('', '', internal=False, refuri=url)
 | 
						|
    reference_node.set_class('remix-link')
 | 
						|
    reference_node += [link_icon_node, link_text_node]
 | 
						|
 | 
						|
    paragraph_node = docutils.nodes.paragraph()
 | 
						|
    paragraph_node.set_class('remix-link-container')
 | 
						|
    paragraph_node += reference_node
 | 
						|
    return paragraph_node
 | 
						|
 | 
						|
 | 
						|
def insert_remix_link(app, doctree, solidity_version):
 | 
						|
    if app.builder.format != 'html' or app.builder.name == 'epub':
 | 
						|
        return
 | 
						|
 | 
						|
    for literal_block_node in doctree.traverse(docutils.nodes.literal_block):
 | 
						|
        assert 'language' in literal_block_node.attributes
 | 
						|
        language = literal_block_node.attributes['language'].lower()
 | 
						|
        if language in ['solidity', 'yul']:
 | 
						|
            text_nodes = list(literal_block_node.traverse(docutils.nodes.Text))
 | 
						|
            assert len(text_nodes) == 1
 | 
						|
 | 
						|
            remix_url = remix_code_url(text_nodes[0], language, solidity_version)
 | 
						|
            url_length = len(remix_url.encode('utf-8'))
 | 
						|
            if url_length > MAX_SAFE_URL_LENGTH:
 | 
						|
                logger.warning(
 | 
						|
                    "Remix URL generated from the code snippet exceeds the maximum safe URL length "
 | 
						|
                    " (%d > %d bytes).",
 | 
						|
                    url_length,
 | 
						|
                    MAX_SAFE_URL_LENGTH,
 | 
						|
                    location=(literal_block_node.source, literal_block_node.line),
 | 
						|
                )
 | 
						|
 | 
						|
            insert_node_before(literal_block_node, build_remix_link_node(remix_url))
 | 
						|
 | 
						|
 | 
						|
def setup(app):
 | 
						|
    # NOTE: Need to access _raw_config here because setup() runs before app.config is ready.
 | 
						|
    solidity_version = app.config._raw_config['version']  # pylint: disable=protected-access
 | 
						|
 | 
						|
    app.connect(
 | 
						|
        'doctree-resolved',
 | 
						|
        lambda app, doctree, docname: insert_remix_link(app, doctree, solidity_version)
 | 
						|
    )
 | 
						|
 | 
						|
    return {
 | 
						|
        'version': solidity_version,
 | 
						|
        'parallel_read_safe': True,
 | 
						|
        'parallel_write_safe': True,
 | 
						|
    }
 |