internal/display/tabwindow.go

   1 package display
   2
   3 import (
   4         "unicode/utf8"
   5
   6         runewidth "github.com/mattn/go-runewidth"
   7         "github.com/zyedidia/micro/internal/buffer"
   8         "github.com/zyedidia/micro/internal/config"
   9         "github.com/zyedidia/micro/internal/screen"
  10         "github.com/zyedidia/micro/internal/util"
  11 )
  12
  13 type TabWindow struct {
  14         Names   []string
  15         active  int
  16         Y       int
  17         width   int
  18         hscroll int
  19 }
  20
  21 func NewTabWindow(w int, y int) *TabWindow {
  22         tw := new(TabWindow)
  23         tw.width = w
  24         tw.Y = y
  25         return tw
  26 }
  27
  28 func (w *TabWindow) Resize(width, height int) {
  29         w.width = width
  30 }
  31
  32 func (w *TabWindow) LocFromVisual(vloc buffer.Loc) int {
  33         x := -w.hscroll
  34
  35         for i, n := range w.Names {
  36                 x++
  37                 s := utf8.RuneCountInString(n)
  38                 if vloc.Y == w.Y && vloc.X < x+s {
  39                         return i
  40                 }
  41                 x += s
  42                 x += 3
  43                 if x >= w.width {
  44                         break
  45                 }
  46         }
  47         return -1
  48 }
  49
  50 func (w *TabWindow) Scroll(amt int) {
  51         w.hscroll += amt
  52         s := w.TotalSize()
  53         w.hscroll = util.Clamp(w.hscroll, 0, s-w.width)
  54
  55         if s-w.width <= 0 {
  56                 w.hscroll = 0
  57         }
  58 }
  59
  60 func (w *TabWindow) TotalSize() int {
  61         sum := 2
  62         for _, n := range w.Names {
  63                 sum += runewidth.StringWidth(n) + 4
  64         }
  65         return sum - 4
  66 }
  67
  68 func (w *TabWindow) Active() int {
  69         return w.active
  70 }
  71
  72 func (w *TabWindow) SetActive(a int) {
  73         w.active = a
  74         x := 2
  75         s := w.TotalSize()
  76
  77         for i, n := range w.Names {
  78                 c := utf8.RuneCountInString(n)
  79                 if i == a {
  80                         if x+c >= w.hscroll+w.width {
  81                                 w.hscroll = util.Clamp(x+c+1-w.width, 0, s-w.width)
  82                         } else if x < w.hscroll {
  83                                 w.hscroll = util.Clamp(x-4, 0, s-w.width)
  84                         }
  85                         break
  86                 }
  87                 x += c + 4
  88         }
  89
  90         if s-w.width <= 0 {
  91                 w.hscroll = 0
  92         }
  93 }
  94
  95 func (w *TabWindow) Display() {
  96         x := -w.hscroll
  97         done := false
  98
  99         draw := func(r rune, n int) {
 100                 for i := 0; i < n; i++ {
 101                         rw := runewidth.RuneWidth(r)
 102                         for j := 0; j < rw; j++ {
 103                                 c := r
 104                                 if j > 0 {
 105                                         c = ' '
 106                                 }
 107                                 if x == w.width-1 && !done {
 108                                         screen.SetContent(w.width-1, w.Y, '>', nil, config.DefStyle.Reverse(true))
 109                                         x++
 110                                         break
 111                                 } else if x == 0 && w.hscroll > 0 {
 112                                         screen.SetContent(0, w.Y, '<', nil, config.DefStyle.Reverse(true))
 113                                 } else if x >= 0 && x < w.width {
 114                                         screen.SetContent(x, w.Y, c, nil, config.DefStyle.Reverse(true))
 115                                 }
 116                                 x++
 117                         }
 118                 }
 119         }
 120
 121         for i, n := range w.Names {
 122                 if i == w.active {
 123                         draw('[', 1)
 124                 } else {
 125                         draw(' ', 1)
 126                 }
 127                 for _, c := range n {
 128                         draw(c, 1)
 129                 }
 130                 if i == len(w.Names)-1 {
 131                         done = true
 132                 }
 133                 if i == w.active {
 134                         draw(']', 1)
 135                         draw(' ', 2)
 136                 } else {
 137                         draw(' ', 3)
 138                 }
 139                 if x >= w.width {
 140                         break
 141                 }
 142         }
 143
 144         if x < w.width {
 145                 draw(' ', w.width-x)
 146         }
 147 }