autogits/gitea-events-rabbitmq-publisher/main.go

package main

import (
	"fmt"
	"log"
	"net/http"
	"os"

	"src.opensuse.org/autogits/common"
)

const (
	ListenAddr          = "[::1]:8002"
	RabbitForwarderPath = "rabbitmq-forwarder"
)

func connectToRabbitMQ() {
	host := os.Getenv("RABBITMQ_HOST")
	username := os.Getenv("RABBITMQ_USERNAME")
	password := os.Getenv("RABBITMQ_PASSWORD")

	if len(host) == 0 || len(username) == 0 || len(password) == 0 {
		fmt.Printf("Missing RABBITMQ_HOST, RABBITMQ_USERNAME, RABBITMQ_PASSWORD")
		os.Exit(1)
	}
	
	go ConnectToExchangeForPublish(host, username, password)
}

func main() {
	common.RequireGiteaSecretToken()
	connectToRabbitMQ()

	http.HandleFunc("POST /rabbitmq-forwarder/{Org}", func(res http.ResponseWriter, req *http.Request) {
		if len(req.Header.Get("Content-Type")) == 0 ||
			req.Header["Content-Type"][0] != "application/json" ||
			req.Method != "POST" {

			res.WriteHeader(http.StatusInternalServerError)
			return
		}

		hdr := req.Header[common.GiteaRequestHeader]
		if len(hdr) != 1 {
			//			h.LogError("Unsupported number of %s headers: %d: %#v\n", GiteaRequestHeader, len(hdr), hdr)
			//			h.WriteError()
			res.WriteHeader(http.StatusInternalServerError)
			return
		}
		reqType := hdr[0]
		err := PublicMessage(reqType, req.PathValue("Org"), req.Body)
		if err != nil {
			log.Printf("hook (%s) processing error: %v\n", reqType, err)
			res.WriteHeader(http.StatusBadRequest)
		} else {
			res.WriteHeader(http.StatusOK)
		}
	})

	log.Fatal(http.ListenAndServe(ListenAddr, nil))
}