mirror of
https://github.com/turbot/steampipe.git
synced 2026-02-20 19:00:11 -05:00
404dd35e21
* Provide feedback for failed prepared statements * Move error functions to error_helpers * Make maintenance client retriable
93 lines
2.0 KiB
Go
93 lines
2.0 KiB
Go
package display
|
|
|
|
import (
|
|
"bufio"
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"os/exec"
|
|
"runtime"
|
|
"strings"
|
|
|
|
"github.com/karrick/gows"
|
|
"github.com/spf13/viper"
|
|
"github.com/turbot/steampipe/pkg/constants"
|
|
"github.com/turbot/steampipe/pkg/error_helpers"
|
|
)
|
|
|
|
// ShowPaged displays the `content` in a system dependent pager
|
|
func ShowPaged(ctx context.Context, content string) {
|
|
if isPagerNeeded(content) && (runtime.GOOS == "darwin" || runtime.GOOS == "linux") {
|
|
nixPager(ctx, content)
|
|
} else {
|
|
nullPager(content)
|
|
}
|
|
}
|
|
|
|
func isPagerNeeded(content string) bool {
|
|
// only show pager in interactive mode
|
|
if !viper.GetBool(constants.ConfigKeyInteractive) {
|
|
return false
|
|
}
|
|
|
|
maxCols, maxRow, _ := gows.GetWinSize()
|
|
|
|
// let's scan through it instead of iterating over it fully
|
|
sc := bufio.NewScanner(strings.NewReader(content))
|
|
|
|
// explicitly allocate a large bugger for the scanner to use - otherwise we may fail for large rows
|
|
buffSize := 256 * 1024
|
|
buff := make([]byte, buffSize)
|
|
sc.Buffer(buff, buffSize)
|
|
|
|
lineCount := 0
|
|
for sc.Scan() {
|
|
line := sc.Text()
|
|
lineCount++
|
|
if lineCount > maxRow {
|
|
return true
|
|
}
|
|
if len(line) > maxCols {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func nullPager(content string) {
|
|
// just dump the whole thing out
|
|
// we will use this for non-tty environments as well
|
|
fmt.Print(content)
|
|
}
|
|
|
|
func nixPager(ctx context.Context, content string) {
|
|
if isLessAvailable() {
|
|
execPager(ctx, exec.Command("less", "-SRXF"), content)
|
|
} else if isMoreAvailable() {
|
|
execPager(ctx, exec.Command("more"), content)
|
|
} else {
|
|
nullPager(content)
|
|
}
|
|
}
|
|
|
|
func isLessAvailable() bool {
|
|
_, err := exec.LookPath("less")
|
|
return err == nil
|
|
}
|
|
|
|
func isMoreAvailable() bool {
|
|
_, err := exec.LookPath("more")
|
|
return err == nil
|
|
}
|
|
|
|
func execPager(ctx context.Context, cmd *exec.Cmd, content string) {
|
|
cmd.Stdout = os.Stdout
|
|
cmd.Stderr = os.Stderr
|
|
cmd.Stdin = strings.NewReader(content)
|
|
// run the command - it will block until the pager is exited
|
|
err := cmd.Run()
|
|
if err != nil {
|
|
error_helpers.ShowErrorWithMessage(ctx, err, "could not display results")
|
|
}
|
|
}
|