You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
118 lines
3.1 KiB
118 lines
3.1 KiB
// Copyright (c) 2013 by Michael Dvorkin. All Rights Reserved.
|
|
// Use of this source code is governed by a MIT-style license that can
|
|
// be found in the LICENSE file.
|
|
|
|
package main
|
|
|
|
import (
|
|
`github.com/michaeldv/mop`
|
|
`github.com/michaeldv/termbox-go`
|
|
`time`
|
|
)
|
|
|
|
const help = `Mop v0.1.0 -- Copyright (c) 2013 Michael Dvorkin. All Rights Reserved.
|
|
NO WARRANTIES OF ANY KIND WHATSOEVER. SEE THE LICENSE FILE FOR DETAILS.
|
|
|
|
<u>Command</u> <u>Description </u>
|
|
+ Add stocks to the list.
|
|
- Remove stocks from the list.
|
|
o Change column sort order.
|
|
g Group stocks by advancing/declining issues.
|
|
? Display this help screen.
|
|
esc Quit mop.
|
|
|
|
Enter comma-delimited list of stock tickers when prompted.
|
|
|
|
<r> Press any key to continue </r>
|
|
`
|
|
|
|
//-----------------------------------------------------------------------------
|
|
func mainLoop(screen *mop.Screen, profile *mop.Profile) {
|
|
var lineEditor *mop.LineEditor
|
|
var columnEditor *mop.ColumnEditor
|
|
|
|
keyboardQueue := make(chan termbox.Event)
|
|
timestampQueue := time.NewTicker(1 * time.Second)
|
|
quotesQueue := time.NewTicker(5 * time.Second)
|
|
marketQueue := time.NewTicker(12 * time.Second)
|
|
showingHelp := false
|
|
|
|
go func() {
|
|
for {
|
|
keyboardQueue <- termbox.PollEvent()
|
|
}
|
|
}()
|
|
|
|
market := new(mop.Market).Initialize()
|
|
quotes := new(mop.Quotes).Initialize(market, profile)
|
|
screen.Draw(market, quotes)
|
|
|
|
loop:
|
|
for {
|
|
select {
|
|
case event := <-keyboardQueue:
|
|
switch event.Type {
|
|
case termbox.EventKey:
|
|
if lineEditor == nil && columnEditor == nil && !showingHelp {
|
|
if event.Key == termbox.KeyEsc {
|
|
break loop
|
|
} else if event.Ch == '+' || event.Ch == '-' {
|
|
lineEditor = new(mop.LineEditor).Initialize(screen, quotes)
|
|
lineEditor.Prompt(event.Ch)
|
|
} else if event.Ch == 'o' || event.Ch == 'O' {
|
|
columnEditor = new(mop.ColumnEditor).Initialize(screen, quotes)
|
|
} else if event.Ch == 'g' || event.Ch == 'G' {
|
|
if profile.Regroup() == nil {
|
|
screen.Draw(quotes)
|
|
}
|
|
} else if event.Ch == '?' || event.Ch == 'h' || event.Ch == 'H' {
|
|
showingHelp = true
|
|
screen.Clear().Draw(help)
|
|
}
|
|
} else if lineEditor != nil {
|
|
if done := lineEditor.Handle(event); done {
|
|
lineEditor = nil
|
|
}
|
|
} else if columnEditor != nil {
|
|
if done := columnEditor.Handle(event); done {
|
|
columnEditor = nil
|
|
}
|
|
} else if showingHelp {
|
|
showingHelp = false
|
|
screen.Clear().Draw(market, quotes)
|
|
}
|
|
case termbox.EventResize:
|
|
screen.Resize()
|
|
if !showingHelp {
|
|
screen.Draw(market, quotes)
|
|
} else {
|
|
screen.Draw(help)
|
|
}
|
|
}
|
|
|
|
case <-timestampQueue.C:
|
|
if !showingHelp {
|
|
screen.Draw(time.Now())
|
|
}
|
|
|
|
case <-quotesQueue.C:
|
|
if !showingHelp {
|
|
screen.Draw(quotes)
|
|
}
|
|
|
|
case <-marketQueue.C:
|
|
if !showingHelp {
|
|
screen.Draw(market)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
func main() {
|
|
screen := new(mop.Screen).Initialize()
|
|
defer screen.Close()
|
|
|
|
profile := new(mop.Profile).Initialize()
|
|
mainLoop(screen, profile)
|
|
}
|
|
|