cointop/cointop/layout.go

package cointop

import (
	"fmt"
	"strings"

	"github.com/jroimartin/gocui"
)

// TODO: break up into small functions

var lastWidth int

// layout sets initial layout
func (ct *Cointop) layout(g *gocui.Gui) error {
	ct.debuglog("layout()")
	maxY := ct.height()
	maxX := ct.ClampedWidth()

	topOffset := 0

	headerHeight := 1
	marketbarHeight := 1
	chartHeight := ct.State.chartHeight
	statusbarHeight := 1

	if ct.State.onlyTable {
		ct.State.hideMarketbar = true
		ct.State.hideChart = true
		ct.State.hideStatusbar = true
	}

	if ct.State.hideMarketbar {
		marketbarHeight = 0
	}

	if ct.State.hideChart {
		chartHeight = 0
	}

	if ct.State.hideStatusbar {
		statusbarHeight = 0
	}

	if !ct.State.hideMarketbar {
		if v, err := g.SetView(ct.Views.Marketbar.Name(), 0, topOffset, maxX, 2); err != nil {
			if err != gocui.ErrUnknownView {
				return err
			}
			ct.Views.Marketbar.SetBacking(v)
			ct.Views.Marketbar.Backing().Frame = false
			ct.colorscheme.SetViewColor(ct.Views.Marketbar.Backing(), "marketbar")
			go func() {
				ct.updateMarketbar()
				_, found := ct.cache.Get(ct.Views.Marketbar.Name())
				if found {
					ct.cache.Delete(ct.Views.Marketbar.Name())
					ct.updateMarketbar()
				}
			}()
		}
	} else {
		if ct.Views.Marketbar.Backing() != nil {
			if err := g.DeleteView(ct.Views.Marketbar.Name()); err != nil {
				return err
			}
			ct.Views.Marketbar.SetBacking(nil)
		}
	}

	topOffset = topOffset + marketbarHeight

	if !ct.State.hideChart {
		if v, err := g.SetView(ct.Views.Chart.Name(), 0, topOffset, maxX, topOffset+chartHeight+marketbarHeight); err != nil {
			if err != gocui.ErrUnknownView {
				return err
			}
			v.Clear()
			ct.Views.Chart.SetBacking(v)
			ct.Views.Chart.Backing().Frame = false
			ct.colorscheme.SetViewColor(ct.Views.Chart.Backing(), "chart")
			go func() {
				ct.UpdateChart()
				cachekey := strings.ToLower(fmt.Sprintf("%s_%s", "globaldata", strings.Replace(ct.State.selectedChartRange, " ", "", -1)))
				_, found := ct.cache.Get(cachekey)
				if found {
					ct.cache.Delete(cachekey)
					ct.UpdateChart()
				}
			}()
		}
	} else {
		if ct.Views.Chart.Backing() != nil {
			if err := g.DeleteView(ct.Views.Chart.Name()); err != nil {
				return err
			}
			ct.Views.Chart.SetBacking(nil)
		}
	}

	topOffset = topOffset + chartHeight
	if v, err := g.SetView(ct.Views.TableHeader.Name(), 0, topOffset, ct.maxTableWidth, topOffset+2); err != nil {
		if err != gocui.ErrUnknownView {
			return err
		}
		ct.Views.TableHeader.SetBacking(v)
		ct.Views.TableHeader.Backing().Frame = false
		ct.colorscheme.SetViewColor(ct.Views.TableHeader.Backing(), "table_header")
		go ct.UpdateTableHeader()
	}

	topOffset = topOffset + headerHeight
	if v, err := g.SetView(ct.Views.Table.Name(), 0, topOffset, ct.maxTableWidth, maxY-statusbarHeight); err != nil {
		if err != gocui.ErrUnknownView {
			return err
		}
		ct.Views.Table.SetBacking(v)
		ct.Views.Table.Backing().Frame = false
		ct.Views.Table.Backing().Highlight = true
		ct.colorscheme.SetViewActiveColor(ct.Views.Table.Backing(), "table_row_active")
		_, found := ct.cache.Get("allCoinsSlugMap")
		if found {
			ct.cache.Delete("allCoinsSlugMap")
		}
		go func() {
			ct.updateCoins()
			ct.UpdateTable()
		}()
	}

	if !ct.State.hideStatusbar {
		if v, err := g.SetView(ct.Views.Statusbar.Name(), 0, maxY-statusbarHeight-1, ct.maxTableWidth, maxY); err != nil {
			if err != gocui.ErrUnknownView {
				return err
			}
			ct.Views.Statusbar.SetBacking(v)
			ct.Views.Statusbar.Backing().Frame = false
			ct.colorscheme.SetViewColor(ct.Views.Statusbar.Backing(), "statusbar")
			go ct.UpdateStatusbar("")
		}
	} else {
		if ct.Views.Statusbar.Backing() != nil {
			if err := g.DeleteView(ct.Views.Statusbar.Name()); err != nil {
				return err
			}
			ct.Views.Statusbar.SetBacking(nil)
		}
	}

	if v, err := g.SetView(ct.Views.SearchField.Name(), 0, maxY-2, ct.maxTableWidth, maxY); err != nil {
		if err != gocui.ErrUnknownView {
			return err
		}
		ct.Views.SearchField.SetBacking(v)
		ct.Views.SearchField.Backing().Editable = true
		ct.Views.SearchField.Backing().Wrap = true
		ct.Views.SearchField.Backing().Frame = false
		ct.colorscheme.SetViewColor(ct.Views.SearchField.Backing(), "searchbar")
	}

	if v, err := g.SetView(ct.Views.Help.Name(), 1, 1, ct.maxTableWidth-1, maxY-1); err != nil {
		if err != gocui.ErrUnknownView {
			return err
		}
		ct.Views.Help.SetBacking(v)
		ct.Views.Help.Backing().Frame = false
		ct.colorscheme.SetViewColor(ct.Views.Help.Backing(), "menu")
	}

	if v, err := g.SetView(ct.Views.PortfolioUpdateMenu.Name(), 1, 1, ct.maxTableWidth-1, maxY-1); err != nil {
		if err != gocui.ErrUnknownView {
			return err
		}
		ct.Views.PortfolioUpdateMenu.SetBacking(v)
		ct.Views.PortfolioUpdateMenu.Backing().Frame = false
		ct.colorscheme.SetViewColor(ct.Views.PortfolioUpdateMenu.Backing(), "menu")
	}

	if v, err := g.SetView(ct.Views.Input.Name(), 3, 6, 30, 8); err != nil {
		if err != gocui.ErrUnknownView {
			return err
		}
		ct.Views.Input.SetBacking(v)
		ct.Views.Input.Backing().Frame = true
		ct.Views.Input.Backing().Editable = true
		ct.Views.Input.Backing().Wrap = true
		ct.colorscheme.SetViewColor(ct.Views.Input.Backing(), "menu")
	}

	if v, err := g.SetView(ct.Views.ConvertMenu.Name(), 1, 1, ct.maxTableWidth-1, maxY-1); err != nil {
		if err != gocui.ErrUnknownView {
			return err
		}
		ct.Views.ConvertMenu.SetBacking(v)
		ct.Views.ConvertMenu.Backing().Frame = false
		ct.colorscheme.SetViewColor(ct.Views.ConvertMenu.Backing(), "menu")

		// run only once on init.
		// this bit of code should be at the bottom
		ct.g = g
		g.SetViewOnBottom(ct.Views.SearchField.Name())         // hide
		g.SetViewOnBottom(ct.Views.Help.Name())                // hide
		g.SetViewOnBottom(ct.Views.ConvertMenu.Name())         // hide
		g.SetViewOnBottom(ct.Views.PortfolioUpdateMenu.Name()) // hide
		g.SetViewOnBottom(ct.Views.Input.Name())               // hide
		ct.SetActiveView(ct.Views.Table.Name())
		ct.intervalFetchData()
	}

	if lastWidth != maxX {
		lastWidth = maxX
		ct.refresh()
	}

	return nil
}