flan.go

package main

import (
	"bufio"
	"errors"
	"fmt"
	"os"
	"strings"
)

/*
	TODO:
	- show historical invocations of command
	- add $arg$ substitution to flannotations
	- parse commands and allow jumping to relevant flags in manpage with tab?
*/

func flanpage(arg string) error {
	manOut, manErr := manOutput(arg)

	cmds, err := readFlanFile()
	if err != nil {
		return err
	}

	flannotations, prs := cmds[arg]
	if prs {
		err := combineFlanAndManOutput(flannotations, manOut, manErr)
		if err != nil {
			return err
		}
	} else {
		return errors.New("no manpage or flannotations found :(")
	}

	return nil
}

func combineFlanAndManOutput(flannotations [][]string, manOut []byte, manErr error) error {
	lessCmd := lessCommand(os.Environ(), os.Stdout, os.Stderr)

	lessIn, err := lessCmd.StdinPipe()
	if err != nil {
		return err
	}

	if err := lessCmd.Start(); err != nil {
		return err
	}

	lessIn.Write([]byte(color("FLANNOTATIONS:\n\n",
		bold, yellow)))

	for _, flanno := range flannotations {

		lessIn.Write([]byte("     "))
		lessIn.Write([]byte(color(flanno[1], yellow)))
		lessIn.Write([]byte("\n"))

		lessIn.Write([]byte("     "))
		lessIn.Write([]byte(color("$ ", green)))

		lessIn.Write([]byte(color(flanno[0], red)))
		lessIn.Write([]byte("\n\n"))
	}

	if manErr != nil {
		lessIn.Write([]byte(color("WARNING:\n\n", bold, red)))
	}

	lessIn.Write(manOut)
	lessIn.Close()

	if err := lessCmd.Wait(); err != nil {
		return err
	}

	return nil
}

func flannotate() error {
	leader := color("> ", bold, green)

	cmds, err := readFlanFile()
	if err != nil {
		return err
	}

	prompt := fmt.Sprintf("Input a command example to flannotate:\n%s", leader)
	fmt.Print(color(prompt, bold, blue))

	reader := bufio.NewReader(os.Stdin)
	input, err := reader.ReadString('\n')

	if err != nil {
		return err
	}

	cmdEx := strings.TrimSpace(input)
	cmdName := strings.Split(cmdEx, " ")[0]

	prompt = fmt.Sprintf("Input a new flannotation for %s:\n%s",
		color(cmdEx, bold, red), leader)

	fmt.Print(color(prompt, bold, blue))

	input, err = reader.ReadString('\n')
	if err != nil {
		return err
	}
	cmdAnno := strings.TrimSpace(input)

	store(cmdName, cmdEx, cmdAnno, cmds)
	if err = writeFlanFile(cmds); err != nil {
		return err
	}

	fmt.Println(color("🍮", magenta))
	return nil
}