Merged with more details logic from https://github.com/Nilsty/whoami
parent
766dfd2b8e
commit
2978d3a7ef
@ -1,25 +1,108 @@
|
|||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"flag"
|
||||||
"fmt"
|
"fmt"
|
||||||
"log"
|
"log"
|
||||||
|
"net"
|
||||||
"net/http"
|
"net/http"
|
||||||
|
"net/url"
|
||||||
"os"
|
"os"
|
||||||
"runtime"
|
"runtime"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"gopkg.in/sirupsen/logrus.v1"
|
||||||
|
"gopkg.in/spf13/viper.v1"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
type DataResponse struct {
|
||||||
|
Hostname string `json:"hostname,omitempty"`
|
||||||
|
Platform string `json:"platform,omitempty"`
|
||||||
|
IP []string `json:"ip,omitempty"`
|
||||||
|
Headers http.Header `json:"header,omitempty"`
|
||||||
|
Environment []string `json:"env,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
var port string
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
flag.StringVar(&port, "port", "8080", "give me a port number")
|
||||||
|
|
||||||
|
lvl, err := logrus.ParseLevel(viper.GetString("loglevel"))
|
||||||
|
if err != nil {
|
||||||
|
lvl = logrus.WarnLevel
|
||||||
|
}
|
||||||
|
logrus.SetLevel(lvl)
|
||||||
|
}
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
port := os.Getenv("PORT")
|
flag.Parse()
|
||||||
if port == "" {
|
|
||||||
port = "8080"
|
http.HandleFunc("/", index)
|
||||||
|
http.HandleFunc("/api", api)
|
||||||
|
|
||||||
|
log.Println("Starting up on port " + port)
|
||||||
|
if err := http.ListenAndServe(":"+port, nil); err != nil {
|
||||||
|
http.ListenAndServe(":"+port, nil)
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func index(w http.ResponseWriter, req *http.Request) {
|
||||||
|
u, _ := url.Parse(req.URL.String())
|
||||||
|
queryParams := u.Query()
|
||||||
|
|
||||||
fmt.Fprintf(os.Stdout, "Listening on :%s\n", port)
|
wait := queryParams.Get("wait")
|
||||||
|
if len(wait) > 0 {
|
||||||
|
duration, err := time.ParseDuration(wait)
|
||||||
|
if err == nil {
|
||||||
|
time.Sleep(duration)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
data := fetchData(req)
|
||||||
|
fmt.Fprintf(os.Stdout, "I'm %s\n", data.Hostname)
|
||||||
|
fmt.Fprintf(w, "I'm %s running on %s\n\n", data.Hostname, data.Platform)
|
||||||
|
|
||||||
|
for _, ip := range data.IP {
|
||||||
|
fmt.Fprintln(w, "IP:", ip)
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, env := range data.Environment {
|
||||||
|
fmt.Fprintln(w, "ENV:", env)
|
||||||
|
}
|
||||||
|
req.Write(w)
|
||||||
|
}
|
||||||
|
|
||||||
|
func api(w http.ResponseWriter, req *http.Request) {
|
||||||
|
data := fetchData(req)
|
||||||
|
json.NewEncoder(w).Encode(data)
|
||||||
|
}
|
||||||
|
|
||||||
|
func fetchData(req *http.Request) DataResponse {
|
||||||
hostname, _ := os.Hostname()
|
hostname, _ := os.Hostname()
|
||||||
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
|
data := DataResponse{
|
||||||
fmt.Fprintf(os.Stdout, "I'm %s\n", hostname)
|
hostname,
|
||||||
fmt.Fprintf(w, "I'm %s running on %s/%s\n", hostname, runtime.GOOS, runtime.GOARCH)
|
fmt.Sprintf("%s/%s", runtime.GOOS, runtime.GOARCH),
|
||||||
})
|
[]string{},
|
||||||
|
req.Header,
|
||||||
|
os.Environ(),
|
||||||
|
}
|
||||||
|
|
||||||
|
ifaces, _ := net.Interfaces()
|
||||||
|
for _, i := range ifaces {
|
||||||
|
addrs, _ := i.Addrs()
|
||||||
|
for _, addr := range addrs {
|
||||||
|
var ip net.IP
|
||||||
|
switch v := addr.(type) {
|
||||||
|
case *net.IPNet:
|
||||||
|
ip = v.IP
|
||||||
|
case *net.IPAddr:
|
||||||
|
ip = v.IP
|
||||||
|
}
|
||||||
|
data.IP = append(data.IP, ip.String())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
log.Fatal(http.ListenAndServe(":"+port, nil))
|
return data
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in new issue