main.go

package main

import (
	"encoding/json"
	"io"
	"net"
	"net/http"
	"net/url"
	"pentest_tools/modules"
)

func main() {
	http.HandleFunc("/api/portscan", portScannerHandler)
	err := http.ListenAndServe(":8000", nil)
	if err != nil {
		panic(err)
	}
}

func portScannerHandler(w http.ResponseWriter, r *http.Request) {
	query := r.URL.Query()
	address := query.Get("address")
	ports := query.Get("ports")

	_, urlErr := url.Parse(address)
	if address == "" || (net.ParseIP(address) == nil && urlErr != nil) {
		io.WriteString(w, "please enter a valid IP or url")
		return
	}

	var options modules.ScanOptions
	options.Address = address
	options.Ports = ports
	openPorts := modules.ScanPorts(options)
	w.Header().Set("Content-Type", "application/json")
	json.NewEncoder(w).Encode(openPorts)
}