mirror of
https://github.com/charmbracelet/glow
synced 2024-11-10 14:14:17 +00:00
fc898831af
Also note that the main stash state and the filter state now operate independently.
1259 lines
31 KiB
Go
1259 lines
31 KiB
Go
package ui
|
||
|
||
import (
|
||
"errors"
|
||
"fmt"
|
||
"io/ioutil"
|
||
"log"
|
||
"sort"
|
||
"strings"
|
||
"time"
|
||
|
||
"github.com/charmbracelet/bubbles/paginator"
|
||
"github.com/charmbracelet/bubbles/spinner"
|
||
"github.com/charmbracelet/bubbles/textinput"
|
||
tea "github.com/charmbracelet/bubbletea"
|
||
"github.com/charmbracelet/charm"
|
||
"github.com/charmbracelet/charm/ui/common"
|
||
"github.com/muesli/reflow/ansi"
|
||
te "github.com/muesli/termenv"
|
||
"github.com/sahilm/fuzzy"
|
||
)
|
||
|
||
const (
|
||
stashIndent = 1
|
||
stashViewItemHeight = 3
|
||
stashViewTopPadding = 5
|
||
stashViewBottomPadding = 4
|
||
stashViewHorizontalPadding = 6
|
||
)
|
||
|
||
var (
|
||
stashHelpItemStyle styleFunc = newFgStyle(common.NewColorPair("#5C5C5C", "#9B9B9B"))
|
||
stashTextInputPromptStyle styleFunc = newFgStyle(common.YellowGreen)
|
||
dividerDot string = te.String(" • ").Foreground(common.NewColorPair("#3C3C3C", "#DDDADA").Color()).String()
|
||
offlineHeaderNote string = te.String("(Offline)").Foreground(common.NewColorPair("#3C3C3C", "#DDDADA").Color()).String()
|
||
)
|
||
|
||
// MSG
|
||
|
||
type fetchedMarkdownMsg *markdown
|
||
type deletedStashedItemMsg int
|
||
type filteredMarkdownMsg []*markdown
|
||
|
||
// MODEL
|
||
|
||
type stashState int
|
||
|
||
const (
|
||
stashStateReady stashState = iota
|
||
stashStatePromptDelete
|
||
stashStateLoadingDocument
|
||
stashStateSettingNote
|
||
stashStateShowingError
|
||
stashStateShowNews
|
||
)
|
||
|
||
type filterState int
|
||
|
||
const (
|
||
unfiltered filterState = iota
|
||
filtering // user is actively setting a filter
|
||
filterApplied // a filter is applied and user is not editing filter
|
||
)
|
||
|
||
type stashModel struct {
|
||
general *general
|
||
state stashState
|
||
filterState filterState
|
||
err error
|
||
spinner spinner.Model
|
||
noteInput textinput.Model
|
||
filterInput textinput.Model
|
||
stashFullyLoaded bool // have we loaded all available stashed documents from the server?
|
||
loadingFromNetwork bool // are we currently loading something from the network?
|
||
loaded DocumentType // load status for news, stash and local files loading; we find out exactly with bitmasking
|
||
|
||
// The master set of markdown documents we're working with.
|
||
markdowns []*markdown
|
||
|
||
// Markdown documents we're currently displaying. Filtering, toggles and so
|
||
// on will alter this slice so we can show what is relevant. For that
|
||
// reason, this field should be considered ephemeral.
|
||
filteredMarkdowns []*markdown
|
||
|
||
// Paths to files being stashed. We treat this like a set, ignoring the
|
||
// value portion with an empty struct.
|
||
filesStashing map[string]struct{}
|
||
|
||
// This is just the selected item in relation to the current page in view.
|
||
// To get the index of the selected item as it relates to the full set of
|
||
// documents we've fetched use the markdownIndex() method on this struct.
|
||
index int
|
||
|
||
// This handles the local pagination, which is different than the page
|
||
// we're fetching from on the server side
|
||
paginator paginator.Model
|
||
|
||
// Page we're fetching stash items from on the server, which is different
|
||
// from the local pagination. Generally, the server will return more items
|
||
// than we can display at a time so we can paginate locally without having
|
||
// to fetch every time.
|
||
page int
|
||
|
||
showStatusMessage bool
|
||
statusMessage string
|
||
statusMessageTimer *time.Timer
|
||
}
|
||
|
||
func (m stashModel) localOnly() bool {
|
||
return m.general.cfg.DocumentTypes == LocalDocument
|
||
}
|
||
|
||
func (m stashModel) stashedOnly() bool {
|
||
return m.general.cfg.DocumentTypes&LocalDocument == 0
|
||
}
|
||
|
||
func (m stashModel) loadingDone() bool {
|
||
// Do the types loaded match the types we want to have?
|
||
return m.loaded == m.general.cfg.DocumentTypes
|
||
}
|
||
|
||
func (m *stashModel) setSize(width, height int) {
|
||
m.general.width = width
|
||
m.general.height = height
|
||
|
||
// Update the paginator
|
||
m.setTotalPages()
|
||
|
||
m.noteInput.Width = width - stashViewHorizontalPadding*2 - ansi.PrintableRuneWidth(m.noteInput.Prompt)
|
||
m.filterInput.Width = width - stashViewHorizontalPadding*2 - ansi.PrintableRuneWidth(m.filterInput.Prompt)
|
||
}
|
||
|
||
func (m *stashModel) resetFiltering() {
|
||
m.filterState = unfiltered
|
||
m.filterInput.Reset()
|
||
sort.Stable(markdownsByLocalFirst(m.markdowns))
|
||
m.filteredMarkdowns = nil
|
||
m.setTotalPages()
|
||
}
|
||
|
||
// Is a filter currently being applied?
|
||
func (m stashModel) isFiltering() bool {
|
||
return m.filterState != unfiltered
|
||
}
|
||
|
||
// Should we be updating the filter?
|
||
func (m stashModel) shouldUpdateFilter() bool {
|
||
// If we're in the middle of setting a note don't update the filter so that
|
||
// the focus won't jump around.
|
||
return m.isFiltering() && m.state != stashStateSettingNote
|
||
}
|
||
|
||
// Sets the total paginator pages according to the amount of markdowns for the
|
||
// current state.
|
||
func (m *stashModel) setTotalPages() {
|
||
m.paginator.PerPage = max(1, (m.general.height-stashViewTopPadding-stashViewBottomPadding)/stashViewItemHeight)
|
||
|
||
if pages := len(m.getVisibleMarkdowns()); pages < 1 {
|
||
m.paginator.SetTotalPages(1)
|
||
} else {
|
||
m.paginator.SetTotalPages(pages)
|
||
}
|
||
|
||
// Make sure the page stays in bounds
|
||
if m.paginator.Page >= m.paginator.TotalPages-1 {
|
||
m.paginator.Page = max(0, m.paginator.TotalPages-1)
|
||
}
|
||
}
|
||
|
||
// MarkdownIndex returns the index of the currently selected markdown item.
|
||
func (m stashModel) markdownIndex() int {
|
||
return m.paginator.Page*m.paginator.PerPage + m.index
|
||
}
|
||
|
||
// Return the current selected markdown in the stash.
|
||
func (m stashModel) selectedMarkdown() *markdown {
|
||
i := m.markdownIndex()
|
||
|
||
mds := m.getVisibleMarkdowns()
|
||
if i < 0 || len(mds) == 0 || len(mds) <= i {
|
||
return nil
|
||
}
|
||
|
||
return mds[i]
|
||
}
|
||
|
||
// Adds markdown documents to the model.
|
||
func (m *stashModel) addMarkdowns(mds ...*markdown) {
|
||
if len(mds) > 0 {
|
||
m.markdowns = append(m.markdowns, mds...)
|
||
if !m.isFiltering() {
|
||
sort.Stable(markdownsByLocalFirst(m.markdowns))
|
||
}
|
||
m.setTotalPages()
|
||
}
|
||
}
|
||
|
||
// Find a local markdown by its path and replace it
|
||
func (m *stashModel) replaceLocalMarkdown(localPath string, newMarkdown *markdown) error {
|
||
var found bool
|
||
|
||
// Look for local markdown
|
||
for i, md := range m.markdowns {
|
||
if md.localPath == localPath {
|
||
m.markdowns[i] = newMarkdown
|
||
found = true
|
||
break
|
||
}
|
||
}
|
||
|
||
if !found {
|
||
err := fmt.Errorf("could't find local markdown %s; not removing from stash", localPath)
|
||
if debug {
|
||
log.Println(err)
|
||
}
|
||
return err
|
||
}
|
||
|
||
if m.isFiltering() {
|
||
found = false
|
||
for i, md := range m.filteredMarkdowns {
|
||
if md.localPath == localPath {
|
||
m.filteredMarkdowns[i] = newMarkdown
|
||
found = true
|
||
break
|
||
}
|
||
}
|
||
if !found {
|
||
err := fmt.Errorf("warning: found local markdown %s in the master markdown list, but not in the filter results", localPath)
|
||
if debug {
|
||
log.Println(err)
|
||
}
|
||
return err
|
||
}
|
||
}
|
||
|
||
return nil
|
||
}
|
||
|
||
// Return the number of markdown documents of a given type.
|
||
func (m stashModel) countMarkdowns(t DocumentType) (found int) {
|
||
mds := m.getVisibleMarkdowns()
|
||
if len(mds) == 0 {
|
||
return
|
||
}
|
||
for i := 0; i < len(mds); i++ {
|
||
if mds[i].markdownType == t {
|
||
found++
|
||
}
|
||
}
|
||
return
|
||
}
|
||
|
||
// Sift through the master markdown collection for the specified types.
|
||
func (m stashModel) getMarkdownByType(types ...DocumentType) []*markdown {
|
||
var agg []*markdown
|
||
|
||
if len(m.markdowns) == 0 {
|
||
return agg
|
||
}
|
||
|
||
for _, t := range types {
|
||
for _, md := range m.markdowns {
|
||
if md.markdownType == t {
|
||
agg = append(agg, md)
|
||
}
|
||
}
|
||
}
|
||
|
||
sort.Sort(markdownsByLocalFirst(agg))
|
||
return agg
|
||
}
|
||
|
||
// Returns the markdowns that should be currently shown.
|
||
func (m stashModel) getVisibleMarkdowns() []*markdown {
|
||
if m.isFiltering() {
|
||
return m.filteredMarkdowns
|
||
}
|
||
|
||
if m.state == stashStateShowNews {
|
||
return m.getMarkdownByType(NewsDocument)
|
||
}
|
||
|
||
return m.getMarkdownByType(LocalDocument, StashedDocument, ConvertedDocument)
|
||
}
|
||
|
||
// Return the markdowns eligible to be filtered.
|
||
func (m stashModel) getFilterableMarkdowns() []*markdown {
|
||
if m.state == stashStateShowNews {
|
||
return m.getMarkdownByType(NewsDocument)
|
||
}
|
||
|
||
return m.getMarkdownByType(LocalDocument, StashedDocument, ConvertedDocument)
|
||
}
|
||
|
||
// Command for opening a markdown document in the pager. Note that this also
|
||
// alters the model.
|
||
func (m *stashModel) openMarkdown(md *markdown) tea.Cmd {
|
||
var cmd tea.Cmd
|
||
m.state = stashStateLoadingDocument
|
||
|
||
if md.markdownType == LocalDocument {
|
||
cmd = loadLocalMarkdown(md)
|
||
} else {
|
||
cmd = loadRemoteMarkdown(m.general.cc, md.ID, md.markdownType)
|
||
}
|
||
|
||
return tea.Batch(cmd, spinner.Tick)
|
||
}
|
||
|
||
func (m *stashModel) hideStatusMessage() {
|
||
m.showStatusMessage = false
|
||
if m.statusMessageTimer != nil {
|
||
m.statusMessageTimer.Stop()
|
||
}
|
||
}
|
||
|
||
func (m *stashModel) moveCursorUp() {
|
||
m.index--
|
||
if m.index < 0 && m.paginator.Page == 0 {
|
||
// Stop
|
||
m.index = 0
|
||
return
|
||
}
|
||
|
||
if m.index >= 0 {
|
||
return
|
||
}
|
||
// Go to previous page
|
||
m.paginator.PrevPage()
|
||
|
||
m.index = m.paginator.ItemsOnPage(len(m.getVisibleMarkdowns())) - 1
|
||
}
|
||
|
||
func (m *stashModel) moveCursorDown() {
|
||
itemsOnPage := m.paginator.ItemsOnPage(len(m.getVisibleMarkdowns()))
|
||
|
||
m.index++
|
||
if m.index < itemsOnPage {
|
||
return
|
||
}
|
||
|
||
if !m.paginator.OnLastPage() {
|
||
m.paginator.NextPage()
|
||
m.index = 0
|
||
return
|
||
}
|
||
|
||
// During filtering the cursor position can exceed the number of
|
||
// itemsOnPage. It's more intuitive to start the cursor at the
|
||
// topmost position when moving it down in this scenario.
|
||
if m.index > itemsOnPage {
|
||
m.index = 0
|
||
return
|
||
}
|
||
m.index = itemsOnPage - 1
|
||
}
|
||
|
||
// INIT
|
||
|
||
func newStashModel(general *general) stashModel {
|
||
sp := spinner.NewModel()
|
||
sp.Spinner = spinner.Line
|
||
sp.ForegroundColor = common.SpinnerColor.String()
|
||
sp.HideFor = time.Millisecond * 50
|
||
sp.MinimumLifetime = time.Millisecond * 180
|
||
sp.Start()
|
||
|
||
p := paginator.NewModel()
|
||
p.Type = paginator.Dots
|
||
p.InactiveDot = common.Subtle("•")
|
||
|
||
ni := textinput.NewModel()
|
||
ni.Prompt = stashTextInputPromptStyle("Memo: ")
|
||
ni.CursorColor = common.Fuschia.String()
|
||
ni.CharLimit = noteCharacterLimit
|
||
ni.Focus()
|
||
|
||
si := textinput.NewModel()
|
||
si.Prompt = stashTextInputPromptStyle("Filter: ")
|
||
si.CursorColor = common.Fuschia.String()
|
||
si.CharLimit = noteCharacterLimit
|
||
si.Focus()
|
||
|
||
m := stashModel{
|
||
general: general,
|
||
spinner: sp,
|
||
noteInput: ni,
|
||
filterInput: si,
|
||
page: 1,
|
||
paginator: p,
|
||
loadingFromNetwork: true,
|
||
filesStashing: make(map[string]struct{}),
|
||
}
|
||
|
||
return m
|
||
}
|
||
|
||
// UPDATE
|
||
|
||
func (m stashModel) update(msg tea.Msg) (stashModel, tea.Cmd) {
|
||
var cmds []tea.Cmd
|
||
|
||
switch msg := msg.(type) {
|
||
case errMsg:
|
||
m.err = msg
|
||
|
||
case stashLoadErrMsg:
|
||
m.err = msg.err
|
||
m.loaded |= StashedDocument // still done, albeit unsuccessfully
|
||
m.stashFullyLoaded = true
|
||
m.loadingFromNetwork = false
|
||
|
||
case newsLoadErrMsg:
|
||
m.err = msg.err
|
||
m.loaded |= NewsDocument // still done, albeit unsuccessfully
|
||
|
||
case localFileSearchFinished:
|
||
// We're finished searching for local files
|
||
m.loaded |= LocalDocument
|
||
|
||
case gotStashMsg, gotNewsMsg:
|
||
// Stash or news results have come in from the server.
|
||
//
|
||
// With the stash, this doesn't mean the whole stash listing is loaded,
|
||
// but now know it can load, at least, so mark the stash as loaded here.
|
||
var docs []*markdown
|
||
|
||
switch msg := msg.(type) {
|
||
case gotStashMsg:
|
||
m.loaded |= StashedDocument
|
||
m.loadingFromNetwork = false
|
||
docs = wrapMarkdowns(StashedDocument, msg)
|
||
|
||
if len(msg) == 0 {
|
||
// If the server comes back with nothing then we've got
|
||
// everything
|
||
m.stashFullyLoaded = true
|
||
} else {
|
||
// Load the next page
|
||
m.page++
|
||
cmds = append(cmds, loadStash(m))
|
||
}
|
||
|
||
case gotNewsMsg:
|
||
m.loaded |= NewsDocument
|
||
docs = wrapMarkdowns(NewsDocument, msg)
|
||
}
|
||
|
||
// If we're filtering build filter indexes immediately so any
|
||
// matching results will show up in the filter.
|
||
if m.isFiltering() {
|
||
for _, md := range docs {
|
||
md.buildFilterValue()
|
||
}
|
||
}
|
||
if m.shouldUpdateFilter() {
|
||
cmds = append(cmds, filterMarkdowns(m))
|
||
}
|
||
|
||
m.addMarkdowns(docs...)
|
||
|
||
case filteredMarkdownMsg:
|
||
m.filteredMarkdowns = msg
|
||
return m, nil
|
||
|
||
case spinner.TickMsg:
|
||
condition := !m.loadingDone() ||
|
||
m.loadingFromNetwork ||
|
||
m.state == stashStateLoadingDocument ||
|
||
len(m.filesStashing) > 0 ||
|
||
m.spinner.Visible()
|
||
|
||
if condition {
|
||
newSpinnerModel, cmd := m.spinner.Update(msg)
|
||
m.spinner = newSpinnerModel
|
||
cmds = append(cmds, cmd)
|
||
}
|
||
|
||
// A note was set on a document. This may have happened in the pager so
|
||
// we'll find the corresponding document here and update accordingly.
|
||
case noteSavedMsg:
|
||
for i := range m.markdowns {
|
||
if m.markdowns[i].ID == msg.ID {
|
||
m.markdowns[i].Note = msg.Note
|
||
}
|
||
}
|
||
|
||
// Something was stashed. Add it to the stash listing.
|
||
case stashSuccessMsg:
|
||
md := markdown(msg)
|
||
delete(m.filesStashing, md.localPath) // remove from the things-we're-stashing list
|
||
|
||
_ = m.replaceLocalMarkdown(md.localPath, &md)
|
||
|
||
m.showStatusMessage = true
|
||
m.statusMessage = "Stashed!"
|
||
if m.statusMessageTimer != nil {
|
||
m.statusMessageTimer.Stop()
|
||
}
|
||
m.statusMessageTimer = time.NewTimer(statusMessageTimeout)
|
||
cmds = append(cmds, waitForStatusMessageTimeout(stashContext, m.statusMessageTimer))
|
||
|
||
case statusMessageTimeoutMsg:
|
||
if applicationContext(msg) == stashContext {
|
||
m.hideStatusMessage()
|
||
}
|
||
}
|
||
|
||
if m.filterState == filtering {
|
||
cmds = append(cmds, m.handleFiltering(msg))
|
||
return m, tea.Batch(cmds...)
|
||
}
|
||
|
||
// Updates per the current state
|
||
switch m.state {
|
||
case stashStateReady, stashStateShowNews:
|
||
cmds = append(cmds, m.handleDocumentBrowsing(msg))
|
||
case stashStatePromptDelete:
|
||
cmds = append(cmds, m.handleDeleteConfirmation(msg))
|
||
case stashStateSettingNote:
|
||
cmds = append(cmds, m.handleNoteInput(msg))
|
||
case stashStateShowingError:
|
||
// Any key exists the error view
|
||
if _, ok := msg.(tea.KeyMsg); ok {
|
||
m.state = stashStateReady
|
||
}
|
||
}
|
||
|
||
return m, tea.Batch(cmds...)
|
||
}
|
||
|
||
// Updates for when a user is browsing the markdown listing
|
||
func (m *stashModel) handleDocumentBrowsing(msg tea.Msg) tea.Cmd {
|
||
var cmds []tea.Cmd
|
||
|
||
pages := len(m.getVisibleMarkdowns())
|
||
|
||
switch msg := msg.(type) {
|
||
// Handle keys
|
||
case tea.KeyMsg:
|
||
switch msg.String() {
|
||
case "k", "ctrl+k", "up", "shift+tab":
|
||
m.moveCursorUp()
|
||
|
||
case "j", "ctrl+j", "down", "tab":
|
||
m.moveCursorDown()
|
||
|
||
// Go to the very start
|
||
case "home", "g":
|
||
m.paginator.Page = 0
|
||
m.index = 0
|
||
|
||
// Go to the very end
|
||
case "end", "G":
|
||
m.paginator.Page = m.paginator.TotalPages - 1
|
||
m.index = m.paginator.ItemsOnPage(pages) - 1
|
||
|
||
case "esc":
|
||
m.resetFiltering()
|
||
m.state = stashStateReady
|
||
|
||
// Open document
|
||
case "enter":
|
||
m.hideStatusMessage()
|
||
|
||
if pages == 0 {
|
||
break
|
||
}
|
||
|
||
// Load the document from the server. We'll handle the message
|
||
// that comes back in the main update function.
|
||
md := m.selectedMarkdown()
|
||
cmds = append(cmds, m.openMarkdown(md))
|
||
|
||
// Filter your notes
|
||
case "/":
|
||
m.hideStatusMessage()
|
||
|
||
// Build values we'll filter against
|
||
for _, md := range m.markdowns {
|
||
md.buildFilterValue()
|
||
}
|
||
|
||
m.filteredMarkdowns = m.getFilterableMarkdowns()
|
||
|
||
m.paginator.Page = 0
|
||
m.index = 0
|
||
m.filterState = filtering
|
||
m.filterInput.CursorEnd()
|
||
m.filterInput.Focus()
|
||
return textinput.Blink
|
||
|
||
// Set note
|
||
case "m":
|
||
m.hideStatusMessage()
|
||
|
||
if pages == 0 {
|
||
break
|
||
}
|
||
|
||
md := m.selectedMarkdown()
|
||
isUserMarkdown := md.markdownType == StashedDocument || md.markdownType == ConvertedDocument
|
||
isSettingNote := m.state == stashStateSettingNote
|
||
isPromptingDelete := m.state == stashStatePromptDelete
|
||
|
||
if isUserMarkdown && !isSettingNote && !isPromptingDelete {
|
||
m.state = stashStateSettingNote
|
||
m.noteInput.SetValue(md.Note)
|
||
m.noteInput.CursorEnd()
|
||
return textinput.Blink
|
||
}
|
||
|
||
// Show news
|
||
case "n":
|
||
if m.state == stashStateShowNews {
|
||
// Exit news
|
||
m.state = stashStateReady
|
||
m.resetFiltering()
|
||
} else {
|
||
// Show news
|
||
m.hideStatusMessage()
|
||
m.paginator.Page = 0
|
||
m.index = 0
|
||
m.state = stashStateShowNews
|
||
m.setTotalPages()
|
||
}
|
||
|
||
return nil
|
||
|
||
// Stash
|
||
case "s":
|
||
if pages == 0 || m.general.authStatus != authOK || m.selectedMarkdown() == nil {
|
||
break
|
||
}
|
||
|
||
md := m.selectedMarkdown()
|
||
|
||
_, isBeingStashed := m.filesStashing[md.localPath]
|
||
isLocalMarkdown := md.markdownType == LocalDocument
|
||
markdownPathMissing := md.localPath == ""
|
||
|
||
if isBeingStashed || !isLocalMarkdown || markdownPathMissing {
|
||
if debug && isBeingStashed {
|
||
log.Printf("refusing to stash markdown; we're already stashing %s", md.localPath)
|
||
} else if debug && isLocalMarkdown && markdownPathMissing {
|
||
log.Printf("refusing to stash markdown; local path is empty: %#v", md)
|
||
}
|
||
break
|
||
}
|
||
|
||
// Checks passed; perform the stash
|
||
m.filesStashing[md.localPath] = struct{}{}
|
||
cmds = append(cmds, stashDocument(m.general.cc, *md))
|
||
|
||
if m.loadingDone() && !m.spinner.Visible() {
|
||
m.spinner.Start()
|
||
cmds = append(cmds, spinner.Tick)
|
||
}
|
||
|
||
// Prompt for deletion
|
||
case "x":
|
||
m.hideStatusMessage()
|
||
|
||
if pages == 0 {
|
||
break
|
||
}
|
||
|
||
t := m.selectedMarkdown().markdownType
|
||
isUserMarkdown := t == StashedDocument || t == ConvertedDocument
|
||
isValidState := m.state != stashStateSettingNote
|
||
|
||
if isUserMarkdown && isValidState {
|
||
m.state = stashStatePromptDelete
|
||
}
|
||
|
||
// Show errors
|
||
case "!":
|
||
if m.err != nil && m.state == stashStateReady {
|
||
m.state = stashStateShowingError
|
||
return nil
|
||
}
|
||
}
|
||
}
|
||
|
||
// Update paginator. Pagination key handling is done here, but it could
|
||
// also be moved up to this level, in which case we'd use model methods
|
||
// like model.PageUp().
|
||
newPaginatorModel, cmd := m.paginator.Update(msg)
|
||
m.paginator = newPaginatorModel
|
||
cmds = append(cmds, cmd)
|
||
|
||
// Extra paginator keystrokes
|
||
if key, ok := msg.(tea.KeyMsg); ok {
|
||
switch key.String() {
|
||
case "b", "u":
|
||
m.paginator.PrevPage()
|
||
case "f", "d":
|
||
m.paginator.NextPage()
|
||
}
|
||
}
|
||
|
||
// Keep the index in bounds when paginating
|
||
itemsOnPage := m.paginator.ItemsOnPage(len(m.getVisibleMarkdowns()))
|
||
if m.index > itemsOnPage-1 {
|
||
m.index = max(0, itemsOnPage-1)
|
||
}
|
||
|
||
// If we're on the last page and we haven't loaded everything, get
|
||
// more stuff.
|
||
if m.paginator.OnLastPage() && !m.loadingFromNetwork && !m.stashFullyLoaded {
|
||
m.page++
|
||
m.loadingFromNetwork = true
|
||
cmds = append(cmds, loadStash(*m))
|
||
}
|
||
|
||
return tea.Batch(cmds...)
|
||
}
|
||
|
||
// Updates for when a user is being prompted whether or not to delete a
|
||
// markdown item.
|
||
func (m *stashModel) handleDeleteConfirmation(msg tea.Msg) tea.Cmd {
|
||
if msg, ok := msg.(tea.KeyMsg); ok {
|
||
switch msg.String() {
|
||
|
||
// Confirm deletion
|
||
case "y":
|
||
if m.state != stashStatePromptDelete {
|
||
break
|
||
}
|
||
|
||
smd := m.selectedMarkdown()
|
||
for i, md := range m.markdowns {
|
||
if md != smd {
|
||
continue
|
||
}
|
||
|
||
if md.markdownType == ConvertedDocument {
|
||
// If document was stashed during this session, convert it
|
||
// back to a local file.
|
||
md.markdownType = LocalDocument
|
||
md.Note = stripAbsolutePath(m.markdowns[i].localPath, m.general.cwd)
|
||
} else {
|
||
// Delete optimistically and remove the stashed item
|
||
// before we've received a success response.
|
||
if m.isFiltering() {
|
||
mds, _ := deleteMarkdown(m.filteredMarkdowns, m.markdowns[i])
|
||
m.filteredMarkdowns = mds
|
||
}
|
||
mds, _ := deleteMarkdown(m.markdowns, m.markdowns[i])
|
||
m.markdowns = mds
|
||
}
|
||
}
|
||
|
||
// Update pagination
|
||
m.setTotalPages()
|
||
|
||
return deleteStashedItem(m.general.cc, smd.ID)
|
||
|
||
default:
|
||
// Any other keys cancels deletion
|
||
m.state = stashStateReady
|
||
|
||
}
|
||
}
|
||
|
||
return nil
|
||
}
|
||
|
||
// Updates for when a user is in the filter editing interface
|
||
func (m *stashModel) handleFiltering(msg tea.Msg) tea.Cmd {
|
||
var cmds []tea.Cmd
|
||
|
||
// Handle keys
|
||
if msg, ok := msg.(tea.KeyMsg); ok {
|
||
switch msg.String() {
|
||
case "esc":
|
||
// Cancel filtering
|
||
m.resetFiltering()
|
||
case "enter", "tab", "shift+tab", "ctrl+k", "up", "ctrl+j", "down":
|
||
m.hideStatusMessage()
|
||
|
||
if len(m.markdowns) == 0 {
|
||
break
|
||
}
|
||
|
||
h := m.getVisibleMarkdowns()
|
||
|
||
// If we've filtered down to nothing, clear the filter
|
||
if len(h) == 0 {
|
||
m.state = stashStateReady
|
||
m.resetFiltering()
|
||
break
|
||
}
|
||
|
||
// When there's only one filtered markdown left we can just
|
||
// "open" it directly
|
||
if len(h) == 1 {
|
||
m.state = stashStateReady
|
||
m.resetFiltering()
|
||
cmds = append(cmds, m.openMarkdown(h[0]))
|
||
break
|
||
}
|
||
|
||
m.filterInput.Blur()
|
||
|
||
m.filterState = filterApplied
|
||
if m.filterInput.Value() == "" {
|
||
m.resetFiltering()
|
||
}
|
||
}
|
||
}
|
||
|
||
// Update the filter text input component
|
||
newFilterInputModel, inputCmd := m.filterInput.Update(msg)
|
||
currentFilterVal := m.filterInput.Value()
|
||
newFilterVal := newFilterInputModel.Value()
|
||
m.filterInput = newFilterInputModel
|
||
cmds = append(cmds, inputCmd)
|
||
|
||
// If the filtering input has changed, request updated filtering
|
||
if newFilterVal != currentFilterVal {
|
||
cmds = append(cmds, filterMarkdowns(*m))
|
||
}
|
||
|
||
// Update pagination
|
||
m.setTotalPages()
|
||
|
||
return tea.Batch(cmds...)
|
||
}
|
||
|
||
func (m *stashModel) handleNoteInput(msg tea.Msg) tea.Cmd {
|
||
var cmds []tea.Cmd
|
||
|
||
if msg, ok := msg.(tea.KeyMsg); ok {
|
||
switch msg.String() {
|
||
case "esc":
|
||
// Cancel note
|
||
m.noteInput.Reset()
|
||
m.state = stashStateReady
|
||
case "enter":
|
||
// Set new note
|
||
md := m.selectedMarkdown()
|
||
newNote := m.noteInput.Value()
|
||
cmd := saveDocumentNote(m.general.cc, md.ID, newNote)
|
||
md.Note = newNote
|
||
m.noteInput.Reset()
|
||
m.state = stashStateReady
|
||
return cmd
|
||
}
|
||
}
|
||
|
||
if m.shouldUpdateFilter() {
|
||
cmds = append(cmds, filterMarkdowns(*m))
|
||
}
|
||
|
||
// Update the note text input component
|
||
newNoteInputModel, noteInputCmd := m.noteInput.Update(msg)
|
||
m.noteInput = newNoteInputModel
|
||
cmds = append(cmds, noteInputCmd)
|
||
|
||
return tea.Batch(cmds...)
|
||
}
|
||
|
||
// VIEW
|
||
|
||
func (m stashModel) view() string {
|
||
var s string
|
||
switch m.state {
|
||
case stashStateShowingError:
|
||
return errorView(m.err, false)
|
||
case stashStateLoadingDocument:
|
||
s += " " + m.spinner.View() + " Loading document..."
|
||
case stashStateReady, stashStateSettingNote, stashStatePromptDelete, stashStateShowNews:
|
||
|
||
loadingIndicator := " "
|
||
if !m.localOnly() && (!m.loadingDone() || m.loadingFromNetwork || m.spinner.Visible()) {
|
||
loadingIndicator = m.spinner.View()
|
||
}
|
||
|
||
// We need to fill any empty height with newlines so the footer reaches
|
||
// the bottom.
|
||
numBlankLines := max(0, (m.general.height-stashViewTopPadding-stashViewBottomPadding)%stashViewItemHeight)
|
||
blankLines := ""
|
||
if numBlankLines > 0 {
|
||
blankLines = strings.Repeat("\n", numBlankLines)
|
||
}
|
||
|
||
var header string
|
||
if m.showStatusMessage {
|
||
header = greenFg(m.statusMessage)
|
||
} else {
|
||
switch m.state {
|
||
case stashStatePromptDelete:
|
||
header = redFg("Delete this item from your stash? ") + faintRedFg("(y/N)")
|
||
case stashStateSettingNote:
|
||
header = yellowFg("Set the memo for this item?")
|
||
}
|
||
}
|
||
|
||
// Only draw the normal header if we're not using the header area for
|
||
// something else (like a prompt or status message)
|
||
if header == "" {
|
||
header = stashHeaderView(m)
|
||
}
|
||
|
||
logoOrFilter := glowLogoView(" Glow ")
|
||
|
||
// If we're filtering we replace the logo with the filter field
|
||
if m.isFiltering() {
|
||
logoOrFilter = m.filterInput.View()
|
||
} else if m.state == stashStateShowNews {
|
||
logoOrFilter += newsTitleStyle(" News ")
|
||
}
|
||
|
||
var pagination string
|
||
if m.paginator.TotalPages > 1 {
|
||
pagination = m.paginator.View()
|
||
|
||
// If the dot pagination is wider than the width of the window
|
||
// switch to the arabic paginator.
|
||
if ansi.PrintableRuneWidth(pagination) > m.general.width-stashViewHorizontalPadding {
|
||
m.paginator.Type = paginator.Arabic
|
||
pagination = common.Subtle(m.paginator.View())
|
||
}
|
||
|
||
// We could also look at m.stashFullyLoaded and add an indicator
|
||
// showing that we don't actually know how many more pages there
|
||
// are.
|
||
}
|
||
|
||
s += fmt.Sprintf(
|
||
"%s %s\n\n %s\n\n%s\n\n%s %s\n\n %s",
|
||
loadingIndicator,
|
||
logoOrFilter,
|
||
header,
|
||
stashPopulatedView(m),
|
||
blankLines,
|
||
pagination,
|
||
stashHelpView(m),
|
||
)
|
||
}
|
||
return "\n" + indent(s, stashIndent)
|
||
}
|
||
|
||
func glowLogoView(text string) string {
|
||
return te.String(text).
|
||
Bold().
|
||
Foreground(glowLogoTextColor).
|
||
Background(common.Fuschia.Color()).
|
||
String()
|
||
}
|
||
|
||
func stashHeaderView(m stashModel) string {
|
||
loading := !m.loadingDone()
|
||
noMarkdowns := len(m.markdowns) == 0
|
||
|
||
if m.general.authStatus == authFailed && m.stashedOnly() {
|
||
return common.Subtle("Can’t load stash. Are you offline?")
|
||
}
|
||
|
||
var maybeOffline string
|
||
if m.general.authStatus == authFailed {
|
||
maybeOffline = " " + offlineHeaderNote
|
||
}
|
||
|
||
// Still loading. We haven't found files, stashed items, or news yet.
|
||
if loading && noMarkdowns {
|
||
if m.stashedOnly() {
|
||
return common.Subtle("Loading your stash...")
|
||
} else {
|
||
return common.Subtle("Looking for stuff...") + maybeOffline
|
||
}
|
||
}
|
||
|
||
localItems := m.countMarkdowns(LocalDocument)
|
||
stashedItems := m.countMarkdowns(StashedDocument) + m.countMarkdowns(ConvertedDocument)
|
||
newsItems := m.countMarkdowns(NewsDocument)
|
||
|
||
// Loading's finished and all we have is news.
|
||
if !loading && localItems == 0 && stashedItems == 0 && newsItems == 0 {
|
||
if m.stashedOnly() {
|
||
return common.Subtle("No stashed markdown files found.") + maybeOffline
|
||
} else {
|
||
return common.Subtle("No local or stashed markdown files found.") + maybeOffline
|
||
}
|
||
}
|
||
|
||
// There are local and/or stashed files, so display counts.
|
||
var s string
|
||
if localItems > 0 {
|
||
s += common.Subtle(fmt.Sprintf("%d Local", localItems))
|
||
}
|
||
if stashedItems > 0 {
|
||
var divider string
|
||
if localItems > 0 {
|
||
divider = dividerDot
|
||
}
|
||
si := common.Subtle(fmt.Sprintf("%d Stashed", stashedItems))
|
||
s += fmt.Sprintf("%s%s", divider, si)
|
||
}
|
||
if newsItems > 0 {
|
||
var divider string
|
||
if localItems > 0 || stashedItems > 0 {
|
||
divider = dividerDot
|
||
}
|
||
si := common.Subtle(fmt.Sprintf("%d News", newsItems))
|
||
|
||
s += fmt.Sprintf("%s%s", divider, si)
|
||
}
|
||
return common.Subtle(s) + maybeOffline
|
||
}
|
||
|
||
func stashPopulatedView(m stashModel) string {
|
||
var b strings.Builder
|
||
|
||
mds := m.getVisibleMarkdowns()
|
||
if len(mds) > 0 {
|
||
start, end := m.paginator.GetSliceBounds(len(mds))
|
||
docs := mds[start:end]
|
||
|
||
for i, md := range docs {
|
||
stashItemView(&b, m, i, md)
|
||
if i != len(docs)-1 {
|
||
fmt.Fprintf(&b, "\n\n")
|
||
}
|
||
}
|
||
}
|
||
|
||
// If there aren't enough items to fill up this page (always the last page)
|
||
// then we need to add some newlines to fill up the space where stash items
|
||
// would have been.
|
||
itemsOnPage := m.paginator.ItemsOnPage(len(mds))
|
||
if itemsOnPage < m.paginator.PerPage {
|
||
n := (m.paginator.PerPage - itemsOnPage) * stashViewItemHeight
|
||
if len(mds) == 0 {
|
||
n -= stashViewItemHeight - 1
|
||
}
|
||
for i := 0; i < n; i++ {
|
||
fmt.Fprint(&b, "\n")
|
||
}
|
||
}
|
||
|
||
return b.String()
|
||
}
|
||
|
||
func stashHelpView(m stashModel) string {
|
||
var (
|
||
h []string
|
||
isStashed bool
|
||
isLocal bool
|
||
numDocs = len(m.getVisibleMarkdowns())
|
||
)
|
||
|
||
if numDocs > 0 {
|
||
md := m.selectedMarkdown()
|
||
isStashed = md != nil && md.markdownType == StashedDocument
|
||
isLocal = md != nil && md.markdownType == LocalDocument
|
||
}
|
||
|
||
if m.state == stashStateSettingNote {
|
||
h = append(h, "enter: confirm", "esc: cancel")
|
||
} else if m.state == stashStatePromptDelete {
|
||
h = append(h, "y: delete", "n: cancel")
|
||
} else if m.filterState == filtering && numDocs == 1 {
|
||
h = append(h, "enter: open", "esc: cancel")
|
||
} else if m.filterState == filtering && numDocs == 0 {
|
||
h = append(h, "enter/esc: cancel")
|
||
} else if m.filterState == filtering {
|
||
h = append(h, "enter: confirm", "esc: cancel", "ctrl+j/ctrl+k, ↑/↓: choose")
|
||
} else if m.state == stashStateShowNews {
|
||
h = append(h, "enter: open", "esc: return", "j/k, ↑/↓: choose", "q: quit")
|
||
} else {
|
||
if len(m.markdowns) > 0 {
|
||
h = append(h, "enter: open")
|
||
}
|
||
if m.filterState == filterApplied {
|
||
h = append(h, "esc: clear filter")
|
||
}
|
||
if len(m.markdowns) > 1 {
|
||
h = append(h, "j/k, ↑/↓: choose")
|
||
}
|
||
if m.paginator.TotalPages > 1 {
|
||
h = append(h, "h/l, ←/→: page")
|
||
}
|
||
if isStashed {
|
||
h = append(h, "x: delete", "m: set memo")
|
||
} else if isLocal && m.general.authStatus == authOK {
|
||
h = append(h, "s: stash")
|
||
}
|
||
if m.err != nil {
|
||
h = append(h, "!: errors")
|
||
}
|
||
h = append(h, "/: filter")
|
||
h = append(h, "q: quit")
|
||
}
|
||
return stashHelpViewBuilder(m.general.width, h...)
|
||
}
|
||
|
||
// builds the help view from various sections pieces, truncating it if the view
|
||
// would otherwise wrap to two lines.
|
||
func stashHelpViewBuilder(windowWidth int, sections ...string) string {
|
||
if len(sections) == 0 {
|
||
return ""
|
||
}
|
||
|
||
const truncationWidth = 1 // width of "…"
|
||
|
||
var (
|
||
s string
|
||
next string
|
||
maxWidth = windowWidth - stashViewHorizontalPadding - truncationWidth
|
||
)
|
||
|
||
for i := 0; i < len(sections); i++ {
|
||
// If we need this more often we'll formalize something rather than
|
||
// use an if clause/switch here.
|
||
switch sections[i] {
|
||
case "s: stash":
|
||
next = greenFg(sections[i])
|
||
default:
|
||
next = stashHelpItemStyle(sections[i])
|
||
}
|
||
|
||
if i < len(sections)-1 {
|
||
next += dividerDot
|
||
}
|
||
|
||
// Only this (and the following) help text items if we have the
|
||
// horizontal space
|
||
if ansi.PrintableRuneWidth(s)+ansi.PrintableRuneWidth(next) >= maxWidth {
|
||
s += common.Subtle("…")
|
||
break
|
||
}
|
||
|
||
s += next
|
||
}
|
||
return s
|
||
}
|
||
|
||
// COMMANDS
|
||
|
||
func loadRemoteMarkdown(cc *charm.Client, id int, t DocumentType) tea.Cmd {
|
||
return func() tea.Msg {
|
||
var (
|
||
md *charm.Markdown
|
||
err error
|
||
)
|
||
|
||
if t == StashedDocument || t == ConvertedDocument {
|
||
md, err = cc.GetStashMarkdown(id)
|
||
} else {
|
||
md, err = cc.GetNewsMarkdown(id)
|
||
}
|
||
|
||
if err != nil {
|
||
if debug {
|
||
log.Println("error loading remote markdown:", err)
|
||
}
|
||
return errMsg{err}
|
||
}
|
||
|
||
return fetchedMarkdownMsg(&markdown{
|
||
markdownType: t,
|
||
Markdown: *md,
|
||
})
|
||
}
|
||
}
|
||
|
||
func loadLocalMarkdown(md *markdown) tea.Cmd {
|
||
return func() tea.Msg {
|
||
if md.markdownType != LocalDocument {
|
||
return errMsg{errors.New("could not load local file: not a local file")}
|
||
}
|
||
if md.localPath == "" {
|
||
return errMsg{errors.New("could not load file: missing path")}
|
||
}
|
||
|
||
data, err := ioutil.ReadFile(md.localPath)
|
||
if err != nil {
|
||
if debug {
|
||
log.Println("error reading local markdown:", err)
|
||
}
|
||
return errMsg{err}
|
||
}
|
||
md.Body = string(data)
|
||
return fetchedMarkdownMsg(md)
|
||
}
|
||
}
|
||
|
||
func deleteStashedItem(cc *charm.Client, id int) tea.Cmd {
|
||
return func() tea.Msg {
|
||
err := cc.DeleteMarkdown(id)
|
||
if err != nil {
|
||
if debug {
|
||
log.Println("could not delete stashed item:", err)
|
||
}
|
||
return errMsg{err}
|
||
}
|
||
return deletedStashedItemMsg(id)
|
||
}
|
||
}
|
||
|
||
func filterMarkdowns(m stashModel) tea.Cmd {
|
||
return func() tea.Msg {
|
||
if m.filterInput.Value() == "" || !m.isFiltering() {
|
||
return filteredMarkdownMsg(m.getFilterableMarkdowns()) // return everything
|
||
}
|
||
|
||
targets := []string{}
|
||
mds := m.getFilterableMarkdowns()
|
||
|
||
for _, t := range mds {
|
||
targets = append(targets, t.filterValue)
|
||
}
|
||
|
||
ranks := fuzzy.Find(m.filterInput.Value(), targets)
|
||
sort.Stable(ranks)
|
||
|
||
filtered := []*markdown{}
|
||
for _, r := range ranks {
|
||
filtered = append(filtered, mds[r.Index])
|
||
}
|
||
|
||
return filteredMarkdownMsg(filtered)
|
||
}
|
||
}
|
||
|
||
// ETC
|
||
|
||
// Delete a markdown from a slice of markdowns
|
||
func deleteMarkdown(markdowns []*markdown, target *markdown) ([]*markdown, error) {
|
||
index := -1
|
||
|
||
for i, v := range markdowns {
|
||
switch target.markdownType {
|
||
case LocalDocument, ConvertedDocument:
|
||
if v.localPath == target.localPath {
|
||
index = i
|
||
}
|
||
case StashedDocument, NewsDocument:
|
||
if v.ID == target.ID {
|
||
index = i
|
||
}
|
||
default:
|
||
return nil, errors.New("unknown markdown type")
|
||
}
|
||
}
|
||
|
||
if index == -1 {
|
||
err := fmt.Errorf("could not find markdown to delete")
|
||
if debug {
|
||
log.Println(err)
|
||
}
|
||
return nil, err
|
||
}
|
||
|
||
return append(markdowns[:index], markdowns[index+1:]...), nil
|
||
}
|