diff options
Diffstat (limited to 'Mist/ext_app.go')
-rw-r--r-- | Mist/ext_app.go | 145 |
1 files changed, 145 insertions, 0 deletions
diff --git a/Mist/ext_app.go b/Mist/ext_app.go new file mode 100644 index 000000000..514084c97 --- /dev/null +++ b/Mist/ext_app.go @@ -0,0 +1,145 @@ +package main + +import ( + "encoding/json" + + "github.com/ethereum/eth-go/ethchain" + "github.com/ethereum/eth-go/ethpipe" + "github.com/ethereum/eth-go/ethreact" + "github.com/ethereum/eth-go/ethstate" + "github.com/ethereum/go-ethereum/javascript" + "gopkg.in/qml.v1" +) + +type AppContainer interface { + Create() error + Destroy() + + Window() *qml.Window + Engine() *qml.Engine + + NewBlock(*ethchain.Block) + NewWatcher(chan bool) + Messages(ethstate.Messages, string) + Post(string, int) +} + +type ExtApplication struct { + *ethpipe.JSPipe + eth ethchain.EthManager + + blockChan chan ethreact.Event + messageChan chan ethreact.Event + quitChan chan bool + watcherQuitChan chan bool + + filters map[string]*ethchain.Filter + + container AppContainer + lib *UiLib +} + +func NewExtApplication(container AppContainer, lib *UiLib) *ExtApplication { + app := &ExtApplication{ + ethpipe.NewJSPipe(lib.eth), + lib.eth, + make(chan ethreact.Event, 100), + make(chan ethreact.Event, 100), + make(chan bool), + make(chan bool), + make(map[string]*ethchain.Filter), + container, + lib, + } + + return app +} + +func (app *ExtApplication) run() { + // Set the "eth" api on to the containers context + context := app.container.Engine().Context() + context.SetVar("eth", app) + context.SetVar("ui", app.lib) + + err := app.container.Create() + if err != nil { + logger.Errorln(err) + return + } + + // Call the main loop + go app.mainLoop() + + // Subscribe to events + reactor := app.lib.eth.Reactor() + reactor.Subscribe("newBlock", app.blockChan) + reactor.Subscribe("messages", app.messageChan) + + app.container.NewWatcher(app.watcherQuitChan) + + win := app.container.Window() + win.Show() + win.Wait() + + app.stop() +} + +func (app *ExtApplication) stop() { + // Clean up + reactor := app.lib.eth.Reactor() + reactor.Unsubscribe("newBlock", app.blockChan) + + // Kill the main loop + app.quitChan <- true + app.watcherQuitChan <- true + + close(app.blockChan) + close(app.quitChan) + + app.container.Destroy() +} + +func (app *ExtApplication) mainLoop() { +out: + for { + select { + case <-app.quitChan: + break out + case block := <-app.blockChan: + if block, ok := block.Resource.(*ethchain.Block); ok { + app.container.NewBlock(block) + } + case msg := <-app.messageChan: + if messages, ok := msg.Resource.(ethstate.Messages); ok { + for id, filter := range app.filters { + msgs := filter.FilterMessages(messages) + if len(msgs) > 0 { + app.container.Messages(msgs, id) + } + } + } + } + } + +} + +func (self *ExtApplication) Watch(filterOptions map[string]interface{}, identifier string) { + self.filters[identifier] = ethchain.NewFilterFromMap(filterOptions, self.eth) +} + +func (self *ExtApplication) GetMessages(object map[string]interface{}) string { + filter := ethchain.NewFilterFromMap(object, self.eth) + + messages := filter.Find() + var msgs []javascript.JSMessage + for _, m := range messages { + msgs = append(msgs, javascript.NewJSMessage(m)) + } + + b, err := json.Marshal(msgs) + if err != nil { + return "{\"error\":" + err.Error() + "}" + } + + return string(b) +} |