feat: inital commit

- added template subcmd which generates frc project templates

cmd/template/configUi.go

@@ -0,0 +1,159 @@
+package template
+
+import (
+	"errors"
+	"fmt"
+	"log/slog"
+	"os"
+	"strconv"
+
+	"github.com/charmbracelet/huh"
+)
+
+var _dummyGroup = huh.NewGroup(huh.NewNote().Title("Dummy"))
+
+type _fieldWrapper struct {
+	Visible func() (bool)
+	Field huh.Field
+}
+
+func _buildGroups(groups ...*huh.Group) []*huh.Group {
+	var visibleGroups []*huh.Group
+
+	for _, g := range groups {
+		if g != _dummyGroup {
+			visibleGroups = append(visibleGroups, g)
+		}
+	}
+
+	return visibleGroups
+}
+
+func _buildGroup(fields ..._fieldWrapper) *huh.Group {
+	var visibleFields []huh.Field
+
+	for _, f := range fields {
+		if f.Visible() {
+			visibleFields = append(visibleFields, f.Field)
+		}
+	}
+
+	if len(visibleFields) > 0 {
+		return huh.NewGroup(visibleFields...)
+	}
+	return _dummyGroup
+}
+
+func openConfigUi(opts TemplateOptions) (TemplateOptions, error) {
+	var lang string = opts.Lang
+	var projectType string = opts.ProjectType
+	var dir string = opts.Dir
+	var team string
+	tmp := strconv.FormatUint(opts.Team, 10)
+	if tmp != "0" {
+		team = tmp
+	}
+	var desktopSupport bool
+	if opts.DesktopSupport != nil {
+		desktopSupport = *opts.DesktopSupport
+	}
+
+	// Display form with selected theme.
+	err := huh.NewForm(
+		_buildGroups(
+			_buildGroup(
+				_fieldWrapper{
+					Visible: func() bool { return lang == "" },
+					Field: huh.NewSelect[string]().
+						OptionsFunc(func() []huh.Option[string] {
+							langs, err := GetLangs()
+							if err != nil {
+								slog.Error("Unable to get languages", "error", err)
+								os.Exit(1)
+							}
+
+							opts := make([]huh.Option[string], len(langs))
+							for i, e := range langs {
+								opts[i] = huh.Option[string]{Value: e, Key: e}
+							}
+							return opts
+						}, nil).
+						Description("Choose your desired language").
+						Value(&lang).
+						Height(4),
+				},
+				_fieldWrapper{
+					Visible: func() bool { return projectType == "" },
+					Field: huh.NewSelect[string]().
+						OptionsFunc(func() []huh.Option[string] {
+							if lang == "" {
+								return []huh.Option[string]{}
+							}
+
+							projects, err := GetProjects(lang)
+							if err != nil {
+								slog.Error("Unable to get project types", "error", err)
+								os.Exit(1)
+							}
+
+							opts := make([]huh.Option[string], len(projects))
+							for i, e := range projects {
+								opts[i] = huh.Option[string]{Value: e, Key: e}
+							}
+							return opts
+						}, &lang).
+						Description("Choose your project type").
+						Value(&projectType),
+				},
+				).Title("Project Language & Type"),
+
+			_buildGroup(
+				_fieldWrapper{
+					Visible: func() bool { return dir == "" },
+					Field: huh.NewInput().
+						Title("Project Path").
+						Placeholder("This is the full path to the project").
+						Value(&dir),
+				},
+				_fieldWrapper{
+					Visible: func() bool { return team == "" },
+					Field: huh.NewInput().
+						Title("Team Number").
+						Placeholder("Your teams number").
+						Value(&team).
+						Validate(func(s string) error {
+							_, err := strconv.Atoi(s)
+							if err != nil {
+								return errors.New("must be a number")
+							}
+							return nil
+						}),
+				},
+				_fieldWrapper{
+					Visible: func() bool { return opts.DesktopSupport == nil },
+					Field: huh.NewConfirm().
+						Title("Enable Desktop Support?").
+						Value(&desktopSupport),
+				},
+				).Title("Project Setup Information"),
+			)...,
+		).Run()
+
+	if err != nil {
+	if err == huh.ErrUserAborted {
+		os.Exit(130)
+	}
+		fmt.Println(err)
+		os.Exit(1)
+	}
+
+	teamnr, _ := strconv.ParseUint(team, 10, 64)
+
+	return TemplateOptions{
+		Lang: lang,
+		ProjectType: projectType,
+		Dir: dir,
+		Team: teamnr,
+		DesktopSupport: &desktopSupport,
+	}, nil
+}

cmd/template/downloadUi.go

@@ -0,0 +1,76 @@
+package template
+
+import (
+	"strings"
+	"time"
+
+	"github.com/charmbracelet/bubbles/progress"
+	tea "github.com/charmbracelet/bubbletea"
+)
+
+const (
+	padding = 2
+	maxWidth = 80
+)
+
+type progressMsg float64
+type progressErrMsg struct{ err error }
+
+func finalPause() tea.Cmd {
+	return tea.Tick(time.Millisecond*750, func(_ time.Time) tea.Msg {
+		return nil
+	})
+}
+
+type downloadModel struct {
+	pw       *progressWriter
+	progress progress.Model
+	err      error
+}
+
+func (m downloadModel) Init() tea.Cmd {
+	return nil
+}
+
+func (m downloadModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
+	switch msg := msg.(type) {
+	case tea.KeyMsg:
+		return m, tea.Quit
+
+	case tea.WindowSizeMsg:
+		m.progress.Width = min(msg.Width - padding * 2 - 4, maxWidth)
+		return m, nil
+
+	case progressErrMsg:
+		m.err = msg.err
+		return m, tea.Quit
+
+	case progressMsg:
+		var cmds []tea.Cmd
+
+		if msg >= 1.0 {
+			cmds = append(cmds, tea.Sequence(finalPause(), tea.Quit))
+		}
+
+		cmds = append(cmds, m.progress.SetPercent(float64(msg)))
+		return m, tea.Batch(cmds...)
+
+	// FrameMsg is sent when the progress bar wants to animate itself
+	case progress.FrameMsg:
+		progressModel, cmd := m.progress.Update(msg)
+		m.progress = progressModel.(progress.Model)
+		return m, cmd
+
+	default:
+		return m, nil
+	}
+}
+
+func (m downloadModel) View() string {
+	if m.err != nil {
+		return "Error downloading: " + m.err.Error() + "\n"
+	}
+
+	pad := strings.Repeat(" ", padding)
+	return "\n" + pad + "Downloading template.zip to " + m.pw.file.Name() + "\n\n" + pad + m.progress.View() + "\n\n" + pad + "Press any key to quit"
+}

cmd/template/remote.go

@@ -0,0 +1,152 @@
+package template
+
+import (
+	"encoding/json"
+	"io"
+	"log/slog"
+	"net/http"
+	"os"
+	"path/filepath"
+	"strconv"
+
+	"rph/state"
+)
+
+type progressWriter struct {
+	total      int
+	downloaded int
+	file       *os.File
+	reader     io.Reader
+	onProgress func(float64)
+}
+
+func (pw *progressWriter) Start() {
+	// TeeReader calls pw.Write() each time a new response is received
+	_, err := io.Copy(pw.file, io.TeeReader(pw.reader, pw))
+	if err != nil {
+		slog.Error("Error in progress writer", "error", progressErrMsg{err})
+	}
+}
+
+func (pw *progressWriter) Write(p []byte) (int, error) {
+	pw.downloaded += len(p)
+	if pw.total > 0 && pw.onProgress != nil {
+		pw.onProgress(float64(pw.downloaded) / float64(pw.total))
+	}
+	return len(p), nil
+}
+
+type asset struct {
+	Name               string `json:"name"`
+	BrowserDownloadURL string `json:"browser_download_url"`
+}
+
+type release struct {
+	TagName string  `json:"tag_name"`
+	Assets  []asset `json:"assets"`
+}
+
+func getTemplateArchive(filename string, force bool, version string) {
+	const url = "https://api.github.com/repos/wpilibsuite/vscode-wpilib/releases/"
+	path := filepath.Join(state.CachePath, filename)
+
+	currentVersion, err := LoadArchiveVersion()
+	// default the version to the latest version if no version is currently
+	// installed and the user wants to keep the current version
+	if err != nil && version == "keep" {
+		version = "latest"
+	} else if currentVersion != "" && version == "keep" {
+		version = currentVersion
+	}
+
+	// use tags to select the version when we're not just getting the latest one
+	if version != "latest" {
+		version = "tags/" + version
+	}
+
+	resp, err := http.Get(url + version)
+	if err != nil {
+		slog.Error("Error fetching release:", "error", err)
+		os.Exit(1)
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode != http.StatusOK {
+		body, _ := io.ReadAll(resp.Body)
+		slog.Error("GitHub API error", "status", resp.Status, "body", string(body))
+		os.Exit(1)
+	}
+
+	var release release
+	if err := json.NewDecoder(resp.Body).Decode(&release); err != nil {
+		slog.Error("Error decoding JSON", "error", err)
+		os.Exit(1)
+	}
+
+	_, ferr := os.Stat(path)
+	currentVersion, err = LoadArchiveVersion()
+	if !force && err == nil && currentVersion == release.TagName && ferr == nil {
+		slog.Info("Template archive is already installed", "version", currentVersion)
+		slog.Info("If you would like to install a different version try: rph template fetch -h")
+		return
+	}
+
+	var downloadURL string
+	for _, asset := range release.Assets {
+		if asset.Name == "templates.zip" {
+			downloadURL = asset.BrowserDownloadURL
+			break
+		}
+	}
+
+	if downloadURL == "" {
+		slog.Warn("templates.zip not found in release version.", "version", version)
+		return
+	}
+
+	err = downloadFile(downloadURL)
+	if err != nil {
+		slog.Error("Error downloading archive file", "error", err)
+		os.Exit(1)
+	}
+
+	err = saveArchiveVersion(release.TagName)
+	if err != nil {
+		slog.Warn("Failed to save version information", "error", err)
+	} else {
+		slog.Info("Downloaded new template file", "version", release.TagName)
+	}
+}
+
+func ListTemplateArchiveVersions(results uint8) []string {
+	const url = "https://api.github.com/repos/wpilibsuite/vscode-wpilib/releases?per_page="
+	resp, err := http.Get(url + strconv.Itoa(int(results)))
+	if err != nil {
+		slog.Error("Error fetching release:", "error", err)
+		os.Exit(1)
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode != http.StatusOK {
+		body, _ := io.ReadAll(resp.Body)
+		slog.Error("GitHub API error", "status", resp.Status, "body", string(body))
+		os.Exit(1)
+	}
+
+	var releases []release
+	if err := json.NewDecoder(resp.Body).Decode(&releases); err != nil {
+		slog.Error("Error decoding JSON", "error", err)
+		os.Exit(1)
+	}
+
+	var versions []string
+	for _, r := range releases {
+		for _, a := range r.Assets {
+			if a.Name == "templates.zip" {
+				versions = append(versions, r.TagName)
+			}
+		}
+	}
+
+	return versions
+}

cmd/template/store.go

@@ -0,0 +1,145 @@
+package template
+
+import (
+	"context"
+	"errors"
+	"io/fs"
+	"log/slog"
+	"net/http"
+	"os"
+	"path/filepath"
+	"rph/state"
+
+	"github.com/charmbracelet/bubbles/progress"
+	tea "github.com/charmbracelet/bubbletea"
+	"github.com/mholt/archives"
+)
+
+const dataFile = "templates.bin"
+const zipFile = "templates.zip"
+
+func saveArchiveVersion(version string) error {
+	return os.WriteFile(
+		filepath.Join(state.CachePath, dataFile),
+		[]byte(version),
+		0644,
+	)
+}
+
+func LoadArchiveVersion() (string, error) {
+	data, err := os.ReadFile(filepath.Join(state.CachePath, dataFile))
+	if err != nil {
+		return "", err
+	}
+	return string(data), nil
+}
+
+func downloadFile(url string) error {
+	var progressBar = true
+
+	// Create the file
+	out, err := os.Create(filepath.Join(state.CachePath, zipFile))
+	if err != nil {
+		return err
+	}
+	defer out.Close()
+
+	// Get the data
+	resp, err := http.Get(url)
+	if err != nil {
+		return err
+	}
+	defer resp.Body.Close()
+
+
+	if resp.ContentLength <= 0 {
+		slog.Warn("Can't parse content length, no progress bar will be shown.")
+		progressBar = false
+	}
+
+	var p *tea.Program
+	pw := &progressWriter{
+		total:  int(resp.ContentLength),
+		file:   out,
+		reader: resp.Body,
+		onProgress: func(ratio float64) {
+			p.Send(progressMsg(ratio))
+		},
+	}
+
+	m := downloadModel{
+		pw:       pw,
+		progress: progress.New(progress.WithDefaultGradient()),
+	}
+	p = tea.NewProgram(m)
+
+	// start the download
+	go pw.Start()
+
+	if progressBar {
+		if _, err := p.Run(); err != nil {
+			slog.Error("Error starting the progress bar", "error", err)
+		}
+	}
+
+	return err
+}
+
+func OpenArchive(ctx context.Context) (fsys fs.FS, err error) {
+	fsys, err = archives.FileSystem(ctx, filepath.Join(state.CachePath, zipFile), nil)
+	if err != nil {
+		return nil, err
+	}
+
+	return fsys, nil
+}
+
+func GetLangs() ([]string, error) {
+	var langs []string
+	fsys, err := OpenArchive(context.Background())
+	if err != nil {
+		slog.Error("Failed to open archive", "err", err)
+		return nil, err
+	}
+
+	entries, err := fs.ReadDir(fsys, ".")
+	if err != nil {
+		slog.Error("No files found in fsys", "err", err)
+		return nil, err
+	}
+
+	for _, entry := range entries {
+		if entry.IsDir() {
+			langs = append(langs, entry.Name())
+		}
+	}
+
+	return langs, nil
+}
+
+func GetProjects(lang string) ([]string, error) {
+	var projects []string
+	if lang == "" {
+		return nil, errors.New("lang must be set")
+	}
+
+	fsys, err := OpenArchive(context.Background())
+	if err != nil {
+		slog.Error("Failed to open archive", "err", err)
+		return nil, err
+	}
+
+	entries, err := fs.ReadDir(fsys, filepath.Join(".", lang))
+	if err != nil {
+		slog.Error("No files found in fsys", "err", err)
+		return nil, err
+	}
+
+	for _, entry := range entries {
+		if entry.IsDir() {
+			projects = append(projects, entry.Name())
+		}
+	}
+
+	return projects, nil
+}

cmd/template/template.go

@@ -0,0 +1,143 @@
+package template
+
+import (
+	"context"
+	"encoding/json"
+	"io/fs"
+	"log/slog"
+	"os"
+	"path/filepath"
+	"regexp"
+	"rph/state"
+)
+
+type TemplateOptions struct {
+	Lang string
+	ProjectType string
+	Dir string
+	Team uint64
+	DesktopSupport *bool
+}
+
+type wpilibPreferences struct {
+	CppIntellisense bool `json:"enableCppIntellisense"`
+	Lang string `json:"currentLanguage"`
+	Year string `json:"projectYear"`
+	Team int `json:"teamNumber"`
+}
+
+// Fetch fetch the latest template zip that's distributed by vscode-wpilib
+func Fetch(force bool, version string) {
+	os.MkdirAll(filepath.Join(state.CachePath), 0755)
+	getTemplateArchive(zipFile, force, version);
+}
+
+func GenerateProject(opts TemplateOptions) {
+	opts, err := openConfigUi(opts)
+	if err != nil {
+		slog.Error("Failed to run interactive ui", "error", err)
+		return
+	}
+
+	err = os.Mkdir(opts.Dir, 0755)
+	if err != nil {
+		slog.Error("Failed to create directory", "path", opts.Dir, "error", err)
+		// TODO: how should we handle if the directory already exists?
+	}
+
+	fsys, err := OpenArchive(context.Background())
+	if err != nil {
+		slog.Error("Failed to open archive", "err", err)
+		return
+	}
+
+	subFS, err := fs.Sub(fsys, filepath.Join(opts.Lang, opts.ProjectType))
+	if err != nil {
+		slog.Error("Unable to find project template", "template", opts.ProjectType, "error", err)
+	}
+
+	err = os.CopyFS(opts.Dir, subFS)
+	if err != nil {
+		slog.Error("Failed to copy template to destination", "template",
+			opts.ProjectType, "destination", opts.Dir, "error", err)
+		os.Exit(1)
+	}
+
+	// Configure the project
+
+	err = os.Chmod(filepath.Join(opts.Dir, "gradlew"), 0755)
+	if err != nil {
+		slog.Error("Unable to make gradlew executable", "error", err)
+	}
+
+	{
+		jsonFile := filepath.Join(opts.Dir, ".wpilib", "wpilib_preferences.json")
+		file, err := os.Open(jsonFile)
+		if err != nil {
+			slog.Error(
+				"Failed to open wpilib preferences file\n\nYou need to put your team number into " +
+				opts.Dir + "/.wpilib/wpilib_preferences.json",
+				"error",
+				err,
+				)
+			goto PostJson
+		}
+		defer file.Close()
+
+		{ // scope it so I can jump over it
+			var p wpilibPreferences
+			decoder := json.NewDecoder(file)
+			err = decoder.Decode(&p)
+			if err != nil {
+				slog.Error("Failed to decode json", "error", err)
+				goto PostJson
+			}
+
+			p.Team = int(opts.Team)
+
+			jsonData, err := json.MarshalIndent(p, "", "  ")
+			if err != nil {
+				slog.Error("Error marshaling JSON:", "error", err)
+				goto PostJson
+			}
+
+			err = os.WriteFile(jsonFile, jsonData, 0644)
+			if err != nil {
+				slog.Error("Error writing to file:", "error", err)
+				goto PostJson
+			}
+		}
+
+		PostJson:
+	}
+
+	// This is a big one
+	{
+		buildGradleFile := filepath.Join(opts.Dir, "build.gradle")
+    in, err := os.ReadFile(buildGradleFile)
+		if err != nil {
+			slog.Error("Failed to open gradle build file", "error", err)
+			goto PostGradle
+		}
+
+		{ // scope it so I can jump over it
+			value := "false"
+			if *opts.DesktopSupport {
+				value = "true"
+			}
+
+			// this regex was translated from:
+			// https://github.com/wpilibsuite/vscode-wpilib/blob/df7fc8bb9db453cbc9ccc32d3c5f81ef53f5e93a/vscode-wpilib/src/shared/generator.ts#L390
+			re := regexp.MustCompile(`(?m)^(\s*def\s+includeDesktopSupport\s*=\s*)(true|false)\b`)
+			out := re.ReplaceAllString(string(in), "${1}" + value)
+
+			err = os.WriteFile(buildGradleFile, []byte(out), 0644)
+			if err != nil {
+				slog.Error("failed to write to gradle build file", "error", err)
+				goto PostGradle
+			}
+		}
+
+		PostGradle:
+	}
+}