Refactored code

1 files changed, 18 insertions, 0 deletions

diff --git a/utils/compile.go b/utils/compile.go
index e5ea50ad4..894fc2d09 100644
--- a/utils/compile.go
+++ b/utils/compile.go
@@ -22,3 +22,21 @@ func Compile(script string) ([]byte, error) {
 
 	return ethutil.Assemble(asm...), nil
 }
+
+func CompileScript(script string) ([]byte, []byte, error) {
+	// Preprocess
+	mainInput, initInput := ethutil.PreProcess(script)
+	// Compile main script
+	mainScript, err := Compile(mainInput)
+	if err != nil {
+		return nil, nil, err
+	}
+
+	// Compile init script
+	initScript, err := Compile(initInput)
+	if err != nil {
+		return nil, nil, err
+	}
+
+	return mainScript, initScript, nil
+}