included the location handler into the metrics handler

1 files changed, 116 insertions, 87 deletions

diff --git a/main.go b/main.go
index abe0710..dbf8f93 100644
--- a/main.go
+++ b/main.go
@@ -1,19 +1,20 @@
 package main
 
 import (
-    "fmt"
-    "net/http"
-    "log"
-    //"net"
-    "io/ioutil"
-    "encoding/json"
-    "strings"
-    "github.com/gliderlabs/ssh"
+	"encoding/json"
+	"fmt"
+	"github.com/gliderlabs/ssh"
+	//"github.com/gorilla/mux"
+	"io/ioutil"
+	"log"
+	"net/http"
+	"strings"
 )
 
 var (
-    metrics_num_passwords int
-    metrics_country_num map[string]int
+	metrics_num_passwords int
+	metrics_city_num      map[string]int
+	cities                map[string]location
 )
 
 type geoipresult struct {
@@ -33,93 +34,121 @@ type geoipresult struct {
 	As          string  `json:"as"`
 }
 
+type location struct {
+	key       string  `json:"key"`
+	latitude  float64 `json:"latitude"`
+	longitude float64 `json:"longitude"`
+	name      string  `json:"name"`
+}
+
 func main() {
-    metrics_country_num = make(map[string]int)
-
-    // start the ssh server
-    log.Println("Starting SSH listener")
-    go func() {
-        listenErr := ssh.ListenAndServe(":2222", nil, ssh.PasswordAuth(handlePass))
-        if listenErr != nil {
-            log.Fatalln(listenErr.Error())
-        }
-    }()
-
-    // start the http server logging the metrics
-    log.Println("Starting HTTP metrics listener")
-    http.HandleFunc("/", indexHandler)
-    http.HandleFunc("/metrics", metricsHandler)
-    listenErr := http.ListenAndServe(":8080", nil)
-    if listenErr != nil {
-        log.Fatalln(listenErr.Error())
-    }
+	metrics_city_num = make(map[string]int)
+	cities = make(map[string]location)
+
+	// start the ssh server
+	log.Println("Starting SSH listener")
+	go func() {
+		listenErr := ssh.ListenAndServe(":2222", nil, ssh.PasswordAuth(handlePass))
+		if listenErr != nil {
+			log.Fatalln(listenErr.Error())
+		}
+	}()
+
+	// start the http server logging the metrics
+	log.Println("Starting HTTP metrics listener")
+	http.HandleFunc("/", indexHandler)
+	http.HandleFunc("/metrics", metricsHandler)
+	http.HandleFunc("/locations", locationHandlerEndpoint)
+	listenErr := http.ListenAndServe(":8084", nil)
+	if listenErr != nil {
+		log.Fatalln(listenErr.Error())
+	}
+}
+
+func locationHandlerEndpoint(w http.ResponseWriter, r *http.Request) {
+	w.Header().Set("Content-Type", "application/json")
+	w.Header().Set("Access-Control-Allow-Origin", "https://grafana.nbg1.emile.space")
+	fmt.Fprintf(w, "%s", "[")
+	var i int = 0
+	for _, v := range cities {
+		fmt.Fprintf(w, "{")
+		fmt.Fprintf(w, "\"key\": \"%s\",", v.key)
+		fmt.Fprintf(w, "\"latitude\": %f,", v.latitude)
+		fmt.Fprintf(w, "\"longitude\": %f,", v.longitude)
+		fmt.Fprintf(w, "\"name\": \"%s\"", v.name)
+		if i == len(cities) - 1 {
+			fmt.Fprintf(w, "}")
+		} else {
+			fmt.Fprintf(w, "},")
+		}
+		i++
+	}
+	fmt.Fprintf(w, "%s", "]")
 }
 
-// Handling incoming SSH connections
+// Handling incoming SSH conn95.216.207.95/32ections
 func handlePass(ctx ssh.Context, pass string) bool {
-    metrics_num_passwords++
-    log.Printf("%s@%s: '%s'", ctx.User(), ctx.RemoteAddr().String(), pass)
-
-    log.Printf("Parsing ip address...")
-    //ip := net.ParseIP(ctx.RemoteAddr().String())
-    stringip := strings.Split(ctx.RemoteAddr().String(), ":")[0]
-    log.Printf("Remote ip: %s", stringip)
-    log.Printf("Done")
-
-    // Define the request string
-    requestString := fmt.Sprintf("%s%s", "http://ip-api.com/json/", stringip)
-
-    fmt.Println(requestString)
-
-    // Send the GET request
-    resp, err := http.Get(requestString)
-    if err != nil {
-        log.Fatal(err)
-    }
-
-    if resp.StatusCode != 200 {
-        return false
-    }
-
-    // Read the response
-    body, err := ioutil.ReadAll(resp.Body)
-    if err != nil {
-        log.Fatal(err)
-    }
-
-    // Unmarshal the response to json
-    var result geoipresult
-    err = json.Unmarshal(body, &result)
-    if err != nil {
-        fmt.Println("JSON ERROR, abort mission!")
-        log.Fatal(err)
-    }
-
-    log.Printf("country: %s", result.CountryCode)
-
-    if metrics_country_num[result.CountryCode] == 0 {
-        log.Println("Map is empty, initializing...")
-        metrics_country_num[result.CountryCode] = 1
-        log.Println("Done")
-    } else {
-        log.Println("Map is initialized, incrementing...")
-        metrics_country_num[result.CountryCode] += 1
-        log.Println("Done")
-    }
-    log.Printf("Done")
-
-    return false
+	metrics_num_passwords++
+	log.Printf("%s@%s: '%s'", ctx.User(), ctx.RemoteAddr().String(), pass)
+
+	stringip := strings.Split(ctx.RemoteAddr().String(), ":")[0]
+
+	// Define the request string
+	requestString := fmt.Sprintf("%s%s", "http://ip-api.com/json/", stringip)
+
+	// Send the GET request
+	resp, err := http.Get(requestString)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	if resp.StatusCode != 200 {
+		return false
+	}
+
+	// Read the response
+	body, err := ioutil.ReadAll(resp.Body)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	// Unmarshal the response to json
+	var result geoipresult
+	err = json.Unmarshal(body, &result)
+	if err != nil {
+		fmt.Println("JSON ERROR, abort mission!")
+		log.Fatal(err)
+	}
+
+	if metrics_city_num[result.City] == 0 {
+		metrics_city_num[result.City] = 1
+	} else {
+		metrics_city_num[result.City] += 1
+	}
+
+	if (cities[result.City] == location{}) {
+		newCity := location{
+			key:       strings.ToLower(result.City),
+			latitude:  result.Lat,
+			longitude: result.Lon,
+			name:      result.City,
+		}
+
+		cities[result.City] = newCity
+	}
+
+	return false
 }
 
 // Handle HTTP /metrics requests
 func metricsHandler(w http.ResponseWriter, req *http.Request) {
-    fmt.Fprintf(w, "num_passwords %d\n", metrics_num_passwords)
+	fmt.Fprintf(w, "num_passwords %d\n", metrics_num_passwords)
 
-    for k, v := range metrics_country_num {
-        fmt.Fprintf(w, "a_metric{country=\"%s\"} %d\n", k, v)
-    }
+	for k, v := range metrics_city_num {
+		fmt.Fprintf(w, "a_metric{city=\"%s\"} %d\n", strings.ToLower(k), v)
+	}
 }
 
 func indexHandler(w http.ResponseWriter, req *http.Request) {
-    _, _ = fmt.Fprintf(w, "<a href='/metrics'>metrics</a>")
+	_, _ = fmt.Fprintf(w, "<a href='/metrics'>metrics</a>")
 }