Added bento :)
This commit is contained in:
+243
-173
@@ -8,17 +8,19 @@ import (
|
||||
tea "github.com/charmbracelet/bubbletea"
|
||||
"github.com/charmbracelet/lipgloss"
|
||||
|
||||
"github.com/atridad/gitea-wrapped/pkg/gitea"
|
||||
"github.com/atridad/gitea-wrapped/pkg/models"
|
||||
"github.com/atridad/gitea-wrapped/pkg/stats"
|
||||
"github.com/atridad/wrapped-cli/pkg/config"
|
||||
"github.com/atridad/wrapped-cli/pkg/gitea"
|
||||
"github.com/atridad/wrapped-cli/pkg/models"
|
||||
"github.com/atridad/wrapped-cli/pkg/stats"
|
||||
)
|
||||
|
||||
type screenType int
|
||||
|
||||
const (
|
||||
inputScreenType screenType = iota
|
||||
loadingScreenType
|
||||
reportScreenType
|
||||
menuScreenType screenType = iota
|
||||
inputScreenType
|
||||
loadingScreenType
|
||||
reportScreenType
|
||||
)
|
||||
|
||||
type tickMsg time.Time
|
||||
@@ -26,215 +28,283 @@ type progressMsg string
|
||||
type errorMsg string
|
||||
|
||||
type doneMsg struct {
|
||||
repos []gitea.GiteaRepo
|
||||
commits []models.Commit
|
||||
username string
|
||||
repos []gitea.GiteaRepo
|
||||
commits []models.Commit
|
||||
username string
|
||||
}
|
||||
|
||||
type App struct {
|
||||
screen screenType
|
||||
inputScreen *InputScreen
|
||||
reportScreen *ReportScreen
|
||||
loadingMsg string
|
||||
err error
|
||||
giteaClient *gitea.Client
|
||||
analyzer *stats.Analyzer
|
||||
spinnerFrame int
|
||||
repos []gitea.GiteaRepo
|
||||
username string
|
||||
screen screenType
|
||||
menuScreen *MenuScreen
|
||||
inputScreen *InputScreen
|
||||
reportScreen *ReportScreen
|
||||
loadingMsg string
|
||||
err error
|
||||
giteaClient *gitea.Client
|
||||
analyzer *stats.Analyzer
|
||||
spinnerFrame int
|
||||
repos []gitea.GiteaRepo
|
||||
username string
|
||||
config *config.Config
|
||||
}
|
||||
|
||||
func NewApp() *App {
|
||||
return &App{
|
||||
screen: inputScreenType,
|
||||
inputScreen: NewInputScreen(),
|
||||
analyzer: stats.NewAnalyzer(),
|
||||
}
|
||||
cfg, err := config.LoadConfig()
|
||||
if err != nil || cfg == nil {
|
||||
cfg = &config.Config{Connections: []config.Connection{}}
|
||||
}
|
||||
|
||||
screen := menuScreenType
|
||||
var inputScreen *InputScreen
|
||||
if len(cfg.Connections) == 0 {
|
||||
screen = inputScreenType
|
||||
inputScreen = NewInputScreen()
|
||||
}
|
||||
|
||||
return &App{
|
||||
screen: screen,
|
||||
menuScreen: NewMenuScreen(cfg),
|
||||
inputScreen: inputScreen,
|
||||
config: cfg,
|
||||
analyzer: stats.NewAnalyzer(),
|
||||
}
|
||||
}
|
||||
|
||||
func (a *App) Init() tea.Cmd {
|
||||
return a.inputScreen.Init()
|
||||
if a.screen == inputScreenType && a.inputScreen != nil {
|
||||
return a.inputScreen.Init()
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (a *App) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
||||
switch msg := msg.(type) {
|
||||
case tea.KeyMsg:
|
||||
if msg.Type == tea.KeyCtrlC {
|
||||
return a, tea.Quit
|
||||
}
|
||||
case tickMsg:
|
||||
if a.screen == loadingScreenType {
|
||||
a.spinnerFrame = (a.spinnerFrame + 1) % 4
|
||||
return a, tea.Tick(time.Millisecond*200, func(t time.Time) tea.Msg {
|
||||
return tickMsg(t)
|
||||
})
|
||||
}
|
||||
}
|
||||
switch msg := msg.(type) {
|
||||
case tea.KeyMsg:
|
||||
if msg.Type == tea.KeyCtrlC {
|
||||
return a, tea.Quit
|
||||
}
|
||||
case tickMsg:
|
||||
if a.screen == loadingScreenType {
|
||||
a.spinnerFrame = (a.spinnerFrame + 1) % 4
|
||||
return a, tea.Tick(time.Millisecond*200, func(t time.Time) tea.Msg {
|
||||
return tickMsg(t)
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
switch a.screen {
|
||||
case inputScreenType:
|
||||
_, cmd := a.inputScreen.Update(msg)
|
||||
switch a.screen {
|
||||
case menuScreenType:
|
||||
_, cmd := a.menuScreen.Update(msg)
|
||||
|
||||
if a.inputScreen.IsDone() {
|
||||
url, username, token := a.inputScreen.GetCredentials()
|
||||
if a.menuScreen.IsNewConnection() {
|
||||
a.screen = inputScreenType
|
||||
a.menuScreen.ResetSelection()
|
||||
a.inputScreen = NewInputScreen()
|
||||
return a, a.inputScreen.Init()
|
||||
}
|
||||
|
||||
if url == "" {
|
||||
a.inputScreen.err = fmt.Errorf("URL is required")
|
||||
a.inputScreen.done = false
|
||||
return a, cmd
|
||||
}
|
||||
if username == "" {
|
||||
a.inputScreen.err = fmt.Errorf("username is required")
|
||||
a.inputScreen.done = false
|
||||
return a, cmd
|
||||
}
|
||||
if token == "" {
|
||||
a.inputScreen.err = fmt.Errorf("token is required")
|
||||
a.inputScreen.done = false
|
||||
return a, cmd
|
||||
}
|
||||
if selected := a.menuScreen.GetSelectedName(); selected != "" {
|
||||
conn := a.config.GetConnection(selected)
|
||||
if conn != nil {
|
||||
a.menuScreen.ResetSelection()
|
||||
a.giteaClient = gitea.NewClient(conn.URL, conn.Token)
|
||||
a.username = conn.Username
|
||||
a.screen = loadingScreenType
|
||||
a.loadingMsg = "Connecting to Gitea..."
|
||||
a.spinnerFrame = 0
|
||||
|
||||
a.giteaClient = gitea.NewClient(url, token)
|
||||
a.username = username
|
||||
a.screen = loadingScreenType
|
||||
a.loadingMsg = "Connecting to Gitea..."
|
||||
a.spinnerFrame = 0
|
||||
return a, tea.Batch(
|
||||
tea.Tick(time.Millisecond*200, func(t time.Time) tea.Msg {
|
||||
return tickMsg(t)
|
||||
}),
|
||||
a.testConnection())
|
||||
}
|
||||
}
|
||||
|
||||
return a, tea.Batch(cmd,
|
||||
tea.Tick(time.Millisecond*200, func(t time.Time) tea.Msg {
|
||||
return tickMsg(t)
|
||||
}),
|
||||
a.testConnection())
|
||||
}
|
||||
return a, cmd
|
||||
|
||||
return a, cmd
|
||||
case inputScreenType:
|
||||
_, cmd := a.inputScreen.Update(msg)
|
||||
|
||||
case loadingScreenType:
|
||||
switch msg := msg.(type) {
|
||||
case progressMsg:
|
||||
a.loadingMsg = string(msg)
|
||||
if a.inputScreen.IsDone() {
|
||||
url, username, token, connName := a.inputScreen.GetCredentials()
|
||||
|
||||
switch string(msg) {
|
||||
case "Fetching repositories...":
|
||||
return a, a.fetchRepos()
|
||||
case "Fetching commits...":
|
||||
return a, a.fetchCommits()
|
||||
}
|
||||
return a, nil
|
||||
if url == "" {
|
||||
a.inputScreen.err = fmt.Errorf("URL is required")
|
||||
a.inputScreen.done = false
|
||||
return a, cmd
|
||||
}
|
||||
if username == "" {
|
||||
a.inputScreen.err = fmt.Errorf("username is required")
|
||||
a.inputScreen.done = false
|
||||
return a, cmd
|
||||
}
|
||||
if token == "" {
|
||||
a.inputScreen.err = fmt.Errorf("token is required")
|
||||
a.inputScreen.done = false
|
||||
return a, cmd
|
||||
}
|
||||
|
||||
case errorMsg:
|
||||
a.err = fmt.Errorf("%s", msg)
|
||||
a.loadingMsg = "Error"
|
||||
return a, nil
|
||||
if connName != "" {
|
||||
a.config.AddConnection(config.Connection{
|
||||
Name: connName,
|
||||
URL: url,
|
||||
Username: username,
|
||||
Token: token,
|
||||
})
|
||||
a.config.Save()
|
||||
}
|
||||
|
||||
case doneMsg:
|
||||
a.analyzer.AddRepos(msg.repos)
|
||||
a.analyzer.AddCommits(msg.commits)
|
||||
userStats := a.analyzer.Generate(msg.username)
|
||||
a.reportScreen = NewReportScreen(userStats)
|
||||
a.screen = reportScreenType
|
||||
return a, nil
|
||||
}
|
||||
a.giteaClient = gitea.NewClient(url, token)
|
||||
a.username = username
|
||||
a.screen = loadingScreenType
|
||||
a.loadingMsg = "Connecting to Gitea..."
|
||||
a.spinnerFrame = 0
|
||||
|
||||
case reportScreenType:
|
||||
if a.reportScreen != nil {
|
||||
_, cmd := a.reportScreen.Update(msg)
|
||||
return a, cmd
|
||||
}
|
||||
}
|
||||
return a, tea.Batch(cmd,
|
||||
tea.Tick(time.Millisecond*200, func(t time.Time) tea.Msg {
|
||||
return tickMsg(t)
|
||||
}),
|
||||
a.testConnection())
|
||||
}
|
||||
|
||||
return a, nil
|
||||
return a, cmd
|
||||
|
||||
case loadingScreenType:
|
||||
switch msg := msg.(type) {
|
||||
case progressMsg:
|
||||
a.loadingMsg = string(msg)
|
||||
|
||||
switch string(msg) {
|
||||
case "Fetching repositories...":
|
||||
return a, a.fetchRepos()
|
||||
case "Fetching commits...":
|
||||
return a, a.fetchCommits()
|
||||
}
|
||||
return a, nil
|
||||
|
||||
case errorMsg:
|
||||
a.err = fmt.Errorf("%s", msg)
|
||||
a.loadingMsg = "Error"
|
||||
return a, nil
|
||||
|
||||
case doneMsg:
|
||||
a.analyzer.AddRepos(msg.repos)
|
||||
a.analyzer.AddCommits(msg.commits)
|
||||
userStats := a.analyzer.Generate(msg.username)
|
||||
a.reportScreen = NewReportScreen(userStats)
|
||||
a.screen = reportScreenType
|
||||
return a, nil
|
||||
}
|
||||
|
||||
case reportScreenType:
|
||||
switch msg := msg.(type) {
|
||||
case tea.KeyMsg:
|
||||
if msg.String() == "q" {
|
||||
a.screen = menuScreenType
|
||||
a.menuScreen = NewMenuScreen(a.config)
|
||||
a.reportScreen = nil
|
||||
a.analyzer = stats.NewAnalyzer()
|
||||
return a, nil
|
||||
}
|
||||
}
|
||||
if a.reportScreen != nil {
|
||||
_, cmd := a.reportScreen.Update(msg)
|
||||
return a, cmd
|
||||
}
|
||||
}
|
||||
|
||||
return a, nil
|
||||
}
|
||||
|
||||
func (a *App) View() string {
|
||||
switch a.screen {
|
||||
case inputScreenType:
|
||||
return a.inputScreen.View()
|
||||
case loadingScreenType:
|
||||
return a.renderLoadingScreen()
|
||||
case reportScreenType:
|
||||
if a.reportScreen != nil {
|
||||
return a.reportScreen.View()
|
||||
}
|
||||
}
|
||||
return ""
|
||||
switch a.screen {
|
||||
case menuScreenType:
|
||||
return a.menuScreen.View()
|
||||
case inputScreenType:
|
||||
return a.inputScreen.View()
|
||||
case loadingScreenType:
|
||||
return a.renderLoadingScreen()
|
||||
case reportScreenType:
|
||||
if a.reportScreen != nil {
|
||||
return a.reportScreen.View()
|
||||
}
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
func (a *App) renderLoadingScreen() string {
|
||||
var s string
|
||||
s += lipgloss.NewStyle().
|
||||
Foreground(lipgloss.Color("212")).
|
||||
Bold(true).
|
||||
Render("🎵 Gitea Wrapped") + "\n\n"
|
||||
var s string
|
||||
s += lipgloss.NewStyle().
|
||||
Foreground(lipgloss.Color("212")).
|
||||
Bold(true).
|
||||
Render("Gitea Wrapped") + "\n\n"
|
||||
|
||||
spinners := []string{"⠋", "⠙", "⠹", "⠸"}
|
||||
spinner := spinners[a.spinnerFrame%4]
|
||||
spinners := []string{"⠋", "⠙", "⠹", "⠸"}
|
||||
spinner := spinners[a.spinnerFrame%4]
|
||||
|
||||
s += fmt.Sprintf("%s %s\n", spinner, a.loadingMsg)
|
||||
s += fmt.Sprintf("%s %s\n", spinner, a.loadingMsg)
|
||||
|
||||
if a.err != nil {
|
||||
s += "\n" + lipgloss.NewStyle().
|
||||
Foreground(lipgloss.Color("1")).
|
||||
Render(fmt.Sprintf("Error: %v", a.err)) + "\n"
|
||||
if a.err != nil {
|
||||
s += "\n" + lipgloss.NewStyle().
|
||||
Foreground(lipgloss.Color("1")).
|
||||
Render(fmt.Sprintf("Error: %v", a.err)) + "\n"
|
||||
}
|
||||
|
||||
return s
|
||||
}
|
||||
|
||||
return s
|
||||
}
|
||||
|
||||
// testConnection tests API connection
|
||||
func (a *App) testConnection() tea.Cmd {
|
||||
return func() tea.Msg {
|
||||
if err := a.giteaClient.TestConnection(); err != nil {
|
||||
return errorMsg(fmt.Sprintf("failed to connect: %v", err))
|
||||
}
|
||||
return progressMsg("Fetching repositories...")
|
||||
}
|
||||
return func() tea.Msg {
|
||||
if err := a.giteaClient.TestConnection(); err != nil {
|
||||
return errorMsg(fmt.Sprintf("failed to connect: %v", err))
|
||||
}
|
||||
return progressMsg("Fetching repositories...")
|
||||
}
|
||||
}
|
||||
|
||||
// fetchRepos fetches all repositories
|
||||
func (a *App) fetchRepos() tea.Cmd {
|
||||
return func() tea.Msg {
|
||||
repos, err := a.giteaClient.GetUserRepos()
|
||||
if err != nil {
|
||||
return errorMsg(fmt.Sprintf("failed to fetch repos: %v", err))
|
||||
return func() tea.Msg {
|
||||
repos, err := a.giteaClient.GetUserRepos()
|
||||
if err != nil {
|
||||
return errorMsg(fmt.Sprintf("failed to fetch repos: %v", err))
|
||||
}
|
||||
|
||||
a.repos = repos
|
||||
return progressMsg("Fetching commits...")
|
||||
}
|
||||
}
|
||||
|
||||
a.repos = repos
|
||||
return progressMsg("Fetching commits...")
|
||||
}
|
||||
}
|
||||
|
||||
// fetchCommits fetches commits from all repos
|
||||
func (a *App) fetchCommits() tea.Cmd {
|
||||
return func() tea.Msg {
|
||||
var commits []models.Commit
|
||||
return func() tea.Msg {
|
||||
var repoRefs []gitea.RepoRef
|
||||
for _, repo := range a.repos {
|
||||
parts := strings.Split(repo.FullName, "/")
|
||||
if len(parts) < 2 {
|
||||
continue
|
||||
}
|
||||
repoRefs = append(repoRefs, gitea.RepoRef{
|
||||
Owner: parts[0],
|
||||
Name: repo.Name,
|
||||
})
|
||||
}
|
||||
|
||||
for i, repo := range a.repos {
|
||||
giteaCommits, err := a.giteaClient.GetReposCommitsParallel(repoRefs)
|
||||
if err != nil {
|
||||
return errorMsg(fmt.Sprintf("failed to fetch commits: %v", err))
|
||||
}
|
||||
|
||||
parts := strings.Split(repo.FullName, "/")
|
||||
if len(parts) < 2 {
|
||||
continue
|
||||
}
|
||||
|
||||
owner := parts[0]
|
||||
giteaCommits, err := a.giteaClient.GetRepoCommits(owner, repo.Name)
|
||||
if err != nil {
|
||||
continue
|
||||
}
|
||||
|
||||
|
||||
for _, gc := range giteaCommits {
|
||||
commits = append(commits, models.Commit{
|
||||
SHA: gc.SHA,
|
||||
Message: gc.Commit.Message,
|
||||
Author: gc.Commit.Author.Name,
|
||||
Timestamp: gc.Commit.Author.Date,
|
||||
RepoName: repo.Name,
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
return doneMsg{a.repos, commits, a.username}
|
||||
}
|
||||
var commits []models.Commit
|
||||
for _, gc := range giteaCommits {
|
||||
commits = append(commits, models.Commit{
|
||||
SHA: gc.SHA,
|
||||
Message: gc.Commit.Message,
|
||||
Author: gc.Commit.Author.Name,
|
||||
AuthorEmail: gc.Commit.Author.Email,
|
||||
Timestamp: gc.Commit.Author.Date,
|
||||
RepoName: gc.RepoName,
|
||||
})
|
||||
}
|
||||
return doneMsg{a.repos, commits, a.username}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user