-
Notifications
You must be signed in to change notification settings - Fork 1
/
main.go
76 lines (60 loc) · 1.51 KB
/
main.go
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
package main
import (
"fmt"
"os"
"strconv"
"time"
"github.com/TheSp1der/goerror"
"golang.org/x/crypto/ssh/terminal"
)
func main() {
sData := make(chan iex)
// get current prices
go updateStockData(sData)
// output to console
if !cmdLnNoConsole {
go outputConsole(sData)
}
// output to webserver
if cmdLnHTTPPort > 0 && cmdLnHTTPPort < 65535 {
go webListener(sData, cmdLnHTTPPort)
}
// end of day e-mail
if cmdLnEmailAddress != "" && cmdLnEmailFrom != "" && cmdLnEmailHost != "" {
go notifyViaMail(sData)
}
// if everything is fine, loop indefinitely
if !cmdLnNoConsole || (cmdLnHTTPPort > 0 && cmdLnHTTPPort < 65535) || (cmdLnEmailAddress != "" && cmdLnEmailFrom != "" && cmdLnEmailHost != "") {
for {
time.Sleep(time.Duration(time.Second * 5))
}
}
}
func notifyViaMail(sData chan iex) {
for {
open, sleepTime := marketStatus()
if !open {
time.Sleep(time.Duration(time.Minute * 5))
if err := basicMailSend(cmdLnEmailHost+":"+strconv.Itoa(cmdLnEmailPort), cmdLnEmailAddress, cmdLnEmailFrom, "Stock Alert", displayHTML(<-sData)); err != nil {
goerror.Warning(err)
}
}
time.Sleep(time.Duration(sleepTime))
}
}
func outputConsole(sData chan iex) {
var hData iex
for {
cData := <-sData
// set cursor to top left position
if terminal.IsTerminal(int(os.Stdout.Fd())) {
fmt.Printf("\033[0;0H")
}
// display output
fmt.Print(displayTerminal(hData, cData))
// update historical data
hData = cData
// sleep
time.Sleep(time.Duration(time.Second * 5))
}
}