correct bugs AG

internal/api/handlers.go

@@ -58,6 +58,7 @@ func (s *Server) SetupRoutes() *http.ServeMux {
 	mux.HandleFunc("/api/ultrabeam/frequency", s.handleUltrabeamFrequency)
 	mux.HandleFunc("/api/ultrabeam/retract", s.handleUltrabeamRetract)
 	mux.HandleFunc("/api/ultrabeam/autotrack", s.handleUltrabeamAutoTrack)
+	mux.HandleFunc("/api/ultrabeam/direction", s.handleUltrabeamDirection)
 
 	// Tuner endpoints
 	mux.HandleFunc("/api/tuner/operate", s.handleTunerOperate)
@@ -66,6 +67,7 @@ func (s *Server) SetupRoutes() *http.ServeMux {
 
 	// Antenna Genius endpoints
 	mux.HandleFunc("/api/antenna/select", s.handleAntennaSelect)
+	mux.HandleFunc("/api/antenna/deselect", s.handleAntennaDeselect)
 	mux.HandleFunc("/api/antenna/reboot", s.handleAntennaReboot)
 
 	// Power Genius endpoints
@@ -336,6 +338,33 @@ func (s *Server) handleAntennaSelect(w http.ResponseWriter, r *http.Request) {
 	s.sendJSON(w, map[string]string{"status": "ok"})
 }
 
+func (s *Server) handleAntennaDeselect(w http.ResponseWriter, r *http.Request) {
+	if r.Method != http.MethodPost {
+		http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
+		return
+	}
+
+	var req struct {
+		Port    int `json:"port"`
+		Antenna int `json:"antenna"`
+	}
+
+	if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
+		http.Error(w, "Invalid request body", http.StatusBadRequest)
+		return
+	}
+
+	log.Printf("Deselecting antenna %d from port %d", req.Antenna, req.Port)
+	if err := s.deviceManager.AntennaGenius().DeselectAntenna(req.Port, req.Antenna); err != nil {
+		log.Printf("Failed to deselect antenna: %v", err)
+		http.Error(w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+	log.Printf("Successfully deselected antenna %d from port %d", req.Antenna, req.Port)
+
+	s.sendJSON(w, map[string]string{"status": "ok"})
+}
+
 func (s *Server) handleAntennaReboot(w http.ResponseWriter, r *http.Request) {
 	if r.Method != http.MethodPost {
 		http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
@@ -414,6 +443,9 @@ func (s *Server) handleUltrabeamFrequency(w http.ResponseWriter, r *http.Request
 		return
 	}
 
+	// Save direction for auto-track to use
+	s.deviceManager.SetUltrabeamDirection(req.Direction)
+
 	if err := s.deviceManager.Ultrabeam().SetFrequency(req.Frequency, req.Direction); err != nil {
 		http.Error(w, err.Error(), http.StatusInternalServerError)
 		return
@@ -457,6 +489,27 @@ func (s *Server) handleUltrabeamAutoTrack(w http.ResponseWriter, r *http.Request
 	s.sendJSON(w, map[string]string{"status": "ok"})
 }
 
+func (s *Server) handleUltrabeamDirection(w http.ResponseWriter, r *http.Request) {
+	if r.Method != http.MethodPost {
+		http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
+		return
+	}
+
+	var req struct {
+		Direction int `json:"direction"` // 0=normal, 1=180°, 2=bi-dir
+	}
+
+	if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
+		http.Error(w, "Invalid request body", http.StatusBadRequest)
+		return
+	}
+
+	// Just save the direction preference for auto-track to use
+	s.deviceManager.SetUltrabeamDirection(req.Direction)
+
+	s.sendJSON(w, map[string]string{"status": "ok"})
+}
+
 func (s *Server) sendJSON(w http.ResponseWriter, data interface{}) {
 	w.Header().Set("Content-Type", "application/json")
 	json.NewEncoder(w).Encode(data)