diff --git a/Sense/server/cmd/migrate/migration/version/2026081623100_sense_layout.go b/Sense/server/cmd/migrate/migration/version/2026081623100_sense_layout.go new file mode 100644 index 0000000..3b48a6a --- /dev/null +++ b/Sense/server/cmd/migrate/migration/version/2026081623100_sense_layout.go @@ -0,0 +1,85 @@ +package version + +import ( + "fmt" + "runtime" + + "gorm.io/gorm" + + "git.ilapage.cn/ila/yovision/Sense/server/cmd/migrate/migration" + migrationModels "git.ilapage.cn/ila/yovision/Sense/server/cmd/migrate/migration/models" + common "git.ilapage.cn/ila/yovision/Sense/server/common/models" +) + +const senseLayoutMenuName = "SenseManage" + +var sensePageMenus = []migrationModels.SysMenu{ + {MenuName: "SenseDeviceManage", Title: "设备管理", Icon: "monitor", Path: "devices", MenuType: "C", Permission: "sense:device:list", Component: "/sense/device/index", Sort: 1, Visible: "0", IsFrame: "1"}, + {MenuName: "SenseAdmission", Title: "视频接入", Icon: "video-camera", Path: "admission", MenuType: "C", Permission: "sense:admission:list", Component: "/sense/admission/index", Sort: 2, Visible: "0", IsFrame: "1"}, + {MenuName: "SenseMedia", Title: "视频服务", Icon: "video-play", Path: "media", MenuType: "C", Permission: "sense:media:list", Component: "/sense/media/index", Sort: 3, Visible: "0", IsFrame: "1"}, + {MenuName: "SenseLiveview", Title: "实时监看", Icon: "eye-open", Path: "liveview", MenuType: "C", Permission: "sense:liveview:view", Component: "/sense/liveview/index", Sort: 4, Visible: "0", IsFrame: "1"}, + {MenuName: "SenseArea", Title: "区域与警戒线", Icon: "guide", Path: "area", MenuType: "C", Permission: "sense:area:list", Component: "/sense/area/index", Sort: 5, Visible: "0", IsFrame: "1"}, +} + +func init() { + _, fileName, _, _ := runtime.Caller(0) + migration.Migrate.SetVersion(migration.GetFilename(fileName), migrateSenseLayout) +} + +func migrateSenseLayout(db *gorm.DB, version string) error { + return db.Transaction(func(tx *gorm.DB) error { + if err := alignSenseLayout(tx); err != nil { + return err + } + return tx.Create(&common.Migration{Version: version}).Error + }) +} + +func alignSenseLayout(tx *gorm.DB) error { + root, err := ensureDeviceMenu(tx, migrationModels.SysMenu{ + MenuName: senseLayoutMenuName, + Title: "视频感知", + Icon: "video-camera", + Path: "/sense", + MenuType: "M", + ParentId: 0, + Component: "Layout", + Sort: 5, + Visible: "0", + IsFrame: "1", + }) + if err != nil { + return fmt.Errorf("ensure Sense layout menu: %w", err) + } + + for _, desired := range sensePageMenus { + desired.ParentId = root.MenuId + desired.Paths = fmt.Sprintf("/0/%d", root.MenuId) + if _, err = ensureDeviceMenu(tx, desired); err != nil { + return fmt.Errorf("align Sense page menu %s: %w", desired.MenuName, err) + } + } + + if err = rebuildSenseMenuPaths(tx, root.MenuId, "/0"); err != nil { + return err + } + return nil +} + +func rebuildSenseMenuPaths(tx *gorm.DB, menuID int, parentPath string) error { + path := fmt.Sprintf("%s/%d", parentPath, menuID) + if err := tx.Model(&migrationModels.SysMenu{}).Where("menu_id = ?", menuID).Update("paths", path).Error; err != nil { + return fmt.Errorf("update menu %d paths: %w", menuID, err) + } + + var children []migrationModels.SysMenu + if err := tx.Where("parent_id = ?", menuID).Order("sort, menu_id").Find(&children).Error; err != nil { + return fmt.Errorf("list children of menu %d: %w", menuID, err) + } + for _, child := range children { + if err := rebuildSenseMenuPaths(tx, child.MenuId, path); err != nil { + return err + } + } + return nil +} diff --git a/Sense/server/cmd/migrate/migration/version/2026081623100_sense_layout_test.go b/Sense/server/cmd/migrate/migration/version/2026081623100_sense_layout_test.go new file mode 100644 index 0000000..3779a42 --- /dev/null +++ b/Sense/server/cmd/migrate/migration/version/2026081623100_sense_layout_test.go @@ -0,0 +1,208 @@ +package version + +import ( + "fmt" + "os" + "testing" + + "gorm.io/driver/postgres" + "gorm.io/driver/sqlite" + "gorm.io/gorm" + + migrationModels "git.ilapage.cn/ila/yovision/Sense/server/cmd/migrate/migration/models" + common "git.ilapage.cn/ila/yovision/Sense/server/common/models" +) + +func TestAlignSenseLayoutPreservesShellAndExistingPages(t *testing.T) { + db := openSenseLayoutTestDB(t) + seedLegacySenseMenus(t, db) + associationsBefore := countRoleMenuAssociations(t, db) + + if err := alignSenseLayout(db); err != nil { + t.Fatal(err) + } + // The alignment helper is intentionally idempotent because partially upgraded + // deployments may rerun the repair before the migration version is recorded. + if err := alignSenseLayout(db); err != nil { + t.Fatalf("repeat alignment: %v", err) + } + + root := requireMenu(t, db, senseLayoutMenuName) + if root.ParentId != 0 || root.Component != "Layout" || root.Path != "/sense" || root.MenuType != "M" { + t.Fatalf("unexpected root menu: %#v", root) + } + if root.Paths != fmt.Sprintf("/0/%d", root.MenuId) { + t.Fatalf("root paths=%q", root.Paths) + } + + for _, desired := range sensePageMenus { + page := requireMenu(t, db, desired.MenuName) + if page.ParentId != root.MenuId || page.Path != desired.Path || page.Component != desired.Component || page.Permission != desired.Permission { + t.Errorf("unexpected page %s: %#v", desired.MenuName, page) + } + wantPagePaths := fmt.Sprintf("/0/%d/%d", root.MenuId, page.MenuId) + if page.Paths != wantPagePaths { + t.Errorf("page %s paths=%q want %q", desired.MenuName, page.Paths, wantPagePaths) + } + + var buttons []migrationModels.SysMenu + if err := db.Where("parent_id = ? AND menu_type = ?", page.MenuId, "F").Find(&buttons).Error; err != nil { + t.Fatal(err) + } + if len(buttons) != 1 { + t.Fatalf("page %s button count=%d", desired.MenuName, len(buttons)) + } + wantButtonPaths := fmt.Sprintf("%s/%d", wantPagePaths, buttons[0].MenuId) + if buttons[0].Paths != wantButtonPaths { + t.Errorf("button %s paths=%q want %q", buttons[0].MenuName, buttons[0].Paths, wantButtonPaths) + } + } + + var rootCount int64 + if err := db.Model(&migrationModels.SysMenu{}).Where("menu_name = ?", senseLayoutMenuName).Count(&rootCount).Error; err != nil { + t.Fatal(err) + } + if rootCount != 1 { + t.Fatalf("Sense layout menu count=%d", rootCount) + } + if associationsAfter := countRoleMenuAssociations(t, db); associationsAfter != associationsBefore { + t.Fatalf("role-menu associations changed from %d to %d", associationsBefore, associationsAfter) + } +} + +func TestMigrateSenseLayoutRollsBackWhenRequiredPageIsMissing(t *testing.T) { + db := openSenseLayoutTestDB(t) + seedLegacySenseMenus(t, db) + if err := db.Where("menu_name = ?", "SenseArea").Delete(&migrationModels.SysMenu{}).Error; err != nil { + t.Fatal(err) + } + + // A missing page is recreated by the repair, so force a failure when the + // migration records its version and verify the menu transaction also rolls back. + version := "2026081623100" + if err := db.Create(&common.Migration{Version: version}).Error; err != nil { + t.Fatal(err) + } + if err := migrateSenseLayout(db, version); err == nil { + t.Fatal("expected duplicate migration version error") + } + + var rootCount, areaCount int64 + db.Model(&migrationModels.SysMenu{}).Where("menu_name = ?", senseLayoutMenuName).Count(&rootCount) + db.Model(&migrationModels.SysMenu{}).Where("menu_name = ?", "SenseArea").Count(&areaCount) + if rootCount != 0 || areaCount != 0 { + t.Fatalf("transaction left partial menus: root=%d area=%d", rootCount, areaCount) + } +} + +func openSenseLayoutTestDB(t *testing.T) *gorm.DB { + t.Helper() + db, err := gorm.Open(sqlite.Open("file:"+t.Name()+"?mode=memory&cache=shared"), &gorm.Config{}) + if err != nil { + t.Fatal(err) + } + if err = db.AutoMigrate(&migrationModels.SysRole{}, &migrationModels.SysMenu{}, &common.Migration{}); err != nil { + t.Fatal(err) + } + return db +} + +func TestSenseLayoutMigrationOnPostgres(t *testing.T) { + dsn := os.Getenv("SENSE_LAYOUT_MIGRATION_TEST_DATABASE_URL") + if dsn == "" { + t.Skip("set SENSE_LAYOUT_MIGRATION_TEST_DATABASE_URL to run the PostgreSQL migration test") + } + db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{}) + if err != nil { + t.Fatal(err) + } + const schema = "sense_layout_101_test" + if err = db.Exec("DROP SCHEMA IF EXISTS " + schema + " CASCADE").Error; err != nil { + t.Fatal(err) + } + if err = db.Exec("CREATE SCHEMA " + schema).Error; err != nil { + t.Fatal(err) + } + t.Cleanup(func() { db.Exec("DROP SCHEMA IF EXISTS " + schema + " CASCADE") }) + sqlDB, err := db.DB() + if err != nil { + t.Fatal(err) + } + sqlDB.SetMaxOpenConns(1) + if err = db.Exec("SET search_path TO " + schema).Error; err != nil { + t.Fatal(err) + } + if err = db.AutoMigrate(&migrationModels.SysRole{}, &migrationModels.SysMenu{}, &common.Migration{}); err != nil { + t.Fatal(err) + } + seedLegacySenseMenus(t, db) + + const version = "2026081623100" + if err = migrateSenseLayout(db, version); err != nil { + t.Fatal(err) + } + root := requireMenu(t, db, senseLayoutMenuName) + if root.Component != "Layout" || root.Path != "/sense" { + t.Fatalf("unexpected PostgreSQL root: %#v", root) + } + var pages, versions int64 + db.Model(&migrationModels.SysMenu{}).Where("parent_id = ? AND menu_type = ?", root.MenuId, "C").Count(&pages) + db.Model(&common.Migration{}).Where("version = ?", version).Count(&versions) + if pages != int64(len(sensePageMenus)) || versions != 1 { + t.Fatalf("pages=%d versions=%d", pages, versions) + } +} + +func seedLegacySenseMenus(t *testing.T, db *gorm.DB) { + t.Helper() + role := migrationModels.SysRole{RoleName: "site_admin", RoleKey: "site_admin", Status: "2"} + if err := db.Create(&role).Error; err != nil { + t.Fatal(err) + } + assigned := make([]migrationModels.SysMenu, 0, len(sensePageMenus)*2) + for index, desired := range sensePageMenus { + legacy := desired + legacy.Path = "/sense/" + desired.Path + legacy.ParentId = 0 + legacy.Sort = index + 5 + legacy.Paths = "" + if err := db.Create(&legacy).Error; err != nil { + t.Fatal(err) + } + button := migrationModels.SysMenu{ + MenuName: desired.MenuName + "Action", + Title: "测试操作", + MenuType: "F", + Permission: desired.Permission + ":action", + ParentId: legacy.MenuId, + Paths: fmt.Sprintf("/0/%d", legacy.MenuId), + Visible: "1", + IsFrame: "1", + } + if err := db.Create(&button).Error; err != nil { + t.Fatal(err) + } + assigned = append(assigned, legacy, button) + } + if err := db.Model(&role).Association("SysMenu").Append(assigned); err != nil { + t.Fatal(err) + } +} + +func countRoleMenuAssociations(t *testing.T, db *gorm.DB) int64 { + t.Helper() + var count int64 + if err := db.Table("sys_role_menu").Count(&count).Error; err != nil { + t.Fatal(err) + } + return count +} + +func requireMenu(t *testing.T, db *gorm.DB, name string) migrationModels.SysMenu { + t.Helper() + var menu migrationModels.SysMenu + if err := db.Where("menu_name = ?", name).First(&menu).Error; err != nil { + t.Fatal(err) + } + return menu +}