package app import ( "math" "git.sr.ht/~rjarry/aerc/config" "git.sr.ht/~rjarry/aerc/lib/ui" "github.com/gdamore/tcell/v2" "github.com/mattn/go-runewidth" ) type PartSwitcher struct { Scrollable parts []*PartViewer selected int height int offset int uiConfig *config.UIConfig } func (ps *PartSwitcher) PreviousPart() { for { ps.selected-- if ps.selected < 0 { ps.selected = len(ps.parts) - 1 } if ps.parts[ps.selected].part.MIMEType != "multipart" { break } } } func (ps *PartSwitcher) NextPart() { for { ps.selected++ if ps.selected >= len(ps.parts) { ps.selected = 0 } if ps.parts[ps.selected].part.MIMEType != "multipart" { break } } } func (ps *PartSwitcher) SelectedPart() *PartViewer { return ps.parts[ps.selected] } func (ps *PartSwitcher) AttachmentParts(all bool) []*PartInfo { var attachments []*PartInfo for _, p := range ps.parts { if p.part.Disposition == "attachment" || (all && p.part.FileName() != "") { pi := &PartInfo{ Index: p.index, Msg: p.msg.MessageInfo(), Part: p.part, } attachments = append(attachments, pi) } } return attachments } func (ps *PartSwitcher) Invalidate() { ui.Invalidate() } func (ps *PartSwitcher) Focus(focus bool) { if ps.parts[ps.selected].term != nil { ps.parts[ps.selected].term.Focus(focus) } } func (ps *PartSwitcher) Show(visible bool) { if ps.parts[ps.selected].term != nil { ps.parts[ps.selected].term.Show(visible) } } func (ps *PartSwitcher) Event(event tcell.Event) bool { return ps.parts[ps.selected].Event(event) } func (ps *PartSwitcher) Draw(ctx *ui.Context) { uiConfig := ps.uiConfig n := len(ps.parts) if n == 1 && !config.Viewer.AlwaysShowMime { ps.parts[ps.selected].Draw(ctx) return } ps.height = config.Viewer.MaxMimeHeight if ps.height <= 0 || n < ps.height { ps.height = n } if ps.height > ctx.Height()/2 { ps.height = ctx.Height() / 2 } ps.UpdateScroller(ps.height, n) ps.EnsureScroll(ps.selected) var styleSwitcher, styleFile, styleMime tcell.Style scrollbarWidth := 0 if ps.NeedScrollbar() { scrollbarWidth = 1 } ps.offset = ctx.Height() - ps.height y := ps.offset row := ps.offset ctx.Fill(0, y, ctx.Width(), ps.height, ' ', uiConfig.GetStyle(config.STYLE_PART_SWITCHER)) for i := ps.Scroll(); i < n; i++ { part := ps.parts[i] if ps.selected == i { styleSwitcher = uiConfig.GetStyleSelected(config.STYLE_PART_SWITCHER) styleFile = uiConfig.GetStyleSelected(config.STYLE_PART_FILENAME) styleMime = uiConfig.GetStyleSelected(config.STYLE_PART_MIMETYPE) } else { styleSwitcher = uiConfig.GetStyle(config.STYLE_PART_SWITCHER) styleFile = uiConfig.GetStyle(config.STYLE_PART_FILENAME) styleMime = uiConfig.GetStyle(config.STYLE_PART_MIMETYPE) } ctx.Fill(0, row, ctx.Width(), 1, ' ', styleSwitcher) left := len(part.index) * 2 if part.part.FileName() != "" { name := runewidth.Truncate(part.part.FileName(), ctx.Width()-left-1, "…") left += ctx.Printf(left, row, styleFile, "%s ", name) } t := "(" + part.part.FullMIMEType() + ")" t = runewidth.Truncate(t, ctx.Width()-left-scrollbarWidth, "…") ctx.Printf(left, row, styleMime, "%s", t) row++ if (i - ps.Scroll()) >= ps.height { break } } if ps.NeedScrollbar() { ps.drawScrollbar(ctx.Subcontext(ctx.Width()-1, y, 1, ps.height)) } ps.parts[ps.selected].Draw(ctx.Subcontext( 0, 0, ctx.Width(), ctx.Height()-ps.height)) } func (ps *PartSwitcher) drawScrollbar(ctx *ui.Context) { uiConfig := ps.uiConfig gutterStyle := uiConfig.GetStyle(config.STYLE_MSGLIST_GUTTER) pillStyle := uiConfig.GetStyle(config.STYLE_MSGLIST_PILL) // gutter ctx.Fill(0, 0, 1, ctx.Height(), ' ', gutterStyle) // pill pillSize := int(math.Ceil(float64(ctx.Height()) * ps.PercentVisible())) pillOffset := int(math.Floor(float64(ctx.Height()) * ps.PercentScrolled())) ctx.Fill(0, pillOffset, 1, pillSize, ' ', pillStyle) } func (ps *PartSwitcher) MouseEvent(localX int, localY int, event tcell.Event) { if localY < ps.offset && ps.parts[ps.selected].term != nil { ps.parts[ps.selected].term.MouseEvent(localX, localY, event) return } e, ok := event.(*tcell.EventMouse) if !ok { return } if ps.parts[ps.selected].term != nil { ps.parts[ps.selected].term.Focus(false) } switch e.Buttons() { case tcell.Button1: i := localY - ps.offset + ps.Scroll() if i < 0 || i >= len(ps.parts) { break } if ps.parts[i].part.MIMEType == "multipart" { break } ps.selected = i ps.Invalidate() case tcell.WheelDown: ps.NextPart() ps.Invalidate() case tcell.WheelUp: ps.PreviousPart() ps.Invalidate() } if ps.parts[ps.selected].term != nil { ps.parts[ps.selected].term.Focus(true) } } func (ps *PartSwitcher) Cleanup() { for _, partViewer := range ps.parts { partViewer.Cleanup() } }