home-automation-backend/src/components/homeassistant/homeassistant.go

package homeassistant

import (
	"encoding/json"
	"fmt"
	"log/slog"
	"net/http"
	"strings"
	"time"

	"github.com/spf13/viper"
	"github.com/t-liu93/home-automation-backend/util/ticktickutil"
)

type haMessage struct {
	Target  string `json:"target"`
	Action  string `json:"action"`
	Content string `json:"content"`
}

type HomeAssistant struct {
	ticktickUtil ticktickutil.TicktickUtil
}

type actionTask struct {
	Action  string `json:"action"`
	DueHour int    `json:"due_hour"`
}

func NewHomeAssistant(ticktick ticktickutil.TicktickUtil) *HomeAssistant {
	return &HomeAssistant{
		ticktickUtil: ticktick,
	}
}

func (ha *HomeAssistant) HandleHaMessage(w http.ResponseWriter, r *http.Request) {
	var message haMessage
	decoder := json.NewDecoder(r.Body)
	decoder.DisallowUnknownFields()
	err := decoder.Decode(&message)
	if err != nil {
		slog.Warn(fmt.Sprintln("homeassistant.HandleHaMessage: Error decoding request body", err))
		http.Error(w, "", http.StatusInternalServerError)
		return
	}

	switch message.Target {
	case "poo_recorder":
		res := ha.handlePooRecorderMsg(message)
		if !res {
			slog.Warn(fmt.Sprintln("homeassistant.HandleHaMessage: Error handling poo recorder message"))
			http.Error(w, "", http.StatusInternalServerError)
		}
	case "location_recorder":
		res := ha.handleLocationRecorderMsg(message)
		if !res {
			slog.Warn(fmt.Sprintln("homeassistant.HandleHaMessage: Error handling location recorder message"))
			http.Error(w, "", http.StatusInternalServerError)
		}
	case "ticktick":
		res := ha.handleTicktickMsg(message)
		if !res {
			slog.Warn(fmt.Sprintln("homeassistant.HandleHaMessage: Error handling ticktick message"))
			http.Error(w, "", http.StatusInternalServerError)
		}
	default:
		slog.Warn(fmt.Sprintln("homeassistant.HandleHaMessage: Unknown target", message.Target))
		http.Error(w, "", http.StatusInternalServerError)
	}
}

func (ha *HomeAssistant) handlePooRecorderMsg(message haMessage) bool {
	switch message.Action {
	case "get_latest":
		return ha.handleGetLatestPoo()
	default:
		slog.Warn(fmt.Sprintln("homeassistant.handlePooRecorderMsg: Unknown action", message.Action))
		return false
	}
}

func (ha *HomeAssistant) handleLocationRecorderMsg(message haMessage) bool {
	if message.Action == "record" {
		port := viper.GetString("port")
		client := &http.Client{
			Timeout: time.Second * 1,
		}
		_, err := client.Post("http://localhost:"+port+"/location/record", "application/json", strings.NewReader(strings.ReplaceAll(message.Content, "'", "\"")))
		if err != nil {
			slog.Warn(fmt.Sprintln("homeassistant.handleLocationRecorderMsg: Error sending request to location recorder", err))
			return false
		}
	} else {
		slog.Warn(fmt.Sprintln("homeassistant.handleLocationRecorderMsg: Unknown action", message.Action))
		return false
	}
	return true
}

func (ha *HomeAssistant) handleTicktickMsg(message haMessage) bool {
	switch message.Action {
	case "create_action_task":
		return ha.createActionTask(message)
	default:
		slog.Warn(fmt.Sprintln("homeassistant.handleTicktickMsg: Unknown action", message.Action))
		return false
	}
}

func (ha *HomeAssistant) handleGetLatestPoo() bool {
	client := &http.Client{
		Timeout: time.Second * 1,
	}
	port := viper.GetString("port")
	_, err := client.Get("http://localhost:" + port + "/poo/latest")
	if err != nil {
		slog.Warn(fmt.Sprintln("homeassistant.handleGetLatestPoo: Error sending request to poo recorder", err))
		return false
	}

	return true
}

func (ha *HomeAssistant) createActionTask(message haMessage) bool {
	if !viper.IsSet("homeassistant.actionTaskProjectId") {
		slog.Warn("homeassistant.createActionTask: actionTaskProjectId not found in config file")
		return false
	}
	projectId := viper.GetString("homeassistant.actionTaskProjectId")
	detail := strings.ReplaceAll(message.Content, "'", "\"")
	var task actionTask
	err := json.Unmarshal([]byte(detail), &task)
	if err != nil {
		slog.Warn(fmt.Sprintln("homeassistant.createActionTask: Error unmarshalling", err))
		return false
	}
	dueHour := task.DueHour
	due := time.Now().Add(time.Hour * time.Duration(dueHour))
	dueNextMidnight := time.Date(due.Year(), due.Month(), due.Day(), 0, 0, 0, 0, time.Local).AddDate(0, 0, 1)
	dueTicktick := dueNextMidnight.UTC().Format(ticktickutil.DateTimeLayout)
	ticktickTask := ticktickutil.Task{
		ProjectId: projectId,
		Title:     task.Action,
		DueDate:   dueTicktick,
	}
	err = ha.ticktickUtil.CreateTask(ticktickTask)
	if err != nil {
		slog.Warn(fmt.Sprintf("homeassistant.createActionTask: Error creating task %s", err))
		return false
	}
	return true
}