Acid
104 lines
2.1 KiB
Go
104 lines
2.1 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"net"
|
|
"net/http"
|
|
"os"
|
|
"text/template"
|
|
|
|
"github.com/joho/godotenv"
|
|
)
|
|
|
|
type ipResponse struct {
|
|
IP string `json:"ip"`
|
|
Asn string `json:"asn"`
|
|
AsName string `json:"as_name"`
|
|
AsDomain string `json:"as_domain"`
|
|
CountryCode string `json:"country_code"`
|
|
Country string `json:"country"`
|
|
ContinentCode string `json:"continent_code"`
|
|
Continent string `json:"continent"`
|
|
}
|
|
|
|
func main() {
|
|
mux := http.NewServeMux()
|
|
|
|
godotenv.Load()
|
|
|
|
// routes
|
|
mux.HandleFunc("/", handleRoot)
|
|
mux.HandleFunc("/robots.txt", serveRobots)
|
|
mux.HandleFunc("/*", handleRoot)
|
|
|
|
mux.Handle("/static/", http.StripPrefix("/static/", http.FileServer(http.Dir("static"))))
|
|
|
|
fmt.Println("listening on 8080")
|
|
http.ListenAndServe(":8080", mux)
|
|
}
|
|
|
|
func handleRoot(w http.ResponseWriter, r *http.Request) {
|
|
templ, err := template.ParseFiles("templates/home.html")
|
|
if err != nil {
|
|
http.Error(w, "template not found", http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
host := r.Header.Get("CF-Connecting-IP")
|
|
if host == "" {
|
|
host = r.Header.Get("X-Real-IP")
|
|
}
|
|
if host == "" {
|
|
host = r.Header.Get("X-Forwarded-For")
|
|
}
|
|
if host == "" {
|
|
host, _, _ = net.SplitHostPort(r.RemoteAddr)
|
|
}
|
|
|
|
dox, err := callApi(host)
|
|
if err != nil {
|
|
http.Error(w, "api call failed", http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
templ.Execute(w, dox)
|
|
}
|
|
|
|
func serveRobots(w http.ResponseWriter, r *http.Request) {
|
|
robots, err := os.ReadFile("templates/robots.txt")
|
|
if err != nil {
|
|
fmt.Fprintf(w, "sucks to suck")
|
|
return
|
|
}
|
|
|
|
fmt.Fprint(w, string(robots))
|
|
}
|
|
|
|
func callApi(ip string) (ipResponse, error) {
|
|
api := os.Getenv("IPAPI")
|
|
fmt.Printf("api: %v\n", api)
|
|
|
|
var values ipResponse
|
|
|
|
if api == " " {
|
|
return values, errors.New("No api key")
|
|
}
|
|
|
|
url := fmt.Sprintf("https://api.ipinfo.io/lite/%s?token=%s", ip, api)
|
|
|
|
request, err := http.Get(url)
|
|
if err != nil {
|
|
return values, errors.New("request failed")
|
|
}
|
|
defer request.Body.Close()
|
|
|
|
body, _ := io.ReadAll(request.Body)
|
|
|
|
err = json.Unmarshal(body, &values)
|
|
|
|
return values, nil
|
|
}
|