feat: Add --tui Flag to Open TUI-mode on File Argument

main.go

@@ -10,6 +10,7 @@ import (
 	"os/exec"
 	"path/filepath"
 	"strings"
+	"time"
 
 	tea "github.com/charmbracelet/bubbletea"
 	"github.com/charmbracelet/glamour"
@@ -31,6 +32,7 @@ var (
 	readmeNames  = []string{"README.md", "README"}
 	configFile   string
 	pager        bool
+	tui          bool
 	style        string
 	width        uint
 	showAllFiles bool
@@ -140,6 +142,7 @@ func validateOptions(cmd *cobra.Command) error {
 	localOnly = viper.GetBool("local")
 	mouse = viper.GetBool("mouse")
 	pager = viper.GetBool("pager")
+	tui = viper.GetBool("tui")
 
 	// validate the glamour style
 	style = viper.GetString("style")
@@ -219,6 +222,16 @@ func execute(cmd *cobra.Command, args []string) error {
 				return runTUI(p, false)
 			}
 		}
+
+		useTUI := tui || cmd.Flags().Changed("tui")
+		if err == nil && useTUI && !info.IsDir() {
+			// Is file; open in TUI-mode
+			p, err := filepath.Abs(args[0])
+			if err == nil {
+				return runTUI(p, false, true)
+			}
+		}
+
 		fallthrough
 
 	// CLI
@@ -313,7 +326,29 @@ func executeCLI(cmd *cobra.Command, src *source, w io.Writer) error {
 	return nil
 }
 
-func runTUI(workingDirectory string, stashedOnly bool) error {
+func runTUI(path string, stashedOnly bool, withFile ...bool) error {
+	var (
+		workingDirectory string
+		filePath         string
+		createdAt        time.Time
+	)
+
+	if len(withFile) == 1 && withFile[0] == true {
+		// Open TUI-mode with file
+		workingDirectory = filepath.Dir(path)
+		filePath = path
+		fileInfo, err := os.Stat(filePath)
+		if err != nil {
+			createdAt = time.Now()
+		} else {
+			createdAt = fileInfo.ModTime()
+		}
+	} else {
+		workingDirectory = path
+		filePath = ""
+		createdAt = time.Now()
+	}
+
 	// Read environment to get debugging stuff
 	var cfg ui.Config
 	if err := babyenv.Parse(&cfg); err != nil {
@@ -330,6 +365,8 @@ func runTUI(workingDirectory string, stashedOnly bool) error {
 	}
 
 	cfg.WorkingDirectory = workingDirectory
+	cfg.FilePath = filePath
+	cfg.FileCreatedAt = createdAt
 	cfg.DocumentTypes = ui.NewDocTypeSet()
 	cfg.ShowAllFiles = showAllFiles
 	cfg.GlamourMaxWidth = width
@@ -372,6 +409,7 @@ func init() {
 	// "Glow Classic" cli arguments
 	rootCmd.PersistentFlags().StringVar(&configFile, "config", "", fmt.Sprintf("config file (default %s)", defaultConfigFile))
 	rootCmd.Flags().BoolVarP(&pager, "pager", "p", false, "display with pager")
+	rootCmd.Flags().BoolVarP(&tui, "tui", "t", false, "display with TUI-mode")
 	rootCmd.Flags().StringVarP(&style, "style", "s", "auto", "style name or JSON path")
 	rootCmd.Flags().UintVarP(&width, "width", "w", 0, "word-wrap at width")
 	rootCmd.Flags().BoolVarP(&showAllFiles, "all", "a", false, "show system files and directories (TUI-mode only)")

ui/config.go

@@ -1,5 +1,7 @@
 package ui
 
+import "time"
+
 // Config contains TUI-specific configuration.
 type Config struct {
 	ShowAllFiles    bool
@@ -12,6 +14,10 @@ type Config struct {
 	// Which directory should we start from?
 	WorkingDirectory string
 
+	// (Optional) If not "", open TUI-mode with this file open
+	FilePath      string
+	FileCreatedAt time.Time
+
 	// Which document types shall we show?
 	DocumentTypes DocTypeSet
 

ui/ui.go

@@ -53,12 +53,19 @@ func NewProgram(cfg Config) *tea.Program {
 		log.Println("Bubble Tea now initializing...")
 		debug = true
 	}
+
 	config = cfg
+
 	opts := []tea.ProgramOption{tea.WithAltScreen()}
 	if cfg.EnableMouse {
 		opts = append(opts, tea.WithMouseCellMotion())
 	}
-	return tea.NewProgram(newModel(cfg), opts...)
+
+	m := newModel(cfg)
+
+	program := tea.NewProgram(m, opts...)
+
+	return program
 }
 
 type errMsg struct{ err error }
@@ -220,13 +227,22 @@ func newModel(cfg Config) tea.Model {
 		filesStashing: make(map[ksuid.KSUID]struct{}),
 	}
 
-	return model{
+	m := model{
 		common:      &common,
 		state:       stateShowStash,
 		keygenState: keygenUnstarted,
 		pager:       newPagerModel(&common),
 		stash:       newStashModel(&common),
 	}
+
+	if cfg.FilePath != "" {
+		// Open file passed in with TUI-mode flag
+		m.pager.currentDocument = *localFileToMarkdown(cfg.WorkingDirectory, cfg.FilePath, cfg.FileCreatedAt, true) // TODO: Fix time
+		m.state = stateShowDocument
+		m.pager.update(keyEnter)
+	}
+
+	return m
 }
 
 func (m model) Init() tea.Cmd {
@@ -359,6 +375,7 @@ func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 
 	case fetchedMarkdownMsg:
 		// We've loaded a markdown file's contents for rendering
+		println("Update markdown!")
 		m.pager.currentDocument = *msg
 		msg.Body = string(utils.RemoveFrontmatter([]byte(msg.Body)))
 		cmds = append(cmds, renderWithGlamour(m.pager, msg.Body))
@@ -383,7 +400,8 @@ func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 		return m, cmd
 
 	case foundLocalFileMsg:
-		newMd := localFileToMarkdown(m.common.cwd, gitcha.SearchResult(msg))
+		res := gitcha.SearchResult(msg)
+		newMd := localFileToMarkdown(m.common.cwd, res.Path, res.Info.ModTime())
 		m.stash.addMarkdowns(newMd)
 		if m.stash.filterApplied() {
 			newMd.buildFilterValue()
@@ -723,13 +741,23 @@ func waitForStatusMessageTimeout(appCtx applicationContext, t *time.Timer) tea.C
 // Convert a Gitcha result to an internal representation of a markdown
 // document. Note that we could be doing things like checking if the file is
 // a directory, but we trust that gitcha has already done that.
-func localFileToMarkdown(cwd string, res gitcha.SearchResult) *markdown {
+func localFileToMarkdown(cwd string, path string, createdAt time.Time, loadBody ...bool) *markdown {
+	body := ""
+	if len(loadBody) == 1 && loadBody[0] {
+		content, err := os.ReadFile(path)
+		if err == nil {
+			// Convert the byte slice to a string
+			body = string(content)
+		}
+	}
+
 	md := &markdown{
 		docType:   LocalDoc,
-		localPath: res.Path,
+		localPath: path,
 		Markdown: charm.Markdown{
-			Note:      stripAbsolutePath(res.Path, cwd),
-			CreatedAt: res.Info.ModTime(),
+			Body:      body,
+			Note:      stripAbsolutePath(path, cwd),
+			CreatedAt: createdAt,
 		},
 	}