sampler-fork/main.go

151 lines
4.8 KiB
Go
Raw Normal View History

2019-01-28 23:09:52 +00:00
package main
import (
"fmt"
2019-03-14 03:01:44 +00:00
ui "github.com/gizak/termui/v3"
"github.com/sqshq/sampler/asset"
2019-05-24 02:58:46 +00:00
"github.com/sqshq/sampler/client"
2019-02-24 04:49:09 +00:00
"github.com/sqshq/sampler/component"
"github.com/sqshq/sampler/component/asciibox"
"github.com/sqshq/sampler/component/barchart"
2019-02-26 04:36:23 +00:00
"github.com/sqshq/sampler/component/gauge"
2019-03-08 04:04:46 +00:00
"github.com/sqshq/sampler/component/layout"
2019-02-24 04:49:09 +00:00
"github.com/sqshq/sampler/component/runchart"
"github.com/sqshq/sampler/component/sparkline"
2019-04-10 02:30:21 +00:00
"github.com/sqshq/sampler/component/textbox"
2019-02-02 14:45:53 +00:00
"github.com/sqshq/sampler/config"
"github.com/sqshq/sampler/console"
"github.com/sqshq/sampler/data"
"github.com/sqshq/sampler/event"
2019-05-24 02:58:46 +00:00
"github.com/sqshq/sampler/metadata"
"runtime/debug"
"time"
2019-01-28 23:09:52 +00:00
)
2019-03-17 23:55:24 +00:00
type Starter struct {
player *asset.AudioPlayer
lout *layout.Layout
palette console.Palette
opt config.Options
cfg config.Config
}
2019-06-07 05:39:11 +00:00
func (s *Starter) startAll() []*data.Sampler {
samplers := make([]*data.Sampler, 0)
for _, c := range s.cfg.RunCharts {
cpt := runchart.NewRunChart(c, s.palette)
2019-06-07 05:39:11 +00:00
samplers = append(samplers, s.start(cpt, cpt.Consumer, c.ComponentConfig, c.Items, c.Triggers))
}
for _, c := range s.cfg.SparkLines {
cpt := sparkline.NewSparkLine(c, s.palette)
2019-06-07 05:39:11 +00:00
samplers = append(samplers, s.start(cpt, cpt.Consumer, c.ComponentConfig, []config.Item{c.Item}, c.Triggers))
}
for _, c := range s.cfg.BarCharts {
cpt := barchart.NewBarChart(c, s.palette)
2019-06-07 05:39:11 +00:00
samplers = append(samplers, s.start(cpt, cpt.Consumer, c.ComponentConfig, c.Items, c.Triggers))
}
for _, c := range s.cfg.Gauges {
cpt := gauge.NewGauge(c, s.palette)
2019-06-07 05:39:11 +00:00
samplers = append(samplers, s.start(cpt, cpt.Consumer, c.ComponentConfig, []config.Item{c.Cur, c.Min, c.Max}, c.Triggers))
}
for _, c := range s.cfg.AsciiBoxes {
cpt := asciibox.NewAsciiBox(c, s.palette)
2019-06-07 05:39:11 +00:00
samplers = append(samplers, s.start(cpt, cpt.Consumer, c.ComponentConfig, []config.Item{c.Item}, c.Triggers))
}
for _, c := range s.cfg.TextBoxes {
cpt := textbox.NewTextBox(c, s.palette)
2019-06-07 05:39:11 +00:00
samplers = append(samplers, s.start(cpt, cpt.Consumer, c.ComponentConfig, []config.Item{c.Item}, c.Triggers))
}
2019-06-07 05:39:11 +00:00
return samplers
2019-03-17 23:55:24 +00:00
}
2019-06-07 05:39:11 +00:00
func (s *Starter) start(drawable ui.Drawable, consumer *data.Consumer, componentConfig config.ComponentConfig, itemsConfig []config.Item, triggersConfig []config.TriggerConfig) *data.Sampler {
2019-04-07 15:09:24 +00:00
cpt := component.NewComponent(drawable, consumer, componentConfig)
2019-03-21 02:23:08 +00:00
triggers := data.NewTriggers(triggersConfig, consumer, s.opt, s.player)
2019-04-09 02:04:08 +00:00
items := data.NewItems(itemsConfig, *componentConfig.RateMs)
2019-03-17 23:55:24 +00:00
s.lout.AddComponent(cpt)
time.Sleep(10 * time.Millisecond) // desync coroutines
2019-06-07 05:39:11 +00:00
return data.NewSampler(consumer, items, triggers, s.opt, s.cfg.Variables, *componentConfig.RateMs)
2019-03-17 23:55:24 +00:00
}
2019-01-28 23:09:52 +00:00
func main() {
cfg, opt := config.LoadConfig()
bc := client.NewBackendClient()
statistics := metadata.GetStatistics(cfg)
license := metadata.GetLicense()
if opt.LicenseKey != nil {
registerLicense(statistics, opt, bc)
}
2019-01-28 23:09:52 +00:00
console.Init()
defer console.Close()
2019-01-30 00:21:57 +00:00
player := asset.NewAudioPlayer()
if player != nil {
defer player.Close()
}
2019-07-29 00:32:15 +00:00
defer handleCrash(statistics, opt, bc)
defer updateStatistics(cfg, time.Now())
2019-05-24 02:58:46 +00:00
palette := console.GetPalette(*cfg.Theme)
2019-07-30 02:39:17 +00:00
lout := layout.NewLayout(component.NewStatusBar(*opt.ConfigFile, palette, license),
component.NewMenu(palette), component.NewIntro(palette), component.NewNagWindow(palette))
2019-03-17 23:55:24 +00:00
if statistics.LaunchCount == 0 {
if !opt.DisableTelemetry {
go bc.ReportInstallation(statistics)
}
lout.StartWithIntro()
} else if statistics.LaunchCount%20 == 0 { // once in a while
if license == nil || !license.Valid {
lout.StartWithNagWindow()
} else {
go verifyLicense(license, bc)
}
if !opt.DisableTelemetry {
go bc.ReportStatistics(statistics)
}
2019-05-21 03:18:23 +00:00
}
starter := &Starter{player, lout, palette, opt, *cfg}
2019-06-07 05:39:11 +00:00
samplers := starter.startAll()
2019-05-24 02:58:46 +00:00
2019-06-07 05:39:11 +00:00
handler := event.NewHandler(samplers, opt, lout)
2019-05-24 02:58:46 +00:00
handler.HandleEvents()
}
func handleCrash(statistics *metadata.Statistics, opt config.Options, bc *client.BackendClient) {
if rec := recover(); rec != nil {
err := rec.(error)
if !opt.DisableTelemetry {
bc.ReportCrash(fmt.Sprintf("%s\n%s", err.Error(), string(debug.Stack())), statistics)
}
panic(err)
}
}
2019-07-29 00:27:14 +00:00
func updateStatistics(cfg *config.Config, startTime time.Time) {
metadata.PersistStatistics(cfg, time.Since(startTime))
}
func registerLicense(statistics *metadata.Statistics, opt config.Options, bc *client.BackendClient) {
lc, err := bc.RegisterLicenseKey(*opt.LicenseKey, statistics)
if err != nil {
console.Exit("License registration failed: " + err.Error())
} else {
metadata.SaveLicense(*lc)
console.Exit("License successfully verified, Sampler can be restarted without --license flag now. Thank you.")
2019-04-10 02:30:21 +00:00
}
2019-01-28 23:09:52 +00:00
}
func verifyLicense(license *metadata.License, bc *client.BackendClient) {
verifiedLicense, _ := bc.VerifyLicenseKey(*license.Key)
if verifiedLicense != nil {
metadata.SaveLicense(*verifiedLicense)
}
}