Merge pull request #1 from ohmydevops/add-availableServices

Add site selection

Author: ohmydevops

main.go

@@ -2,10 +2,12 @@ package main
 
 import (
 	"encoding/json"
+	"flag"
 	"fmt"
 	"io"
 	"net/http"
 	"net/http/cookiejar"
+	"os"
 	"time"
 
 	"github.com/gen2brain/beeep"
@@ -14,11 +16,21 @@ import (
 const baseURL = "https://radar.arvancloud.ir/api/v1/internet-monitoring?isp="
 
 var (
-	errorCounts = make(map[string]int)
-	erroredISPs = make(map[string]bool)
+	errorCounts      = make(map[string]int)
+	erroredISPs      = make(map[string]bool)
+	serviceIndicator string
 )
 
-var serviceIndicator string = "google"
+var availableServices = []string{
+	"google",
+	"wikipedia",
+	"playstation",
+	"bing",
+	"github",
+	"digikala",
+	"divar",
+	"aparat",
+}
 
 var isps = []string{
 	"sindad-buf",
@@ -36,6 +48,34 @@ var isps = []string{
 	"irancell",
 }
 
+func printServices() {
+	fmt.Println("Usage:")
+	fmt.Println("  --service=N       Run directly without prompt, where N is the service number (see below)")
+	fmt.Println("  --help            Show this help message and available services")
+	fmt.Println()
+	fmt.Println("Available services:")
+	for i, s := range availableServices {
+		fmt.Printf("  %2d) %s\n", i+1, s)
+	}
+	fmt.Println()
+	fmt.Println("Examples:")
+	fmt.Println("  ./radar-linux                # Interactive mode (asks for service)")
+	fmt.Println("  ./radar-linux --service=3    # Monitor playstation directly")
+	fmt.Println("  ./radar-linux --help         # Show this help message")
+}
+
+func chooseServiceInteractive() string {
+	printServices()
+	var choice int
+	fmt.Print("Enter number: ")
+	_, err := fmt.Scanln(&choice)
+	if err != nil || choice < 1 || choice > len(availableServices) {
+		fmt.Println("❌ Invalid choice. Defaulting to 'google'")
+		return "google"
+	}
+	return availableServices[choice-1]
+}
+
 func fetchData(client *http.Client, isp string) (float64, error) {
 	url := baseURL + isp
 
@@ -60,12 +100,12 @@ func fetchData(client *http.Client, isp string) (float64, error) {
 		return 0, fmt.Errorf("❌ JSON parse error: %v", err)
 	}
 
-	googleValues, ok := data[serviceIndicator]
-	if !ok || googleValues == nil || len(googleValues) == 0 {
+	values, ok := data[serviceIndicator]
+	if !ok || values == nil || len(values) == 0 {
 		return 0, fmt.Errorf("-")
 	}
 
-	return googleValues[len(googleValues)-1], nil
+	return values[len(values)-1], nil
 }
 
 func checkStatus(isp string, value float64) {
@@ -74,19 +114,19 @@ func checkStatus(isp string, value float64) {
 	if value != 0 {
 		errorCounts[isp]++
 		if errorCounts[isp] >= 3 && !erroredISPs[isp] {
-			err := beeep.Notify("🔴 Internet Outage", fmt.Sprintf("Google unreachable from %s", isp), "./icon.png")
+			err := beeep.Notify("🔴 Internet Outage", fmt.Sprintf("%s unreachable from %s", serviceIndicator, isp), "./icon.png")
 			if err != nil {
 				fmt.Printf("[%s] ❌ Notification error: %v\n", isp, err)
 			}
 			erroredISPs[isp] = true
 		}
 	} else {
 		if erroredISPs[isp] {
-			err := beeep.Notify("🟢 Internet Outage fixed", fmt.Sprintf("Google reachable from %s", isp), "./icon.png")
+			err := beeep.Notify("🟢 Internet Restored", fmt.Sprintf("%s is reachable again from %s", serviceIndicator, isp), "./icon.png")
 			if err != nil {
 				fmt.Printf("[%s] ❌ Notification error: %v\n", isp, err)
 			}
-			fmt.Printf("[%s] 🟢 Google is reachable again\n", isp)
+			fmt.Printf("[%s] 🟢 %s is reachable again\n", isp, serviceIndicator)
 		}
 		errorCounts[isp] = 0
 		erroredISPs[isp] = false
@@ -100,7 +140,27 @@ func waitUntilNextMinute() {
 }
 
 func main() {
-	fmt.Println("Arvan Cloud Outage Notification started ...")
+	fmt.Println("📡 Arvan Cloud Radar Monitor")
+	serviceFlag := flag.Int("service", 0, "Service number to monitor (e.g. 1 for google, 2 for wikipedia...)")
+	helpFlag := flag.Bool("help", false, "Show available services")
+	flag.Parse()
+
+	if *helpFlag {
+		printServices()
+		return
+	}
+
+	if *serviceFlag > 0 && *serviceFlag <= len(availableServices) {
+		serviceIndicator = availableServices[*serviceFlag-1]
+	} else if *serviceFlag != 0 {
+		fmt.Println("❌ Invalid service number. Use --help to see available options.")
+		os.Exit(1)
+	} else {
+		serviceIndicator = chooseServiceInteractive()
+	}
+
+	fmt.Printf("✅ Monitoring service: %s\n", serviceIndicator)
+
 	jar, _ := cookiejar.New(nil)
 	client := &http.Client{Jar: jar}