commit e03e883e3af0de0611eb7f9a1b37fe9c131d9080
Author: Suvodip Ghosh <suvodip@siliconpin.com>
Date:   Mon Jun 23 16:38:18 2025 +0530

    first commit

diff --git a/.env b/.env
new file mode 100644
index 0000000..81020f4
--- /dev/null
+++ b/.env
@@ -0,0 +1,2 @@
+PORT=8080
+URL_PREFIX=http://0.0.0.0:8000/image_asset
\ No newline at end of file
diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..7087f7f
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1 @@
+image_asset
\ No newline at end of file
diff --git a/go.mod b/go.mod
new file mode 100644
index 0000000..a3d17fb
--- /dev/null
+++ b/go.mod
@@ -0,0 +1,11 @@
+module image_manager
+
+go 1.18
+
+require (
+	github.com/google/uuid v1.6.0
+	github.com/joho/godotenv v1.5.1
+	github.com/mattn/go-sqlite3 v1.14.28
+)
+
+require github.com/rs/cors v1.11.1 // indirect
diff --git a/go.sum b/go.sum
new file mode 100644
index 0000000..ae4eccd
--- /dev/null
+++ b/go.sum
@@ -0,0 +1,8 @@
+github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
+github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
+github.com/joho/godotenv v1.5.1 h1:7eLL/+HRGLY0ldzfGMeQkb7vMd0as4CfYvUVzLqw0N0=
+github.com/joho/godotenv v1.5.1/go.mod h1:f4LDr5Voq0i2e/R5DDNOoa2zzDfwtkZa6DnEwAbqwq4=
+github.com/mattn/go-sqlite3 v1.14.28 h1:ThEiQrnbtumT+QMknw63Befp/ce/nUPgBPMlRFEum7A=
+github.com/mattn/go-sqlite3 v1.14.28/go.mod h1:Uh1q+B4BYcTPb+yiD3kU8Ct7aC0hY9fxUwlHK0RXw+Y=
+github.com/rs/cors v1.11.1 h1:eU3gRzXLRK57F5rKMGMZURNdIG4EoAmX8k94r9wXWHA=
+github.com/rs/cors v1.11.1/go.mod h1:XyqrcTp5zjWr1wsJ8PIRZssZ8b/WMcMf71DJnit4EMU=
diff --git a/image_asset.sqlite b/image_asset.sqlite
new file mode 100644
index 0000000..797b9d0
Binary files /dev/null and b/image_asset.sqlite differ
diff --git a/main.go b/main.go
new file mode 100644
index 0000000..624dbb2
--- /dev/null
+++ b/main.go
@@ -0,0 +1,236 @@
+package main
+
+import (
+	"database/sql"
+	"encoding/json"
+	"fmt"
+	"log"
+	"net/http"
+	"os"
+	"path/filepath"
+	"strings"
+	"github.com/joho/godotenv"
+	_ "github.com/mattn/go-sqlite3"
+	"github.com/google/uuid"
+	"github.com/rs/cors"
+)
+
+type Config struct {
+	Port      string
+	URLPrefix string
+}
+
+type Image struct {
+	ID        int
+	Name      string
+	Delivered bool
+	Score     int
+	SecretKey string
+	User      string
+}
+
+var db *sql.DB
+var config Config
+
+func main() {
+	// Load environment variables
+	err := godotenv.Load()
+	if err != nil {
+		log.Fatal("Error loading .env file")
+	}
+
+	config = Config{
+		Port:      os.Getenv("PORT"),
+		URLPrefix: os.Getenv("URL_PREFIX"),
+	}
+
+	if len(os.Args) < 2 {
+		log.Fatal("Usage: ./image_manager [save_name|serve]")
+	}
+
+	command := os.Args[1]
+
+	// Initialize database
+	initDB()
+
+	switch command {
+	case "save_name":
+		saveImageNames()
+	case "serve":
+		serveAPI()
+	default:
+		log.Fatal("Unknown command. Use 'save_name' or 'serve'")
+	}
+}
+
+func initDB() {
+	var err error
+	db, err = sql.Open("sqlite3", "./image_asset.sqlite")
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	// Create table if not exists
+	_, err = db.Exec(`
+		CREATE TABLE IF NOT EXISTS images (
+			id INTEGER PRIMARY KEY AUTOINCREMENT,
+			name TEXT NOT NULL UNIQUE,
+			delivered INTEGER DEFAULT 0,
+			score INTEGER DEFAULT 0,
+			secret_key TEXT,
+			user TEXT
+		)
+	`)
+	if err != nil {
+		log.Fatal(err)
+	}
+}
+
+func saveImageNames() {
+	// Get all files in image_asset directory
+	files, err := os.ReadDir("./image_asset")
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	// Insert image names into database
+	for _, file := range files {
+		if !file.IsDir() {
+			// Check if the file is an image (simple extension check)
+			ext := strings.ToLower(filepath.Ext(file.Name()))
+			if ext == ".jpg" || ext == ".jpeg" || ext == ".png" || ext == ".gif" {
+				_, err := db.Exec("INSERT OR IGNORE INTO images (name) VALUES (?)", file.Name())
+				if err != nil {
+					log.Printf("Error inserting %s: %v", file.Name(), err)
+				} else {
+					log.Printf("Saved: %s", file.Name())
+				}
+			}
+		}
+	}
+}
+
+func serveAPI() {
+	// Create a new CORS handler
+	c := cors.New(cors.Options{
+		AllowedOrigins:   []string{"http://localhost:4321"}, // Your frontend origin
+		AllowedMethods:   []string{"GET", "POST", "OPTIONS"},
+		AllowedHeaders:   []string{"Content-Type"},
+		AllowCredentials: true,
+		Debug:           true, // Remove in production
+	})
+
+	// Create a new mux router
+	router := http.NewServeMux()
+	router.HandleFunc("/get-image", getImageHandler)
+	router.HandleFunc("/save-score", saveScoreHandler)
+
+	// Wrap your router with the CORS handler
+	handler := c.Handler(router)
+
+	log.Printf("Server starting on port %s", config.Port)
+	log.Fatal(http.ListenAndServe(":"+config.Port, handler))
+}
+
+func getImageHandler(w http.ResponseWriter, r *http.Request) {
+	if r.Method != http.MethodGet {
+		http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
+		return
+	}
+
+	// Get a random undelivered image
+	var img Image
+	err := db.QueryRow(`
+		SELECT id, name FROM images 
+		WHERE delivered = 0 
+		ORDER BY RANDOM() 
+		LIMIT 1
+	`).Scan(&img.ID, &img.Name)
+	if err != nil {
+		if err == sql.ErrNoRows {
+			http.Error(w, "No images available", http.StatusNotFound)
+			return
+		}
+		http.Error(w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+
+	// Generate secret key
+	secretKey := uuid.New().String()
+
+	// Update the image with delivered status and secret key
+	_, err = db.Exec(`
+		UPDATE images 
+		SET delivered = 1, secret_key = ?
+		WHERE id = ?
+	`, secretKey, img.ID)
+	if err != nil {
+		http.Error(w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+
+	// Prepare response
+	response := map[string]interface{}{
+		"url":       fmt.Sprintf("%s/%s", config.URLPrefix, img.Name),
+		"secretKey": secretKey,
+	}
+
+	w.Header().Set("Content-Type", "application/json")
+	json.NewEncoder(w).Encode(response)
+}
+
+func saveScoreHandler(w http.ResponseWriter, r *http.Request) {
+	if r.Method != http.MethodPost {
+		http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
+		return
+	}
+
+	type Request struct {
+		FileName  string `json:"fileName"`
+		Score     int    `json:"score"`
+		User      string `json:"user,omitempty"`
+		SecretKey string `json:"secretKey"`
+	}
+
+	var req Request
+	err := json.NewDecoder(r.Body).Decode(&req)
+	if err != nil {
+		http.Error(w, err.Error(), http.StatusBadRequest)
+		return
+	}
+
+	// Validate request
+	if req.FileName == "" || req.SecretKey == "" {
+		http.Error(w, "fileName and secretKey are required", http.StatusBadRequest)
+		return
+	}
+
+	// Verify secret key and update score
+	var img Image
+	err = db.QueryRow(`
+		SELECT id FROM images 
+		WHERE name = ? AND secret_key = ?
+	`, req.FileName, req.SecretKey).Scan(&img.ID)
+	if err != nil {
+		if err == sql.ErrNoRows {
+			http.Error(w, "Invalid fileName or secretKey", http.StatusUnauthorized)
+			return
+		}
+		http.Error(w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+
+	// Update score and user
+	_, err = db.Exec(`
+		UPDATE images 
+		SET score = ?, user = ?
+		WHERE id = ?
+	`, req.Score, req.User, img.ID)
+	if err != nil {
+		http.Error(w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+
+	w.WriteHeader(http.StatusOK)
+	w.Write([]byte("Score saved successfully"))
+}
\ No newline at end of file