main.go

package main

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

	"github.com/go-co-op/gocron"
	"github.com/prometheus/client_golang/prometheus/promhttp"
)

func runCronJobs(apiKey, apiEmail, orgId, s3Bucket string, lookBack int) {
	s := gocron.NewScheduler(time.UTC)

	s.Every(lookBack).Minute().Do(func() {
		getAuditLogs(apiKey, apiEmail, orgId, s3Bucket, lookBack)
	})

	s.StartBlocking()
	fmt.Println()
}

func main() {
	apiEmail := os.Getenv("CLOUDFLARE_API_EMAIL")
	apiKey := os.Getenv("CLOUDFLARE_API_KEY")
	orgId := os.Getenv("CLOUDFLARE_ORGANIZATION_ID")
	interval := os.Getenv("CLOUDFLARE_LOOK_BACK_INTERVAL")
	s3Bucket := os.Getenv("AWS_S3_BUCKET_NAME")

	if apiEmail == "" {
		log.Fatal("Must specify CLOUDFLARE_API_EMAIL")
	}
	if apiKey == "" {
		log.Fatal("Must specify CLOUDFLARE_API_KEY")
	}
	if orgId == "" {
		log.Fatal("Must specify CLOUDFLARE_ORGANIZATION_ID")
	}

	// Convert user supplied look back value to an integer otherwise set a default
	var lookBack int
	if interval != "" {
		envToInt, err := getEnvInt(interval)
		if err != nil {
			log.Fatal("If CLOUDFLARE_LOOK_BACK_INTERVAL it must be an integer")
		}
		lookBack = envToInt
	}
	lookBack = 5

	http.Handle("/metrics", promhttp.Handler())
	go func() {
		http.ListenAndServe(":2112", nil)
	}()
	runCronJobs(apiKey, apiEmail, orgId, s3Bucket, lookBack)
}