infrastructure/docker/webpull/main.go

package main

import (
	"fmt"
	"io"
	"os/exec"
	"os"
	"log"
	"net/http"
	"strings"
)

func myexec(w io.Writer, main string, params ...string) error {
	cmd := exec.Command(main, params...)
	cmd.Stdout = w
	cmd.Stderr = w
	err := cmd.Run()
	if err != nil {
		fmt.Fprintf(w, "Failed to run: %s %s\n", main, strings.Join(params, " "))
	}
	return err
}

func update(w io.Writer) {
	fmt.Fprintf(w, "Start update...\n")
	_, err := os.Stat("./.git")
	if err != nil {
		fmt.Fprintf(w, ".git folder does not exist, creating it...\n")
		err := myexec(w, "git", "init")
		if err != nil {
			return
		}
	}

	err = myexec(w, "git", "remote", "get-url", "origin")
	if err != nil {
		repo, exists := os.LookupEnv("WEBPULL_REPO")
		if !exists {
			fmt.Fprintf(w, "You must define WEBPULL_REPO env variable...\n")
			return
		}
		fmt.Fprintf(w, "git remote is not yet set...\n")
		err := myexec(w, "git", "remote", "add", "origin", repo)
		if err != nil {
			return
		}
	}

	err = myexec(w, "git", "pull", "origin", "master")
	if err != nil {
		fmt.Fprintf(w, "Failed to pull...\n")
		return
	}

	_, err = os.Stat("./.webpool")
	if err != nil {
		fmt.Fprintf(w, "You must create an executable file named '.webpool' at the root of your repository.\nIf you have nothing to run, just create an empty bash script...\n")
		return
	}

	err = myexec(w, "./.webpool")
	if err != nil {
		fmt.Fprintf(w, "An error occured during script execution\n")
		return
	}

	fmt.Fprintf(w, "Success.\n")
}

func main() {
	
	update(os.Stdout)
	fs := http.FileServer(http.Dir("./static"))
	http.HandleFunc("/update", func(w http.ResponseWriter, r *http.Request) {
		update(w)
	})
	http.Handle("/", fs)

	log.Fatal(http.ListenAndServe(":8080", nil))
}