iterm2ui.go

package mkrg

import (
	"bytes"
	"encoding/base64"
	"fmt"
	"image"
	"image/png"
	"os"
	"time"
)

type iterm2UI struct {
	height, width     int
	column, maxColumn int
	from, until       time.Time
	img               *image.RGBA
}

func newIterm2UI(height, width, maxColumn int, from, until time.Time) *iterm2UI {
	return &iterm2UI{height, width, 0, maxColumn, from, until, nil}
}

func (ui *iterm2UI) output(graph graph, ms metricsByName) error {
	imgHeight, imgWidth, padding := ui.height*20, ui.width*12, ui.width/5
	if ui.column == 0 {
		ui.img = image.NewRGBA(image.Rect(0, 0, (imgWidth+padding)*ui.maxColumn-padding, imgHeight+padding*2))
	}
	printImage(&imageWithMargins{ui.img, padding, (imgWidth + padding) * ui.column}, graph, ms, imgHeight, imgWidth, ui.from, ui.until)
	if ui.column == ui.maxColumn-1 {
		if err := ui.cleanup(); err != nil {
			return err
		}
		ui.column = 0
	} else {
		ui.column++
	}
	return nil
}

func (ui *iterm2UI) cleanup() error {
	if ui.column > 0 {
		buf := new(bytes.Buffer)
		if err := png.Encode(buf, ui.img); err != nil {
			return err
		}
		printOSC()
		fmt.Print("1337;File=inline=1;preserveAspectRatio=1;width=100%:")
		fmt.Print(base64.StdEncoding.EncodeToString(buf.Bytes()))
		printST()
	}
	return nil
}

func printOSC() {
	if os.Getenv("TMUX") != "" {
		fmt.Print("\x1bPtmux;\x1b\x1b]")
	} else {
		fmt.Print("\x1b]")
	}
}

func printST() {
	if os.Getenv("TMUX") != "" {
		fmt.Print("\x07\x1b\\\n")
	} else {
		fmt.Print("\x07\n")
	}
}