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": handlePooRecorderMsg(message) case "location_recorder": handleLocationRecorderMsg(message) 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) { switch message.Action { case "get_latest": handleGetLatestPoo() } } func handleLocationRecorderMsg(message haMessage) { if message.Action == "record" { port := viper.GetString("port") req, err := http.NewRequest("POST", "http://localhost:"+port+"/location/record", strings.NewReader(strings.ReplaceAll(message.Content, "'", "\""))) if err != nil { slog.Warn(fmt.Sprintln("handleLocationRecorderMsg Error creating request to location recorder", err)) return } req.Header.Set("Content-Type", "application/json") client := &http.Client{ Timeout: time.Second * 1, } _, err = client.Do(req) if err != nil { slog.Warn(fmt.Sprintln("handleLocationRecorderMsg Error sending request to location recorder", err)) } } else { slog.Warn(fmt.Sprintln("handleLocationRecorderMsg Unknown action", message.Action)) } } 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)) } }