octopus_exporter/cmd/octopus_exporter/meters.go

package main

import (
	"fmt"
	"log"
	"os"
)

type meterKind string

const (
	electricity meterKind = "electricity"
	gas         meterKind = "gas"
)

type meterCandidate struct {
	kind     meterKind
	mpan     string
	mprn     string
	serial   string
	deviceID string
}

type resolvedMeter struct {
	deviceID string
	mpan     string // electricity
	mprn     string // gas
	serial   string
}

func getMeters(token string) ([]meterCandidate, error) {
	result, err := doGraphQL(gqlRequest{
		Query: `{ viewer { accounts { ... on AccountType { properties {
			electricityMeterPoints {
				mpan
				meters {
					serialNumber
					smartDevices { deviceId }
				}
			}
			gasMeterPoints {
				mprn
				meters {
					serialNumber
					smartDevices { deviceId }
				}
			}
		} } } } }`,
	}, token)
	if err != nil {
		return nil, err
	}

	var candidates []meterCandidate

	accounts, _ := result["data"].(map[string]any)["viewer"].(map[string]any)["accounts"].([]any)
	for _, a := range accounts {
		props, _ := a.(map[string]any)["properties"].([]any)
		for _, p := range props {
			pm, ok := p.(map[string]any)
			if !ok {
				continue
			}

			for _, mp := range toSlice(pm["electricityMeterPoints"]) {
				mpan, _ := mp.(map[string]any)["mpan"].(string)
				for _, m := range toSlice(mp.(map[string]any)["meters"]) {
					serial, _ := m.(map[string]any)["serialNumber"].(string)
					for _, d := range toSlice(m.(map[string]any)["smartDevices"]) {
						deviceID, _ := d.(map[string]any)["deviceId"].(string)
						if deviceID != "" {
							candidates = append(candidates, meterCandidate{kind: electricity, mpan: mpan, serial: serial, deviceID: deviceID})
						}
					}
					// Include meters without smart devices so we can still use the REST consumption endpoint.
					if len(toSlice(m.(map[string]any)["smartDevices"])) == 0 && serial != "" {
						candidates = append(candidates, meterCandidate{kind: electricity, mpan: mpan, serial: serial})
					}
				}
			}

			for _, mp := range toSlice(pm["gasMeterPoints"]) {
				mprn, _ := mp.(map[string]any)["mprn"].(string)
				for _, m := range toSlice(mp.(map[string]any)["meters"]) {
					serial, _ := m.(map[string]any)["serialNumber"].(string)
					for _, d := range toSlice(m.(map[string]any)["smartDevices"]) {
						deviceID, _ := d.(map[string]any)["deviceId"].(string)
						if deviceID != "" {
							candidates = append(candidates, meterCandidate{kind: gas, mprn: mprn, serial: serial, deviceID: deviceID})
						}
					}
					if len(toSlice(m.(map[string]any)["smartDevices"])) == 0 && serial != "" {
						candidates = append(candidates, meterCandidate{kind: gas, mprn: mprn, serial: serial})
					}
				}
			}
		}
	}

	return candidates, nil
}

// resolveMeter finds the meter matching the env var filters for the given kind.
// Returns (nil, nil) if no meter of that kind exists on the account.
func resolveMeter(candidates []meterCandidate, kind meterKind) (*resolvedMeter, error) {
	var wantDeviceID, wantID, wantSerial string
	switch kind {
	case electricity:
		wantDeviceID = os.Getenv("OCTOPUS_DEVICE_ID")
		wantID = os.Getenv("OCTOPUS_MPAN")
		wantSerial = os.Getenv("OCTOPUS_SERIAL")
	case gas:
		wantDeviceID = os.Getenv("OCTOPUS_GAS_DEVICE_ID")
		wantID = os.Getenv("OCTOPUS_GAS_MPRN")
		wantSerial = os.Getenv("OCTOPUS_GAS_SERIAL")
	}

	for _, c := range candidates {
		if c.kind != kind {
			continue
		}
		if wantDeviceID != "" && c.deviceID != wantDeviceID {
			continue
		}
		if wantID != "" {
			if kind == electricity && c.mpan != wantID {
				continue
			}
			if kind == gas && c.mprn != wantID {
				continue
			}
		}
		if wantSerial != "" && c.serial != wantSerial {
			continue
		}

		m := &resolvedMeter{deviceID: c.deviceID, mpan: c.mpan, mprn: c.mprn, serial: c.serial}
		switch kind {
		case electricity:
			log.Printf("using electricity meter: MPAN=%s serial=%s deviceID=%s", m.mpan, m.serial, m.deviceID)
		case gas:
			log.Printf("using gas meter: MPRN=%s serial=%s deviceID=%s", m.mprn, m.serial, m.deviceID)
		}
		return m, nil
	}

	if wantDeviceID != "" || wantID != "" || wantSerial != "" {
		return nil, fmt.Errorf("no %s meter matched the specified filters", kind)
	}
	return nil, nil
}