From 56cb4e5d0890d0edf2b0317cff3b21a028c87d9d Mon Sep 17 00:00:00 2001
From: Timam <mail@timam.io>
Date: Mon, 23 Oct 2023 16:37:55 +0600
Subject: [PATCH] bring back old code

---
 svc/watcher/main.go | 105 +++++++++++++++++++++-----------------------
 1 file changed, 51 insertions(+), 54 deletions(-)

diff --git a/svc/watcher/main.go b/svc/watcher/main.go
index 5c3e5d1..da20076 100644
--- a/svc/watcher/main.go
+++ b/svc/watcher/main.go
@@ -1,64 +1,61 @@
 package main
 
 import (
+	"errors"
+	"github.com/prometheus/client_golang/prometheus/promhttp"
+	"github.com/statuzproj/statuz/utils/evictedconf"
 	"github.com/statuzproj/statuz/utils/healthz"
 	"log"
 	"net/http"
+	"sync"
 )
 
-//
-//func main() {
-//	var wg sync.WaitGroup
-//
-//	// Start the HTTP server in a goroutine
-//	wg.Add(1)
-//	go func() {
-//		defer wg.Done()
-//
-//		http.HandleFunc("/healthz", healthz.HealthCheck)
-//		http.Handle("/metrics", promhttp.Handler())
-//		err := http.ListenAndServe(":8080", nil)
-//		if err != nil {
-//			log.Fatalf("HTTP server error: %v", err)
-//		}
-//	}()
-//
-//	// Start the watcher in another goroutine
-//	wg.Add(1)
-//	go func() {
-//		defer wg.Done()
-//
-//		err := start()
-//		if err != nil {
-//			log.Fatalf("Error starting watcher: %v", err)
-//		}
-//	}()
-//
-//	// Wait for all goroutines to complete before exiting
-//	wg.Wait()
-//}
-//
-//func start() error {
-//	env, err := evictedconf.GetEnvVars()
-//	if err != nil {
-//		return err
-//	}
-//
-//	if env.Type == "webpage" {
-//		err := watchPage(env.Endpoint, env.Interval)
-//		if err != nil {
-//			return err
-//		}
-//	} else if env.Type == "ip" {
-//		watchIp(env.Endpoint, env.Port, env.Interval)
-//	} else {
-//		return errors.New("Unsupported watch type: " + env.Type)
-//	}
-//	return nil
-//}
-
 func main() {
-	log.Println("Work in Progress!!")
-	http.HandleFunc("/healthz", healthz.HealthCheck)
-	http.ListenAndServe(":8080", nil)
+	var wg sync.WaitGroup
+
+	// Start the HTTP server in a goroutine
+	wg.Add(1)
+	go func() {
+		defer wg.Done()
+
+		http.HandleFunc("/healthz", healthz.HealthCheck)
+		http.Handle("/metrics", promhttp.Handler())
+		err := http.ListenAndServe(":8080", nil)
+		if err != nil {
+			log.Fatalf("HTTP server error: %v", err)
+		}
+	}()
+
+	// Start the watcher in another goroutine
+	wg.Add(1)
+	go func() {
+		defer wg.Done()
+
+		err := start()
+		if err != nil {
+			log.Fatalf("Error starting watcher: %v", err)
+		}
+	}()
+
+	// Wait for all goroutines to complete before exiting
+	wg.Wait()
+}
+
+func start() error {
+	env, err := evictedconf.GetEnvVars()
+	if err != nil {
+		return err
+	}
+
+	if env.Type == "webpage" {
+		err := watchPage(env.Endpoint, env.Interval)
+		if err != nil {
+			return err
+		}
+	} else if env.Type == "ip" {
+		watchIp(env.Endpoint, env.Port, env.Interval)
+	} else {
+		return errors.New("Unsupported watch type: " + env.Type)
+	}
+	return nil
 }