tliu93/home-automation-backend
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

wip

Browse Source
This commit is contained in:
Tianyu Liu
2025-05-19 20:25:53 +02:00
parent 234323c766
commit 0a76c5feca
4 changed files with 99 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files

79
src/components/homeassistant/homeassistant.go
View File

@@ -18,60 +18,68 @@ type haMessage struct {
Content string `json:"content"`
}
type HomeAssistant struct {
ticktickUtil ticktickutil.TicktickUtil
}
type actionTask struct {
Action string `json:"action"`
DueHour int `json:"due_hour"`
}
var (
ticktickUtil ticktickutil.TicktickUtil
)
func Init(ticktick ticktickutil.TicktickUtil) {
ticktickUtil = ticktick
func NewHomeAssistant(ticktick ticktickutil.TicktickUtil) *HomeAssistant {
return &HomeAssistant{
ticktickUtil: ticktick,
}
}
func HandleHaMessage(w http.ResponseWriter, r *http.Request) {
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("HandleHaMessage: Error decoding request body", err))
http.Error(w, err.Error(), http.StatusBadRequest)
slog.Warn(fmt.Sprintln("homeassistant.HandleHaMessage: Error decoding request body", err))
http.Error(w, "", http.StatusInternalServerError)
return
}
switch message.Target {
case "poo_recorder":
res := handlePooRecorderMsg(message)
res := ha.handlePooRecorderMsg(message)
if !res {
http.Error(w, "Error handling poo recorder message", http.StatusInternalServerError)
slog.Warn(fmt.Sprintln("homeassistant.HandleHaMessage: Error handling poo recorder message"))
http.Error(w, "", http.StatusInternalServerError)
}
case "location_recorder":
res := handleLocationRecorderMsg(message)
res := ha.handleLocationRecorderMsg(message)
if !res {
http.Error(w, "Error handling location recorder message", http.StatusInternalServerError)
slog.Warn(fmt.Sprintln("homeassistant.HandleHaMessage: Error handling location recorder message"))
http.Error(w, "", http.StatusInternalServerError)
}
case "ticktick":
handleTicktickMsg(message)
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("HandleHaMessage Unknown target", message.Target))
http.Error(w, "Unknown target", http.StatusBadRequest)
slog.Warn(fmt.Sprintln("homeassistant.HandleHaMessage: Unknown target", message.Target))
http.Error(w, "", http.StatusInternalServerError)
}
}
func handlePooRecorderMsg(message haMessage) bool {
func (ha *HomeAssistant) handlePooRecorderMsg(message haMessage) bool {
switch message.Action {
case "get_latest":
return handleGetLatestPoo()
return ha.handleGetLatestPoo()
default:
slog.Warn(fmt.Sprintln("handlePooRecorderMsg: Unknown action", message.Action))
slog.Warn(fmt.Sprintln("homeassistant.handlePooRecorderMsg: Unknown action", message.Action))
return false
}
}
func handleLocationRecorderMsg(message haMessage) bool {
func (ha *HomeAssistant) handleLocationRecorderMsg(message haMessage) bool {
if message.Action == "record" {
port := viper.GetString("port")
client := &http.Client{
@@ -79,49 +87,52 @@ func handleLocationRecorderMsg(message haMessage) bool {
}
_, 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))
slog.Warn(fmt.Sprintln("homeassistant.handleLocationRecorderMsg: Error sending request to location recorder", err))
return false
}
} else {
slog.Warn(fmt.Sprintln("handleLocationRecorderMsg: Unknown action", message.Action))
slog.Warn(fmt.Sprintln("homeassistant.handleLocationRecorderMsg: Unknown action", message.Action))
return false
}
return true
}
func handleTicktickMsg(message haMessage) {
func (ha *HomeAssistant) handleTicktickMsg(message haMessage) bool {
switch message.Action {
case "create_action_task":
createActionTask(message)
return ha.createActionTask(message)
default:
slog.Warn(fmt.Sprintln("homeassistant.handleTicktickMsg: Unknown action", message.Action))
return false
}
}
func handleGetLatestPoo() bool {
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("handleGetLatestPoo: Error sending request to poo recorder", err))
slog.Warn(fmt.Sprintln("homeassistant.handleGetLatestPoo: Error sending request to poo recorder", err))
return false
}
return true
}
func createActionTask(message haMessage) {
func (ha *HomeAssistant) createActionTask(message haMessage) bool {
if !viper.InConfig("homeassistant.actionTaskProjectId") {
slog.Warn("Homeassistant.createActionTask actionTaskProjectId not found in config file")
return
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
slog.Warn(fmt.Sprintln("homeassistant.createActionTask: Error unmarshalling", err))
return false
}
dueHour := task.DueHour
due := time.Now().Add(time.Hour * time.Duration(dueHour))
@@ -132,8 +143,10 @@ func createActionTask(message haMessage) {
Title: task.Action,
DueDate: dueTicktick,
}
err = ticktickUtil.CreateTask(ticktickTask)
err = ha.ticktickUtil.CreateTask(ticktickTask)
if err != nil {
slog.Warn(fmt.Sprintf("Homeassistant.createActionTask Error creating task %s", err))
slog.Warn(fmt.Sprintf("homeassistant.createActionTask: Error creating task %s", err))
return false
}
return true
}

70
src/components/homeassistant/homeassistant_test.go
View File

@@ -10,18 +10,44 @@ import (
"github.com/spf13/viper"
"github.com/stretchr/testify/assert"
"github.com/t-liu93/home-automation-backend/util/ticktickutil"
)
var (
loggerText = new(bytes.Buffer)
)
func SetupTearDown(t *testing.T) func() {
type MockTicktickUtil struct {
// Mock methods for testing
}
func (m *MockTicktickUtil) HandleAuthCode(w http.ResponseWriter, r *http.Request) {
// Mock implementation
}
func (m *MockTicktickUtil) GetTasks(projectId string) []ticktickutil.Task {
// Mock implementation
return []ticktickutil.Task{}
}
func (m *MockTicktickUtil) HasDuplicateTask(projectId string, taskTitle string) bool {
// Mock implementation
return false
}
func (m *MockTicktickUtil) CreateTask(task ticktickutil.Task) error {
// Mock implementation
return nil
}
func SetupTearDown(t *testing.T) (func(), *HomeAssistant) {
loggertearDown := loggerSetupTeardown()
mockTicktick := &MockTicktickUtil{}
ha := NewHomeAssistant(mockTicktick)
return func() {
loggertearDown()
}
}, ha
}
func loggerSetupTeardown() func() {
@@ -36,20 +62,20 @@ func loggerSetupTeardown() func() {
}
func TestHandleHaMessageJsonDecodeError(t *testing.T) {
teardown := SetupTearDown(t)
teardown, ha := SetupTearDown(t)
defer teardown()
invalidRequestBody := ` { "target": "poo_recorder", "action": "get_latest", "content": " }`
req := httptest.NewRequest(http.MethodPost, "/homeassistant/publish", strings.NewReader(invalidRequestBody))
w := httptest.NewRecorder()
HandleHaMessage(w, req)
assert.Equal(t, http.StatusBadRequest, w.Code)
assert.Contains(t, loggerText.String(), "HandleHaMessage: Error decoding request body")
ha.HandleHaMessage(w, req)
assert.Equal(t, http.StatusInternalServerError, w.Code)
assert.Contains(t, loggerText.String(), "homeassistant.HandleHaMessage: Error decoding request body")
}
func TestHandlePooRecorderMsgGetLatest(t *testing.T) {
teardown := SetupTearDown(t)
teardown, ha := SetupTearDown(t)
defer teardown()
requestBody := `{"target": "poo_recorder", "action": "get_latest", "content": ""}`
req := httptest.NewRequest(http.MethodPost, "/homeassistant/publish", strings.NewReader(requestBody))
@@ -62,26 +88,26 @@ func TestHandlePooRecorderMsgGetLatest(t *testing.T) {
port := strings.Split(server.URL, ":")[2]
viper.Set("port", port)
HandleHaMessage(w, req)
ha.HandleHaMessage(w, req)
assert.Equal(t, http.StatusOK, w.Code)
assert.Empty(t, loggerText.String())
}
func TestHandlePooRecorderMsgUnknownAction(t *testing.T) {
teardown := SetupTearDown(t)
teardown, ha := SetupTearDown(t)
defer teardown()
requestBody := `{"target": "poo_recorder", "action": "unknown_action", "content": ""}`
req := httptest.NewRequest(http.MethodPost, "/homeassistant/publish", strings.NewReader(requestBody))
w := httptest.NewRecorder()
HandleHaMessage(w, req)
ha.HandleHaMessage(w, req)
assert.Equal(t, http.StatusInternalServerError, w.Code)
assert.Contains(t, loggerText.String(), "handlePooRecorderMsg: Unknown action")
assert.Contains(t, loggerText.String(), "homeassistant.handlePooRecorderMsg: Unknown action")
}
func TestHandlePooRecorderMsgGetLatestError(t *testing.T) {
teardown := SetupTearDown(t)
teardown, ha := SetupTearDown(t)
defer teardown()
requestBody := `{"target": "poo_recorder", "action": "get_latest", "content": ""}`
@@ -91,13 +117,13 @@ func TestHandlePooRecorderMsgGetLatestError(t *testing.T) {
port := "invalid port"
viper.Set("port", port)
HandleHaMessage(w, req)
ha.HandleHaMessage(w, req)
assert.Equal(t, http.StatusInternalServerError, w.Code)
assert.Contains(t, loggerText.String(), "handleGetLatestPoo: Error sending request to poo recorder")
assert.Contains(t, loggerText.String(), "homeassistant.handleGetLatestPoo: Error sending request to poo recorder")
}
func TestHandleLocationRecorderMsg(t *testing.T) {
teardown := SetupTearDown(t)
teardown, ha := SetupTearDown(t)
defer teardown()
requestBody := `{"target": "location_recorder", "action": "record", "content": "{'person': 'test', 'latitude': '1.0', 'longitude': '2.0', 'altitude': '3.0'}"}`
@@ -113,26 +139,26 @@ func TestHandleLocationRecorderMsg(t *testing.T) {
port := strings.Split(server.URL, ":")[2]
viper.Set("port", port)
HandleHaMessage(w, req)
ha.HandleHaMessage(w, req)
assert.Equal(t, http.StatusOK, w.Code)
assert.Empty(t, loggerText.String())
}
func TestHandleLocationRecorderMsgUnknownAction(t *testing.T) {
teardown := SetupTearDown(t)
teardown, ha := SetupTearDown(t)
defer teardown()
requestBody := `{"target": "location_recorder", "action": "unknown_action", "content": ""}`
req := httptest.NewRequest(http.MethodPost, "/homeassistant/publish", strings.NewReader(requestBody))
w := httptest.NewRecorder()
HandleHaMessage(w, req)
ha.HandleHaMessage(w, req)
assert.Equal(t, http.StatusInternalServerError, w.Code)
assert.Contains(t, loggerText.String(), "handleLocationRecorderMsg: Unknown action")
assert.Contains(t, loggerText.String(), "homeassistant.handleLocationRecorderMsg: Unknown action")
}
func TestHandleLocationRecorderMsgRequestErr(t *testing.T) {
teardown := SetupTearDown(t)
teardown, ha := SetupTearDown(t)
defer teardown()
requestBody := `{"target": "location_recorder", "action": "record", "content": "{'person': 'test', 'latitude': '1.0', 'longitude': '2.0', 'altitude': '3.0'}"}`
@@ -142,7 +168,7 @@ func TestHandleLocationRecorderMsgRequestErr(t *testing.T) {
port := "invalid port"
viper.Set("port", port)
HandleHaMessage(w, req)
ha.HandleHaMessage(w, req)
assert.Equal(t, http.StatusInternalServerError, w.Code)
assert.Contains(t, loggerText.String(), "handleLocationRecorderMsg: Error sending request to location recorder")
assert.Contains(t, loggerText.String(), "homeassistant.handleLocationRecorderMsg: Error sending request to location recorder")
}