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)) } }