upload config, manage volumes, and gen architecture improvements

This commit is contained in:
Zoe
2024-12-09 02:53:26 -06:00
parent f88a5b3db5
commit ee8e3a253f
9 changed files with 720 additions and 598 deletions

View File

@@ -15,7 +15,6 @@ import (
"path/filepath"
"strconv"
"strings"
"syscall"
"time"
"github.com/briandowns/spinner"
@@ -31,12 +30,26 @@ type Config struct {
DeamonURL string `json:"deamon_url"`
}
func compressDirectory() ([]byte, error) {
func compressDirectory(compression pkg.Compression) ([]byte, error) {
var buf bytes.Buffer
gzWriter := gzip.NewWriter(&buf)
tarWriter := tar.NewWriter(gzWriter)
var err error
err := filepath.Walk(".", func(path string, info os.FileInfo, err error) error {
var gzWriter *gzip.Writer
if compression.Enabled {
gzWriter, err = gzip.NewWriterLevel(&buf, compression.Level)
if err != nil {
return nil, err
}
}
var tarWriter *tar.Writer
if gzWriter != nil {
tarWriter = tar.NewWriter(gzWriter)
} else {
tarWriter = tar.NewWriter(&buf)
}
err = filepath.Walk(".", func(path string, info os.FileInfo, err error) error {
if err != nil {
return err
}
@@ -51,7 +64,7 @@ func compressDirectory() ([]byte, error) {
}
header.Name = path
if err := tarWriter.WriteHeader(header); err != nil {
if err = tarWriter.WriteHeader(header); err != nil {
return err
}
@@ -62,7 +75,7 @@ func compressDirectory() ([]byte, error) {
}
defer file.Close()
if _, err := io.Copy(tarWriter, file); err != nil {
if _, err = io.Copy(tarWriter, file); err != nil {
return err
}
}
@@ -74,55 +87,56 @@ func compressDirectory() ([]byte, error) {
return nil, err
}
if err := tarWriter.Close(); err != nil {
if err = tarWriter.Close(); err != nil {
return nil, err
}
if err := gzWriter.Close(); err != nil {
return nil, err
if gzWriter != nil {
if err = gzWriter.Close(); err != nil {
return nil, err
}
}
return buf.Bytes(), nil
}
func getProjectName() string {
func getProjectName(command string, args []string) (string, error) {
var projectName string
if len(os.Args) < 3 {
if len(args) == 0 {
if _, err := os.Stat("flux.json"); err != nil {
fmt.Printf("Usage: flux %[1]s <app name>, or run flux %[1]s in the project directory\n", os.Args[1])
os.Exit(1)
return "", fmt.Errorf("Usage: flux %[1]s <app name>, or run flux %[1]s in the project directory\n", command)
}
fluxConfigFile, err := os.Open("flux.json")
if err != nil {
fmt.Printf("Failed to open flux.json: %v\n", err)
os.Exit(1)
return "", fmt.Errorf("Failed to open flux.json: %v\n", err)
}
defer fluxConfigFile.Close()
var config pkg.ProjectConfig
if err := json.NewDecoder(fluxConfigFile).Decode(&config); err != nil {
fmt.Printf("Failed to decode flux.json: %v\n", err)
os.Exit(1)
return "", fmt.Errorf("Failed to decode flux.json: %v\n", err)
}
projectName = config.Name
} else {
projectName = os.Args[2]
projectName = args[0]
}
return projectName
return projectName, nil
}
func main() {
func runCommand(command string, args []string, config Config, info pkg.Info) error {
loadingSpinner := spinner.New(spinner.CharSets[14], 100*time.Millisecond)
if len(os.Args) < 2 {
fmt.Println("Usage: flux <command>")
os.Exit(1)
}
defer func() {
if loadingSpinner.Active() {
loadingSpinner.Stop()
}
}()
signalChannel := make(chan os.Signal, 1)
signal.Notify(signalChannel, syscall.SIGINT, syscall.SIGTERM)
signal.Notify(signalChannel, os.Interrupt)
go func() {
<-signalChannel
if loadingSpinner.Active() {
@@ -132,7 +146,297 @@ func main() {
os.Exit(0)
}()
command := os.Args[1]
switch command {
case "deploy":
if _, err := os.Stat("flux.json"); err != nil {
return fmt.Errorf("No flux.json found, please run flux init first\n")
}
loadingSpinner.Suffix = " Deploying"
loadingSpinner.Start()
buf, err := compressDirectory(info.Compression)
if err != nil {
return fmt.Errorf("Failed to compress directory: %v\n", err)
}
body := &bytes.Buffer{}
writer := multipart.NewWriter(body)
configPart, err := writer.CreateFormFile("config", "flux.json")
if err != nil {
return fmt.Errorf("Failed to create config part: %v\n", err)
}
fluxConfigFile, err := os.Open("flux.json")
if err != nil {
return fmt.Errorf("Failed to open flux.json: %v\n", err)
}
defer fluxConfigFile.Close()
if _, err := io.Copy(configPart, fluxConfigFile); err != nil {
return fmt.Errorf("Failed to write config part: %v\n", err)
}
codePart, err := writer.CreateFormFile("code", "code.tar.gz")
if err != nil {
return fmt.Errorf("Failed to create code part: %v\n", err)
}
if _, err := codePart.Write(buf); err != nil {
return fmt.Errorf("Failed to write code part: %v\n", err)
}
if err := writer.Close(); err != nil {
return fmt.Errorf("Failed to close writer: %v\n", err)
}
resp, err := http.Post(config.DeamonURL+"/deploy", "multipart/form-data; boundary="+writer.Boundary(), body)
if err != nil {
return fmt.Errorf("Failed to send request: %v\n", err)
}
defer resp.Body.Close()
if resp.StatusCode != http.StatusOK {
responseBody, err := io.ReadAll(resp.Body)
if err != nil {
return fmt.Errorf("error reading response body: %v\n", err)
}
if len(responseBody) > 0 && responseBody[len(responseBody)-1] == '\n' {
responseBody = responseBody[:len(responseBody)-1]
}
return fmt.Errorf("Deploy failed: %s\n", responseBody)
}
loadingSpinner.Stop()
fmt.Println("Deployed successfully!")
case "stop":
projectName, err := getProjectName(command, args)
if err != nil {
return err
}
req, err := http.Post(config.DeamonURL+"/stop/"+projectName, "application/json", nil)
if err != nil {
return fmt.Errorf("Failed to stop app: %v\n", err)
}
defer req.Body.Close()
if req.StatusCode != http.StatusOK {
responseBody, err := io.ReadAll(req.Body)
if err != nil {
return fmt.Errorf("error reading response body: %v\n", err)
}
if len(responseBody) > 0 && responseBody[len(responseBody)-1] == '\n' {
responseBody = responseBody[:len(responseBody)-1]
}
return fmt.Errorf("Stop failed: %s\n", responseBody)
}
fmt.Printf("Successfully stopped %s\n", projectName)
case "start":
projectName, err := getProjectName(command, args)
if err != nil {
return err
}
req, err := http.Post(config.DeamonURL+"/start/"+projectName, "application/json", nil)
if err != nil {
return fmt.Errorf("Failed to start app: %v\n", err)
}
defer req.Body.Close()
if req.StatusCode != http.StatusOK {
responseBody, err := io.ReadAll(req.Body)
if err != nil {
return fmt.Errorf("error reading response body: %v\n", err)
}
if len(responseBody) > 0 && responseBody[len(responseBody)-1] == '\n' {
responseBody = responseBody[:len(responseBody)-1]
}
return fmt.Errorf("Start failed: %s\n", responseBody)
}
fmt.Printf("Successfully started %s\n", projectName)
case "delete":
if len(args) == 1 {
if args[0] == "all" {
var response string
fmt.Print("Are you sure you want to delete all projects? this will delete all volumes and containers associated and cannot be undone. \n[y/N] ")
fmt.Scanln(&response)
if strings.ToLower(response) != "y" {
fmt.Println("Aborting...")
return nil
}
response = ""
fmt.Printf("Are you really sure you want to delete all projects? \n[y/N] ")
fmt.Scanln(&response)
if strings.ToLower(response) != "y" {
fmt.Println("Aborting...")
return nil
}
req, err := http.NewRequest("DELETE", config.DeamonURL+"/deployments", nil)
if err != nil {
return fmt.Errorf("Failed to delete deployments: %v\n", err)
}
resp, err := http.DefaultClient.Do(req)
if err != nil {
return fmt.Errorf("failed to delete deployments: %v", err)
}
defer resp.Body.Close()
if resp.StatusCode != http.StatusOK {
responseBody, err := io.ReadAll(resp.Body)
if err != nil {
return fmt.Errorf("error reading response body: %v\n", err)
}
if len(responseBody) > 0 && responseBody[len(responseBody)-1] == '\n' {
responseBody = responseBody[:len(responseBody)-1]
}
return fmt.Errorf("delete failed: %s", responseBody)
}
fmt.Printf("Successfully deleted all projects\n")
return nil
}
}
projectName, err := getProjectName(command, args)
if err != nil {
return err
}
// ask for confirmation
fmt.Printf("Are you sure you want to delete %s? this will delete all volumes and containers associated with the deployment, and cannot be undone. \n[y/N] ", projectName)
var response string
fmt.Scanln(&response)
if strings.ToLower(response) != "y" {
fmt.Println("Aborting...")
return nil
}
req, err := http.NewRequest("DELETE", config.DeamonURL+"/deployments/"+projectName, nil)
if err != nil {
return fmt.Errorf("failed to delete app: %v", err)
}
resp, err := http.DefaultClient.Do(req)
if err != nil {
return fmt.Errorf("failed to delete app: %v", err)
}
defer resp.Body.Close()
if resp.StatusCode != http.StatusOK {
responseBody, err := io.ReadAll(resp.Body)
if err != nil {
return fmt.Errorf("error reading response body: %v", err)
}
if len(responseBody) > 0 && responseBody[len(responseBody)-1] == '\n' {
responseBody = responseBody[:len(responseBody)-1]
}
return fmt.Errorf("delete failed: %s", responseBody)
}
fmt.Printf("Successfully deleted %s\n", projectName)
case "list":
resp, err := http.Get(config.DeamonURL + "/apps")
if err != nil {
return fmt.Errorf("failed to get apps: %v", err)
}
if resp.StatusCode != http.StatusOK {
responseBody, err := io.ReadAll(resp.Body)
if err != nil {
return fmt.Errorf("error reading response body: %v", err)
}
if len(responseBody) > 0 && responseBody[len(responseBody)-1] == '\n' {
responseBody = responseBody[:len(responseBody)-1]
}
return fmt.Errorf("list failed: %s", responseBody)
}
var apps []pkg.App
if err := json.NewDecoder(resp.Body).Decode(&apps); err != nil {
return fmt.Errorf("failed to decode apps: %v", err)
}
if len(apps) == 0 {
fmt.Println("No apps found")
return nil
}
for _, app := range apps {
fmt.Printf("%s (%s)\n", app.Name, app.DeploymentStatus)
}
case "init":
var projectConfig pkg.ProjectConfig
var response string
if len(args) > 1 {
response = args[0]
} else {
fmt.Println("What is the name of your project?")
fmt.Scanln(&response)
}
projectConfig.Name = response
fmt.Println("What URL should your project listen to?")
fmt.Scanln(&response)
if strings.HasPrefix(response, "http") {
strings.TrimPrefix(response, "http://")
strings.TrimPrefix(response, "https://")
}
response = strings.Split(response, "/")[0]
projectConfig.Url = response
fmt.Println("What port does your project listen to?")
fmt.Scanln(&response)
port, err := strconv.ParseUint(response, 10, 16)
projectConfig.Port = uint16(port)
if err != nil || projectConfig.Port < 1 || projectConfig.Port > 65535 {
return fmt.Errorf("That doesnt look like a valid port, try a number between 1 and 65535")
}
configBytes, err := json.MarshalIndent(projectConfig, "", " ")
if err != nil {
return fmt.Errorf("failed to parse project config: %v", err)
}
os.WriteFile("flux.json", configBytes, 0644)
fmt.Printf("Successfully initialized project %s\n", projectConfig.Name)
default:
return fmt.Errorf("unknown command: %s", command)
}
return nil
}
func main() {
if len(os.Args) < 2 {
fmt.Println("Usage: flux <command>")
os.Exit(1)
}
if _, err := os.Stat(filepath.Join(configPath, "config.json")); err != nil {
if err := os.MkdirAll(configPath, 0755); err != nil {
@@ -158,325 +462,35 @@ func main() {
os.Exit(1)
}
switch command {
case "deploy":
if _, err := os.Stat("flux.json"); err != nil {
fmt.Printf("No flux.json found, please run flux init first\n")
os.Exit(1)
}
command := os.Args[1]
args := os.Args[2:]
loadingSpinner.Suffix = " Deploying"
loadingSpinner.Start()
resp, err := http.Get(config.DeamonURL + "/heartbeat")
if err != nil {
fmt.Println("Failed to connect to daemon")
os.Exit(1)
}
defer resp.Body.Close()
if resp.StatusCode != http.StatusOK {
fmt.Println("Failed to connect to daemon")
os.Exit(1)
}
buf, err := compressDirectory()
if err != nil {
loadingSpinner.Stop()
var info pkg.Info
err = json.NewDecoder(resp.Body).Decode(&info)
if err != nil {
fmt.Printf("Failed to decode info: %v\n", err)
os.Exit(1)
}
fmt.Printf("Failed to compress directory: %v\n", err)
os.Exit(1)
}
if resp.StatusCode != http.StatusOK {
fmt.Println("Failed to connect to daemon")
os.Exit(1)
}
body := &bytes.Buffer{}
writer := multipart.NewWriter(body)
configPart, err := writer.CreateFormFile("config", "flux.json")
if err != nil {
loadingSpinner.Stop()
fmt.Printf("Failed to create config part: %v\n", err)
os.Exit(1)
}
fluxConfigFile, err := os.Open("flux.json")
if err != nil {
loadingSpinner.Stop()
fmt.Printf("Failed to open flux.json: %v\n", err)
os.Exit(1)
}
defer fluxConfigFile.Close()
if _, err := io.Copy(configPart, fluxConfigFile); err != nil {
loadingSpinner.Stop()
fmt.Printf("Failed to write config part: %v\n", err)
os.Exit(1)
}
codePart, err := writer.CreateFormFile("code", "code.tar.gz")
if err != nil {
loadingSpinner.Stop()
fmt.Printf("Failed to create code part: %v\n", err)
os.Exit(1)
}
if _, err := codePart.Write(buf); err != nil {
loadingSpinner.Stop()
fmt.Printf("Failed to write code part: %v\n", err)
os.Exit(1)
}
if err := writer.Close(); err != nil {
loadingSpinner.Stop()
fmt.Printf("Failed to close writer: %v\n", err)
os.Exit(1)
}
resp, err := http.Post(config.DeamonURL+"/deploy", "multipart/form-data; boundary="+writer.Boundary(), body)
if err != nil {
loadingSpinner.Stop()
fmt.Printf("Failed to send request: %v\n", err)
os.Exit(1)
}
defer resp.Body.Close()
if resp.StatusCode != http.StatusOK {
loadingSpinner.Stop()
responseBody, err := io.ReadAll(resp.Body)
if err != nil {
fmt.Printf("error reading response body: %v\n", err)
os.Exit(1)
}
if len(responseBody) > 0 && responseBody[len(responseBody)-1] == '\n' {
responseBody = responseBody[:len(responseBody)-1]
}
fmt.Printf("Deploy failed: %s\n", responseBody)
os.Exit(1)
}
loadingSpinner.Stop()
fmt.Println("Deployed successfully!")
case "stop":
projectName := getProjectName()
req, err := http.Post(config.DeamonURL+"/stop/"+projectName, "application/json", nil)
if err != nil {
fmt.Printf("Failed to stop app: %v\n", err)
os.Exit(1)
}
defer req.Body.Close()
if req.StatusCode != http.StatusOK {
responseBody, err := io.ReadAll(req.Body)
if err != nil {
fmt.Printf("error reading response body: %v\n", err)
os.Exit(1)
}
if len(responseBody) > 0 && responseBody[len(responseBody)-1] == '\n' {
responseBody = responseBody[:len(responseBody)-1]
}
fmt.Printf("Stop failed: %s\n", responseBody)
os.Exit(1)
}
fmt.Printf("Successfully stopped %s\n", projectName)
case "start":
projectName := getProjectName()
req, err := http.Post(config.DeamonURL+"/start/"+projectName, "application/json", nil)
if err != nil {
fmt.Printf("Failed to start app: %v\n", err)
os.Exit(1)
}
defer req.Body.Close()
if req.StatusCode != http.StatusOK {
responseBody, err := io.ReadAll(req.Body)
if err != nil {
fmt.Printf("error reading response body: %v\n", err)
os.Exit(1)
}
if len(responseBody) > 0 && responseBody[len(responseBody)-1] == '\n' {
responseBody = responseBody[:len(responseBody)-1]
}
fmt.Printf("Start failed: %s\n", responseBody)
os.Exit(1)
}
fmt.Printf("Successfully started %s\n", projectName)
case "delete":
if len(os.Args) == 3 {
if os.Args[2] == "all" {
var response string
fmt.Print("Are you sure you want to delete all projects? this will delete all volumes and containers associated and cannot be undone. \n[y/N] ")
fmt.Scanln(&response)
if strings.ToLower(response) != "y" {
fmt.Println("Aborting...")
os.Exit(0)
}
response = ""
fmt.Printf("Are you really sure you want to delete all projects? \n[y/N] ")
fmt.Scanln(&response)
if strings.ToLower(response) != "y" {
fmt.Println("Aborting...")
os.Exit(0)
}
req, err := http.NewRequest("DELETE", config.DeamonURL+"/deployments", nil)
if err != nil {
fmt.Printf("Failed to delete deployments: %v\n", err)
os.Exit(1)
}
resp, err := http.DefaultClient.Do(req)
if err != nil {
fmt.Printf("Failed to delete deployments: %v\n", err)
os.Exit(1)
}
defer resp.Body.Close()
if resp.StatusCode != http.StatusOK {
responseBody, err := io.ReadAll(resp.Body)
if err != nil {
fmt.Printf("error reading response body: %v\n", err)
os.Exit(1)
}
if len(responseBody) > 0 && responseBody[len(responseBody)-1] == '\n' {
responseBody = responseBody[:len(responseBody)-1]
}
fmt.Printf("Delete failed: %s\n", responseBody)
os.Exit(1)
}
fmt.Printf("Successfully deleted all projects\n")
return
}
}
projectName := getProjectName()
// ask for confirmation
fmt.Printf("Are you sure you want to delete %s? this will delete all volumes and containers associated with the deployment, and cannot be undone. \n[y/N] ", projectName)
var response string
fmt.Scanln(&response)
if strings.ToLower(response) != "y" {
fmt.Println("Aborting...")
os.Exit(0)
}
req, err := http.NewRequest("DELETE", config.DeamonURL+"/deployments/"+projectName, nil)
if err != nil {
fmt.Printf("Failed to delete app: %v\n", err)
os.Exit(1)
}
resp, err := http.DefaultClient.Do(req)
if err != nil {
fmt.Printf("Failed to delete app: %v\n", err)
os.Exit(1)
}
defer resp.Body.Close()
if resp.StatusCode != http.StatusOK {
responseBody, err := io.ReadAll(resp.Body)
if err != nil {
fmt.Printf("error reading response body: %v\n", err)
os.Exit(1)
}
if len(responseBody) > 0 && responseBody[len(responseBody)-1] == '\n' {
responseBody = responseBody[:len(responseBody)-1]
}
fmt.Printf("Delete failed: %s\n", responseBody)
os.Exit(1)
}
fmt.Printf("Successfully deleted %s\n", projectName)
case "list":
resp, err := http.Get(config.DeamonURL + "/apps")
if err != nil {
fmt.Printf("Failed to get apps: %v\n", err)
os.Exit(1)
}
if resp.StatusCode != http.StatusOK {
responseBody, err := io.ReadAll(resp.Body)
if err != nil {
fmt.Printf("error reading response body: %v\n", err)
os.Exit(1)
}
if len(responseBody) > 0 && responseBody[len(responseBody)-1] == '\n' {
responseBody = responseBody[:len(responseBody)-1]
}
fmt.Printf("List failed: %s\n", responseBody)
os.Exit(1)
}
var apps []pkg.App
if err := json.NewDecoder(resp.Body).Decode(&apps); err != nil {
fmt.Printf("Failed to decode apps: %v\n", err)
os.Exit(1)
}
if len(apps) == 0 {
fmt.Println("No apps found")
os.Exit(0)
}
for _, app := range apps {
fmt.Printf("%s (%s)\n", app.Name, app.DeploymentStatus)
}
case "init":
var projectConfig pkg.ProjectConfig
var response string
if len(os.Args) > 2 {
response = os.Args[2]
} else {
fmt.Println("What is the name of your project?")
fmt.Scanln(&response)
projectConfig.Name = response
}
fmt.Println("What URL should your project listen to?")
fmt.Scanln(&response)
if strings.HasPrefix(response, "http") {
strings.TrimPrefix(response, "http://")
strings.TrimPrefix(response, "https://")
}
response = strings.Split(response, "/")[0]
projectConfig.Url = response
fmt.Println("What port does your project listen to?")
fmt.Scanln(&response)
port, err := strconv.ParseUint(response, 10, 16)
projectConfig.Port = uint16(port)
if err != nil || projectConfig.Port < 1 || projectConfig.Port > 65535 {
fmt.Println("That doesnt look like a valid port", err)
os.Exit(1)
}
configBytes, err := json.MarshalIndent(projectConfig, "", " ")
if err != nil {
fmt.Printf("Failed to marshal project config: %v\n", err)
os.Exit(1)
}
os.WriteFile("flux.json", configBytes, 0644)
fmt.Printf("Successfully initialized project %s\n", projectConfig.Name)
default:
fmt.Println("Unknown command:", command)
err = runCommand(command, args, config, info)
if err != nil {
fmt.Printf("%v\n", err)
os.Exit(1)
}
}