2014-05-02 04:55:43 -07:00
|
|
|
package ethutil
|
|
|
|
|
|
|
|
import (
|
|
|
|
"fmt"
|
|
|
|
"github.com/obscuren/mutan"
|
2014-06-25 08:26:34 -07:00
|
|
|
"github.com/obscuren/mutan/backends"
|
2014-06-20 17:40:25 -07:00
|
|
|
"github.com/obscuren/serpent-go"
|
2014-05-02 04:55:43 -07:00
|
|
|
"strings"
|
|
|
|
)
|
|
|
|
|
|
|
|
// General compile function
|
2014-06-20 17:40:25 -07:00
|
|
|
func Compile(script string) (ret []byte, err error) {
|
2014-07-04 04:05:07 -07:00
|
|
|
if len(script) > 2 {
|
|
|
|
line := strings.Split(script, "\n")[0]
|
2014-06-20 17:40:25 -07:00
|
|
|
|
2014-07-04 04:05:07 -07:00
|
|
|
if line[0:2] == "#!" {
|
|
|
|
switch line {
|
|
|
|
case "#!serpent":
|
|
|
|
byteCode, err := serpent.Compile(script)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
return byteCode, nil
|
|
|
|
}
|
|
|
|
} else {
|
2014-06-20 17:40:25 -07:00
|
|
|
|
2014-07-04 04:05:07 -07: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-20 17:40:25 -07:00
|
|
|
}
|
2014-07-04 04:05:07 -07:00
|
|
|
return nil, fmt.Errorf("%v", errs)
|
2014-05-02 04:55:43 -07:00
|
|
|
}
|
|
|
|
|
2014-07-04 04:05:07 -07:00
|
|
|
return byteCode, nil
|
|
|
|
}
|
2014-06-20 17:40:25 -07:00
|
|
|
}
|
2014-07-04 04:05:07 -07:00
|
|
|
|
|
|
|
return nil, nil
|
2014-05-02 04:55:43 -07:00
|
|
|
}
|