You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

122 lines
3.6 KiB

#!/usr/bin/env python3
# Import local files
from whitespace import *
from command import *
from util import *
import builtin_commands
from panflute import *
import re
import sys
from typing import List
from mj_show import show
flags = ["dog"]
commands = {}
def executeCommand(source: Code, element: Element) -> List[Element]:
mode = 'empty'
text = ""
content = []
def print(s: str):
nonlocal mode, text
if mode == 'elements':
raise SyntaxError("Cannot use `print` and `appendChild` in one command at the same time.")
mode = 'text'
text += s
def println(s: str):
print(s+"\n")
def appendChild(e: Element):
nonlocal mode, content
if mode == 'text':
raise SyntaxError("Cannot use `print` and `appendChild` in one command at the same time.")
mode = 'elements'
content.append(e)
def appendChildren(l: List[Element]):
for e in l:
appendChild(e)
exec(source)
if mode == 'text':
return convert_text(text)
if mode == 'elements':
return content
return []
def transform(e: Element, doc: Doc) -> Element: # Returns next sibling element to transform
"""Transform the AST, making format-agnostic changes."""
if isinstance(e, Whitespace) and bavlna(e):
e = NBSP()
if hasattr(e, "attributes"):
# `if` attribute. Only show this element if flag is set.
if "if" in e.attributes:
if not e.attributes["if"] in flags:
return Null()
# `ifn` attribute. Only show this element if flag is NOT set
if "ifn" in e.attributes:
if e.attributes["ifn"] in flags:
return Null()
# `c` attribute. Execute a command with the name saved in this attribute.
if (isinstance(e, Div) or isinstance(e, Span)) and "c" in e.attributes:
if isinstance(e, Div):
e = MultilineCommand(*e.content, identifier=e.identifier, classes=e.classes, attributes=e.attributes)
else:
e = InlineCommand(*e.content, identifier=e.identifier, classes=e.classes, attributes=e.attributes)
print(f"Created inline command {e}")
# Execute python code inside source code block
if isinstance(e, CodeBlock) and hasattr(e, "classes") and "python" in e.classes and "run" in e.classes:
exec(e.text)
return Null()
# Define commands
# TODO: def/longdef?
if isinstance(e, CodeBlock) and hasattr(e, "classes") and "python" in e.classes and hasattr(e, "attributes"):
if "define" in e.attributes:
if not e.attributes["define"] in commands:
commands[e.attributes["define"]] = compile(e.text, '<string>', 'exec')
return Null()
else:
raise NameError(f"Command already defined: '{e.attributes['define']}'")
if "redefine" in e.attributes:
commands[e.attributes["redefine"]] = compile(e.text, '<string>', 'exec')
return Null()
# Handle special command shorthand [!commandname]{}
if isinstance(e, Span) and len(e.content) == 1 and isinstance(e.content[0], Str) and re.match(r"^![\w.]+$", e.content[0].text):
e = InlineCommand(identifier=e.identifier, classes=e.classes, attributes={**e.attributes, "c": e.content[0].text[1:]})
# Execute commands
# Walk transforms the children first, then the root element,
# so the content of the element the command receives is
# already transformed.
# TODO: Transform content that comes out of commands.
if isinstance(e, Command):
if not e.attributes["c"] in commands:
raise NameError(f"Command not defined '{e.attributes['c']}'.")
e = e.replaceSelf(executeCommand(commands[e.attributes["c"]], e))
return e
doc = convert_text(open(sys.argv[1], "r").read(), standalone=True, extra_args=["-f", "markdown+fenced_code_attributes"])
print(show(doc))
doc = doc.walk(transform)
print("---------------------")
#print(show(doc))
print(convert_text(doc, input_format="panflute", output_format="markdown"))