From afcbcdd1a72aed542d18a1ee7ee5dd7e118deb74 Mon Sep 17 00:00:00 2001
From: Noah <noah.matthew@metzger.cc>
Date: Fri, 8 Mar 2024 14:51:58 -0600
Subject: [PATCH] Restart lapis server on receiving a webhook (fix #2)

---
 .gitignore |  1 +
 main.go    | 51 +++++++++++++++++++++++++++++++++++++--------------
 2 files changed, 38 insertions(+), 14 deletions(-)

diff --git a/.gitignore b/.gitignore
index 9780db9..4f52390 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1 +1,2 @@
 lapisdeploy
+sites/
diff --git a/main.go b/main.go
index 1e551f8..9acc727 100644
--- a/main.go
+++ b/main.go
@@ -6,15 +6,22 @@ import (
 	"log"
 	"net/http"
 	"os"
+	"os/exec"
+	"strings"
+	"syscall"
+
 	"github.com/gogs/git-module"
 )
 
+const sites_dir = "./sites"
+
 type DeployError struct {
 	code int
 	where string
 	details string
+	command_output string
 }
-func newDeployError(code int, where string, details string) DeployError {
+func newDeployError(code int, where string, details string, command_output string) DeployError {
 	return DeployError{ code: code, where: where, details: details }
 }
 
@@ -27,26 +34,41 @@ func fileExists(path string) (bool, error) {
 }
 
 // startServer attempts to start a Lapis server in the given repository.
-func startServer(name string) DeployError {
-	
+func restartServer(name string) DeployError {
+	log.Printf("Restarting Lapis server on repository %s...", name)
+
+	old_cwd, _ := os.Getwd()
+	defer syscall.Chdir(old_cwd)
+	syscall.Chdir(sites_dir+"/"+name)
+
+	out, err := exec.Command("lapis", "build").CombinedOutput()
+	if err != nil {
+		return newDeployError(1, "restartServer", fmt.Sprintf("Failed to run 'lapis build' in repository '%s'", name), string(out))
+	}
+	log.Printf("[lapis build] %s", out)
+	if !strings.Contains(string(out), "HUP") {
+		return newDeployError(1, "restartServer", "'lapis build' command returned unexpected value!", string(out))
+	}
 	return DeployError{}
 }
 
 // pullRepo attempts to pull or clone a repository using the given name and url
 func pullRepo(url string, name string) DeployError  {	
-	exists, err := fileExists(name)
+	exists, err := fileExists(sites_dir+"/"+name)
 	if err != nil {
 		return newDeployError(1, "fileExists",
-			fmt.Sprintf("Failed to check whether folder '%s' exists while pulling down repository '%s'!", name, name))
+			fmt.Sprintf("Failed to check whether folder '%s' exists while pulling down repository '%s'!", name, name), "")
 	}
-	if !exists {	
-		if err = git.Clone(url, name); err != nil {
-			return newDeployError(1, "pullRepo", fmt.Sprintf("Failed to pull down repository '%s'", name))
+	if !exists {
+		log.Printf("Cloning repository %s...", name)
+		if err = git.Clone(url, sites_dir+"/"+name); err != nil {
+			return newDeployError(1, "pullRepo", fmt.Sprintf("Failed to pull down repository '%s'", name), "")
 		}
 	} else {
-		repo,err := git.Open(name)	
+		log.Printf("Pulling down repository %s...", name)
+		repo,err := git.Open(sites_dir+"/"+name)
 		if err != nil {
-			return newDeployError(1, "pullRepo", fmt.Sprintf("Failed to open git repository '%s'", name))
+			return newDeployError(1, "pullRepo", fmt.Sprintf("Failed to open git repository '%s'", name), "")
 		}
 		repo.Pull()
 	}
@@ -56,15 +78,16 @@ func pullRepo(url string, name string) DeployError  {
 func handler(data map[string]interface{}) {
 	repository := data["repository"].(map[string]interface{})
 	log.Default().Printf("Repo: %s", repository["full_name"])
-	log.Default().Printf("Ref: %s", data["ref"])
-
 
 	deploy_error := pullRepo(repository["ssh_url"].(string), repository["name"].(string))
 	if deploy_error.code != 0 {
 		log.Default().Printf("Error in %s! [Code: %d] (%s)", deploy_error.where, deploy_error.code, deploy_error.details)
 	}
-	//syscall.Chdir(repository["name"].(string))
-	//syscall.Exec("lapis", []string{"serve"}, []string{"production"})
+	deploy_error = restartServer(repository["name"].(string))
+	if deploy_error.code != 0 {
+		log.Printf("Error in %s! [Code: %d] (%s) {Command output: '%s'}",
+			deploy_error.where, deploy_error.code, deploy_error.details, deploy_error.command_output)
+	}	
 }
 
 func main()  {