Added registration code display in logs

start.sh

@@ -622,10 +622,15 @@ EOF
 package main
 
 import (
+    "encoding/json"
     "fmt"
+    "io"
     "log"
+    "math/rand"
     "net/http"
     "os"
+    "strconv"
+    "strings"
     "time"
 )
 
@@ -641,21 +646,149 @@ func main() {
     log.Println("PGPORT:", os.Getenv("PGPORT"))
     log.Println("API_ENDPOINT:", os.Getenv("ENTE_API_ENDPOINT"))
     
+    // Create a logger that logs to both stdout and a file
+    logFile, err := os.OpenFile("/app/data/logs/api_requests.log", os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0644)
+    if err != nil {
+        log.Printf("Error opening log file: %v", err)
+    }
+    defer logFile.Close()
+    
+    multiWriter := io.MultiWriter(os.Stdout, logFile)
+    logger := log.New(multiWriter, "", log.LstdFlags)
+    
+    rand.Seed(time.Now().UnixNano())
+    
+    // Map to store verification codes
+    verificationCodes := make(map[string]string)
+    
     http.HandleFunc("/health", func(w http.ResponseWriter, r *http.Request) {
         w.Header().Set("Content-Type", "application/json")
         fmt.Fprintf(w, \`{"status":"ok","version":"mock-1.0.0","time":"%s"}\`, time.Now().Format(time.RFC3339))
     })
     
+    // Handle registration requests
+    http.HandleFunc("/users", func(w http.ResponseWriter, r *http.Request) {
+        if r.Method == "POST" {
+            body, err := io.ReadAll(r.Body)
+            if err != nil {
+                http.Error(w, "Error reading request body", http.StatusBadRequest)
+                return
+            }
+            
+            logger.Printf("REGISTRATION REQUEST: %s", string(body))
+            
+            // Extract email from request - simplified parsing
+            emailStart := strings.Index(string(body), "\"email\":\"")
+            var email string
+            if emailStart >= 0 {
+                emailStart += 9 // Length of "\"email\":\""
+                emailEnd := strings.Index(string(body)[emailStart:], "\"")
+                if emailEnd >= 0 {
+                    email = string(body)[emailStart : emailStart+emailEnd]
+                }
+            }
+            
+            // Generate verification code
+            verificationCode := strconv.Itoa(100000 + rand.Intn(900000)) // 6-digit code
+            if email != "" {
+                verificationCodes[email] = verificationCode
+                logger.Printf("===================================================")
+                logger.Printf("⚠️  VERIFICATION CODE for %s: %s", email, verificationCode)
+                logger.Printf("===================================================")
+            }
+            
+            // Return a success response
+            w.Header().Set("Content-Type", "application/json")
+            w.WriteHeader(http.StatusOK)
+            fmt.Fprintf(w, \`{"status":"ok","message":"Verification code sent (check logs)"}\`)
+        } else {
+            // Just handle other methods with a generic response
+            w.Header().Set("Content-Type", "application/json")
+            fmt.Fprintf(w, \`{"status":"mock","endpoint":"%s","method":"%s"}\`, r.URL.Path, r.Method)
+        }
+    })
+    
+    // Handle verification endpoint
+    http.HandleFunc("/users/verification", func(w http.ResponseWriter, r *http.Request) {
+        if r.Method == "POST" {
+            body, err := io.ReadAll(r.Body)
+            if err != nil {
+                http.Error(w, "Error reading request body", http.StatusBadRequest)
+                return
+            }
+            
+            logger.Printf("VERIFICATION REQUEST: %s", string(body))
+            
+            // Extract email and code
+            var email, code string
+            
+            emailStart := strings.Index(string(body), "\"email\":\"")
+            if emailStart >= 0 {
+                emailStart += 9
+                emailEnd := strings.Index(string(body)[emailStart:], "\"")
+                if emailEnd >= 0 {
+                    email = string(body)[emailStart : emailStart+emailEnd]
+                }
+            }
+            
+            codeStart := strings.Index(string(body), "\"code\":\"")
+            if codeStart >= 0 {
+                codeStart += 8
+                codeEnd := strings.Index(string(body)[codeStart:], "\"")
+                if codeEnd >= 0 {
+                    code = string(body)[codeStart : codeStart+codeEnd]
+                }
+            }
+            
+            // Verify the code
+            isValid := false
+            if email != "" && code != "" {
+                expectedCode, exists := verificationCodes[email]
+                if exists && expectedCode == code {
+                    isValid = true
+                    logger.Printf("✅ SUCCESSFUL VERIFICATION for %s with code %s", email, code)
+                } else {
+                    logger.Printf("❌ FAILED VERIFICATION for %s with code %s (expected %s)", 
+                        email, code, expectedCode)
+                }
+            }
+            
+            w.Header().Set("Content-Type", "application/json")
+            if isValid {
+                // Return a successful verification response
+                w.WriteHeader(http.StatusOK)
+                fmt.Fprintf(w, \`{"status":"ok","token":"mock-token-12345"}\`)
+            } else {
+                // Return an error
+                w.WriteHeader(http.StatusBadRequest)
+                fmt.Fprintf(w, \`{"status":"error","message":"Invalid verification code"}\`)
+            }
+        } else {
+            // Handle other methods with a generic response
+            w.Header().Set("Content-Type", "application/json")
+            fmt.Fprintf(w, \`{"status":"mock","endpoint":"%s","method":"%s"}\`, r.URL.Path, r.Method)
+        }
+    })
+    
+    // Generic handler for all other requests
     http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
-        log.Printf("Received request for %s via %s", r.URL.Path, r.Method)
+        logger.Printf("Received request for %s via %s", r.URL.Path, r.Method)
+        
+        if r.Method == "POST" || r.Method == "PUT" {
+            body, _ := io.ReadAll(r.Body)
+            if len(body) > 0 {
+                logger.Printf("Request body: %s", string(body))
+            }
+        }
+        
         w.Header().Set("Content-Type", "application/json")
         fmt.Fprintf(w, \`{"status":"mock","endpoint":"%s","method":"%s","time":"%s"}\`, 
             r.URL.Path, r.Method, time.Now().Format(time.RFC3339))
     })
     
-    log.Printf("Mock Ente API server listening on port %s\n", port)
+    logger.Printf("Mock Ente API server listening on port %s\n", port)
     if err := http.ListenAndServe("0.0.0.0:" + port, nil); err != nil {
-        log.Fatalf("Server failed: %v", err)
+        logger.Fatalf("Server failed: %v", err)
     }
 }
 EOT
@@ -1153,8 +1286,18 @@ else
     echo "==> Response: $CONFIG_JS_TEST"
 fi
 
-echo "==> Entering wait state - press Ctrl+C to stop"
+echo "==> Entering wait state - watching logs for registration codes"
-# Wait for all background processes to complete (or for user to interrupt)
+echo "==> Registration verification codes will appear in the logs below"
+echo "==> Press Ctrl+C to stop"
+
+# Set up a tail process to show logs in real-time while maintaining the wait state
+tail -f /app/data/logs/api_requests.log &
+TAIL_PID=$!
+
+# Trap to kill the tail process when the script exits
+trap 'kill -TERM $TAIL_PID; kill -TERM $SERVER_PID; kill -TERM $PUBLIC_SERVER_PID; kill -TERM $CADDY_PID; exit' TERM INT
+
+# Wait for all processes
 wait $SERVER_PID
 wait $PUBLIC_SERVER_PID
 wait $CADDY_PID