406 lines
9.7 KiB
Go
406 lines
9.7 KiB
Go
|
// cctuip is a Co-op Cloud TUI.
|
||
|
package main
|
||
|
|
||
|
import (
|
||
|
"flag"
|
||
|
"fmt"
|
||
|
"log"
|
||
|
"os"
|
||
|
"sort"
|
||
|
"strings"
|
||
|
|
||
|
abraClient "coopcloud.tech/abra/pkg/client"
|
||
|
"coopcloud.tech/abra/pkg/config"
|
||
|
"coopcloud.tech/abra/pkg/recipe"
|
||
|
"coopcloud.tech/tagcmp"
|
||
|
"github.com/charmbracelet/bubbles/key"
|
||
|
"github.com/charmbracelet/bubbles/spinner"
|
||
|
tea "github.com/charmbracelet/bubbletea"
|
||
|
"github.com/charmbracelet/lipgloss"
|
||
|
dockerClient "github.com/docker/docker/client"
|
||
|
"github.com/evertras/bubble-table/table"
|
||
|
"golang.org/x/exp/slices"
|
||
|
"golang.org/x/term"
|
||
|
)
|
||
|
|
||
|
// help is the cctuip CLI help output.
|
||
|
const help = `cctuip [options]
|
||
|
|
||
|
cctuip is a Co-op Cloud TUI.
|
||
|
|
||
|
Options:
|
||
|
-h output help
|
||
|
`
|
||
|
|
||
|
var helpFlag bool
|
||
|
|
||
|
// handleCliFlags parses CLI flags.
|
||
|
func handleCliFlags() error {
|
||
|
flag.BoolVar(&helpFlag, "h", false, "output help")
|
||
|
flag.Parse()
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
// getClient retrieves a docker client from the abra API.
|
||
|
func getClient(server string) (*dockerClient.Client, error) {
|
||
|
cl, err := abraClient.New(server)
|
||
|
if err != nil {
|
||
|
return nil, fmt.Errorf("getClient: %s", err)
|
||
|
}
|
||
|
return cl, nil
|
||
|
}
|
||
|
|
||
|
// getApps retrieves app metadata from the abra API.
|
||
|
func getApps() ([]config.App, error) {
|
||
|
appFiles, err := config.LoadAppFiles("")
|
||
|
if err != nil {
|
||
|
return []config.App{}, fmt.Errorf("getApps: %s", err)
|
||
|
}
|
||
|
|
||
|
apps, err := config.GetApps(appFiles, "")
|
||
|
if err != nil {
|
||
|
return []config.App{}, fmt.Errorf("getApps: %s", err)
|
||
|
}
|
||
|
|
||
|
sort.Sort(config.ByName(apps))
|
||
|
|
||
|
return apps, nil
|
||
|
}
|
||
|
|
||
|
// getNumServersAndRecipes totals servers and recipes.
|
||
|
func getNumServersAndRecipes(apps []config.App) (int, int) {
|
||
|
var (
|
||
|
servers []string
|
||
|
recipes []string
|
||
|
)
|
||
|
|
||
|
for _, app := range apps {
|
||
|
if !slices.Contains(servers, app.Server) {
|
||
|
servers = append(servers, app.Server)
|
||
|
}
|
||
|
if !slices.Contains(recipes, app.Recipe) {
|
||
|
recipes = append(recipes, app.Recipe)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return len(servers), len(recipes)
|
||
|
}
|
||
|
|
||
|
// errorMsg delivers errors to the UI.
|
||
|
type errorMsg struct{ err error }
|
||
|
|
||
|
// Error implements error output rendering.
|
||
|
func (e errorMsg) Error() string { return e.err.Error() }
|
||
|
|
||
|
// appsDeployStatusMsg delivers the deployment status of all apps to the UI.
|
||
|
type appsDeployStatusMsg map[string]map[string]string
|
||
|
|
||
|
// getAppsDeployStatus retrieves apps deployment status from a server.
|
||
|
func getAppsDeployStatus(m model) tea.Msg {
|
||
|
var apps []config.App
|
||
|
|
||
|
for _, row := range m.table.GetVisibleRows() {
|
||
|
apps = append(apps, row.Data["app"].(config.App))
|
||
|
}
|
||
|
|
||
|
statuses, err := config.GetAppStatuses(apps, true)
|
||
|
if err != nil {
|
||
|
return errorMsg{err}
|
||
|
}
|
||
|
|
||
|
catl, err := recipe.ReadRecipeCatalogue()
|
||
|
if err != nil {
|
||
|
return errorMsg{err}
|
||
|
}
|
||
|
|
||
|
for _, app := range apps {
|
||
|
var newUpdates []string
|
||
|
if status, ok := statuses[app.StackName()]; ok {
|
||
|
if version, ok := status["version"]; ok {
|
||
|
updates, err := recipe.GetRecipeCatalogueVersions(app.Recipe, catl)
|
||
|
if err != nil {
|
||
|
return errorMsg{err}
|
||
|
}
|
||
|
|
||
|
parsedVersion, err := tagcmp.Parse(version)
|
||
|
if err != nil {
|
||
|
statuses[app.StackName()]["updates"] = "🤷"
|
||
|
continue
|
||
|
}
|
||
|
|
||
|
for _, update := range updates {
|
||
|
parsedUpdate, err := tagcmp.Parse(update)
|
||
|
if err != nil {
|
||
|
statuses[app.StackName()]["updates"] = "🤷"
|
||
|
continue
|
||
|
}
|
||
|
|
||
|
if update != version && parsedUpdate.IsGreaterThan(parsedVersion) {
|
||
|
newUpdates = append(newUpdates, update)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if len(newUpdates) == 0 {
|
||
|
statuses[app.StackName()]["updates"] = "✅"
|
||
|
} else {
|
||
|
statuses[app.StackName()]["updates"] = fmt.Sprintf("%v", len(newUpdates))
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return appsDeployStatusMsg(statuses)
|
||
|
}
|
||
|
|
||
|
func renderAppsDeployStatus(m *model, appStatuses appsDeployStatusMsg) table.Model {
|
||
|
for _, row := range m.table.GetVisibleRows() {
|
||
|
app := row.Data["app"].(config.App)
|
||
|
appStatus := appStatuses[app.StackName()]
|
||
|
var (
|
||
|
version = appStatus["version"]
|
||
|
updates = appStatus["updates"]
|
||
|
status = appStatus["status"]
|
||
|
chaos = appStatus["chaos"]
|
||
|
chaosVersion = appStatus["chaosVersion"]
|
||
|
autoUpdate = appStatus["autoUpdate"]
|
||
|
)
|
||
|
if status == "" {
|
||
|
row.Data["status"] = "🤷"
|
||
|
} else {
|
||
|
row.Data["status"] = status
|
||
|
}
|
||
|
if version == "" {
|
||
|
row.Data["version"] = "🤷"
|
||
|
} else {
|
||
|
row.Data["version"] = version
|
||
|
row.Data["updates"] = updates
|
||
|
}
|
||
|
if chaos == "" {
|
||
|
row.Data["chaos"] = "🤷"
|
||
|
} else {
|
||
|
row.Data["chaos"] = chaos
|
||
|
}
|
||
|
if chaosVersion == "" {
|
||
|
row.Data["chaos-version"] = "🤷"
|
||
|
} else {
|
||
|
row.Data["chaos-version"] = chaosVersion
|
||
|
}
|
||
|
if autoUpdate == "" {
|
||
|
row.Data["autoUpdate"] = "🤷"
|
||
|
} else {
|
||
|
row.Data["autoUpdate"] = autoUpdate
|
||
|
}
|
||
|
}
|
||
|
return m.table
|
||
|
}
|
||
|
|
||
|
type initTableMsg struct{ table table.Model }
|
||
|
|
||
|
// initTable loads the table layout from local-first data sources (~/.abra).
|
||
|
func initTable(m model) tea.Msg {
|
||
|
var rows []table.Row
|
||
|
|
||
|
for _, app := range m.apps {
|
||
|
rows = append(rows, table.NewRow(table.RowData{
|
||
|
"domain": app.Domain,
|
||
|
"server": app.Server,
|
||
|
"recipe": app.Recipe,
|
||
|
"status": "🤷",
|
||
|
"version": "🤷",
|
||
|
"updates": "🤷",
|
||
|
"chaos": "🤷",
|
||
|
"chaos-version": "🤷",
|
||
|
"auto-update": "🤷",
|
||
|
"app": app, // attach app itself for faster lookups
|
||
|
}))
|
||
|
}
|
||
|
|
||
|
colStyle := lipgloss.NewStyle().Align(lipgloss.Left)
|
||
|
columns := []table.Column{
|
||
|
table.NewFlexColumn("domain", "Domain", 2).WithFiltered(true).WithStyle(colStyle),
|
||
|
table.NewFlexColumn("server", "Server", 1).WithFiltered(true).WithStyle(colStyle),
|
||
|
table.NewFlexColumn("recipe", "Recipe", 1).WithFiltered(true).WithStyle(colStyle),
|
||
|
table.NewFlexColumn("status", "Status", 1).WithStyle(colStyle),
|
||
|
table.NewFlexColumn("version", "Version", 2).WithStyle(colStyle),
|
||
|
table.NewFlexColumn("updates", "Updates", 2).WithStyle(colStyle),
|
||
|
table.NewFlexColumn("chaos", "Chaos", 1).WithStyle(colStyle),
|
||
|
table.NewFlexColumn("chaos-version", "Chaos version", 2).WithStyle(colStyle),
|
||
|
table.NewFlexColumn("auto-update", "Auto-update", 1).WithStyle(colStyle),
|
||
|
}
|
||
|
|
||
|
width, height, err := term.GetSize(0)
|
||
|
if err != nil {
|
||
|
log.Fatal(err) // TODO
|
||
|
}
|
||
|
|
||
|
keymap := table.DefaultKeyMap()
|
||
|
keymap.Filter = key.NewBinding(key.WithKeys("/", "f"))
|
||
|
keymap.PageDown = key.NewBinding(key.WithKeys("right", "l", "pgdown", "ctrl+d"))
|
||
|
keymap.PageUp = key.NewBinding(key.WithKeys("left", "h", "pgup", "ctrl+u"))
|
||
|
|
||
|
t := table.
|
||
|
New(columns).
|
||
|
Filtered(true).
|
||
|
Focused(true).
|
||
|
WithPageSize(height - 10).
|
||
|
WithRows([]table.Row(rows)).
|
||
|
WithTargetWidth(width).
|
||
|
WithKeyMap(keymap)
|
||
|
|
||
|
return initTableMsg{table: t}
|
||
|
}
|
||
|
|
||
|
// model is the TUI application state.
|
||
|
type model struct {
|
||
|
apps []config.App
|
||
|
|
||
|
numApps int
|
||
|
numServers int
|
||
|
numRecipes int
|
||
|
|
||
|
numFilteredApps int
|
||
|
numFilteredServers int
|
||
|
numFilteredRecipes int
|
||
|
|
||
|
table table.Model
|
||
|
spinner spinner.Model
|
||
|
pollingStatus bool
|
||
|
|
||
|
err error
|
||
|
}
|
||
|
|
||
|
// getFilteredApps retrieves all visible apps.
|
||
|
func (m model) getFilteredApps() []config.App {
|
||
|
var servers []config.App
|
||
|
for _, row := range m.table.GetVisibleRows() {
|
||
|
servers = append(servers, row.Data["app"].(config.App))
|
||
|
}
|
||
|
return servers
|
||
|
}
|
||
|
|
||
|
// updateCount updates the apps/servers/recipes count.
|
||
|
func (m *model) updateCount() {
|
||
|
if m.table.GetIsFilterActive() {
|
||
|
apps := m.getFilteredApps()
|
||
|
m.numFilteredApps = len(apps)
|
||
|
m.numFilteredServers, m.numFilteredRecipes = getNumServersAndRecipes(apps)
|
||
|
} else {
|
||
|
m.numFilteredApps = m.numApps
|
||
|
m.numFilteredServers = m.numServers
|
||
|
m.numFilteredRecipes = m.numRecipes
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// Init initialises a new model. All I/O happens here and the results are fed
|
||
|
// into the UI for further updates and rendering.
|
||
|
func (m model) Init() tea.Cmd {
|
||
|
return tea.Batch(
|
||
|
func() tea.Msg { return initTable(m) },
|
||
|
m.spinner.Tick,
|
||
|
)
|
||
|
}
|
||
|
|
||
|
// Update handles updates to the TUI via I/O and the user.
|
||
|
func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
||
|
var cmd tea.Cmd
|
||
|
var cmds []tea.Cmd
|
||
|
|
||
|
m.table, cmd = m.table.Update(msg)
|
||
|
cmds = append(cmds, cmd)
|
||
|
|
||
|
m.spinner, cmd = m.spinner.Update(msg)
|
||
|
cmds = append(cmds, cmd)
|
||
|
|
||
|
switch msg := msg.(type) {
|
||
|
case tea.KeyMsg:
|
||
|
m.updateCount()
|
||
|
|
||
|
switch msg.String() {
|
||
|
case "q":
|
||
|
return m, tea.Quit
|
||
|
case "s":
|
||
|
if !m.table.GetIsFilterInputFocused() {
|
||
|
m.pollingStatus = true
|
||
|
return m, func() tea.Msg { return getAppsDeployStatus(m) }
|
||
|
}
|
||
|
}
|
||
|
case initTableMsg:
|
||
|
m.table = msg.table
|
||
|
case appsDeployStatusMsg:
|
||
|
m.pollingStatus = false
|
||
|
m.table = renderAppsDeployStatus(&m, msg)
|
||
|
case tea.WindowSizeMsg:
|
||
|
m.table = m.table.WithTargetWidth(msg.Width)
|
||
|
m.table = m.table.WithPageSize(msg.Height - 10)
|
||
|
case errorMsg:
|
||
|
m.err = msg // TODO
|
||
|
}
|
||
|
|
||
|
return m, tea.Batch(cmds...)
|
||
|
}
|
||
|
|
||
|
// View renders the UI.
|
||
|
func (m model) View() string {
|
||
|
if m.err != nil {
|
||
|
// TODO
|
||
|
return fmt.Sprintf("\nWe had some trouble: %v\n\n", m.err)
|
||
|
}
|
||
|
|
||
|
body := strings.Builder{}
|
||
|
|
||
|
body.WriteString(fmt.Sprintf(
|
||
|
"Servers: %v, Apps: %v, Recipes: %v\n",
|
||
|
m.numFilteredServers, m.numFilteredApps, m.numFilteredRecipes,
|
||
|
))
|
||
|
|
||
|
body.WriteString(m.table.View() + "\n")
|
||
|
body.WriteString("cctuip v0.1.0")
|
||
|
|
||
|
if m.pollingStatus {
|
||
|
body.WriteString(fmt.Sprintf(" %s querying app status...", m.spinner.View()))
|
||
|
}
|
||
|
|
||
|
return body.String()
|
||
|
}
|
||
|
|
||
|
// main is the command-line entrypoint.
|
||
|
func main() {
|
||
|
handleCliFlags()
|
||
|
|
||
|
if helpFlag {
|
||
|
fmt.Print(help)
|
||
|
os.Exit(0)
|
||
|
}
|
||
|
|
||
|
apps, err := getApps()
|
||
|
if err != nil {
|
||
|
log.Fatal(err)
|
||
|
}
|
||
|
|
||
|
numServers, numRecipes := getNumServersAndRecipes(apps)
|
||
|
if err != nil {
|
||
|
log.Fatal(err)
|
||
|
}
|
||
|
|
||
|
s := spinner.New()
|
||
|
s.Spinner = spinner.Dot
|
||
|
s.Style = lipgloss.NewStyle().Foreground(lipgloss.Color("205"))
|
||
|
|
||
|
numApps := len(apps)
|
||
|
m := model{
|
||
|
apps: apps,
|
||
|
numApps: numApps,
|
||
|
numServers: numServers,
|
||
|
numRecipes: numRecipes,
|
||
|
numFilteredApps: numApps,
|
||
|
numFilteredServers: numServers,
|
||
|
numFilteredRecipes: numRecipes,
|
||
|
spinner: s,
|
||
|
}
|
||
|
|
||
|
p := tea.NewProgram(m, tea.WithAltScreen())
|
||
|
if _, err := p.Run(); err != nil {
|
||
|
log.Fatalf("oops, cctuip exploded: %s", err)
|
||
|
}
|
||
|
}
|