autogits/gitea-events-rabbitmq-publisher/main.go
2024-08-14 17:58:22 +02:00

62 lines
1.5 KiB
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))
}