chore: refactor

Author: madonuko

anitnya.go

@@ -3,12 +3,15 @@ package main
 import (
 	"crypto/tls"
 	"crypto/x509"
+	"io"
 	"log"
 	"os"
 
 	"github.com/streadway/amqp"
 )
 
+var l = log.New(os.Stdout, "[anitnya] ", 0)
+
 type FedoraAuth struct {
 }
 
@@ -17,12 +20,53 @@ func (auth *FedoraAuth) Mechanism() string {
 }
 
 func (auth *FedoraAuth) Response() string {
-	return "guest\nguest"
+	return ""
 }
 
-func anitnya_conn() {
-	l := log.New(os.Stdout, "[anitnya] ", 0)
+func exists(path string) (bool, error) {
+    _, err := os.Stat(path)
+    if err == nil { return true, nil }
+    if os.IsNotExist(err) { return false, nil }
+    return false, err
+}
+
+func readf(path string) chan []byte {
+	ch := make(chan []byte)
+	go func() {
+		f, err := os.Open(path)
+		if err != nil {
+			l.Fatal(err)
+		}
+		defer f.Close()
+		s, err := io.ReadAll(f)
+		if err != nil {
+			l.Fatal(err)
+		}
+		ch <- s
+	}()
+	return ch
+}
 
+func init_tls() (tls.Certificate, *x509.CertPool) {
+	x, err := exists("/etc/fedora-messaging/")
+	if err != nil {
+		l.Println("Fail to check /etc/fedora-messaging/: " + err.Error())
+	}
+	if x {
+		l.Println("using keys and certs in /etc/fedora-messaging/")
+		f1 := readf("/etc/fedora-messaging/fedora-cert.pem")
+		f2 := readf("/etc/fedora-messaging/fedora-key.pem")
+		f3 := readf("/etc/fedora-messaging/cacert.pem")
+		cert, err := tls.X509KeyPair(<-f1, <-f2)
+		if err != nil {
+			l.Fatal(err)
+		}
+	
+		l.Println("cfg cacert")
+		certpool := x509.NewCertPool()
+		certpool.AppendCertsFromPEM(<-f3)
+		return cert, certpool
+	}
 	l.Println("dl cert, prikey, cacert")
 	dl1 := dl("https://raw.githubusercontent.com/fedora-infra/fedora-messaging/stable/configs/fedora-cert.pem")
 	dl2 := dl("https://raw.githubusercontent.com/fedora-infra/fedora-messaging/stable/configs/fedora-key.pem")
@@ -35,6 +79,11 @@ func anitnya_conn() {
 	l.Println("cfg cacert")
 	certpool := x509.NewCertPool()
 	certpool.AppendCertsFromPEM([]byte(<-dl3))
+	return cert, certpool
+}
+
+func anitnya_conn() {
+	cert, certpool := init_tls()
 
 	l.Println("conn")
 	conn, err := amqp.DialConfig("amqps://rabbitmq.fedoraproject.org/%2Fpublic_pubsub", amqp.Config{
@@ -55,8 +104,8 @@ func anitnya_conn() {
 		l.Fatal(err)
 	}
 	defer conn.Close()
-	// "org.release-monitoring.prod.anitya.project.version.update.v2",
-	for _, topic := range []string{"org.fedoraproject.prod.copr"} {
+	// "org.release-monitoring.prod.anitya.project.version.update.v2", "org.fedoraproject.prod.copr"
+	for _, topic := range []string{"org.fedoraproject.prod.ci.productmd-compose.test.complete"} {
 		l.Println("xdcl " + topic)
 		err = ch.ExchangeDeclare(topic, "topic", true, false, false, false, nil)
 		if err != nil {