73 lines
2.0 KiB
Go
73 lines
2.0 KiB
Go
package main
|
|
|
|
import (
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"ai-media-hub/backend/handlers"
|
|
"ai-media-hub/backend/models"
|
|
"ai-media-hub/backend/services"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
func main() {
|
|
root := envOrDefault("APP_ROOT", "/app")
|
|
dbPath := envOrDefault("SQLITE_PATH", filepath.Join(root, "db", "media.db"))
|
|
downloadsDir := envOrDefault("DOWNLOADS_DIR", filepath.Join(root, "downloads"))
|
|
frontendDir := envOrDefault("FRONTEND_DIR", filepath.Join(root, "frontend"))
|
|
workerScript := envOrDefault("WORKER_SCRIPT", filepath.Join(root, "worker", "downloader.py"))
|
|
|
|
db, err := models.InitDB(dbPath)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
defer db.Close()
|
|
|
|
if err := handlers.EnsurePaths(downloadsDir, workerScript); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
app := &handlers.App{
|
|
DB: db,
|
|
DownloadsDir: downloadsDir,
|
|
WorkerScript: workerScript,
|
|
SearchService: services.NewSearchService(
|
|
os.Getenv("VERTEX_AI_SEARCH_API_KEY"),
|
|
os.Getenv("VERTEX_AI_SEARCH_PROJECT_ID"),
|
|
os.Getenv("VERTEX_AI_SEARCH_LOCATION"),
|
|
os.Getenv("VERTEX_AI_SEARCH_DATA_STORE_ID"),
|
|
os.Getenv("VERTEX_AI_SEARCH_SERVING_CONFIG"),
|
|
),
|
|
GeminiService: services.NewGeminiService(os.Getenv("GEMINI_API_KEY")),
|
|
Hub: handlers.NewHub(),
|
|
}
|
|
|
|
router := gin.Default()
|
|
handlers.RegisterRoutes(router, app)
|
|
router.StaticFile("/", filepath.Join(frontendDir, "index.html"))
|
|
router.StaticFile("/app.js", filepath.Join(frontendDir, "app.js"))
|
|
router.StaticFile("/style.css", filepath.Join(frontendDir, "style.css"))
|
|
router.NoRoute(func(c *gin.Context) {
|
|
c.File(filepath.Join(frontendDir, "index.html"))
|
|
})
|
|
router.NoMethod(func(c *gin.Context) {
|
|
c.JSON(http.StatusMethodNotAllowed, gin.H{"error": "method not allowed"})
|
|
})
|
|
|
|
addr := envOrDefault("APP_ADDR", ":8080")
|
|
log.Printf("server listening on %s", addr)
|
|
if err := router.Run(addr); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func envOrDefault(key, fallback string) string {
|
|
if value := os.Getenv(key); value != "" {
|
|
return value
|
|
}
|
|
return fallback
|
|
}
|