initial commit

server/deploy.go

@@ -0,0 +1,177 @@
+package server
+
+import (
+	"database/sql"
+	"encoding/json"
+	"fmt"
+	"log"
+	"mime/multipart"
+	"net/http"
+	"os/exec"
+	"strings"
+)
+
+type DeployRequest struct {
+	Config multipart.File `form:"config"`
+	Code   multipart.File `form:"code"`
+}
+
+type DeployResponse struct {
+	AppID int64 `json:"app_id"`
+}
+
+type ProjectConfig struct {
+	Name        string   `json:"name"`
+	Urls        []string `json:"urls"`
+	Port        int      `json:"port"`
+	EnvFile     string   `json:"env_file"`
+	Environment []string `json:"environment"`
+}
+
+func (s *FluxServer) DeployHandler(w http.ResponseWriter, r *http.Request) {
+	err := r.ParseMultipartForm(10 << 30) // 10 GiB
+	if err != nil {
+		http.Error(w, err.Error(), http.StatusBadRequest)
+		return
+	}
+
+	// bind to DeployRequest struct
+	var deployRequest DeployRequest
+	deployRequest.Config, _, err = r.FormFile("config")
+	if err != nil {
+		http.Error(w, "No flux.json found", http.StatusBadRequest)
+		return
+	}
+	defer deployRequest.Config.Close()
+
+	deployRequest.Code, _, err = r.FormFile("code")
+	if err != nil {
+		http.Error(w, "No code archive found", http.StatusBadRequest)
+		return
+	}
+	defer deployRequest.Code.Close()
+
+	var projectConfig ProjectConfig
+	if err := json.NewDecoder(deployRequest.Config).Decode(&projectConfig); err != nil {
+		log.Printf("Failed to decode config: %v", err)
+		http.Error(w, "Invalid flux.json", http.StatusBadRequest)
+		return
+	}
+
+	if projectConfig.Name == "" {
+		http.Error(w, "No project name specified", http.StatusBadRequest)
+		return
+	}
+
+	if projectConfig.Urls == nil || len(projectConfig.Urls) == 0 {
+		http.Error(w, "No deployment urls specified", http.StatusBadRequest)
+		return
+	}
+
+	if projectConfig.Port == 0 {
+		http.Error(w, "No port specified", http.StatusBadRequest)
+		return
+	}
+
+	log.Printf("Deploying project %s to %s", projectConfig.Name, projectConfig.Urls)
+
+	projectPath, err := s.UploadAppCode(deployRequest.Code, projectConfig)
+	if err != nil {
+		log.Printf("Failed to upload code: %v", err)
+		http.Error(w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+
+	prepareCmd := exec.Command("go", "generate")
+	prepareCmd.Dir = projectPath
+	err = prepareCmd.Run()
+	if err != nil {
+		log.Printf("Failed to prepare project: %s", err)
+		http.Error(w, fmt.Sprintf("Failed to prepare project: %s", err), http.StatusInternalServerError)
+		return
+	}
+
+	imageName := fmt.Sprintf("%s-image", projectConfig.Name)
+	buildCmd := exec.Command("pack", "build", imageName, "--builder", s.config.Builder)
+	buildCmd.Dir = projectPath
+	err = buildCmd.Run()
+	if err != nil {
+		log.Printf("Failed to build image: %s", err)
+		http.Error(w, fmt.Sprintf("Failed to build image: %s", err), http.StatusInternalServerError)
+		return
+	}
+
+	containerID, err := s.containerManager.DeployContainer(r.Context(), imageName, projectConfig.Name, projectPath, projectConfig)
+	if err != nil {
+		log.Printf("Failed to deploy container: %v", err)
+		http.Error(w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+
+	deploymentResult, err := s.db.Exec("INSERT INTO deployments (urls) VALUES (?)", strings.Join(projectConfig.Urls, ","))
+	if err != nil {
+		log.Printf("Failed to insert deployment: %v", err)
+		http.Error(w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+
+	deploymentID, err := deploymentResult.LastInsertId()
+	if err != nil {
+		log.Printf("Failed to get deployment id: %v", err)
+		http.Error(w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+
+	_, err = s.db.Exec("INSERT INTO containers (container_id, deployment_id, status) VALUES (?, ?, ?)", containerID, deploymentID, "pending")
+	if err != nil {
+		log.Printf("Failed to get container id: %v", err)
+		http.Error(w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+
+	appExists := s.db.QueryRow("SELECT * FROM apps WHERE name = ?", projectConfig.Name)
+	configBytes, err := json.Marshal(projectConfig)
+	if err != nil {
+		log.Printf("Failed to marshal project config: %v", err)
+		http.Error(w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+	var appResult sql.Result
+
+	if appExists.Err() == sql.ErrNoRows {
+		// create app in the database
+		appResult, err = s.db.Exec("INSERT INTO apps (name, image, project_path, project_config, deployment_id) VALUES (?, ?, ?, ?, ?)", projectConfig.Name, imageName, projectPath, configBytes, deploymentID)
+		if err != nil {
+			log.Printf("Failed to insert app: %v", err)
+			http.Error(w, err.Error(), http.StatusInternalServerError)
+			return
+		}
+	} else {
+		// update app in the database
+		appResult, err = s.db.Exec("UPDATE apps SET project_config = ?, deployment_id = ? WHERE name = ?", configBytes, deploymentID, projectConfig.Name)
+		if err != nil {
+			log.Printf("Failed to update app: %v", err)
+			http.Error(w, err.Error(), http.StatusInternalServerError)
+			return
+		}
+	}
+
+	appId, err := appResult.LastInsertId()
+	if err != nil {
+		log.Printf("Failed to get app id: %v", err)
+		http.Error(w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+
+	json.NewEncoder(w).Encode(DeployResponse{
+		AppID: appId,
+	})
+}
+
+func (s *FluxServer) ListAppsHandler(w http.ResponseWriter, r *http.Request) {
+	// Implement app listing logic
+	apps := s.db.QueryRow("SELECT * FROM apps")
+
+	w.Header().Set("Content-Type", "application/json")
+	json.NewEncoder(w).Encode(apps)
+}