//----------------------------------------------------------------------------- func mainScreen(dry *app.Dry, screen *ui.Screen) { if ok, _ := dry.Ok(); !ok { return } keyboardQueue, done := ui.EventChannel() timestampQueue := time.NewTicker(1 * time.Second) viewClosed := make(chan struct{}, 1) keyboardQueueForView := make(chan termbox.Event) dryOutputChan := dry.OuputChannel() statusBar := ui.NewStatusBar(0) defer timestampQueue.Stop() defer close(done) defer close(keyboardQueueForView) defer close(viewClosed) app.Render(dry, screen, statusBar) //belongs outside the loop var viewMode = false go func(viewMode *bool) { for { dryMessage := <-dryOutputChan if !*viewMode { statusBar.StatusMessage(dryMessage, 10*time.Second) if dry.Changed() { screen.Clear() app.Render(dry, screen, statusBar) } else { statusBar.Render() } screen.Flush() } } }(&viewMode) loop: for { //Used for refresh-forcing events happening outside dry var refresh = false select { case <-timestampQueue.C: if !viewMode { timestamp := time.Now().Format(`15:04:05`) screen.RenderLine(0, 0, `<right><white>`+timestamp+`</></right>`) screen.Flush() } case <-viewClosed: viewMode = false dry.ShowContainers() case event := <-keyboardQueue: switch event.Type { case termbox.EventKey: if !viewMode { if event.Key == termbox.KeyEsc || event.Ch == 'q' || event.Ch == 'Q' { break loop } else if event.Key == termbox.KeyArrowUp { //cursor up screen.ScrollCursorUp() refresh = true } else if event.Key == termbox.KeyArrowDown { // cursor down screen.ScrollCursorDown() refresh = true } else if event.Key == termbox.KeyF1 { //sort dry.Sort() } else if event.Key == termbox.KeyF2 { //show all containers dry.ToggleShowAllContainers() } else if event.Key == termbox.KeyF5 { // refresh dry.Refresh() } else if event.Key == termbox.KeyF10 { // docker info dry.ShowInfo() viewMode = true go less(dry, screen, keyboardQueueForView, viewClosed) } else if event.Ch == '?' || event.Ch == 'h' || event.Ch == 'H' { //help viewMode = true dry.ShowHelp() go less(dry, screen, keyboardQueueForView, viewClosed) } else if event.Ch == 'e' || event.Ch == 'E' { //remove dry.Rm(screen.CursorPosition()) } else if event.Key == termbox.KeyCtrlE { //remove all stopped dry.RemoveAllStoppedContainers() } else if event.Ch == 'k' || event.Ch == 'K' { //kill dry.Kill(screen.CursorPosition()) } else if event.Ch == 'l' || event.Ch == 'L' { //logs if logs, err := dry.Logs(screen.CursorPosition()); err == nil { viewMode = true go stream(screen, logs, keyboardQueueForView, viewClosed) } } else if event.Ch == 'r' || event.Ch == 'R' { //start dry.StartContainer(screen.CursorPosition()) } else if event.Ch == 's' || event.Ch == 'S' { //stats done, errC, err := dry.Stats(screen.CursorPosition()) if err == nil { viewMode = true go autorefresh(dry, screen, keyboardQueueForView, viewClosed, done, errC) } } else if event.Ch == 't' || event.Ch == 'T' { //stop dry.StopContainer(screen.CursorPosition()) } else if event.Key == termbox.KeyEnter { //inspect dry.Inspect(screen.CursorPosition()) viewMode = true go less(dry, screen, keyboardQueueForView, viewClosed) } } else if viewMode { //The view handles the event keyboardQueueForView <- event } case termbox.EventResize: screen.Resize() refresh = true } } if !viewMode && (refresh || dry.Changed()) { screen.Clear() app.Render(dry, screen, statusBar) } } log.Debug("something broke the loop") }
//RenderLoop renders dry until it quits func RenderLoop(dry *Dry, screen *ui.Screen) { if ok, _ := dry.Ok(); !ok { return } keyboardQueue, done := ui.EventChannel() timestampQueue := time.NewTicker(1 * time.Second) viewClosed := make(chan struct{}, 1) keyboardQueueForView := make(chan termbox.Event) dryOutputChan := dry.OuputChannel() statusBar := ui.NewStatusBar(0) defer timestampQueue.Stop() defer close(done) defer close(keyboardQueueForView) defer close(viewClosed) Render(dry, screen, statusBar) //focus creation belongs outside the loop focus := &focusTracker{&sync.Mutex{}, true} go func(focus *focusTracker) { for { dryMessage, ok := <-dryOutputChan if ok { if focus.hasFocus() { statusBar.StatusMessage(dryMessage, 10*time.Second) if dry.Changed() { screen.Clear() Render(dry, screen, statusBar) } else { statusBar.Render() } screen.Flush() } } else { return } } }(focus) loop: for { //Used for refresh-forcing events happening outside dry var refresh = false select { case <-timestampQueue.C: if focus.hasFocus() { timestamp := time.Now().Format(`15:04:05`) screen.RenderLine(0, 0, `<right><white>`+timestamp+`</></right>`) screen.Flush() } case <-viewClosed: focus.set(true) dry.ShowMainView() refresh = true case event := <-keyboardQueue: switch event.Type { case termbox.EventKey: if focus.hasFocus() { if event.Key == termbox.KeyEsc || event.Ch == 'q' || event.Ch == 'Q' { break loop } else { handler := eventHandlerFactory(dry, screen, keyboardQueueForView, viewClosed) if handler != nil { r, f := handler.handle(event) refresh = r focus.set(f) } else { log.Panic("There is no event handler") } } } else { //Whoever has the focus, handles the event keyboardQueueForView <- event } case termbox.EventResize: screen.Resize() refresh = true } } if focus.hasFocus() && refresh { screen.Clear() Render(dry, screen, statusBar) } } log.Debug("something broke the loop. Time to die") }