basics/network/utils/parse.go

package utils

import (
	"github.com/oneclickvirt/basics/model"
	"strconv"
	"strings"
)

func ParseIpInfo(data map[string]interface{}) *model.IpInfo {
	ipInfo := &model.IpInfo{}
	if ip, ok := data["ip"].(string); ok {
		ipInfo.Ip = ip
	}
	if location, ok := data["location"].(map[string]interface{}); ok {
		if city, ok := location["city"].(string); ok {
			ipInfo.City = city
		}
		if region, ok := location["region"].(map[string]interface{}); ok {
			if name, ok := region["name"].(string); ok {
				ipInfo.Region = name
			}
		}
		if country, ok := location["country"].(map[string]interface{}); ok {
			if name, ok := country["name"].(string); ok {
				ipInfo.Country = name
			}
		}
	}
	if connection, ok := data["connection"].(map[string]interface{}); ok {
		if asn, ok := connection["asn"].(float64); ok {
			ipInfo.ASN = strconv.Itoa(int(asn))
		}
		if org, ok := connection["organization"].(string); ok {
			ipInfo.Org = org
		}
	}
	return ipInfo
}

func ParseSecurityInfo(data map[string]interface{}) *model.SecurityInfo {
	securityInfo := &model.SecurityInfo{}
	if security, ok := data["security"].(map[string]interface{}); ok {
		if isAbuser, ok := security["is_abuser"].(bool); ok {
			securityInfo.IsAbuser = strconv.FormatBool(isAbuser)
		}
		if isAttacker, ok := security["is_attacker"].(bool); ok {
			securityInfo.IsAttacker = strconv.FormatBool(isAttacker)
		}
		if isBogon, ok := security["is_bogon"].(bool); ok {
			securityInfo.IsBogon = strconv.FormatBool(isBogon)
		}
		if isCloudProvider, ok := security["is_cloud_provider"].(bool); ok {
			securityInfo.IsCloudProvider = strconv.FormatBool(isCloudProvider)
		}
		if isProxy, ok := security["is_proxy"].(bool); ok {
			securityInfo.IsProxy = strconv.FormatBool(isProxy)
		}
		if isRelay, ok := security["is_relay"].(bool); ok {
			securityInfo.IsRelay = strconv.FormatBool(isRelay)
		}
		if isTor, ok := security["is_tor"].(bool); ok {
			securityInfo.IsTor = strconv.FormatBool(isTor)
		}
		if isTorExit, ok := security["is_tor_exit"].(bool); ok {
			securityInfo.IsTorExit = strconv.FormatBool(isTorExit)
		}
		if isVpn, ok := security["is_vpn"].(bool); ok {
			securityInfo.IsVpn = strconv.FormatBool(isVpn)
		}
		if isAnonymous, ok := security["is_anonymous"].(bool); ok {
			securityInfo.IsAnonymous = strconv.FormatBool(isAnonymous)
		}
		if isThreat, ok := security["is_threat"].(bool); ok {
			securityInfo.IsThreat = strconv.FormatBool(isThreat)
		}
	}
	return securityInfo
}

// ParseYesNo 检测文本内容含No则返回No，否则返回Yes
func ParseYesNo(text string) string {
	if strings.Contains(strings.ToLower(text), "no") {
		return "No"
	}
	return "Yes"
}