handler_json.go

package macouidatabase

import (
	"encoding/json"
	"fmt"
	"os"
	"strings"
)

var rawMap map[string]struct {
	Manufacturer   string   `json:"manufacturer"`
	Registry       string   `json:"registry"`
	ShortName      string   `json:"short_name"`
	DeviceType     string   `json:"device_type"`
	RegisteredDate string   `json:"registered_date"`
	Address        string   `json:"address"`
	Country        *string  `json:"country"`
	Sources        []string `json:"sources"`
}

type MacJson struct {
	path string
}

func (m *MacJson) opendatabase() ([]byte, error) {
	return os.ReadFile(m.path)
}

func (m *MacJson) FindByOUI(oui string) (*MacAddressOUI, error) {
	content, err := m.opendatabase()
	if err != nil {
		return nil, fmt.Errorf("error in open database, error = %v", err)
	}
	err = json.Unmarshal(content, &rawMap)
	if err != nil {
		return nil, fmt.Errorf("error in select data, error = %v", err)
	}
	if val, ok := rawMap[oui]; ok {
		return &MacAddressOUI{
			OUI:            oui,
			Manufacturer:   val.Manufacturer,
			Registry:       val.Registry,
			ShortName:      val.ShortName,
			DeviceType:     val.DeviceType,
			RegisteredDate: val.RegisteredDate,
			Address:        val.Address,
			Sources:        strings.Join(val.Sources, "+"),
		}, nil
	} else {
		return nil, fmt.Errorf("OUI not exists")
	}
}

func (m *MacJson) FindByManufacturer(manufacturer string) ([]MacAddressOUI, error) {
	var macAddresses []MacAddressOUI
	content, err := m.opendatabase()
	if err != nil {
		return nil, fmt.Errorf("error in open database, error = %v", err)
	}
	err = json.Unmarshal(content, &rawMap)
	if err != nil {
		return nil, fmt.Errorf("error in select data, error = %v", err)
	}
	for oui, item := range rawMap {
		if strings.Contains(strings.ToLower(item.Manufacturer), strings.ToLower(manufacturer)) {
			macAddresses = append(macAddresses, MacAddressOUI{
				OUI:            oui,
				Manufacturer:   item.Manufacturer,
				Registry:       item.Registry,
				ShortName:      item.ShortName,
				DeviceType:     item.DeviceType,
				RegisteredDate: item.RegisteredDate,
				Address:        item.Address,
				Sources:        strings.Join(item.Sources, "+"),
			})
		}
	}
	if len(macAddresses) == 0 {
		return nil, fmt.Errorf("manufacture does not exists")
	} else {
		return macAddresses, nil
	}
}

func (m *MacJson) FindByAddress(address string) ([]MacAddressOUI, error) {
	var macAddresses []MacAddressOUI
	content, err := m.opendatabase()
	if err != nil {
		return nil, fmt.Errorf("error in open database, error = %v", err)
	}
	err = json.Unmarshal(content, &rawMap)
	if err != nil {
		return nil, fmt.Errorf("error in select data, error = %v", err)
	}
	for oui, item := range rawMap {
		if strings.Contains(strings.ToLower(item.Address), strings.ToLower(address)) {
			macAddresses = append(macAddresses, MacAddressOUI{
				OUI:            oui,
				Manufacturer:   item.Manufacturer,
				Registry:       item.Registry,
				ShortName:      item.ShortName,
				DeviceType:     item.DeviceType,
				RegisteredDate: item.RegisteredDate,
				Address:        item.Address,
				Sources:        strings.Join(item.Sources, "+"),
			})
		}
	}
	if len(macAddresses) == 0 {
		return nil, fmt.Errorf("address does not exists")
	} else {
		return macAddresses, nil
	}
}

func (m *MacJson) FindByShortName(short_name string) ([]MacAddressOUI, error) {
	var macAddresses []MacAddressOUI
	content, err := m.opendatabase()
	if err != nil {
		return nil, fmt.Errorf("error in open database, error = %v", err)
	}
	err = json.Unmarshal(content, &rawMap)
	if err != nil {
		return nil, fmt.Errorf("error in select data, error = %v", err)
	}
	for oui, item := range rawMap {
		if strings.Contains(strings.ToLower(item.ShortName), strings.ToLower(short_name)) {
			macAddresses = append(macAddresses, MacAddressOUI{
				OUI:            oui,
				Manufacturer:   item.Manufacturer,
				Registry:       item.Registry,
				ShortName:      item.ShortName,
				DeviceType:     item.DeviceType,
				RegisteredDate: item.RegisteredDate,
				Address:        item.Address,
				Sources:        strings.Join(item.Sources, "+"),
			})
		}
	}
	if len(macAddresses) == 0 {
		return nil, fmt.Errorf("short_name does not exists")
	} else {
		return macAddresses, nil
	}
}

func (m *MacJson) AllMacAddresses() ([]MacAddressOUI, error) {
	var macAddresses []MacAddressOUI
	content, err := m.opendatabase()
	if err != nil {
		return nil, fmt.Errorf("error in open database, error = %v", err)
	}
	err = json.Unmarshal(content, &rawMap)
	if err != nil {
		return nil, fmt.Errorf("error in select data, error = %v", err)
	}
	for oui, item := range rawMap {
		macAddresses = append(macAddresses, MacAddressOUI{
			OUI:            oui,
			Manufacturer:   item.Manufacturer,
			Registry:       item.Registry,
			ShortName:      item.ShortName,
			DeviceType:     item.DeviceType,
			RegisteredDate: item.RegisteredDate,
			Address:        item.Address,
			Sources:        strings.Join(item.Sources, "+"),
		})
	}
	if len(macAddresses) == 0 {
		return nil, fmt.Errorf("database is empty")
	} else {
		return macAddresses, nil
	}
}