initial commit

hooks/hooks.go

@@ -0,0 +1,146 @@
+package hooks
+
+import (
+	"encoding/json"
+	"io/ioutil"
+
+	"github.com/adnanh/webhook/rules"
+)
+
+// Hook is a structure that contains command to be executed
+// and the current working directory name where that command should be executed
+type Hook struct {
+	ID      string     `json:"id"`
+	Command string     `json:"command"`
+	Cwd     string     `json:"cwd"`
+	Secret  string     `json:"secret"`
+	Rule    rules.Rule `json:"trigger-rule"`
+}
+
+// Hooks represents structure that contains list of Hook objects
+// and the name of file which is correspondingly mapped to it
+type Hooks struct {
+	fileName string
+	list     []Hook
+}
+
+// UnmarshalJSON implementation for a single hook
+func (h *Hook) UnmarshalJSON(j []byte) error {
+	m := make(map[string]interface{})
+	err := json.Unmarshal(j, &m)
+
+	if err != nil {
+		return err
+	}
+
+	if v, ok := m["id"]; ok {
+		h.ID = v.(string)
+	}
+
+	if v, ok := m["command"]; ok {
+		h.Command = v.(string)
+	}
+
+	if v, ok := m["cwd"]; ok {
+		h.Cwd = v.(string)
+	}
+
+	if v, ok := m["trigger-rule"]; ok {
+		rule := v.(map[string]interface{})
+
+		if ruleValue, ok := rule["match"]; ok {
+			ruleString, _ := json.Marshal(ruleValue)
+			rulePtr := new(rules.MatchRule)
+
+			err = json.Unmarshal(ruleString, rulePtr)
+			if err != nil {
+				return err
+			}
+
+			h.Rule = *rulePtr
+
+		} else if ruleValue, ok := rule["not"]; ok {
+			ruleString, _ := json.Marshal(ruleValue)
+			rulePtr := new(rules.NotRule)
+
+			err = json.Unmarshal(ruleString, rulePtr)
+			if err != nil {
+				return err
+			}
+
+			h.Rule = *rulePtr
+
+		} else if ruleValue, ok := rule["and"]; ok {
+			ruleString, _ := json.Marshal(ruleValue)
+			rulePtr := new(rules.AndRule)
+
+			err = json.Unmarshal(ruleString, rulePtr)
+			if err != nil {
+				return err
+			}
+
+			h.Rule = *rulePtr
+
+		} else if ruleValue, ok := rule["or"]; ok {
+			ruleString, _ := json.Marshal(ruleValue)
+			rulePtr := new(rules.OrRule)
+
+			err = json.Unmarshal(ruleString, rulePtr)
+			if err != nil {
+				return err
+			}
+
+			h.Rule = *rulePtr
+		}
+
+	}
+
+	return nil
+}
+
+// New creates an instance of Hooks, tries to unmarshal contents of hookFile
+// and returns a pointer to the newly created instance
+func New(hookFile string) (*Hooks, error) {
+	h := &Hooks{fileName: hookFile}
+
+	// parse hook file for hooks
+	file, e := ioutil.ReadFile(hookFile)
+
+	if e != nil {
+		return h, e
+	}
+
+	e = json.Unmarshal(file, &(h.list))
+
+	h.SetDefaults()
+
+	return h, e
+}
+
+// Match looks for the hook with the given id in the list of hooks
+// and returns the pointer to the hook if it exists, or nil if it doesn't exist
+func (h *Hooks) Match(id string, params interface{}) *Hook {
+	for i := range h.list {
+		if h.list[i].ID == id {
+			if h.list[i].Rule.Evaluate(params) {
+				return &h.list[i]
+			}
+		}
+	}
+
+	return nil
+}
+
+// Count returns number of hooks in the list
+func (h *Hooks) Count() int {
+	return len(h.list)
+}
+
+// SetDefaults sets default values that were ommited for hooks in JSON file
+func (h *Hooks) SetDefaults() {
+	for i := range h.list {
+		if h.list[i].Cwd == "" {
+			h.list[i].Cwd = "."
+		}
+	}
+}