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,
 | |
|     }
 |