feat: add stats screen, already played screen and implement a blacklist of common usernames to not record stats for those names

Author: f-gillmann

.dockerignore

@@ -32,3 +32,8 @@ go.work
 # Environment files
 .env
 .env.local
+
+# Database files
+*.db
+*.sqlite
+*.sqlite3

.gitignore

@@ -1,3 +1,4 @@
 .idea/
 .ssh/
 bin/
+wordle-stats.db

Dockerfile

@@ -1,30 +1,34 @@
 # --- Build ---
 FROM golang:1.23-alpine AS builder
 
+RUN apk --no-cache add gcc musl-dev sqlite-dev
+
 WORKDIR /app
 
 COPY go.mod go.sum ./
 RUN go mod download
 
 COPY . .
 
-RUN CGO_ENABLED=0 GOOS=linux go build -a -installsuffix cgo -o wordle-ssh .
+RUN CGO_ENABLED=1 GOOS=linux go build -a -ldflags '-linkmode external -extldflags "-static"' -o wordle-ssh ./cmd/main.go
 
 # --- Run ---
 FROM alpine:latest
 
-RUN apk --no-cache add ca-certificates
+RUN apk --no-cache add ca-certificates sqlite-libs
 
 WORKDIR /root/
 
 COPY --from=builder /app/wordle-ssh .
 
-RUN mkdir -p .ssh
+# Create directories for SSH keys and database
+RUN mkdir -p .ssh /data
 
 EXPOSE 23234
 
 ENV WORDLE_SSH_HOST=0.0.0.0
 ENV WORDLE_SSH_PORT=23234
 ENV WORDLE_SSH_HOST_KEY_PATH=.ssh/id_ed25519
+ENV WORDLE_SSH_DB_PATH=/data/wordle-stats.db
 
 CMD ["./wordle-ssh"]

cmd/main.go

@@ -11,15 +11,27 @@ func main() {
 	// Load configuration
 	config := server.LoadConfigFromEnv()
 
+	// Initialize logger
+	logger := log.NewWithOptions(os.Stderr, log.Options{
+		ReportTimestamp: true,
+		ReportCaller:    config.LogLevel == log.DebugLevel,
+		TimeFormat:      "2006/01/02 15:04:05",
+		Prefix:          "[wordle-ssh]",
+		Level:           config.LogLevel,
+	})
+
+	// Set the logger in config
+	config.Logger = logger
+
 	// Create and start the server
 	srv, err := server.New(config)
 	if err != nil {
-		log.Fatal(err)
+		logger.Fatal("Failed to create server", "error", err)
 		os.Exit(1)
 	}
 
 	if err := srv.Start(); err != nil {
-		log.Fatal(err)
+		logger.Fatal("Failed to start server", "error", err)
 		os.Exit(1)
 	}
 }

docker-compose.yml

@@ -10,9 +10,13 @@ services:
       - WORDLE_SSH_HOST=0.0.0.0
       - WORDLE_SSH_PORT=23234
       - WORDLE_SSH_HOST_KEY_PATH=.ssh/id_ed25519
+      - WORDLE_SSH_DB_PATH=/data/wordle-stats.db
+      - WORDLE_SSH_LOG_LEVEL=info
     volumes:
       - ssh-keys:/root/.ssh
+      - wordle-data:/data
     restart: unless-stopped
 
 volumes:
   ssh-keys:
+  wordle-data:

go.mod

@@ -8,6 +8,7 @@ require (
 	github.com/charmbracelet/log v0.4.2
 	github.com/charmbracelet/ssh v0.0.0-20250826160808-ebfa259c7309
 	github.com/charmbracelet/wish v1.4.7
+	github.com/mattn/go-sqlite3 v1.14.32
 )
 
 require (

go.sum

@@ -48,6 +48,8 @@ github.com/mattn/go-localereader v0.0.1 h1:ygSAOl7ZXTx4RdPYinUpg6W99U8jWvWi9Ye2J
 github.com/mattn/go-localereader v0.0.1/go.mod h1:8fBrzywKY7BI3czFoHkuzRoWE9C+EiG4R1k4Cjx5p88=
 github.com/mattn/go-runewidth v0.0.16 h1:E5ScNMtiwvlvB5paMFdw9p4kSQzbXFikJ5SQO6TULQc=
 github.com/mattn/go-runewidth v0.0.16/go.mod h1:Jdepj2loyihRzMpdS35Xk/zdY8IAYHsh153qUoGf23w=
+github.com/mattn/go-sqlite3 v1.14.32 h1:JD12Ag3oLy1zQA+BNn74xRgaBbdhbNIDYvQUEuuErjs=
+github.com/mattn/go-sqlite3 v1.14.32/go.mod h1:Uh1q+B4BYcTPb+yiD3kU8Ct7aC0hY9fxUwlHK0RXw+Y=
 github.com/muesli/ansi v0.0.0-20230316100256-276c6243b2f6 h1:ZK8zHtRHOkbHy6Mmr5D264iyp3TiX5OmNcI5cIARiQI=
 github.com/muesli/ansi v0.0.0-20230316100256-276c6243b2f6/go.mod h1:CJlz5H+gyd6CUWT45Oy4q24RdLyn7Md9Vj2/ldJBSIo=
 github.com/muesli/cancelreader v0.2.2 h1:3I4Kt4BQjOR54NavqnDogx/MIoWBFa0StPA8ELUXHmA=

internal/server/server.go

@@ -14,6 +14,7 @@ import (
 	"github.com/charmbracelet/wish"
 	"github.com/charmbracelet/wish/bubbletea"
 	"github.com/charmbracelet/wish/logging"
+	"github.com/f-gillmann/wordle-ssh/internal/stats"
 	"github.com/f-gillmann/wordle-ssh/internal/ui"
 	"github.com/f-gillmann/wordle-ssh/internal/wordle"
 )
@@ -22,13 +23,15 @@ const (
 	defaultHost        = "0.0.0.0"
 	defaultPort        = "23234"
 	defaultHostKeyPath = ".ssh/id_ed25519"
+	defaultDBPath      = "./wordle-stats.db"
 )
 
 // Config holds the server configuration
 type Config struct {
 	Host        string
 	Port        string
 	HostKeyPath string
+	DBPath      string
 	Logger      *log.Logger
 	LogLevel    log.Level
 }
@@ -50,6 +53,11 @@ func LoadConfigFromEnv() Config {
 		hostKeyPath = defaultHostKeyPath
 	}
 
+	dbPath := os.Getenv("WORDLE_SSH_DB_PATH")
+	if dbPath == "" {
+		dbPath = defaultDBPath
+	}
+
 	logLevel := os.Getenv("WORDLE_SSH_LOG_LEVEL")
 	var level log.Level
 
@@ -70,6 +78,7 @@ func LoadConfigFromEnv() Config {
 		Host:        host,
 		Port:        port,
 		HostKeyPath: hostKeyPath,
+		DBPath:      dbPath,
 		LogLevel:    level,
 	}
 }
@@ -80,6 +89,7 @@ type Server struct {
 	wordleWord string
 	wordleDate string
 	wishServer *ssh.Server
+	statsStore *stats.Store
 }
 
 // New creates a new SSH server
@@ -96,20 +106,25 @@ func New(config Config) (*Server, error) {
 		config.HostKeyPath = defaultHostKeyPath
 	}
 
+	if config.DBPath == "" {
+		config.DBPath = defaultDBPath
+	}
+
 	if config.Logger == nil {
-		config.Logger = log.NewWithOptions(os.Stderr, log.Options{
-			ReportTimestamp: true,
-			ReportCaller:    config.LogLevel == log.DebugLevel,
-			TimeFormat:      "2006/01/02 15:04:05",
-			Prefix:          "[wordle-ssh]",
-			Level:           config.LogLevel,
-		})
+		return nil, fmt.Errorf("logger must be provided in config")
 	}
 
 	s := &Server{
 		config: config,
 	}
 
+	// Initialize stats store
+	statsStore, err := stats.NewStore(config.DBPath, config.Logger)
+	if err != nil {
+		return nil, fmt.Errorf("failed to initialize stats store: %w", err)
+	}
+	s.statsStore = statsStore
+
 	// Fetch today's Wordle word
 	if err := s.refreshWordleWord(); err != nil {
 		return nil, fmt.Errorf("failed to fetch wordle word: %w", err)
@@ -153,15 +168,35 @@ func (s *Server) refreshWordleWord() error {
 }
 
 // teaHandler creates a bubbletea program for each SSH session
-func (s *Server) teaHandler(ssh.Session) (tea.Model, []tea.ProgramOption) {
+func (s *Server) teaHandler(sshSession ssh.Session) (tea.Model, []tea.ProgramOption) {
 	// Refresh Wordle word if it's a new day
 	if err := s.refreshWordleWord(); err != nil {
 		s.config.Logger.Error("Failed to refresh Wordle word", "error", err)
 		return nil, nil
 	}
 
-	// Create the app model with the current word
-	m := ui.NewAppModel(s.wordleWord)
+	// Get username from SSH session
+	username := sshSession.User()
+	if username == "" {
+		username = "anonymous"
+	}
+
+	// Check if username is blacklisted
+	isBlacklisted := stats.IsBlacklisted(username)
+
+	// Check if user has already played today (but not for blacklisted users)
+	hasPlayed := false
+	if !isBlacklisted {
+		var err error
+		hasPlayed, err = s.statsStore.HasPlayedToday(username, s.wordleDate)
+
+		if err != nil {
+			s.config.Logger.Error("Failed to check if user played today", "error", err, "username", username)
+		}
+	}
+
+	// Create the app model with the current word, stats store, and logger
+	m := ui.NewAppModel(s.wordleWord, s.wordleDate, username, s.statsStore, hasPlayed, isBlacklisted, s.config.Logger)
 
 	return m, []tea.ProgramOption{tea.WithAltScreen()}
 }
@@ -171,7 +206,7 @@ func (s *Server) Start() error {
 	done := make(chan os.Signal, 1)
 	signal.Notify(done, os.Interrupt, syscall.SIGINT, syscall.SIGTERM)
 
-	s.config.Logger.Info("Starting SSH server", "host", s.config.Host, "port", s.config.Port)
+	s.config.Logger.Info("Starting SSH server", "host", s.config.Host, "port", s.config.Port, "db", s.config.DBPath)
 
 	go func() {
 		if err := s.wishServer.ListenAndServe(); err != nil {
@@ -185,6 +220,11 @@ func (s *Server) Start() error {
 	ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
 	defer cancel()
 
+	// Close stats store
+	if err := s.statsStore.Close(); err != nil {
+		s.config.Logger.Error("Failed to close stats store", "error", err)
+	}
+
 	if err := s.wishServer.Shutdown(ctx); err != nil {
 		return fmt.Errorf("failed to shutdown server: %w", err)
 	}

internal/stats/blacklist.go

@@ -0,0 +1,28 @@
+package stats
+
+// BlacklistedUsernames contains usernames that should never record stats
+var BlacklistedUsernames = map[string]bool{
+	"anonymous": true,
+	"root":      true,
+	"toor":      true,
+	"admin":     true,
+	"user":      true,
+	"guest":     true,
+	"test":      true,
+	"demo":      true,
+	"ubuntu":    true,
+	"debian":    true,
+	"centos":    true,
+	"fedora":    true,
+	"oracle":    true,
+	"pi":        true,
+	"vagrant":   true,
+	"default":   true,
+	"1234":      true,
+	"ftp":       true,
+}
+
+// IsBlacklisted checks if a username is blacklisted
+func IsBlacklisted(username string) bool {
+	return BlacklistedUsernames[username]
+}