2014-05-02 11:55:43 +00:00
|
|
|
package ethutil
|
|
|
|
|
|
|
|
import (
|
|
|
|
"fmt"
|
|
|
|
"github.com/obscuren/mutan"
|
2014-06-25 15:26:34 +00:00
|
|
|
"github.com/obscuren/mutan/backends"
|
2014-06-21 00:40:25 +00:00
|
|
|
"github.com/obscuren/serpent-go"
|
2014-05-02 11:55:43 +00:00
|
|
|
"strings"
|
|
|
|
)
|
|
|
|
|
|
|
|
// General compile function
|
2014-06-21 00:40:25 +00:00
|
|
|
func Compile(script string) (ret []byte, err error) {
|
2014-07-04 11:05:07 +00:00
|
|
|
if len(script) > 2 {
|
|
|
|
line := strings.Split(script, "\n")[0]
|
2014-06-21 00:40:25 +00:00
|
|
|
|
2014-07-04 11:34:50 +00:00
|
|
|
if len(line) > 1 && line[0:2] == "#!" {
|
2014-07-04 11:05:07 +00:00
|
|
|
switch line {
|
|
|
|
case "#!serpent":
|
|
|
|
byteCode, err := serpent.Compile(script)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
return byteCode, nil
|
|
|
|
}
|
|
|
|
} else {
|
2014-06-21 00:40:25 +00:00
|
|
|
|
2014-07-04 11:05:07 +00:00
|
|
|
compiler := mutan.NewCompiler(backend.NewEthereumBackend())
|
|
|
|
byteCode, errors := compiler.Compile(strings.NewReader(script))
|
|
|
|
if len(errors) > 0 {
|
|
|
|
var errs string
|
|
|
|
for _, er := range errors {
|
|
|
|
if er != nil {
|
|
|
|
errs += er.Error()
|
|
|
|
}
|
2014-06-21 00:40:25 +00:00
|
|
|
}
|
2014-07-04 11:05:07 +00:00
|
|
|
return nil, fmt.Errorf("%v", errs)
|
2014-05-02 11:55:43 +00:00
|
|
|
}
|
|
|
|
|
2014-07-04 11:05:07 +00:00
|
|
|
return byteCode, nil
|
|
|
|
}
|
2014-06-21 00:40:25 +00:00
|
|
|
}
|
2014-07-04 11:05:07 +00:00
|
|
|
|
|
|
|
return nil, nil
|
2014-05-02 11:55:43 +00:00
|
|
|
}
|