m5stack/StackChan
1
0
Fork 0
mirror of https://github.com/m5stack/StackChan.git synced 2026-04-27 19:12:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

server code 4/27

Browse Source
This commit is contained in:
袁智鸿
2026-04-27 10:35:03 +08:00
parent f0fa33cd67
commit 2ccbcd01c0
143 changed files with 5812 additions and 756 deletions
Download Patch File Download Diff File Expand all files Collapse all files
server/internal/controller/device/device_new.go
+6
View File
@@ -18,3 +18,9 @@ type ControllerV1 struct{}
func NewV1() device.IDeviceV1 {
return &ControllerV1{}
}
type ControllerV2 struct{}
func NewV2() device.IDeviceV2 {
return &ControllerV2{}
}
server/internal/controller/device/device_v1_create.go
+10 -1
View File
@@ -9,12 +9,21 @@ import (
"context"
"stackChan/api/device/v1"
"stackChan/internal/dao"
"stackChan/internal/model"
"stackChan/internal/model/do"
"github.com/gogf/gf/v2/errors/gcode"
"github.com/gogf/gf/v2/errors/gerror"
"github.com/gogf/gf/v2/frame/g"
)
func (c *ControllerV1) Create(ctx context.Context, req *v1.CreateReq) (res *v1.CreateRes, err error) {
mac := g.RequestFromCtx(ctx).GetCtxVar(model.Mac).String()
if mac == "" {
return nil, gerror.NewCode(gcode.CodeInvalidParameter)
}
insertId, err := dao.Device.Ctx(ctx).Data(do.Device{
Mac: req.Mac,
Mac: mac,
Name: req.Name,
}).InsertAndGetId()
if err != nil {
server/internal/controller/device/device_v1_get_device_info.go
+9 -1
View File
@@ -11,11 +11,19 @@ import (
"stackChan/internal/model"
"stackChan/api/device/v1"
"github.com/gogf/gf/v2/errors/gcode"
"github.com/gogf/gf/v2/errors/gerror"
"github.com/gogf/gf/v2/frame/g"
)
func (c *ControllerV1) GetDeviceInfo(ctx context.Context, req *v1.GetDeviceInfoReq) (res *v1.GetDeviceInfoRes, err error) {
mac := g.RequestFromCtx(ctx).GetCtxVar(model.Mac).String()
if mac == "" {
return nil, gerror.NewCode(gcode.CodeInvalidParameter)
}
var info model.DeviceInfo
err = dao.Device.Ctx(ctx).WherePri(req.Mac).Scan(&info)
err = dao.Device.Ctx(ctx).WherePri(mac).Scan(&info)
if err != nil {
return nil, err
}
server/internal/controller/device/device_v1_get_random_device.go
+15 -1
View File
@@ -9,14 +9,28 @@ import (
"context"
"stackChan/api/device/v1"
"stackChan/internal/dao"
"stackChan/internal/model"
"stackChan/internal/model/entity"
"stackChan/internal/web_socket"
"github.com/gogf/gf/v2/errors/gcode"
"github.com/gogf/gf/v2/errors/gerror"
"github.com/gogf/gf/v2/frame/g"
)
func (c *ControllerV1) GetRandomDevice(ctx context.Context, req *v1.GetRandomDeviceReq) (res *v1.GetRandomDeviceRes, err error) {
mac := g.RequestFromCtx(ctx).GetCtxVar(model.Mac).String()
if mac == "" {
return nil, gerror.NewCode(gcode.CodeInvalidParameter)
}
pageSize := req.PageSize
if pageSize <= 0 {
pageSize = 6
}
// Obtain the list of online StackChan mac addresses (excluding the current user) from the websocket layer.
macList := web_socket.GetRandomStackChanDevice(req.Mac, 6)
macList := web_socket.GetRandomStackChanDevice(mac, pageSize)
if len(macList) == 0 {
res = (*v1.GetRandomDeviceRes)(&[]entity.Device{})
server/internal/controller/device/device_v1_update.go
+10 -1
View File
@@ -8,14 +8,23 @@ package device
import (
"context"
"stackChan/internal/dao"
"stackChan/internal/model"
"stackChan/internal/model/do"
"stackChan/api/device/v1"
"github.com/gogf/gf/v2/errors/gcode"
"github.com/gogf/gf/v2/errors/gerror"
"github.com/gogf/gf/v2/frame/g"
)
func (c *ControllerV1) Update(ctx context.Context, req *v1.UpdateReq) (res *v1.UpdateRes, err error) {
mac := g.RequestFromCtx(ctx).GetCtxVar(model.Mac).String()
if mac == "" {
return nil, gerror.NewCode(gcode.CodeInvalidParameter)
}
_, err = dao.Device.Ctx(ctx).Data(do.Device{
Name: req.Name,
}).WherePri(req.Mac).Update()
}).WherePri(mac).Update()
return
}
server/internal/controller/device/device_v1_update_device_info.go
+10 -1
View File
@@ -8,17 +8,26 @@ package device
import (
"context"
"stackChan/internal/dao"
"stackChan/internal/model"
"stackChan/internal/model/do"
"stackChan/api/device/v1"
"github.com/gogf/gf/v2/errors/gcode"
"github.com/gogf/gf/v2/errors/gerror"
"github.com/gogf/gf/v2/frame/g"
)
func (c *ControllerV1) UpdateDeviceInfo(ctx context.Context, req *v1.UpdateDeviceInfoReq) (res *v1.UpdateDeviceInfoRes, err error) {
mac := g.RequestFromCtx(ctx).GetCtxVar(model.Mac).String()
if mac == "" {
return nil, gerror.NewCode(gcode.CodeInvalidParameter)
}
doDevice := do.Device{}
if req.Name != "" {
doDevice.Name = req.Name
}
_, err = dao.Device.Ctx(ctx).Data(doDevice).WherePri(req.Mac).Update()
_, err = dao.Device.Ctx(ctx).Data(doDevice).WherePri(mac).Update()
if err != nil {
return nil, err
}
server/internal/controller/device/device_v2_agent_restore_default.go
+26
View File
@@ -0,0 +1,26 @@
/*
SPDX-FileCopyrightText: 2026 M5Stack Technology CO LTD
SPDX-License-Identifier: MIT
*/
package device
import (
"context"
"stackChan/api/device/v2"
)
func (c *ControllerV2) AgentRestoreDefault(ctx context.Context, req *v2.AgentRestoreDefaultReq) (res *v2.AgentRestoreDefaultRes, err error) {
return new(v2.AgentRestoreDefaultRes(true)), err
//if req.Mac == "" {
// return nil, gerror.NewCode(gcode.CodeMissingParameter, "Device MAC address cannot be empty")
//}
//restoreResponse, err := service.RestoreDefaultAgent(req.Mac)
//if err != nil {
// return nil, err
//}
//if !restoreResponse {
// return nil, gerror.NewCode(gcode.CodeInternalError, "Failed to restore default configuration")
//}
//return new(v2.AgentRestoreDefaultRes(true)), nil
}
server/internal/controller/device/device_v2_bind_device.go
+41
View File
@@ -0,0 +1,41 @@
/*
SPDX-FileCopyrightText: 2026 M5Stack Technology CO LTD
SPDX-License-Identifier: MIT
*/
package device
import (
"context"
"stackChan/internal/dao"
"stackChan/internal/model"
"stackChan/internal/service"
"github.com/gogf/gf/v2/errors/gcode"
"github.com/gogf/gf/v2/errors/gerror"
"github.com/gogf/gf/v2/frame/g"
"github.com/gogf/gf/v2/os/gtime"
"stackChan/api/device/v2"
)
// BindDevice Device binding interface
func (c *ControllerV2) BindDevice(ctx context.Context, req *v2.BindDeviceReq) (res *v2.BindDeviceRes, err error) {
// 1. Get current logged-in user UID (from context)
_, err = service.CreateMacIfNotExists(ctx, req.Mac)
uid := g.RequestFromCtx(ctx).GetCtxVar(model.Uid).Int64()
if uid == 0 {
return nil, gerror.NewCode(gcode.CodeMissingParameter, "User UID cannot be empty")
}
if req.Mac == "" {
return nil, gerror.NewCode(gcode.CodeMissingParameter, "Device MAC address cannot be empty")
}
_, err = dao.Device.Ctx(ctx).
Where("mac = ?", req.Mac).
Data("uid", uid, "bind_time", gtime.Now().Format("Y-m-d H:i:s")).
Update()
if err != nil {
return nil, gerror.WrapCode(gcode.CodeDbOperationError, err, "Device binding failed")
}
return new(v2.BindDeviceRes(true)), nil
}
server/internal/controller/device/device_v2_get_devices.go
+31
View File
@@ -0,0 +1,31 @@
/*
SPDX-FileCopyrightText: 2026 M5Stack Technology CO LTD
SPDX-License-Identifier: MIT
*/
package device
import (
"context"
"stackChan/internal/dao"
"stackChan/internal/model"
"github.com/gogf/gf/v2/errors/gcode"
"github.com/gogf/gf/v2/errors/gerror"
"github.com/gogf/gf/v2/frame/g"
"stackChan/api/device/v2"
)
func (c *ControllerV2) GetDevices(ctx context.Context, req *v2.GetDevicesReq) (res *v2.GetDevicesRes, err error) {
uid := g.RequestFromCtx(ctx).GetCtxVar(model.Uid).Int64()
if uid == 0 {
return nil, gerror.NewCode(gcode.CodeMissingParameter, "user UID is required")
}
devices := make([]model.DeviceInfo, 0)
err = dao.Device.Ctx(ctx).Where("uid=?", uid).Scan(&devices)
if err != nil {
return nil, err
}
return new(v2.GetDevicesRes(devices)), nil
}
server/internal/controller/device/device_v2_unbind_device.go
+69
View File
@@ -0,0 +1,69 @@
/*
SPDX-FileCopyrightText: 2026 M5Stack Technology CO LTD
SPDX-License-Identifier: MIT
*/
package device
import (
"context"
"stackChan/internal/dao"
"stackChan/internal/model"
"stackChan/internal/service"
"stackChan/internal/xiaozhi"
"github.com/gogf/gf/v2/errors/gcode"
"github.com/gogf/gf/v2/errors/gerror"
"github.com/gogf/gf/v2/frame/g"
"stackChan/api/device/v2"
)
// UnbindDevice Device unbinding interface
func (c *ControllerV2) UnbindDevice(ctx context.Context, req *v2.UnbindDeviceReq) (res *v2.UnbindDeviceRes, err error) {
_, err = service.CreateMacIfNotExists(ctx, req.Mac)
if err != nil {
return nil, gerror.WrapCode(gcode.CodeDbOperationError, err, "Failed to initialize device information")
}
uid := g.RequestFromCtx(ctx).GetCtxVar(model.Uid).Int64()
if uid == 0 {
return nil, gerror.NewCode(gcode.CodeMissingParameter, "User UID cannot be empty")
}
// 3. Validate MAC address parameter
if req.Mac == "" {
return nil, gerror.NewCode(gcode.CodeMissingParameter, "Device MAC address cannot be empty")
}
restoreResponse, err := service.RestoreDefaultAgent(req.Mac)
if err != nil {
return nil, err
}
if !restoreResponse {
return nil, gerror.NewCode(gcode.CodeInternalError, "Failed to restore default configuration")
}
// xiaozhi Unbind Device
unbindResponse, err := xiaozhi.UnbindDevice(&req.Mac)
if err != nil {
return nil, gerror.NewCode(gcode.CodeInternalError)
}
if !unbindResponse {
g.Log().Error(ctx, "xiaozhi Unbind Device failed:")
return nil, gerror.NewCode(gcode.CodeInternalError)
}
// 4. Perform unbind: set uid to 0/NULL (only the current user's own device can be unbound)
_, err = dao.Device.Ctx(ctx).
Where("mac = ?", req.Mac).
Where("uid = ?", uid).
Data("uid", nil, "bind_time", nil).
Update()
if err != nil {
return nil, gerror.WrapCode(gcode.CodeDbOperationError, err, "Device unbinding failed")
}
// 5. Return success response (consistent with bind interface format)
return new(v2.UnbindDeviceRes(true)), nil
}
server/internal/controller/device/device_v2_update_device.go
+67
View File
@@ -0,0 +1,67 @@
/*
SPDX-FileCopyrightText: 2026 M5Stack Technology CO LTD
SPDX-License-Identifier: MIT
*/
package device
import (
"context"
"stackChan/internal/dao"
"stackChan/internal/model"
"github.com/gogf/gf/v2/errors/gcode"
"github.com/gogf/gf/v2/errors/gerror"
"github.com/gogf/gf/v2/frame/g"
"stackChan/api/device/v2"
)
func (c *ControllerV2) UpdateDevice(ctx context.Context, req *v2.UpdateDeviceReq) (res *v2.UpdateDeviceRes, err error) {
// Get current logged-in user UID
uid := g.RequestFromCtx(ctx).GetCtxVar(model.Uid).Int64()
if uid == 0 {
return nil, gerror.NewCode(gcode.CodeMissingParameter, "User UID cannot be empty")
}
// check device exists and belongs to current user
count, err := dao.Device.Ctx(ctx).
Where("mac = ?", req.Mac).
Where("uid = ?", uid).
Count()
if err != nil {
return nil, gerror.WrapCode(gcode.CodeDbOperationError, err, "Failed to query device information")
}
if count == 0 {
return nil, gerror.NewCode(gcode.CodeNotFound, "device not found or not belong to current user")
}
// build update data
updateData := g.Map{}
if req.Name != "" {
updateData["name"] = req.Name
}
if req.Longitude != 0 {
updateData["longitude"] = req.Longitude
}
if req.Latitude != 0 {
updateData["latitude"] = req.Latitude
}
// no need to update
if len(updateData) == 0 {
return new(v2.UpdateDeviceRes(true)), nil
}
// update device information
_, err = dao.Device.Ctx(ctx).
Where("mac = ?", req.Mac).
Where("uid = ?", uid).
Data(updateData).
Update()
if err != nil {
return nil, gerror.WrapCode(gcode.CodeDbOperationError, err, "update device failed")
}
return new(v2.UpdateDeviceRes(true)), nil
}