aboutsummaryrefslogtreecommitdiffstats
path: root/cmd/mist/ext_app.go
blob: 8927a4b256d248d64a3d1a48eb04e58d8fdd0d86 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
package main

import (
    "encoding/json"

    "github.com/ethereum/go-ethereum/ethchain"
    "github.com/ethereum/go-ethereum/ethpipe"
    "github.com/ethereum/go-ethereum/ethstate"
    "github.com/ethereum/go-ethereum/event"
    "github.com/ethereum/go-ethereum/javascript"
    "github.com/ethereum/go-ethereum/ui/qt"
    "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

    events          event.Subscription
    watcherQuitChan chan bool

    filters map[string]*ethchain.Filter

    container AppContainer
    lib       *UiLib
}

func NewExtApplication(container AppContainer, lib *UiLib) *ExtApplication {
    return &ExtApplication{
        JSPipe:          ethpipe.NewJSPipe(lib.eth),
        eth:             lib.eth,
        watcherQuitChan: make(chan bool),
        filters:         make(map[string]*ethchain.Filter),
        container:       container,
        lib:             lib,
    }
}

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
    }

    // Subscribe to events
    mux := app.lib.eth.EventMux()
    app.events = mux.Subscribe(ethchain.NewBlockEvent{}, ethstate.Messages(nil))

    // Call the main loop
    go app.mainLoop()

    app.container.NewWatcher(app.watcherQuitChan)

    win := app.container.Window()
    win.Show()
    win.Wait()

    app.stop()
}

func (app *ExtApplication) stop() {
    app.events.Unsubscribe()

    // Kill the main loop
    app.watcherQuitChan <- true

    app.container.Destroy()
}

func (app *ExtApplication) mainLoop() {
    for ev := range app.events.Chan() {
        switch ev := ev.(type) {
        case ethchain.NewBlockEvent:
            app.container.NewBlock(ev.Block)

        case ethstate.Messages:
            for id, filter := range app.filters {
                msgs := filter.FilterMessages(ev)
                if len(msgs) > 0 {
                    app.container.Messages(msgs, id)
                }
            }
        }
    }
}

func (self *ExtApplication) Watch(filterOptions map[string]interface{}, identifier string) {
    self.filters[identifier] = qt.NewFilterFromMap(filterOptions, self.eth)
}

func (self *ExtApplication) GetMessages(object map[string]interface{}) string {
    filter := qt.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)
}