feat: initial implementation of EnvGuard with improved TUI layout
This commit is contained in:
Jose Luis Montañes Ojados
253
cmd/cli/main.go
Normal file
253
cmd/cli/main.go
Normal file
@@ -0,0 +1,253 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"os"
|
||||
"time"
|
||||
|
||||
"envguard/internal/api"
|
||||
"envguard/internal/models"
|
||||
|
||||
"github.com/charmbracelet/bubbles/table"
|
||||
tea "github.com/charmbracelet/bubbletea"
|
||||
"github.com/charmbracelet/lipgloss"
|
||||
)
|
||||
|
||||
var baseStyle = lipgloss.NewStyle().
|
||||
BorderStyle(lipgloss.NormalBorder()).
|
||||
BorderForeground(lipgloss.Color("240"))
|
||||
|
||||
type model struct {
|
||||
table table.Model
|
||||
services []models.Service
|
||||
user string
|
||||
err error
|
||||
message string
|
||||
width int
|
||||
height int
|
||||
}
|
||||
|
||||
func (m model) Init() tea.Cmd {
|
||||
return fetchServices
|
||||
}
|
||||
|
||||
func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
||||
var cmd tea.Cmd
|
||||
switch msg := msg.(type) {
|
||||
case tea.KeyMsg:
|
||||
switch msg.String() {
|
||||
case "esc", "q", "ctrl+c":
|
||||
return m, tea.Quit
|
||||
case "enter":
|
||||
selected := m.table.SelectedRow()
|
||||
if selected == nil {
|
||||
return m, nil
|
||||
}
|
||||
serviceName := selected[0]
|
||||
// Find service state
|
||||
for _, s := range m.services {
|
||||
if s.Name == serviceName {
|
||||
if s.IsLocked {
|
||||
if s.LockedBy == m.user {
|
||||
// Unlock
|
||||
err := api.UnlockService(serviceName, m.user)
|
||||
if err != nil {
|
||||
m.message = "Error unlocking: " + err.Error()
|
||||
} else {
|
||||
m.message = "Unlocked " + serviceName
|
||||
}
|
||||
} else {
|
||||
m.message = "Cannot unlock service locked by " + s.LockedBy
|
||||
}
|
||||
} else {
|
||||
// Lock
|
||||
err := api.LockService(serviceName, m.user)
|
||||
if err != nil {
|
||||
m.message = "Error locking: " + err.Error()
|
||||
} else {
|
||||
m.message = "Locked " + serviceName
|
||||
}
|
||||
}
|
||||
return m, fetchServices
|
||||
}
|
||||
}
|
||||
case "r":
|
||||
return m, fetchServices
|
||||
}
|
||||
case tea.WindowSizeMsg:
|
||||
m.width = msg.Width
|
||||
m.height = msg.Height
|
||||
|
||||
// Calculate available dimensions
|
||||
// Borders take 2 chars total width (1 left + 1 right)
|
||||
// We add extra safety margin (total -6) to avoid last-column wrap issues
|
||||
availWidth := m.width - 10
|
||||
|
||||
// Adjust column widths dynamically
|
||||
// Total weight = 100%
|
||||
cols := m.table.Columns()
|
||||
// Calculate widths ensuring at least minimums
|
||||
c0 := int(float64(availWidth) * 0.3)
|
||||
c1 := int(float64(availWidth) * 0.2)
|
||||
c2 := int(float64(availWidth) * 0.2)
|
||||
c3 := availWidth - c0 - c1 - c2
|
||||
|
||||
cols[0].Width = c0
|
||||
cols[1].Width = c1
|
||||
cols[2].Width = c2
|
||||
cols[3].Width = c3
|
||||
|
||||
m.table.SetColumns(cols)
|
||||
m.table.SetWidth(availWidth)
|
||||
|
||||
// Header + Footer + padding calculation
|
||||
availableHeight := m.height - 7
|
||||
if availableHeight < 5 {
|
||||
availableHeight = 5
|
||||
}
|
||||
m.table.SetHeight(availableHeight)
|
||||
case []models.Service:
|
||||
m.services = msg
|
||||
rows := []table.Row{}
|
||||
for _, s := range msg {
|
||||
status := "🟢 LIBRE"
|
||||
if s.IsLocked {
|
||||
status = "🔴 LOCKED"
|
||||
}
|
||||
user := s.LockedBy
|
||||
if user == "" {
|
||||
user = "--"
|
||||
}
|
||||
timeStr := "--"
|
||||
if s.IsLocked {
|
||||
duration := time.Since(s.LockedAt)
|
||||
timeStr = fmt.Sprintf("Hace %s", duration.Round(time.Second))
|
||||
}
|
||||
|
||||
rows = append(rows, table.Row{s.Name, status, user, timeStr})
|
||||
}
|
||||
m.table.SetRows(rows)
|
||||
m.message = "Refreshed"
|
||||
case error:
|
||||
m.err = msg
|
||||
}
|
||||
m.table, cmd = m.table.Update(msg)
|
||||
return m, cmd
|
||||
}
|
||||
|
||||
func (m model) View() string {
|
||||
if m.err != nil {
|
||||
return fmt.Sprintf("Error: %v\nPress q to quit", m.err)
|
||||
}
|
||||
|
||||
if m.width == 0 {
|
||||
return "Loading..."
|
||||
}
|
||||
|
||||
userInfo := fmt.Sprintf("User: %s", m.user)
|
||||
if m.message != "" {
|
||||
userInfo += " | " + m.message
|
||||
}
|
||||
|
||||
// Dimensions
|
||||
appWidth := m.width - 2
|
||||
appHeight := m.height - 2
|
||||
if appWidth < 20 {
|
||||
appWidth = 20
|
||||
}
|
||||
if appHeight < 10 {
|
||||
appHeight = 10
|
||||
}
|
||||
|
||||
// Content inside border
|
||||
contentWidth := appWidth - 2
|
||||
|
||||
// Local Styles
|
||||
headerStyle := lipgloss.NewStyle().
|
||||
Bold(true).
|
||||
Foreground(lipgloss.Color("252")).
|
||||
PaddingLeft(1).
|
||||
PaddingBottom(1).
|
||||
Width(contentWidth)
|
||||
|
||||
infoStyle := lipgloss.NewStyle().
|
||||
PaddingLeft(1).
|
||||
Foreground(lipgloss.Color("241")).
|
||||
Width(contentWidth)
|
||||
|
||||
footerStyle := lipgloss.NewStyle().
|
||||
Foreground(lipgloss.Color("240")).
|
||||
Align(lipgloss.Center).
|
||||
Width(contentWidth)
|
||||
|
||||
// Render Content
|
||||
content := lipgloss.JoinVertical(lipgloss.Left,
|
||||
headerStyle.Render("🛡️ ENV-GUARD v1.0"),
|
||||
infoStyle.Render(userInfo),
|
||||
m.table.View(),
|
||||
footerStyle.Render("[↑/↓] Navigate [Enter] Toggle Lock [r] Refresh [q] Quit"),
|
||||
)
|
||||
|
||||
// Create Border Style locally
|
||||
borderStyle := lipgloss.NewStyle().
|
||||
BorderStyle(lipgloss.NormalBorder()).
|
||||
BorderForeground(lipgloss.Color("240")).
|
||||
Width(appWidth).
|
||||
Height(appHeight - 2)
|
||||
|
||||
// Wrap in Border
|
||||
gui := borderStyle.Render(content)
|
||||
|
||||
// Place in center of screen
|
||||
return lipgloss.Place(m.width, m.height, lipgloss.Center, lipgloss.Center, gui)
|
||||
}
|
||||
|
||||
func fetchServices() tea.Msg {
|
||||
services, err := api.GetServices()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return services
|
||||
}
|
||||
|
||||
func main() {
|
||||
user := os.Getenv("USER")
|
||||
if user == "" {
|
||||
user = "guest"
|
||||
}
|
||||
|
||||
columns := []table.Column{
|
||||
{Title: "SERVICE", Width: 20},
|
||||
{Title: "STATUS", Width: 15},
|
||||
{Title: "USER", Width: 15},
|
||||
{Title: "TIME", Width: 20},
|
||||
}
|
||||
|
||||
t := table.New(
|
||||
table.WithColumns(columns),
|
||||
table.WithFocused(true),
|
||||
table.WithHeight(10),
|
||||
)
|
||||
|
||||
s := table.DefaultStyles()
|
||||
s.Header = s.Header.
|
||||
BorderStyle(lipgloss.NormalBorder()).
|
||||
BorderForeground(lipgloss.Color("240")).
|
||||
BorderBottom(true).
|
||||
Bold(true)
|
||||
s.Selected = s.Selected.
|
||||
Foreground(lipgloss.Color("229")).
|
||||
Background(lipgloss.Color("57")).
|
||||
Bold(false)
|
||||
t.SetStyles(s)
|
||||
|
||||
m := model{
|
||||
table: t,
|
||||
user: user,
|
||||
}
|
||||
|
||||
if _, err := tea.NewProgram(m, tea.WithAltScreen()).Run(); err != nil {
|
||||
fmt.Println("Error running program:", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
}
|
||||
112
cmd/server/main.go
Normal file
112
cmd/server/main.go
Normal file
@@ -0,0 +1,112 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"log"
|
||||
"net/http"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"envguard/internal/models"
|
||||
)
|
||||
|
||||
var (
|
||||
services = []models.Service{
|
||||
{Name: "auth-service"},
|
||||
{Name: "payments-api"},
|
||||
{Name: "user-db"},
|
||||
{Name: "notifications"},
|
||||
{Name: "front-web"},
|
||||
}
|
||||
mu sync.Mutex
|
||||
)
|
||||
|
||||
func main() {
|
||||
http.HandleFunc("/services", handleServices)
|
||||
http.HandleFunc("/lock", handleLock)
|
||||
http.HandleFunc("/unlock", handleUnlock)
|
||||
|
||||
fmt.Println("🚦 Lock Server corriendo en :8080")
|
||||
if err := http.ListenAndServe(":8080", nil); err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
}
|
||||
|
||||
func handleServices(w http.ResponseWriter, r *http.Request) {
|
||||
mu.Lock()
|
||||
defer mu.Unlock()
|
||||
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
json.NewEncoder(w).Encode(services)
|
||||
}
|
||||
|
||||
func handleLock(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Method != http.MethodPost {
|
||||
http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
|
||||
return
|
||||
}
|
||||
|
||||
var req models.LockRequest
|
||||
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
|
||||
http.Error(w, err.Error(), http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
|
||||
mu.Lock()
|
||||
defer mu.Unlock()
|
||||
|
||||
for i := range services {
|
||||
if services[i].Name == req.ServiceName {
|
||||
if services[i].IsLocked {
|
||||
http.Error(w, "Service already locked", http.StatusConflict)
|
||||
return
|
||||
}
|
||||
services[i].IsLocked = true
|
||||
services[i].LockedBy = req.User
|
||||
services[i].LockedAt = time.Now()
|
||||
|
||||
w.WriteHeader(http.StatusOK)
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
http.Error(w, "Service not found", http.StatusNotFound)
|
||||
}
|
||||
|
||||
func handleUnlock(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Method != http.MethodPost {
|
||||
http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
|
||||
return
|
||||
}
|
||||
|
||||
var req models.UnlockRequest
|
||||
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
|
||||
http.Error(w, err.Error(), http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
|
||||
mu.Lock()
|
||||
defer mu.Unlock()
|
||||
|
||||
for i := range services {
|
||||
if services[i].Name == req.ServiceName {
|
||||
if !services[i].IsLocked {
|
||||
http.Error(w, "Service is not locked", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
if services[i].LockedBy != req.User {
|
||||
http.Error(w, "You cannot unlock a service locked by someone else", http.StatusForbidden)
|
||||
return
|
||||
}
|
||||
services[i].IsLocked = false
|
||||
services[i].LockedBy = ""
|
||||
services[i].LockedAt = time.Time{}
|
||||
|
||||
w.WriteHeader(http.StatusOK)
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
http.Error(w, "Service not found", http.StatusNotFound)
|
||||
}
|
||||
Reference in New Issue
Block a user