Upload files to "internal/repl"
This commit is contained in:
1 parent
f098896469
commit
d695d05fbd
3 files changed
+752
No files matched your search
@@ -0,0 +1,195 @@
|
||||
/*
|
||||
* Copyright 2026 Safronov Grigorii
|
||||
*
|
||||
* Licensed under the CDDL, Version 1.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
*
|
||||
* You may obtain a copy of the License at
|
||||
* https://opensource.org/licenses/CDDL-1.0
|
||||
*/
|
||||
|
||||
// Файл: internal/repl/pager.go
|
||||
// Назначение: Пейджер для постраничного вывода длинных результатов.
|
||||
// Использует внешний pager (less/more) если доступен, иначе встроенный.
|
||||
// Совместимо с Linux и OpenIndiana.
|
||||
|
||||
package repl
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"fmt"
|
||||
"io"
|
||||
"os"
|
||||
"os/exec"
|
||||
"strings"
|
||||
)
|
||||
|
||||
// Pager предоставляет постраничный вывод.
|
||||
type Pager struct {
|
||||
enabled bool
|
||||
threshold int
|
||||
external string
|
||||
lines int
|
||||
}
|
||||
|
||||
// NewPager создаёт пейджер.
|
||||
func NewPager(enabled bool, threshold int) *Pager {
|
||||
p := &Pager{
|
||||
enabled: enabled,
|
||||
threshold: threshold,
|
||||
lines: 24,
|
||||
}
|
||||
p.external = detectPager()
|
||||
return p
|
||||
}
|
||||
|
||||
// detectPager ищет доступный внешний pager.
|
||||
// Порядок: $PAGER -> less -> more -> "" (встроенный).
|
||||
func detectPager() string {
|
||||
if env := os.Getenv("PAGER"); env != "" {
|
||||
return env
|
||||
}
|
||||
if path, err := exec.LookPath("less"); err == nil {
|
||||
return path
|
||||
}
|
||||
if path, err := exec.LookPath("more"); err == nil {
|
||||
return path
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
// Enabled возвращает, включён ли пейджер.
|
||||
func (p *Pager) Enabled() bool {
|
||||
return p != nil && p.enabled
|
||||
}
|
||||
|
||||
// SetEnabled включает/выключает пейджер.
|
||||
func (p *Pager) SetEnabled(v bool) {
|
||||
if p != nil {
|
||||
p.enabled = v
|
||||
}
|
||||
}
|
||||
|
||||
// SetThreshold устанавливает порог строк.
|
||||
func (p *Pager) SetThreshold(n int) {
|
||||
if p != nil && n > 0 {
|
||||
p.threshold = n
|
||||
}
|
||||
}
|
||||
|
||||
// Page выводит текст через пейджер, если он длиннее порога.
|
||||
func (p *Pager) Page(text string) error {
|
||||
if p == nil || !p.enabled {
|
||||
fmt.Print(text)
|
||||
return nil
|
||||
}
|
||||
|
||||
lineCount := strings.Count(text, "\n")
|
||||
if lineCount < p.threshold {
|
||||
fmt.Print(text)
|
||||
return nil
|
||||
}
|
||||
|
||||
if p.external != "" {
|
||||
if err := p.pageExternal(text); err == nil {
|
||||
return nil
|
||||
}
|
||||
// fallback на встроенный
|
||||
}
|
||||
|
||||
return p.pageBuiltin(text)
|
||||
}
|
||||
|
||||
// pageExternal запускает внешний pager.
|
||||
func (p *Pager) pageExternal(text string) error {
|
||||
args := []string{}
|
||||
base := strings.ToLower(p.external)
|
||||
|
||||
if strings.Contains(base, "less") {
|
||||
// -R: сохранять ANSI-цвета, -F: выйти если всё влезает, -X: не чистить экран
|
||||
args = append(args, "-RFX")
|
||||
}
|
||||
|
||||
cmd := exec.Command(p.external, args...)
|
||||
cmd.Stdout = os.Stdout
|
||||
cmd.Stderr = os.Stderr
|
||||
|
||||
stdin, err := cmd.StdinPipe()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if err := cmd.Start(); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if _, err := io.WriteString(stdin, text); err != nil {
|
||||
stdin.Close()
|
||||
_ = cmd.Wait()
|
||||
return err
|
||||
}
|
||||
stdin.Close()
|
||||
|
||||
return cmd.Wait()
|
||||
}
|
||||
|
||||
// pageBuiltin — встроенный пейджер (fallback).
|
||||
// Управление: Enter — следующая строка, Space — страница, q — выход.
|
||||
func (p *Pager) pageBuiltin(text string) error {
|
||||
scanner := bufio.NewScanner(strings.NewReader(text))
|
||||
buf := make([]byte, 0, 64*1024)
|
||||
scanner.Buffer(buf, 1024*1024)
|
||||
|
||||
lines := []string{}
|
||||
for scanner.Scan() {
|
||||
lines = append(lines, scanner.Text())
|
||||
}
|
||||
if err := scanner.Err(); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
pageSize := p.lines - 1
|
||||
if pageSize < 5 {
|
||||
pageSize = 5
|
||||
}
|
||||
|
||||
reader := bufio.NewReader(os.Stdin)
|
||||
pos := 0
|
||||
|
||||
for pos < len(lines) {
|
||||
end := pos + pageSize
|
||||
if end > len(lines) {
|
||||
end = len(lines)
|
||||
}
|
||||
for i := pos; i < end; i++ {
|
||||
fmt.Println(lines[i])
|
||||
}
|
||||
pos = end
|
||||
|
||||
if pos >= len(lines) {
|
||||
break
|
||||
}
|
||||
|
||||
fmt.Printf("\033[7m--More--\033[0m (Enter/Space/q) ")
|
||||
ch, err := reader.ReadByte()
|
||||
fmt.Print("\r\033[K")
|
||||
if err != nil {
|
||||
break
|
||||
}
|
||||
switch ch {
|
||||
case 'q', 'Q':
|
||||
return nil
|
||||
case '\n', '\r':
|
||||
// одна строка
|
||||
if pos < len(lines) {
|
||||
fmt.Println(lines[pos])
|
||||
pos++
|
||||
}
|
||||
case ' ':
|
||||
// страница
|
||||
default:
|
||||
// игнорируем
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
Reference in new issue
Block a user