58 lines
1.4 KiB
Go
58 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
func main() {
|
|
addr := envOr("LISTEN_ADDR", ":8085")
|
|
writeToken, err := readSecretFile("RECEIVER_WRITE_TOKEN_FILE")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
readToken, err := readSecretFile("RECEIVER_READ_TOKEN_FILE")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
var server *receiverServer
|
|
if os.Getenv("SHADOW_RECEIVER_MODE") == "1" {
|
|
server, err = newShadowReceiverServer(envOr("RECEIPT_DIRECTORY", "/var/lib/notification-receiver"), writeToken, readToken, os.Getenv("CANDIDATE_RELEASE_SET_ID"), os.Getenv("CANDIDATE_TASK_ID"), time.Now)
|
|
} else {
|
|
server, err = newReceiverServer(envOr("RECEIPT_DIRECTORY", "/var/lib/notification-receiver"), writeToken, readToken, time.Now)
|
|
}
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
log.Printf("notification-receiver listening on %s", addr)
|
|
if err = http.ListenAndServe(addr, server.routes()); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func readSecretFile(name string) (string, error) {
|
|
path := os.Getenv(name)
|
|
if path == "" {
|
|
return "", fmt.Errorf("%s is required", name)
|
|
}
|
|
value, err := os.ReadFile(path)
|
|
if err != nil {
|
|
return "", fmt.Errorf("read %s: %w", name, err)
|
|
}
|
|
secret := strings.TrimSpace(string(value))
|
|
if secret == "" {
|
|
return "", fmt.Errorf("%s is empty", name)
|
|
}
|
|
return secret, nil
|
|
}
|
|
func envOr(name, fallback string) string {
|
|
if value := os.Getenv(name); value != "" {
|
|
return value
|
|
}
|
|
return fallback
|
|
}
|