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 }