auth_model.go

package main

import (
	"fmt"

	tea "github.com/charmbracelet/bubbletea"
	"github.com/charmbracelet/huh"
)

type authState int

const (
	authIdle authState = iota
	authPrompt
)

type authModel struct {
	form   *huh.Form
	state  authState
	width  int
	height int
}

type authenticated struct{}

type needSecretsMsg struct{}

func (m authModel) Init() tea.Cmd {
	if cfg.AccessToken == "" {
		return func() tea.Msg { return needSecretsMsg{} }
	}
	return func() tea.Msg { return authenticated{} }
}

func (m authModel) View() string {
	if m.form != nil {
		return m.form.View()
	}
	return "Input required"
}

func (m authModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
	var cmd tea.Cmd
	var cmds []tea.Cmd
	switch msg := msg.(type) {
	case tea.WindowSizeMsg:
		m.width, m.height = msg.Width, msg.Height
	case tea.KeyMsg:
		switch msg.String() {
		case "ctrl+c":
			return m, tea.Quit
		}
		if m.form != nil {
			var f tea.Model
			f, cmd = m.form.Update(msg)
			if form, ok := f.(*huh.Form); ok {
				m.form = form
			}
			cmds = append(cmds, cmd)
			return m, tea.Batch(cmds...)
		}
	case needSecretsMsg:
		m.state = authPrompt
		m.form = huh.NewForm(
			huh.NewGroup(
				huh.NewInput().
					Title("Github Personal Token").
					Value(&cfg.AccessToken).
					Key("access_token").
					Validate(func(s string) error {
						if s == "" {
							return fmt.Errorf("client id required")
						}
						return nil
					}),
			),
		)
		if m.width > 0 && m.height > 0 {
			var f tea.Model
			f, _ = m.form.Update(tea.WindowSizeMsg{Width: m.width, Height: m.height})
			if form, ok := f.(*huh.Form); ok {
				m.form = form
			}
		}
		return m, m.form.Init()
	case infoMsg:
		if msg.variant == info_error {
			log.Errorln(msg.msg)
		}
		return m, nil
	default:
		// only run this block of code when we're submitting the form
		if m.state == authPrompt && m.form != nil {
			var f tea.Model
			f, cmd = m.form.Update(msg)
			if form, ok := f.(*huh.Form); ok {
				m.form = form
			}
			cmds = append(cmds, cmd)

			if m.form.State == huh.StateCompleted {
				access_token := m.form.GetString("access_token")

				if err := cfg.set("AccessToken", access_token); err != nil {
					panic(err)
				}

				client, err := createGhClient()
				if err != nil {
					panic(err)
				}
				githubClient = client
				if err := preloadGists(); err != nil {
					panic(err)
				}
				cmds = append(cmds, func() tea.Msg { return authenticated{} })
				m.state = authIdle
				return m, tea.Batch(cmds...)
			}
		}
	}
	cmds = append(cmds, cmd)
	return m, tea.Batch(cmds...)
}