openpdu/src/board_dummy.go

package main

import (
	"fmt"
	"log"
	"strings"

	"github.com/spf13/viper"
)

type DummyChannel struct {
	ID        string
	Num       uint
	name      string
	MQTTTopic string
	Value     bool
	onboot    string
	parent    *DummyBoard
}

type DummyBoard struct {
	ID           string
	Name         string
	ChannelCount uint
	Channels     []*DummyChannel
}

func newDummyChannel(v *viper.Viper, channelID string) DummyChannel {
	v.SetDefault("name", "unknown")
	v.SetDefault("lastValue", false)
	v.SetDefault("onboot", "off")
	v.SetDefault("mqtttopic", v.GetString("name"))

	value := false
	switch v.GetString("onboot") {
	case "on":
		value = true
	case "last":
		value = v.GetBool("lastValue")
	}
	// newUUID := UUID.New().String()
	// v.SetDefault("id", newUUID)
	return DummyChannel{
		ID:        channelID,
		Num:       v.GetUint("num"),
		name:      v.GetString("name"),
		MQTTTopic: v.GetString("mqtttopic"),
		Value:     value,
		onboot:    v.GetString("onboot"),
	}
}

func newDummyBoard(v *viper.Viper, id string) *DummyBoard {
	var b DummyBoard

	v.SetDefault("name", "board "+id)
	v.SetDefault("type", "dummy")
	v.SetDefault("channelCount", 0)
	v.SetDefault("channels", "")

	if v.GetInt("channelCount") > 0 {
		for i := 0; i < v.GetInt("channelCount"); i++ {
			v.SetDefault("channels."+fmt.Sprint(i)+".num", i)
		}
	}

	b = DummyBoard{
		ID:           id,
		Name:         v.GetString("name"),
		ChannelCount: v.GetUint("channelCount"),
	}

	channels := make([]*DummyChannel, v.GetInt("channelCount"))
	if v.GetInt("channelCount") > 0 {
		channelsConfig := v.Sub("channels")
		if channelsConfig != nil {
			for channelid1 := range channelsConfig.AllSettings() {
				channelid := strings.ToLower(channelid1)
				channelConfig := channelsConfig.Sub(channelid)
				c := newDummyChannel(channelConfig, channelid)
				c.parent = &b
				if c.Num >= v.GetUint("channelCount") {
					continue
				}
				channels[c.Num] = &c
				allChannels[c.ID] = &c
			}
		}
	}
	b.Channels = channels

	return &b
}

func (c *DummyChannel) Toggle() (bool, error) {
	c.Value = !c.Value
	c.OnChange()
	return c.Value, nil
}

func (c *DummyChannel) On() error {
	c.Value = true
	c.OnChange()
	return nil
}

func (c *DummyChannel) Off() error {
	c.Value = true
	c.OnChange()
	return nil
}

func (c *DummyChannel) ToString() string {
	if !c.Value {
		return "off"
	}
	return "on"
}

func (c *DummyChannel) UpdateMQTT() {
	MQTTpublish(c.MQTTTopic, c.ToString())
}

func (c *DummyChannel) OnChange() {
	if c.onboot == "last" {
		s := fmt.Sprintf("boards.%s.channels.%s.lastvalue", c.parent.ID, c.ID)
		viper.Set(s, c.Value)
		viper.WriteConfig()
	}
	c.UpdateMQTT()
}

func (c *DummyChannel) Status() bool {
	return c.Value
}

func (c *DummyChannel) Parent() Board {
	return c.parent
}

func (c *DummyChannel) Dump() {
	log.Printf(" Channel %d (on boot: %s): %s \n", c.Num, c.onboot, c.Name)
}

func (b *DummyBoard) Dump() {
	log.Printf("Board '%s' (id: %s): %d channels\n", b.Name, b.ID, b.ChannelCount)
	for c := range b.Channels {
		b.Channels[c].Dump()
	}
}

func (b *DummyBoard) Init() {
	return
}

func (b *DummyBoard) Channel(num uint64) Channel {
	return b.Channels[num]
}

func (c *DummyChannel) Name() string {
	return c.name
}

func (c *DummyChannel) OnBoot() string {
	return c.onboot
}

func (c *DummyChannel) SetOnBoot(str string) {
	c.onboot = str
	s := fmt.Sprintf("boards.%s.channels.%s.onboot", c.parent.ID, c.ID)
	viper.Set(s, str)
}

func (c *DummyChannel) SetMQTTTopic(str string) {
	c.MQTTTopic = str
	s := fmt.Sprintf("boards.%s.channels.%s.mqtttopic", c.parent.ID, c.ID)
	viper.Set(s, str)
}