Expand psql output

expand.go

/*
	Accepts an unexpanded psql result set on stdin and expands it.
*/
package main

import (
	"bufio"
	"fmt"
	"os"
	"strings"

	"golang.org/x/crypto/ssh/terminal"
)

func main() {
	width := 80
	if terminal.IsTerminal(int(os.Stdout.Fd())) {
		w, _, _ := terminal.GetSize(int(os.Stdout.Fd()))
		if w != 0 {
			width = w
		}
	}

	var header []string
	pad := ""

	scanner := bufio.NewScanner(os.Stdin)
	i := 0
	for scanner.Scan() {
		line := scanner.Text()
		switch i {
		case 0:
			max := 0
			header = strings.Split(line, "|")
			for j, e := range header {
				header[j] = strings.TrimSpace(e)
				if l := len(header[j]); l > max {
					max = l
				}
			}
			pad = fmt.Sprint(max)
		case 1:
			// skip
		default:
			row := strings.Split(line, "|")
			for j, e := range row {
				row[j] = strings.TrimSpace(e)
			}
			hr := fmt.Sprintf("-[ RECORD %d ]", i-1)
			for j := len(hr); j < width; j++ {
				hr += "-"
			}
			fmt.Print(hr + "\n")
			for j, f := range header {
				fmt.Printf("%-"+pad+"s   | %s\n", f, row[j])
			}
		}
		i++
	}
	if err := scanner.Err(); err != nil {
		panic(err)
	}
}