fritz_devicelist.go

package go_fritzbox_api

import (
	"encoding/json"
	"net/http"
	"net/url"
	"strings"
)

type response struct {
	Data struct {
		Topology struct {
			Rootuid string                     `json:"rootuid"`
			Devices map[string]json.RawMessage `json:"devices"`
		}
	}
}

type DeviceList struct {
	Rootuid string `json:"rootuid"`
	Devices []Device
}

func (c *Client) GetCLientList() (clients DeviceList, err error) {
	data := url.Values{
		"sid":  {c.session.Sid},
		"page": {"homeNet"},
	}

	resp, err := c.doRequest(http.MethodPost, "data.lua", data, true)
	if err != nil {
		return
	}

	body, err := getBody(resp)
	if err != nil {
		return
	}

	// remove empty conninfo array, otherwise json.Unmarshal will fail (inconsistent types)
	body = strings.ReplaceAll(body, ",\"conninfo\":[]", "")

	r := response{}
	err = json.Unmarshal([]byte(body), &r)
	if err != nil {
		return
	}

	clients.Devices = []Device{}
	for _, v := range r.Data.Topology.Devices {
		var d Device

		err = json.Unmarshal(v, &d)
		if err != nil {
			// ignore errors because they will happen for devices not relevant here (dect)
			//fmt.Println(string(v))
			//fmt.Println(err)
			continue
		}

		if d.Nameinfo.Name != "" && d.UID != "" {
			clients.Devices = append(clients.Devices, d)
		}
	}
	err = nil

	return
}

func (c *Client) AddMACs(cl *DeviceList) (err error) {
	data := url.Values{
		"sid":   {c.session.Sid},
		"page":  {"netDev"},
		"xhrId": {"all"},
	}

	resp, err := c.doRequest(http.MethodPost, "data.lua", data, true)
	if err != nil {
		return
	}

	body, err := getBody(resp)
	if err != nil {
		return
	}

	type macResponse struct {
		Data struct {
			Active []struct {
				UID string
				Mac string
			}
		}
	}

	r := macResponse{}
	err = json.Unmarshal([]byte(body), &r)
	if err != nil {
		return
	}

	for i, d := range cl.Devices {
		for _, v := range r.Data.Active {
			if d.UID == v.UID {
				cl.Devices[i].MAC = v.Mac
			}
		}
	}

	return
}

// AddProfiles gets the profile for every device and adds it into an already existing Client-list.
// Warning: This will send one request per device, meaning it will take multiple seconds to complete.
// Alternatively, call GetProfileUIDFromDevice only for the devices you need
func (c *Client) AddProfiles(cl *DeviceList) {
	for dI, d := range cl.Devices {
		profiles, err := c.GetAvailableProfiles()
		if err != nil {
			return
		}

		var uid string
		uid, err = c.GetProfileUIDFromDevice(d.UID)
		if err != nil {
			return
		}

		d.Profile = profiles[uid]
		cl.Devices[dI] = d
	}
}