aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--docker/webpull/main.go44
1 files changed, 32 insertions, 12 deletions
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)