kvyper.py

#!/usr/bin/env python3.6

import sys
import os
import subprocess
import re

from scripts.vyper_parser import main as parse # string -> string
from scripts.op2byte import encode as op2byte # string list -> bytes

path = os.path.dirname(os.path.realpath(__file__))


def krun(kdir, pgm): # string * string -> string
    try:
        p = subprocess.run(['krun', '-d', os.path.join(path, kdir), '-cPGM=' + pgm, '-pPGM=kast -e', '--debug'],
                           stdout=subprocess.PIPE, stderr=subprocess.PIPE, encoding='utf-8')
    except FileNotFoundError as e:
        print("Error: subprocess.run() ended with FileNotFoundError for file: " + str(e.filename), file=sys.stderr)
        raise e

    if p.returncode == 0:
        return p.stdout
    else:
        raise RuntimeError(p.stderr)


def viper2lll(ast): # string -> string
    out = krun('vyper-lll', ast)

    if "<k> . </k>" in out:
        lll = re.search(r'<lll> (.*) </lll>', out).group(1)
        return lll
    else:
        raise RuntimeError("vyper-lll computation got stuck:\n\n" + out + "\n\n")


def lll2evm(lll): # string -> string list
    out = krun('lll-evm', lll)

    if "<k> . </k>" in out:
        evm = re.compile(r' \) ListItem \( ').sub(' ', re.search(r'<evm> ListItem \( (.*) \) </evm>', out).group(1))
        return evm.split(' ')
    else:
        raise RuntimeError("lll-evm computation got stuck:\n\n" + out + "\n\n")


def compile(code): # string -> bytes
    ast = parse(code)
    lll = viper2lll(ast)
    evm = lll2evm(lll) # list of opcodes
    return op2byte(evm)


if __name__ == '__main__':
    if len(sys.argv) < 2:
        print("no input file")
        sys.exit(1)
    with open(sys.argv[1], "r") as fin:
        code = fin.read()
    print('0x' + compile(code).hex())