From 2cfaa0db6515cb5ce1f6752bd60bc3f6b6eb0646 Mon Sep 17 00:00:00 2001
From: Lukas Matt <lukas@zauberstuhl.de>
Date: Sat, 17 Mar 2018 17:32:15 +0100
Subject: [PATCH] Add custom build triggers to repositories

---
 frontend.go         | 19 +++++++++++++------
 repo.go             |  2 ++
 templates/auth.html | 19 ++++++++++++++++---
 webhook.go          |  4 ++--
 4 files changed, 33 insertions(+), 11 deletions(-)

diff --git a/frontend.go b/frontend.go
index 316dcd6..2d77045 100644
--- a/frontend.go
+++ b/frontend.go
@@ -104,18 +104,25 @@ func resultPage(w http.ResponseWriter, r *http.Request) {
       return
     }
 
-    var optIn bool
-    if strings.ToUpper(r.URL.Query().Get("optin")) == "ON" {
-      optIn = true
-    }
-
     secret := Secret(16)
     repo := Repo{
       Project: project,
       Slug: repo,
       Token: accessToken,
       Secret: secret,
-      OptIn: optIn,
+    }
+
+    // set the repo to opt-in
+    if strings.ToUpper(r.URL.Query().Get("optin")) == "ON" {
+      repo.OptIn = true
+    }
+    // define custom opt-in flag
+    if r.URL.Query().Get("optinFlag") != "" {
+      repo.OptInFlag = r.URL.Query().Get("optinFlag")
+    }
+    // define custom opt-out flag
+    if r.URL.Query().Get("optoutFlag") != "" {
+      repo.OptOutFlag = r.URL.Query().Get("optoutFlag")
     }
 
     name := "web"
diff --git a/repo.go b/repo.go
index 2e14d0a..911e73f 100644
--- a/repo.go
+++ b/repo.go
@@ -28,6 +28,8 @@ type Repo struct {
   Token string
   Secret string
   OptIn bool
+  OptInFlag string `gorm:"default:'ci'"`
+  OptOutFlag string `gorm:"default:'ci skip'"`
 }
 
 type Repos []Repo
diff --git a/templates/auth.html b/templates/auth.html
index b15865f..27daf07 100644
--- a/templates/auth.html
+++ b/templates/auth.html
@@ -19,11 +19,24 @@
       <input type="checkbox" class="form-check-input" id="optinInput" name="optin" describedby="optinHelp">
       <label class="form-check-label" for="optinInput">Check if you want to opt-in on a new test</label>
       <small id="optinHelp" class="form-text text-muted">
-        That would mean that tests are triggered if you specify <code>[ci]</code> in the PR title, message
+        That would mean that tests are triggered if you specify <code>[ci]</code> in the PR title, body, commit-message
         or if the PR was tagged with <span class="badge badge-primary">ci</span>. If you leave the box unchecked
-        tests will be triggered on every pull-request. However you can skip them by writing <code>[ci skip]</code> in the PR title,
-        message or if the PR was tagged with <span class="badge badge-primary">ci skip</span>.
+        tests will be triggered on every pull-request. However you can skip them by using <code>[ci skip]</code>
+        in the mentioned fields or tag it with <span class="badge badge-primary">ci skip</span>.
       </small>
+      <div class="form-group">
+        <label for="optinFlagInput">(optional) Custom Flags</label>
+        <input type="text" class="form-control" id="optinFlagInput" name="optinFlag" aria-describedby="optinFlagHelp" placeholder="default is 'ci'">
+        <small id="optinFlagHelp" class="form-text text-muted">
+          If you want to use custom build triggers for opt-in you can specify it here.
+        </small>
+      </div>
+      <div class="form-group">
+        <input type="text" class="form-control" name="optoutFlag" aria-describedby="optoutFlagHelp" placeholder="default is 'ci skip'">
+        <small id="optoutFlagHelp" class="form-text text-muted">
+          If you want to use custom build triggers for opt-out you can specify it here.
+        </small>
+      </div>
     </div>
     <button type="submit" class="btn btn-primary">Submit</button>
   </form>
diff --git a/webhook.go b/webhook.go
index 5dc08d5..e1563d0 100644
--- a/webhook.go
+++ b/webhook.go
@@ -81,9 +81,9 @@ func webhook(w http.ResponseWriter, r *http.Request) {
   //}
 
   var flagExists = false
-  var buildFlag = "ci skip"
+  var buildFlag = repo.OptOutFlag
   if repo.OptIn {
-    buildFlag = "ci"
+    buildFlag = repo.OptInFlag
   }
 
   // check PR title and body for [ci] or [ci skip] flag