module: ip_information

modules/ip_information/ip_information.go

@@ -0,0 +1,195 @@
+package ip_information
+
+import (
+	"encoding/json"
+	"fmt"
+	"github.com/graniet/go-pretty/table"
+	"github.com/graniet/operative-framework/session"
+	"io/ioutil"
+	"net/http"
+	"os"
+	"strconv"
+)
+
+type IpInformation struct {
+	session.SessionModule
+	Sess *session.Session `json:"-"`
+}
+
+type Information struct {
+	IP          string  `json:"ip"`
+	CountryCode string  `json:"country_code"`
+	CountryName string  `json:"country_name"`
+	RegionCode  string  `json:"region_code"`
+	RegionName  string  `json:"region_name"`
+	City        string  `json:"city"`
+	ZipCode     string  `json:"zip_code"`
+	TimeZone    string  `json:"time_zone"`
+	Latitude    float64 `json:"latitude"`
+	Longitude   float64 `json:"longitude"`
+	MetroCode   int     `json:"metro_code"`
+}
+
+func PushIpInformationModule(s *session.Session) *IpInformation {
+	mod := IpInformation{
+		Sess: s,
+	}
+
+	mod.CreateNewParam("TARGET", "IP Address", "", true, session.STRING)
+	return &mod
+}
+
+func (module *IpInformation) Name() string {
+	return "ip.info"
+}
+
+func (module *IpInformation) Description() string {
+	return "Get information from IP Address"
+}
+
+func (module *IpInformation) Author() string {
+	return "Tristan Granier"
+}
+
+func (module *IpInformation) GetType() string {
+	return "ip_address"
+}
+
+func (module *IpInformation) GetInformation() session.ModuleInformation {
+	information := session.ModuleInformation{
+		Name:        module.Name(),
+		Description: module.Description(),
+		Author:      module.Author(),
+		Type:        module.GetType(),
+		Parameters:  module.Parameters,
+	}
+	return information
+}
+
+func (module *IpInformation) Start() {
+
+	trg, err := module.GetParameter("TARGET")
+	if err != nil {
+		module.Sess.Stream.Error(err.Error())
+		return
+	}
+
+	target, err := module.Sess.GetTarget(trg.Value)
+	if err != nil {
+		module.Sess.Stream.Error(err.Error())
+		return
+	}
+
+	url := "https://freegeoip.app/json/" + target.Name
+	var IpAddressInformation Information
+
+	req, _ := http.NewRequest("GET", url, nil)
+
+	req.Header.Add("accept", "application/json")
+	req.Header.Add("content-type", "application/json")
+
+	res, _ := http.DefaultClient.Do(req)
+
+	defer res.Body.Close()
+	body, _ := ioutil.ReadAll(res.Body)
+	err = json.Unmarshal(body, &IpAddressInformation)
+	if err != nil {
+		module.Sess.Stream.Error(err.Error())
+	}
+
+	t := module.Sess.Stream.GenerateTable()
+	t.SetOutputMirror(os.Stdout)
+	t.SetAllowedColumnLengths([]int{40, 0})
+	t.AppendRow(table.Row{
+		"IP",
+		IpAddressInformation.IP,
+	})
+	t.AppendRow(table.Row{
+		"CountryCode",
+		IpAddressInformation.CountryCode,
+	})
+	t.AppendRow(table.Row{
+		"CountryName",
+		IpAddressInformation.CountryName,
+	})
+	t.AppendRow(table.Row{
+		"RegionCode",
+		IpAddressInformation.RegionCode,
+	})
+	t.AppendRow(table.Row{
+		"RegionName",
+		IpAddressInformation.RegionName,
+	})
+	t.AppendRow(table.Row{
+		"City",
+		IpAddressInformation.City,
+	})
+	t.AppendRow(table.Row{
+		"ZipCode",
+		IpAddressInformation.ZipCode,
+	})
+	t.AppendRow(table.Row{
+		"TimeZone",
+		IpAddressInformation.TimeZone,
+	})
+	t.AppendRow(table.Row{
+		"Latitude",
+		IpAddressInformation.Latitude,
+	})
+	t.AppendRow(table.Row{
+		"Longitude",
+		IpAddressInformation.Longitude,
+	})
+	t.AppendRow(table.Row{
+		"MetroCode",
+		IpAddressInformation.MetroCode,
+	})
+
+	module.Sess.Stream.Render(t)
+
+	result := session.TargetResults{
+		Header: "IP" +
+			target.GetSeparator() +
+			"CountryCode" +
+			target.GetSeparator() +
+			"CountryName" +
+			target.GetSeparator() +
+			"RegionCode" +
+			target.GetSeparator() +
+			"RegionName" +
+			target.GetSeparator() +
+			"City" +
+			target.GetSeparator() +
+			"ZipCode" +
+			target.GetSeparator() +
+			"TimeZone" +
+			target.GetSeparator() +
+			"Latitude" +
+			target.GetSeparator() +
+			"Longitude" +
+			target.GetSeparator() +
+			"MetroCode",
+		Value: IpAddressInformation.IP +
+			target.GetSeparator() +
+			IpAddressInformation.CountryCode +
+			target.GetSeparator() +
+			IpAddressInformation.CountryName +
+			target.GetSeparator() +
+			IpAddressInformation.RegionCode +
+			target.GetSeparator() +
+			IpAddressInformation.RegionName +
+			target.GetSeparator() +
+			IpAddressInformation.City +
+			target.GetSeparator() +
+			IpAddressInformation.ZipCode +
+			target.GetSeparator() +
+			IpAddressInformation.TimeZone +
+			target.GetSeparator() +
+			fmt.Sprintf("%f", IpAddressInformation.Latitude) +
+			target.GetSeparator() +
+			fmt.Sprintf("%f", IpAddressInformation.Longitude) +
+			target.GetSeparator() +
+			strconv.Itoa(IpAddressInformation.MetroCode),
+	}
+	target.Save(module, result)
+}

modules/modules.go

@@ -12,6 +12,7 @@ import (
 	"github.com/graniet/operative-framework/modules/image_reverse_search"
 	"github.com/graniet/operative-framework/modules/info_greffe"
 	"github.com/graniet/operative-framework/modules/instagram"
+	"github.com/graniet/operative-framework/modules/ip_information"
 	"github.com/graniet/operative-framework/modules/linkedin_search"
 	"github.com/graniet/operative-framework/modules/mac_vendor"
 	"github.com/graniet/operative-framework/modules/metatag_spider"
@@ -55,6 +56,7 @@ func LoadModules(s *session.Session) {
 	s.Modules = append(s.Modules, instagram.PushInstagramCommentsModule(s))
 	s.Modules = append(s.Modules, image_reverse_search.PushImageReverseModule(s))
 	s.Modules = append(s.Modules, info_greffe.PushInfoGreffeRegistrationModule(s))
+	s.Modules = append(s.Modules, ip_information.PushIpInformationModule(s))
 	s.Modules = append(s.Modules, linkedin_search.PushLinkedinSearchModule(s))
 	s.Modules = append(s.Modules, mac_vendor.PushMacVendorModule(s))
 	s.Modules = append(s.Modules, metatag_spider.PushMetaTagModule(s))