mirror of
https://github.com/sstent/garminsync-go.git
synced 2026-02-14 11:21:43 +00:00
checkpoint 1
This commit is contained in:
283
main.go
283
main.go
@@ -2,131 +2,198 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"database/sql"
|
||||
"log"
|
||||
"net/http"
|
||||
"os"
|
||||
"os/signal"
|
||||
"syscall"
|
||||
"time"
|
||||
"context"
|
||||
"database/sql"
|
||||
"fmt"
|
||||
"log"
|
||||
"net/http"
|
||||
"os"
|
||||
"os/signal"
|
||||
"path/filepath"
|
||||
"syscall"
|
||||
"time"
|
||||
|
||||
_ "github.com/mattn/go-sqlite3"
|
||||
"github.com/robfig/cron/v3"
|
||||
"github.com/joho/godotenv"
|
||||
|
||||
"github.com/yourusername/garminsync/internal/database"
|
||||
"github.com/yourusername/garminsync/internal/garmin"
|
||||
"github.com/yourusername/garminsync/internal/sync"
|
||||
"github.com/yourusername/garminsync/internal/web"
|
||||
|
||||
_ "github.com/mattn/go-sqlite3"
|
||||
"github.com/robfig/cron/v3"
|
||||
)
|
||||
|
||||
type App struct {
|
||||
db *sql.DB
|
||||
cron *cron.Cron
|
||||
server *http.Server
|
||||
garmin *GarminClient
|
||||
shutdown chan os.Signal
|
||||
db *database.SQLiteDB
|
||||
cron *cron.Cron
|
||||
server *http.Server
|
||||
garmin *garmin.Client
|
||||
shutdown chan os.Signal
|
||||
syncService *sync.SyncService
|
||||
}
|
||||
|
||||
func main() {
|
||||
app := &App{
|
||||
shutdown: make(chan os.Signal, 1),
|
||||
}
|
||||
|
||||
// Initialize components
|
||||
if err := app.init(); err != nil {
|
||||
log.Fatal("Failed to initialize app:", err)
|
||||
}
|
||||
|
||||
// Start services
|
||||
app.start()
|
||||
|
||||
// Wait for shutdown signal
|
||||
signal.Notify(app.shutdown, os.Interrupt, syscall.SIGTERM)
|
||||
<-app.shutdown
|
||||
|
||||
// Graceful shutdown
|
||||
app.stop()
|
||||
// Load environment variables from .env file
|
||||
if err := godotenv.Load(); err != nil {
|
||||
log.Println("No .env file found, using system environment variables")
|
||||
}
|
||||
|
||||
app := &App{
|
||||
shutdown: make(chan os.Signal, 1),
|
||||
}
|
||||
|
||||
// Initialize components
|
||||
if err := app.init(); err != nil {
|
||||
log.Fatal("Failed to initialize app:", err)
|
||||
}
|
||||
|
||||
// Start services
|
||||
app.start()
|
||||
|
||||
// Wait for shutdown signal
|
||||
signal.Notify(app.shutdown, os.Interrupt, syscall.SIGTERM)
|
||||
<-app.shutdown
|
||||
|
||||
// Graceful shutdown
|
||||
app.stop()
|
||||
}
|
||||
|
||||
func (app *App) init() error {
|
||||
var err error
|
||||
|
||||
// Initialize database
|
||||
app.db, err = initDatabase()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// Initialize Garmin client
|
||||
app.garmin = NewGarminClient()
|
||||
|
||||
// Setup cron scheduler
|
||||
app.cron = cron.New()
|
||||
|
||||
// Setup HTTP server
|
||||
app.server = &http.Server{
|
||||
Addr: ":8888",
|
||||
Handler: app.setupRoutes(),
|
||||
}
|
||||
|
||||
return nil
|
||||
var err error
|
||||
|
||||
// Initialize database
|
||||
dbConn, err := initDatabase()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
app.db = database.NewSQLiteDBFromDB(dbConn)
|
||||
|
||||
// Initialize Garmin client
|
||||
app.garmin = garmin.NewClient()
|
||||
|
||||
// Initialize sync service
|
||||
dataDir := os.Getenv("DATA_DIR")
|
||||
if dataDir == "" {
|
||||
dataDir = "./data"
|
||||
}
|
||||
app.syncService = sync.NewSyncService(app.garmin, database.NewSQLiteDBFromDB(app.db), dataDir)
|
||||
|
||||
// Setup cron scheduler
|
||||
app.cron = cron.New()
|
||||
|
||||
// Setup HTTP server
|
||||
webHandler := web.NewWebHandler(app.db)
|
||||
templateDir := os.Getenv("TEMPLATE_DIR")
|
||||
if templateDir == "" {
|
||||
templateDir = "./internal/web/templates"
|
||||
}
|
||||
if err := webHandler.LoadTemplates(templateDir); err != nil {
|
||||
return fmt.Errorf("failed to load templates: %w", err)
|
||||
}
|
||||
|
||||
app.server = &http.Server{
|
||||
Addr: ":8888",
|
||||
Handler: app.setupRoutes(webHandler),
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (app *App) start() {
|
||||
// Start cron scheduler
|
||||
app.cron.AddFunc("@hourly", func() {
|
||||
log.Println("Starting scheduled sync...")
|
||||
app.syncActivities()
|
||||
})
|
||||
app.cron.Start()
|
||||
|
||||
// Start web server
|
||||
go func() {
|
||||
log.Println("Server starting on http://localhost:8888")
|
||||
if err := app.server.ListenAndServe(); err != http.ErrServerClosed {
|
||||
log.Printf("Server error: %v", err)
|
||||
}
|
||||
}()
|
||||
// Start cron scheduler
|
||||
app.cron.AddFunc("@hourly", func() {
|
||||
log.Println("Starting scheduled sync...")
|
||||
if err := app.syncService.Sync(context.Background()); err != nil {
|
||||
log.Printf("Sync failed: %v", err)
|
||||
}
|
||||
})
|
||||
app.cron.Start()
|
||||
|
||||
// Start web server
|
||||
go func() {
|
||||
log.Println("Server starting on http://localhost:8888")
|
||||
if err := app.server.ListenAndServe(); err != http.ErrServerClosed {
|
||||
log.Printf("Server error: %v", err)
|
||||
}
|
||||
}()
|
||||
}
|
||||
|
||||
func (app *App) stop() {
|
||||
log.Println("Shutting down...")
|
||||
|
||||
// Stop cron
|
||||
app.cron.Stop()
|
||||
|
||||
// Stop web server
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||||
defer cancel()
|
||||
|
||||
if err := app.server.Shutdown(ctx); err != nil {
|
||||
log.Printf("Server shutdown error: %v", err)
|
||||
}
|
||||
|
||||
// Close database
|
||||
if app.db != nil {
|
||||
app.db.Close()
|
||||
}
|
||||
|
||||
log.Println("Shutdown complete")
|
||||
log.Println("Shutting down...")
|
||||
|
||||
// Stop cron
|
||||
app.cron.Stop()
|
||||
|
||||
// Stop web server
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||||
defer cancel()
|
||||
|
||||
if err := app.server.Shutdown(ctx); err != nil {
|
||||
log.Printf("Server shutdown error: %v", err)
|
||||
}
|
||||
|
||||
// Close database
|
||||
if app.db != nil {
|
||||
app.db.Close()
|
||||
}
|
||||
|
||||
log.Println("Shutdown complete")
|
||||
}
|
||||
|
||||
// main.go - Database initialization
|
||||
// Database initialization
|
||||
func initDatabase() (*sql.DB, error) {
|
||||
// Get data directory from environment or use default
|
||||
dataDir := os.Getenv("DATA_DIR")
|
||||
if dataDir == "" {
|
||||
dataDir = "./data"
|
||||
}
|
||||
|
||||
// Create data directory if it doesn't exist
|
||||
if err := os.MkdirAll(dataDir, 0755); err != nil {
|
||||
return nil, fmt.Errorf("failed to create data directory: %v", err)
|
||||
}
|
||||
|
||||
dbPath := filepath.Join(dataDir, "garmin.db")
|
||||
|
||||
// Initialize SQLite database
|
||||
db, err := database.NewSQLiteDB(dbPath)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to initialize database: %v", err)
|
||||
}
|
||||
|
||||
return db.db, nil // Return the underlying *sql.DB
|
||||
// Get database path from environment or use default
|
||||
dbPath := os.Getenv("DB_PATH")
|
||||
if dbPath == "" {
|
||||
// Fallback to DATA_DIR/garmin.db if DB_PATH not set
|
||||
dataDir := os.Getenv("DATA_DIR")
|
||||
if dataDir == "" {
|
||||
dataDir = "./data"
|
||||
}
|
||||
|
||||
// Create data directory if it doesn't exist
|
||||
if err := os.MkdirAll(dataDir, 0755); err != nil {
|
||||
return nil, fmt.Errorf("failed to create data directory: %v", err)
|
||||
}
|
||||
|
||||
dbPath = filepath.Join(dataDir, "garmin.db")
|
||||
}
|
||||
|
||||
// Initialize SQLite database
|
||||
db, err := sql.Open("sqlite3", dbPath+"?_foreign_keys=on")
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to open database: %v", err)
|
||||
}
|
||||
|
||||
// Verify connection
|
||||
if err := db.Ping(); err != nil {
|
||||
return nil, fmt.Errorf("database ping failed: %v", err)
|
||||
}
|
||||
|
||||
// Create tables if they don't exist
|
||||
sqliteDB := &database.SQLiteDB{db: db}
|
||||
if err := sqliteDB.createTables(); err != nil {
|
||||
return nil, fmt.Errorf("failed to create tables: %v", err)
|
||||
}
|
||||
|
||||
return db, nil
|
||||
}
|
||||
|
||||
// Application routes
|
||||
func (app *App) setupRoutes(webHandler *web.WebHandler) *http.ServeMux {
|
||||
mux := http.NewServeMux()
|
||||
|
||||
// Health check
|
||||
mux.HandleFunc("/health", func(w http.ResponseWriter, r *http.Request) {
|
||||
w.WriteHeader(http.StatusOK)
|
||||
w.Write([]byte("OK"))
|
||||
})
|
||||
|
||||
// Web UI routes
|
||||
mux.HandleFunc("/", webHandler.Index)
|
||||
mux.HandleFunc("/activities", webHandler.ActivityList)
|
||||
mux.HandleFunc("/activity", webHandler.ActivityDetail)
|
||||
|
||||
return mux
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user