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 actionTask struct {
	Action  string `json:"action"`
	DueHour int    `json:"due_hour"`
}

var (
	ticktickUtil ticktickutil.TicktickUtil
)

func Init(ticktick ticktickutil.TicktickUtil) {
	ticktickUtil = ticktick
}

func 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("HandleHaMessage: Error decoding request body", err))
		http.Error(w, err.Error(), http.StatusBadRequest)
		return
	}

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

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

func 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("handleLocationRecorderMsg: Error sending request to location recorder", err))
			return false
		}
	} else {
		slog.Warn(fmt.Sprintln("handleLocationRecorderMsg: Unknown action", message.Action))
		return false
	}
	return true
}

func handleTicktickMsg(message haMessage) {
	switch message.Action {
	case "create_action_task":
		createActionTask(message)
	}
}

func 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("handleGetLatestPoo: Error sending request to poo recorder", err))
		return false
	}

	return true
}

func createActionTask(message haMessage) {
	if !viper.InConfig("homeassistant.actionTaskProjectId") {
		slog.Warn("Homeassistant.createActionTask actionTaskProjectId not found in config file")
		return
	}
	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
	}
	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 = ticktickUtil.CreateTask(ticktickTask)
	if err != nil {
		slog.Warn(fmt.Sprintf("Homeassistant.createActionTask Error creating task %s", err))
	}
}