mirror of https://github.com/xrehpicx/pee.git
115 lines
2.7 KiB
Go
115 lines
2.7 KiB
Go
package filepicker
|
|
|
|
import (
|
|
"errors"
|
|
"os"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/charmbracelet/bubbles/filepicker"
|
|
tea "github.com/charmbracelet/bubbletea"
|
|
)
|
|
|
|
type model struct {
|
|
headerMessage string
|
|
selectedMessage string
|
|
filepicker filepicker.Model
|
|
selectedFile string
|
|
quitting bool
|
|
err error
|
|
}
|
|
|
|
type clearErrorMsg struct{}
|
|
|
|
func clearErrorAfter(t time.Duration) tea.Cmd {
|
|
return tea.Tick(t, func(_ time.Time) tea.Msg {
|
|
return clearErrorMsg{}
|
|
})
|
|
}
|
|
|
|
func (m model) Init() tea.Cmd {
|
|
return m.filepicker.Init()
|
|
}
|
|
|
|
func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
|
switch msg := msg.(type) {
|
|
case tea.KeyMsg:
|
|
switch msg.String() {
|
|
case " ":
|
|
m.quitting = true
|
|
return m, tea.Quit
|
|
case "q", "ctrl+c", "esc":
|
|
m.quitting = true
|
|
m.selectedFile = ""
|
|
m.filepicker.FileSelected = ""
|
|
return m, tea.Quit
|
|
}
|
|
case clearErrorMsg:
|
|
m.err = nil
|
|
}
|
|
|
|
var cmd tea.Cmd
|
|
m.filepicker, cmd = m.filepicker.Update(msg)
|
|
|
|
// Did the user select a file?
|
|
if didSelect, path := m.filepicker.DidSelectFile(msg); didSelect {
|
|
// Get the path of the selected file.
|
|
m.selectedFile = path
|
|
}
|
|
|
|
// Did the user select a disabled file?
|
|
// This is only necessary to display an error to the user.
|
|
if didSelect, path := m.filepicker.DidSelectDisabledFile(msg); didSelect {
|
|
// Let's clear the selectedFile and display an error.
|
|
m.err = errors.New(path + " is not valid.")
|
|
m.selectedFile = ""
|
|
return m, tea.Batch(cmd, clearErrorAfter(2*time.Second))
|
|
}
|
|
|
|
return m, cmd
|
|
}
|
|
|
|
func (m model) View() string {
|
|
if m.quitting {
|
|
return ""
|
|
}
|
|
var s strings.Builder
|
|
s.WriteString("\n ")
|
|
if m.err != nil {
|
|
s.WriteString(m.filepicker.Styles.DisabledFile.Render(m.err.Error()))
|
|
} else if m.selectedFile == "" {
|
|
s.WriteString(m.headerMessage)
|
|
} else {
|
|
s.WriteString(m.selectedMessage + m.filepicker.Styles.Selected.Render(m.selectedFile) + " <space> to select")
|
|
}
|
|
s.WriteString("\n\n" + m.filepicker.View() + "\n" + "<esc> or q to quit")
|
|
return s.String()
|
|
}
|
|
|
|
func FilePicker(headerMessage string, selectedMessage string) (string, error) {
|
|
fp := filepicker.New()
|
|
// fp.AllowedTypes = []string{".mod", ".sum", ".go", ".txt", ".md"}
|
|
fp.DirAllowed = true
|
|
fp.FileAllowed = false
|
|
fp.CurrentDirectory, _ = os.Getwd()
|
|
fp.FileSelected, _ = os.Getwd()
|
|
|
|
// Set default values for header and footer messages
|
|
if headerMessage == "" {
|
|
headerMessage = "Select file..."
|
|
}
|
|
if selectedMessage == "" {
|
|
selectedMessage = "Selected file: "
|
|
}
|
|
m := model{
|
|
headerMessage: headerMessage,
|
|
selectedMessage: selectedMessage,
|
|
filepicker: fp,
|
|
selectedFile: fp.FileSelected,
|
|
}
|
|
tm, err := tea.NewProgram(&m, tea.WithOutput(os.Stderr)).Run()
|
|
mm := tm.(model)
|
|
|
|
return mm.selectedFile, err
|
|
}
|