mirror of https://github.com/t1meshift/js.git
119 lines
3.2 KiB
Python
119 lines
3.2 KiB
Python
"""Pylint tells me this module should have a docstring.
|
|
So here it is.
|
|
"""
|
|
import sys
|
|
import argparse
|
|
import logging
|
|
import colorama
|
|
import coloredlogs
|
|
|
|
from jasminesnake import __version__, __snake__, LOG_LEVELS
|
|
from .js_stream import JSBaseStream, JSStringStream, JSFileStream
|
|
from .lex.ErrorListeners import LogErrorListener
|
|
from .ast import to_ascii_tree, from_parse_tree
|
|
|
|
|
|
def create_argument_parser():
|
|
_arg_parser = argparse.ArgumentParser(
|
|
description="Jasmine Snake, another JS interpreter in Python",
|
|
epilog="I hope you don't use it, **especially** in production.",
|
|
)
|
|
|
|
_arg_parser.add_argument("--snake", action="store_true", help="print a snake")
|
|
_arg_parser.add_argument(
|
|
"--ast", choices=["full", "short", "none"], default="none", help="print AST"
|
|
)
|
|
_arg_parser.add_argument(
|
|
"--verbose",
|
|
"-v",
|
|
action="count",
|
|
default=0,
|
|
help="be more verbose. up to 4 (-vvvv) could be handled, more are ignored",
|
|
)
|
|
_arg_parser.add_argument(
|
|
"infile",
|
|
type=str,
|
|
help='JS input file. use "-" to read input from stdin.',
|
|
nargs="?",
|
|
)
|
|
|
|
return _arg_parser
|
|
|
|
|
|
def main():
|
|
# Init colorama
|
|
colorama.init()
|
|
|
|
# Init logging
|
|
log_level = min(args.verbose, 4) # Ignore verbosity values more than 4
|
|
coloredlogs.install(
|
|
level=LOG_LEVELS[log_level]["level"], fmt=LOG_LEVELS[log_level]["format"]
|
|
)
|
|
|
|
# Print the snake if an argument is present
|
|
if args.snake:
|
|
print(colorama.Style.DIM + __snake__ + colorama.Style.RESET_ALL)
|
|
print(
|
|
colorama.Fore.BLACK
|
|
+ colorama.Back.YELLOW
|
|
+ "Don't tread on me!"
|
|
+ colorama.Back.RESET
|
|
+ colorama.Fore.RESET
|
|
)
|
|
|
|
# Read JS code from file or stdin
|
|
if args.infile is not None:
|
|
stream: JSBaseStream
|
|
|
|
if args.infile == "-":
|
|
input_str = sys.stdin.read()
|
|
stream = JSStringStream(input_str)
|
|
|
|
else:
|
|
stream = JSFileStream(args.infile, LogErrorListener())
|
|
|
|
tree = stream.parse()
|
|
|
|
ast_tree = from_parse_tree(tree)
|
|
|
|
logging.info("Got an AST!\n")
|
|
if args.ast != "none":
|
|
ascii_ast = to_ascii_tree(ast_tree, ast_format=args.ast)
|
|
print(ascii_ast)
|
|
|
|
# TODO: run logic
|
|
sys.exit(0)
|
|
|
|
print("Jasmine Snake v{version}".format(version=__version__))
|
|
print(
|
|
colorama.Fore.YELLOW
|
|
+ "Notice that only single-line statements are supported."
|
|
+ colorama.Fore.RESET
|
|
)
|
|
print()
|
|
|
|
try:
|
|
while True:
|
|
input_str = input("> ")
|
|
logging.debug("Got input %s", input_str)
|
|
|
|
stream = JSStringStream(input_str, LogErrorListener())
|
|
tree = stream.parse()
|
|
logging.debug("Got tree %s", tree.toStringTree(stream.parser.ruleNames))
|
|
|
|
ast_tree = from_parse_tree(tree)
|
|
ascii_ast = to_ascii_tree(ast_tree)
|
|
|
|
logging.info("Got an AST!")
|
|
logging.info(ascii_ast)
|
|
# TODO: run logic
|
|
except EOFError:
|
|
print("Ctrl-D received, shutting down...")
|
|
sys.exit(0)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
arg_parser = create_argument_parser()
|
|
args = arg_parser.parse_args()
|
|
main()
|