ProfessionalUwU/echoip
1
0
Fork 0
mirror of https://github.com/mpolden/echoip.git synced 2025-06-25 12:17:52 +02:00
Code Issues Packages Projects Releases Wiki Activity

Only display enabled features on web page

Browse Source
This commit is contained in:
Martin Polden
2016-04-15 21:23:03 +02:00
parent 100131e420
commit 95b5b76ef7
2 changed files with 20 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
api/api.go
View File

@ -29,6 +29,9 @@ type API struct {
lookupCountry func(net.IP) (string, error)
testPort func(net.IP, uint64) error
ipFromRequest func(*http.Request) (net.IP, error)
reverseLookup bool
countryLookup bool
portTesting bool
}
type Response struct {
@ -60,15 +63,18 @@ func (a *API) EnableCountryLookup(filepath string) error {
a.lookupCountry = func(ip net.IP) (string, error) {
return lookupCountry(db, ip)
}
a.countryLookup = true
return nil
}
func (a *API) EnableReverseLookup() {
a.lookupAddr = net.LookupAddr
a.reverseLookup = true
}
func (a *API) EnablePortTesting() {
a.testPort = testPort
a.portTesting = true
}
func ipFromRequest(r *http.Request) (net.IP, error) {
@ -199,7 +205,13 @@ func (a *API) DefaultHandler(w http.ResponseWriter, r *http.Request) *appError {
if err != nil {
return internalServerError(err)
}
if err := t.Execute(w, &response); err != nil {
var data = struct {
Response
ReverseLookup bool
CountryLookup bool
PortTesting bool
}{response, a.reverseLookup, a.countryLookup, a.portTesting}
if err := t.Execute(w, &data); err != nil {
return internalServerError(err)
}
return nil