Initial import, working

4 files changed, 358 insertions, 0 deletions

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..6dbe7bf
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,6 @@
+/bin/
+/pkg/
+/src/
+/mFi Bridge/
+/mfi_homekit
+/config.json
diff --git a/config.example.json b/config.example.json
new file mode 100644
index 0000000..dc13259
--- /dev/null
+++ b/config.example.json
@@ -0,0 +1,14 @@
+{
+        "DEFAULT" : {
+                "Username": "mfi",
+                "Password": "foobar",
+                "Port": 22
+        },
+
+        "Bedroom": {
+                "Host": "192.168.0.2",
+                "Devices": [
+                        "Lamp"
+                ]
+        }
+}
diff --git a/main.go b/main.go
new file mode 100644
index 0000000..4b39e1e
--- /dev/null
+++ b/main.go
@@ -0,0 +1,257 @@
+package main
+
+import (
+        "encoding/json"
+        "errors"
+        "fmt"
+        "github.com/brutella/hc"
+        "github.com/brutella/hc/accessory"
+        "github.com/brutella/hc/characteristic"
+        "golang.org/x/crypto/ssh"
+        "io/ioutil"
+        "net"
+)
+
+const CMD_PATH = "/var/etc/persistent/power_control.sh"
+
+type Device struct {
+        Root     *Device
+        Username string
+        Password string
+        Port     int
+        Host     string
+        Devices  []string
+}
+
+func (d *Device) GetUsername() string {
+        if d.Username != "" {
+                return d.Username
+        } else if d.Root != nil {
+                return d.Root.Username
+        } else {
+                return ""
+        }
+}
+
+func (d *Device) GetPassword() string {
+        if d.Password != "" {
+                return d.Password
+        } else if d.Root != nil {
+                return d.Root.Password
+        } else {
+                return ""
+        }
+}
+
+func (d *Device) GetPort() int {
+        if d.Port != 0 {
+                return d.Port
+        } else if d.Root != nil {
+                return d.Root.Port
+        } else {
+                return 22
+        }
+}
+
+func (d *Device) newSession() (*ssh.Client, *ssh.Session, error) {
+        cfg := &ssh.ClientConfig{
+                User:            d.GetUsername(),
+                Auth:            []ssh.AuthMethod{ssh.Password(d.GetPassword())},
+                HostKeyCallback: ssh.InsecureIgnoreHostKey(),
+        }
+
+        // These devices use really old crufty versions of SSH
+        cfg.Config.KeyExchanges = []string{"diffie-hellman-group1-sha1"}
+        cfg.Config.Ciphers = append(cfg.Config.Ciphers, "aes128-cbc")
+
+        conn, err := ssh.Dial("tcp", fmt.Sprintf("%s:%d", d.Host, d.GetPort()), cfg)
+        if err != nil {
+                return nil, nil, err
+        }
+
+        s, err := conn.NewSession()
+        if err != nil {
+                conn.Close()
+                return nil, nil, err
+        }
+
+        return conn, s, nil
+}
+
+func (d *Device) runCommand(cmd string) ([]DeviceOutput, error) {
+        conn, sess, err := d.newSession()
+        if err != nil {
+                return nil, err
+        }
+        defer conn.Close()
+        defer sess.Close()
+
+        out, err := sess.Output(cmd)
+        if err != nil {
+                return nil, err
+        }
+
+        if out == nil {
+                return nil, nil
+        } else {
+                var r []DeviceOutput
+                err = json.Unmarshal(out, &r)
+                if err != nil {
+                        return nil, err
+                }
+                return r, nil
+        }
+}
+
+func (d *Device) findOutput(name string) (int, error) {
+        for i, e := range d.Devices {
+                if e == name {
+                        return i + 1, nil
+                }
+        }
+
+        return 0, errors.New("Unknown device")
+}
+
+func (d *Device) TurnOn(n string) ([]DeviceOutput, error) {
+        o, err := d.findOutput(n)
+        if err != nil {
+                return nil, err
+        }
+
+        out, err := d.runCommand(fmt.Sprintf("%s on %d", CMD_PATH, o))
+        if err != nil {
+                return nil, err
+        } else {
+                return out, nil
+        }
+}
+
+func (d *Device) TurnOff(n string) ([]DeviceOutput, error) {
+        o, err := d.findOutput(n)
+        if err != nil {
+                return nil, err
+        }
+
+        out, err := d.runCommand(fmt.Sprintf("%s off %d", CMD_PATH, o))
+        if err != nil {
+                return nil, err
+        } else {
+                return out, nil
+        }
+}
+
+func (d *Device) Toggle(n string, on bool) ([]DeviceOutput, error) {
+        if on == true {
+                return d.TurnOn(n)
+        } else {
+                return d.TurnOff(n)
+        }
+}
+
+func (d *Device) GetReport() ([]DeviceOutput, error) {
+        out, err := d.runCommand(fmt.Sprintf("%s report", CMD_PATH))
+        if err != nil {
+                return nil, err
+        } else {
+                return out, nil
+        }
+}
+
+type DeviceOutput struct {
+        Output      int     `json:"output"`
+        Engaged     bool    `json:"engaged"`
+        ActivePower float64 `json:"active_power"`
+        EnergySum   float64 `json:"energy_sum"`
+        CurrentRMS  float64 `json:"current_rms"`
+        VoltageRMS  float64 `json:"voltage_rms"`
+        PowerFactor float64 `json:"power_factor"`
+}
+
+func LoadAppConfig(filename string) (map[string]*Device, error) {
+        raw, err := ioutil.ReadFile(filename)
+        if err != nil {
+                return nil, err
+        }
+
+        var cfg map[string]*Device
+        err = json.Unmarshal(raw, &cfg)
+        if err != nil {
+                return nil, err
+        }
+
+        if def, ok := cfg["DEFAULT"]; ok {
+                delete(cfg, "DEFAULT")
+
+                for _, v := range cfg {
+                        v.Root = def
+                }
+        }
+
+        return cfg, nil
+}
+
+type Output struct {
+        Device *Device
+        Name   string
+}
+
+func (o *Output) Toggle(on bool) error {
+        _, err := o.Device.Toggle(o.Name, on)
+        if err != nil {
+                fmt.Println(err)
+        }
+        return err
+}
+
+func main() {
+        cfg, err := LoadAppConfig("config.json")
+        if err != nil {
+                fmt.Println(err)
+                return
+        }
+
+        reg := []*accessory.Accessory{}
+        accs := make(map[*characteristic.Characteristic]*Output, 10)
+
+        for _, v := range cfg {
+                report, err := v.GetReport()
+                if err != nil {
+                        panic(err)
+                }
+
+                for i, d := range v.Devices {
+                        // Unnamed outputs are unused
+                        if d == "" {
+                                continue
+                        }
+
+                        sw := accessory.NewSwitch(accessory.Info{Name: d})
+                        sw.Switch.On.OnValueUpdateFromConn(func(_ net.Conn, c *characteristic.Characteristic, new, _ interface{}) {
+                                on := new.(bool)
+                                output := accs[c]
+                                fmt.Printf("Client changed switch %s to %t\n", output.Name, on)
+                                output.Toggle(on)
+                        })
+                        accs[sw.Switch.On.Characteristic] = &Output{
+                                Device: v,
+                                Name:   d,
+                        }
+                        sw.Switch.On.UpdateValue(report[i].Engaged)
+                        reg = append(reg, sw.Accessory)
+                }
+        }
+
+        br := accessory.NewBridge(accessory.Info{Name: "mFi Bridge"})
+
+        t, err := hc.NewIPTransport(hc.Config{Pin: "00102003"}, br.Accessory, reg...)
+        if err != nil {
+                fmt.Println(err)
+        }
+
+        hc.OnTermination(func() {
+                <-t.Stop()
+        })
+
+        t.Start()
+}
diff --git a/power_control.sh b/power_control.sh
new file mode 100755
index 0000000..8e995bd
--- /dev/null
+++ b/power_control.sh
@@ -0,0 +1,81 @@
+#!/bin/sh
+
+POWER_PATH="/proc/power"
+NUM_OUTPUTS=$(find $POWER_PATH -name 'output*' | wc -l)
+
+valid_output() {
+        [ "$1" -gt "$NUM_OUTPUTS" ] && return 1
+        [ "$1" -lt "1" ] && return 1
+        return 0
+}
+
+check_output() {
+        ! valid_output $1 &&  error "Invalid output number"
+}
+
+
+error() {
+        echo "{\"error\":\"$1\"}"
+        exit 1
+}
+
+set_output() {
+        echo $2 > $POWER_PATH/relay$1
+}
+
+clear_all() {
+        for i in $(seq 1 $NUM_OUTPUTS); do
+                echo 1 > $POWER_PATH/clear_ae$i
+        done
+}
+
+report() {
+        echo "["
+        for i in $(seq 1 $NUM_OUTPUTS); do
+                relay_state=$(cat $POWER_PATH/relay$i)
+                if [ $relay_state -eq 1 ]; then
+                        state="true"
+                else
+                        state="false"
+                fi
+
+                echo -e "\t{"
+                echo -e "\t\t\"output\": $i,"
+                echo -e "\t\t\"engaged\": $state,"
+                echo -e "\t\t\"active_power\": $(cat $POWER_PATH/active_pwr$i),"
+                echo -e "\t\t\"energy_sum\": $(cat $POWER_PATH/energy_sum$i),"
+                echo -e "\t\t\"current_rms\": $(cat $POWER_PATH/i_rms$i),"
+                echo -e "\t\t\"voltage_rms\": $(cat $POWER_PATH/v_rms$i),"
+                echo -e "\t\t\"power_factor\": $(cat $POWER_PATH/pf$i)"
+
+                if [ $i -eq $NUM_OUTPUTS ]; then
+                        echo -e "\t}"
+                else
+                        echo -e "\t},"
+                fi
+        done
+        echo "]"
+}
+
+
+case $1 in
+        on)
+                check_output $2
+                set_output $2 1
+                report
+                ;;
+        off)
+                check_output $2
+                set_output $2 0
+                report
+                ;;
+        report)
+                report
+                ;;
+        clear)
+                clear_all
+                ;;
+        *)
+                error "Invalid command"
+                ;;
+esac