Added pre processing of script data

This commit is contained in:
obscuren
2014-04-12 00:13:42 -04:00
parent 116516158d
commit 086acd122b
3 changed files with 35 additions and 3 deletions

View File

@ -3,6 +3,7 @@ package ethutil
import (
_ "fmt"
"math/big"
"regexp"
)
// Op codes
@ -132,3 +133,33 @@ func Assemble(instructions ...interface{}) (script []byte) {
return
}
/*
Prepocessing function that takes init and main apart:
init() {
// something
}
main() {
// main something
}
*/
func PreProcess(data string) (mainInput, initInput string) {
reg := "\\(\\)\\s*{([\\d\\w\\W\\n\\s]+?)}"
mainReg := regexp.MustCompile("main" + reg)
initReg := regexp.MustCompile("init" + reg)
main := mainReg.FindStringSubmatch(data)
if len(main) > 0 {
mainInput = main[1]
} else {
mainInput = data
}
init := initReg.FindStringSubmatch(data)
if len(init) > 0 {
initInput = init[1]
}
return
}