support config PromptPageSize

tssh/config.go

@@ -36,6 +36,7 @@ import (
 	"os"
 	"path/filepath"
 	"runtime"
+	"strconv"
 	"strings"
 	"sync"
 
@@ -71,6 +72,7 @@ type tsshConfig struct {
 	exConfigPath        string
 	defaultUploadPath   string
 	defaultDownloadPath string
+	promptPageSize      uint8
 	loadConfig          sync.Once
 	loadExConfig        sync.Once
 	loadHosts           sync.Once
@@ -123,6 +125,13 @@ func parseTsshConfig() {
 			userConfig.defaultUploadPath = resolveHomeDir(value)
 		case name == "defaultdownloadpath" && userConfig.defaultDownloadPath == "":
 			userConfig.defaultDownloadPath = resolveHomeDir(value)
+		case name == "promptpagesize" && userConfig.promptPageSize == 0:
+			pageSize, err := strconv.ParseUint(value, 10, 8)
+			if err != nil {
+				warning("PromptPageSize %s is invalid: %v", value, err)
+			} else {
+				userConfig.promptPageSize = uint8(pageSize)
+			}
 		}
 	}
 
@@ -138,6 +147,9 @@ func parseTsshConfig() {
 	if userConfig.defaultDownloadPath != "" {
 		debug("DefaultDownloadPath = %s", userConfig.defaultDownloadPath)
 	}
+	if userConfig.promptPageSize != 0 {
+		debug("PromptPageSize = %d", userConfig.promptPageSize)
+	}
 }
 
 func initUserConfig(configFile string) error {
@@ -467,3 +479,10 @@ func getSecretConfig(alias, key string) string {
 	}
 	return getExConfig(alias, key)
 }
+
+func getPromptPageSize() int {
+	if userConfig.promptPageSize != 0 {
+		return int(userConfig.promptPageSize)
+	}
+	return 10
+}

tssh/prompt.go

@@ -41,8 +41,6 @@ import (
 var promptCursorIcon = "🧨"
 var promptSelectedIcon = "🍺"
 
-const promptPageSize = 10
-
 const (
 	keyCtrlA     = '\x01'
 	keyCtrlB     = '\x02'
@@ -148,7 +146,7 @@ func (p *sshPrompt) getShortcuts() []string {
 }
 
 func (p *sshPrompt) getPageCount() int {
-	return (len(p.hosts)-1)/promptPageSize + 1
+	return (len(p.hosts)-1)/getPromptPageSize() + 1
 }
 
 func (p *sshPrompt) hasSelected() bool {
@@ -603,7 +601,7 @@ func chooseAlias(keywords string) (string, bool, error) {
 			Label:        "SSH Alias",
 			Items:        hosts,
 			Templates:    templates,
-			Size:         promptPageSize,
+			Size:         getPromptPageSize(),
 			Searcher:     searcher,
 			Stdin:        pipeIn,
 			Stdout:       &bellFilter{os.Stderr},