serve.go

package main

import (
	"encoding/json"
	"log"
	"net/http"
	"os"

	"github.com/binxio/cru/ref"
	"gopkg.in/src-d/go-git.v4/plumbing"
)

type ContainerReferenceUpdateRequest struct {
	CommitMessage   string   `json:"commit-message"`
	ImageReferences []string `json:"image-references"`
	MatchingTag     bool     `json:"matching-tag,omitEmpty"`
}

type ContainerReferenceUpdateResponse struct {
	GitURL string   `json:"git-url,omitempty"`
	Files  []string `json:"files,omitempty"`
	Hash   string   `json:"commit-sha,omitempty"`
}

func (c Cru) ServeHTTP(w http.ResponseWriter, r *http.Request) {
	var hash plumbing.Hash
	var request ContainerReferenceUpdateRequest
	var response ContainerReferenceUpdateResponse

	err := json.NewDecoder(r.Body).Decode(&request)
	if err != nil {
		http.Error(w, err.Error(), http.StatusBadRequest)
		return
	}
	c.CommitMsg = request.CommitMessage
	if c.CommitMsg == "" {
		http.Error(w, "commit message is empty", http.StatusBadRequest)
		return
	}

	c.imageRefs = make(ref.ContainerImageReferences, 0)
	for _, r := range request.ImageReferences {
		r, err := ref.NewContainerImageReference(r)
		if err != nil {
			http.Error(w, err.Error(), http.StatusBadRequest)
			return
		}
		c.imageRefs = append(c.imageRefs, *r)
	}
	if len(c.imageRefs) == 0 {
		http.Error(w, err.Error(), http.StatusBadRequest)
		return
	}

	if err = c.ConnectToRepository(); err != nil {
		http.Error(w, err.Error(), http.StatusInternalServerError)
		return
	}
	c.MatchingTag = request.MatchingTag
	if err = c.Walk(Update); err != nil {
		http.Error(w, err.Error(), http.StatusInternalServerError)
		return
	}

	if len(c.updatedFiles) > 0 {
		log.Printf("INFO: updated a total of %d files", len(c.updatedFiles))
		if hash, err = c.Commit(); err != nil {
			http.Error(w, err.Error(), http.StatusInternalServerError)
			return
		}
		if err = c.Push(); err != nil {
			http.Error(w, err.Error(), http.StatusInternalServerError)
			return
		}
		response = ContainerReferenceUpdateResponse{c.Url, c.updatedFiles, hash.String()}
	} else {
		log.Println("INFO: no files were updated by cru")
		response = ContainerReferenceUpdateResponse{GitURL: c.Url}
	}

	if body, err := json.Marshal(response); err == nil {
		w.Header().Set("Content-Type", "application/json")
		w.Write(body)
		return
	} else {
		http.Error(w, err.Error(), http.StatusInternalServerError)
		return
	}
}

func (c *Cru) ListenAndServe() {
	port := os.Getenv("PORT")
	if port == "" {
		port = "8080"
	}

	if c.Port == "" {
		c.Port = os.Getenv("PORT")
	}
	if c.Port == "" {
		c.Port = "8080"
	}

	log.Printf("Listening on port %s", c.Port)
	if err := http.ListenAndServe(":"+c.Port, c); err != nil {
		log.Fatal(err)
	}
}