Fixed shebang check. Previously it would hang on an unknown shebang

This commit is contained in:
obscuren 2014-07-04 13:05:07 +02:00
parent 633027d980
commit d5bcc01eae

View File

@ -10,16 +10,21 @@ import (
// General compile function
func Compile(script string) (ret []byte, err error) {
c := strings.Split(script, "\n")[0]
if len(script) > 2 {
line := strings.Split(script, "\n")[0]
if c == "#!serpent" {
if line[0:2] == "#!" {
switch line {
case "#!serpent":
byteCode, err := serpent.Compile(script)
if err != nil {
return nil, err
}
return byteCode, nil
}
} else {
compiler := mutan.NewCompiler(backend.NewEthereumBackend())
byteCode, errors := compiler.Compile(strings.NewReader(script))
if len(errors) > 0 {
@ -34,4 +39,7 @@ func Compile(script string) (ret []byte, err error) {
return byteCode, nil
}
}
return nil, nil
}