package sw

import (
	"errors"
	"github.com/gaochao1/gosnmp"
	"log"
	"strconv"
	"time"
)

func MemUtilization(ip, community string, timeout, retry int) (int, error) {
	vendor, err := SysVendor(ip, community, timeout)
	method := "get"
	var oid string

	defer func() {
		if r := recover(); r != nil {
			log.Println(ip+" Recovered in MemUtilization", r)
		}
	}()

	switch vendor {
	case "Cisco_NX":
		oid = "1.3.6.1.4.1.9.9.305.1.1.2.0"
	case "Cisco", "Cisco_IOS_XE", "Cisco_IOS_7200", "Cisco_12K":
		memUsedOid := "1.3.6.1.4.1.9.9.48.1.1.1.5.1"
		snmpMemUsed, _ := RunSnmp(ip, community, memUsedOid, method, timeout)

		memFreeOid := "1.3.6.1.4.1.9.9.48.1.1.1.6.1"
		snmpMemFree, _ := RunSnmp(ip, community, memFreeOid, method, timeout)

		if &snmpMemFree[0] != nil && &snmpMemUsed[0] != nil {
			memUsed := snmpMemUsed[0].Value.(int)
			memFree := snmpMemFree[0].Value.(int)

			if memUsed+memFree != 0 {
				memUtili := float64(memUsed) / float64(memUsed+memFree)
				return int(memUtili * 100), nil
			}
		}
	case "Cisco_IOS_XR":
		return getCisco_IOS_XR_Mem(ip, community, timeout, retry)
	case "Cisco_ASA", "Cisco_ASA_OLD":
		return getCisco_ASA_Mem(ip, community, timeout, retry)
	case "Huawei", "Huawei_V5.70", "Huawei_V5.130":
		oid = "1.3.6.1.4.1.2011.5.25.31.1.1.1.1.7"
		return getH3CHWcpumem(ip, community, oid, timeout, retry)
	case "Huawei_V3.10":
		return getOldHuawei_Mem(ip, community, timeout, retry)
	case "Huawei_ME60":
		return getHuawei_Me60_Mem(ip, community, timeout, retry)
	case "H3C", "H3C_V5", "H3C_V7":
		oid = "1.3.6.1.4.1.25506.2.6.1.1.1.1.8"
		return getH3CHWcpumem(ip, community, oid, timeout, retry)
	case "H3C_S9500":
		oid = "1.3.6.1.4.1.2011.10.2.6.1.1.1.1.8"
		return getH3CHWcpumem(ip, community, oid, timeout, retry)
	case "Juniper":
		oid = "1.3.6.1.4.1.2636.3.1.13.1.11"
		return getH3CHWcpumem(ip, community, oid, timeout, retry)
	case "Ruijie":
		oid = "1.3.6.1.4.1.4881.1.1.10.2.35.1.1.1.3.0"
		return getRuijiecpumem(ip, community, oid, timeout, retry)
	default:
		return 0, err
	}

	var snmpPDUs []gosnmp.SnmpPDU
	for i := 0; i < retry; i++ {
		snmpPDUs, err = RunSnmp(ip, community, oid, method, timeout)
		if len(snmpPDUs) > 0 {
			break
		}
		time.Sleep(100 * time.Millisecond)
	}

	if err == nil {
		for _, pdu := range snmpPDUs {
			return pdu.Value.(int), err
		}
	}

	return 0, err
}
func getCisco_IOS_XR_Mem(ip, community string, timeout, retry int) (int, error) {
	cpuindex := "1.3.6.1.4.1.9.9.109.1.1.1.1.2"
	method := "getnext"
	var snmpPDUs []gosnmp.SnmpPDU
	var err error
	var index string
	for i := 0; i < retry; i++ {
		snmpPDUs, err = RunSnmp(ip, community, cpuindex, method, timeout)
		if len(snmpPDUs) > 0 {
			break
		}
		time.Sleep(100 * time.Millisecond)
	}
	index = strconv.Itoa(snmpPDUs[0].Value.(int))
	method = "get"
	memUsedOid := "1.3.6.1.4.1.9.9.221.1.1.1.1.18." + index + ".1"
	snmpMemUsed, _ := RunSnmp(ip, community, memUsedOid, method, timeout)
	memFreeOid := "1.3.6.1.4.1.9.9.221.1.1.1.1.20." + index + ".1"
	snmpMemFree, _ := RunSnmp(ip, community, memFreeOid, method, timeout)
	if &snmpMemFree[0] != nil && &snmpMemUsed[0] != nil {
		memUsed := snmpMemUsed[0].Value.(uint64)
		memFree := snmpMemFree[0].Value.(uint64)
		if memUsed+memFree != 0 {
			memUtili := float64(memUsed) / float64(memUsed+memFree)
			return int(memUtili * 100), err
		}
	}
	return 0, err
}

func getOldHuawei_Mem(ip, community string, timeout, retry int) (int, error) {
	method := "walk"
	memTotalOid := "1.3.6.1.4.1.2011.6.1.2.1.1.2"
	snmpMemTotal, err := RunSnmp(ip, community, memTotalOid, method, timeout)

	memFreeOid := "1.3.6.1.4.1.2011.6.1.2.1.1.3"
	snmpMemFree, err := RunSnmp(ip, community, memFreeOid, method, timeout)
	if len(snmpMemFree) == 0 || len(snmpMemTotal) == 0 {
		err := errors.New(ip + " No Such Object available on this agent at this OID")
		return 0, err
	} else {
		memTotal := snmpMemTotal[0].Value.(int)
		memFree := snmpMemFree[0].Value.(int)
		if memTotal != 0 {
			memUtili := float64(memTotal-memFree) / float64(memTotal)
			return int(memUtili * 100), err
		}
	}
	return 0, err
}

func getCisco_ASA_Mem(ip, community string, timeout, retry int) (int, error) {
	method := "walk"
	memUsedOid := "1.3.6.1.4.1.9.9.221.1.1.1.1.18"
	snmpMemUsed, err := RunSnmp(ip, community, memUsedOid, method, timeout)

	memFreeOid := "1.3.6.1.4.1.9.9.221.1.1.1.1.20"
	snmpMemFree, err := RunSnmp(ip, community, memFreeOid, method, timeout)
	if &snmpMemFree[0] != nil && &snmpMemUsed[0] != nil {
		memUsed := snmpMemUsed[0].Value.(uint64)
		memFree := snmpMemFree[0].Value.(uint64)
		if memUsed+memFree != 0 {
			memUtili := float64(memUsed) / float64(memUsed+memFree)
			return int(memUtili * 100), nil
		}
	}
	return 0, err
}

func getHuawei_Me60_Mem(ip, community string, timeout, retry int) (int, error) {
	memTotalOid := "1.3.6.1.4.1.2011.6.3.5.1.1.2"

	memTotal, _, err := snmp_walk_sum(ip, community, memTotalOid, timeout, retry)

	memFreeOid := "1.3.6.1.4.1.2011.6.3.5.1.1.3"
	memFree, _, err := snmp_walk_sum(ip, community, memFreeOid, timeout, retry)
	if memTotal != 0 && memFree != 0 {
		memUtili := float64(memTotal-memFree) / float64(memTotal)
		return int(memUtili * 100), nil
	}
	return 0, err
}