diff --git a/docker/webpull/main.go b/docker/webpull/main.go index 2161256..46c90b9 100644 --- a/docker/webpull/main.go +++ b/docker/webpull/main.go @@ -2,6 +2,7 @@ package main import ( "fmt" + "errors" "io" "os/exec" "os" @@ -21,14 +22,14 @@ func myexec(w io.Writer, main string, params ...string) error { return err } -func update(w io.Writer) { +func update(w io.Writer) error { 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 + return err } } @@ -37,41 +38,60 @@ func update(w io.Writer) { repo, exists := os.LookupEnv("WEBPULL_REPO") if !exists { fmt.Fprintf(w, "You must define WEBPULL_REPO env variable...\n") - return + return errors.New("Missing environment variable WEBPULL_REPO") } fmt.Fprintf(w, "git remote is not yet set...\n") err := myexec(w, "git", "remote", "add", "origin", repo) if err != nil { - return + return err } } err = myexec(w, "git", "pull", "origin", "master") if err != nil { fmt.Fprintf(w, "Failed to pull...\n") - return + return err } - _, err = os.Stat("./.webpool") + _, err = os.Stat("./.webpull") 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 + fmt.Fprintf(w, "You must create an executable file named '.webpull' at the root of your repository.\nIf you have nothing to run, just create an empty bash script...\n") + return err } - err = myexec(w, "./.webpool") + err = myexec(w, "./.webpull") if err != nil { fmt.Fprintf(w, "An error occured during script execution\n") - return + return err } fmt.Fprintf(w, "Success.\n") + return nil } func main() { - - update(os.Stdout) + token, exists := os.LookupEnv("WEBPULL_TOKEN") + if !exists { + log.Fatal("Environment variable 'WEBPULL_TOKEN' must be defined") + } + + if update(os.Stdout) != nil { + log.Fatal("Initial 'update' failed") + } + fs := http.FileServer(http.Dir("./static")) http.HandleFunc("/update", func(w http.ResponseWriter, r *http.Request) { + keys, ok := r.URL.Query()["token"] + if !ok || len(keys[0]) < 1 { + http.Error(w, "Missing 'token' query parameter", 401) + return + } + + if keys[0] != token { + http.Error(w, "Wrong token", 401) + return + } + update(w) }) http.Handle("/", fs)