visit/main.go

package main

import (
	"encoding/json"
	"fmt"
	"io/ioutil"
	"math/rand"
	"os"
	"os/exec"
	"path/filepath"
	"time"

	"github.com/AlecAivazis/survey/v2"
	"github.com/charmbracelet/glamour"
)

func main() {
	var err error
	if len(os.Args) > 2 {
		fmt.Println("expected zero or one arguments")
	}
	if len(os.Args) > 1 {
		arg := os.Args[1]
		switch arg {
		case "-r", "--random":
			err = visitRandomUser()
		default:
			err = visitUsername(os.Args[1])
		}
	} else {
		err = visitPrompt()
	}

	if err != nil {
		fmt.Fprintf(os.Stderr, "%s\n", err)
		os.Exit(1)
	}
}

func visitRandomUser() error {
	rand.Seed(time.Now().UnixNano())
	usernames, err := getUsernames()
	if err != nil {
		return err
	}

	return visitUser(usernames[rand.Intn(len(usernames))])
}

func visitUsername(username string) error {
	usernames, err := getUsernames()
	if err != nil {
		return err
	}

	for _, name := range usernames {
		if name == username {
			return visitUser(username)
		}
	}

	return fmt.Errorf("no such user: %s", username)
}

func visitPrompt() error {
	usernames, err := getUsernames()
	if err != nil {
		return err
	}
	username := ""
	prompt := &survey.Select{
		Message: "Choose a townie to visit!",
		Options: usernames,
	}
	survey.AskOne(prompt, &username)
	return visitUser(username)
}

func visitUser(username string) error {
	files, err := ioutil.ReadDir(filepath.Join("/home", username))
	if err != nil {
		return fmt.Errorf("could not read user's home directory: %w", err)
	}

	for _, filename := range files {
		fmt.Println("DEBUG", filename)
		if isReadme(filename) {
			if isMarkdown(filename) {
				// TODO fill in once i know if filename is abs or rel path
				/*
					r, _ := glamour.NewTermRenderer(glamour.WithAutoStyle())
					out, err := r.Render()
				*/

			}
			fmt.Println("TODO print raw file")
			return nil
		}
	}

	fmt.Println("TODO non-readme fallback")

	return nil
}

func isReadme(filename string) bool {
	fn := strings.ToLower(filename)
	return strings.HasPrefix(fn, "readme")
}

func isMarkdown(filename string) bool {
	fn := strings.ToLower(filename)
	return strings.HasSuffix(fn, ".md") || strings.HasSuffix(fn, ".markdown")
}

func getUsernames() ([]string, error) {
	cmd := exec.Command("/bin/bash", "-c", "town stats | jq -r '.users|map(.username)'")
	output, err := cmd.Output()
	if err != nil {
		return nil, err
	}
	usernames := []string{}
	err = json.Unmarshal(output, &usernames)

	if err != nil {
		return nil, err
	}

	return usernames, nil
}