76 lines
3.6 KiB
Go
76 lines
3.6 KiB
Go
package version
|
|
|
|
import (
|
|
"fmt"
|
|
"runtime"
|
|
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/clause"
|
|
|
|
"git.ilapage.cn/ila/yovision/Sense/server/app/sense/provisioning"
|
|
"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"
|
|
)
|
|
|
|
func init() {
|
|
_, fileName, _, _ := runtime.Caller(0)
|
|
migration.Migrate.SetVersion(migration.GetFilename(fileName), migrateSenseProvisioning)
|
|
}
|
|
|
|
func migrateSenseProvisioning(db *gorm.DB, version string) error {
|
|
return db.Transaction(func(tx *gorm.DB) error {
|
|
if err := tx.AutoMigrate(&provisioning.Batch{}, &provisioning.Item{}, &deviceCasbinRule{}); err != nil {
|
|
return err
|
|
}
|
|
root, err := ensureDeviceMenu(tx, migrationModels.SysMenu{MenuName: senseLayoutMenuName, Title: "视频感知", Icon: "video-camera", Path: "/sense", MenuType: "M", Component: "Layout", Sort: 5, Visible: "0", IsFrame: "1"})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
page, err := ensureDeviceMenu(tx, migrationModels.SysMenu{MenuName: "SenseProvisioning", Title: "批量开通", Icon: "upload", Path: "provisioning", Paths: fmt.Sprintf("/0/%d", root.MenuId), MenuType: "C", Permission: "sense:provisioning:list", ParentId: root.MenuId, Component: "/sense/provisioning/index", Sort: 2, Visible: "0", IsFrame: "1"})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
definitions := []struct{ name, title, action, permission string }{
|
|
{"SenseProvisioningImport", "导入批次", "POST", "sense:provisioning:import"},
|
|
{"SenseProvisioningExecute", "执行开通", "POST", "sense:provisioning:execute"},
|
|
{"SenseProvisioningRetry", "重试失败项", "POST", "sense:provisioning:retry"},
|
|
{"SenseProvisioningExport", "导出结果", "GET", "sense:provisioning:export"},
|
|
}
|
|
buttons := make([]migrationModels.SysMenu, 0, len(definitions))
|
|
for index, definition := range definitions {
|
|
button, err := ensureDeviceMenu(tx, migrationModels.SysMenu{MenuName: definition.name, Title: definition.title, MenuType: "F", Action: definition.action, Permission: definition.permission, ParentId: page.MenuId, Paths: fmt.Sprintf("/0/%d/%d", root.MenuId, page.MenuId), Sort: index + 1, Visible: "1", IsFrame: "1"})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
buttons = append(buttons, button)
|
|
}
|
|
allMenus := append([]migrationModels.SysMenu{page}, buttons...)
|
|
for _, role := range []string{"implementation_operator", "site_admin"} {
|
|
if err = attachDeviceRole(tx, role, allMenus); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
if err = attachDeviceRole(tx, "viewer", []migrationModels.SysMenu{page, buttons[3]}); err != nil {
|
|
return err
|
|
}
|
|
read := [][2]string{{"/api/v1/provisioning/batches", "GET"}, {"/api/v1/provisioning/batches/:id", "GET"}, {"/api/v1/provisioning/batches/:id/export", "GET"}}
|
|
write := [][2]string{{"/api/v1/provisioning/batches", "POST"}, {"/api/v1/provisioning/batches/:id/execute", "POST"}, {"/api/v1/provisioning/batches/:id/retry-failed", "POST"}, {"/api/v1/provisioning/batches/:id/items/:itemId/retry", "POST"}}
|
|
for _, role := range []string{"implementation_operator", "site_admin", "viewer"} {
|
|
policies := append([][2]string{}, read...)
|
|
if role != "viewer" {
|
|
policies = append(policies, write...)
|
|
}
|
|
for _, policy := range policies {
|
|
if err = tx.Clauses(clause.OnConflict{DoNothing: true}).Create(&deviceCasbinRule{Ptype: "p", V0: role, V1: policy[0], V2: policy[1]}).Error; err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
if err = rebuildSenseMenuPaths(tx, root.MenuId, "/0"); err != nil {
|
|
return err
|
|
}
|
|
return tx.Create(&common.Migration{Version: version}).Error
|
|
})
|
|
}
|