61 lines
1.4 KiB
Go
61 lines
1.4 KiB
Go
package seeds
|
|
|
|
import (
|
|
"encoding/json"
|
|
"io"
|
|
"os"
|
|
|
|
"github.com/Caknoooo/go-gin-clean-starter/database/entities"
|
|
"github.com/sirupsen/logrus"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type RoleMenuSeed struct {
|
|
RoleName string `json:"role_name"`
|
|
MenuName string `json:"menu_name"`
|
|
}
|
|
|
|
func ListRoleMenuSeeder(db *gorm.DB) error {
|
|
jsonFile, err := os.Open("./database/seeders/json/role_menus.json")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
jsonData, err := io.ReadAll(jsonFile)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var listRoleMenu []RoleMenuSeed
|
|
if err := json.Unmarshal(jsonData, &listRoleMenu); err != nil {
|
|
return err
|
|
}
|
|
hasTable := db.Migrator().HasTable(&entities.M_Role_Menu{})
|
|
if !hasTable {
|
|
if err := db.Migrator().CreateTable(&entities.M_Role_Menu{}); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
for _, data := range listRoleMenu {
|
|
var role entities.M_Role
|
|
var menu entities.M_Menu
|
|
|
|
if err := db.Where("name = ?", data.RoleName).First(&role).Error; err != nil {
|
|
logrus.Error("Role not found: ", err)
|
|
return err // role tidak ditemukan
|
|
}
|
|
if err := db.Where("name = ?", data.MenuName).First(&menu).Error; err != nil {
|
|
logrus.Error("Menu not found: ", err)
|
|
return err // menu tidak ditemukan
|
|
}
|
|
var roleMenu entities.M_Role_Menu
|
|
if err := db.FirstOrCreate(&roleMenu, entities.M_Role_Menu{
|
|
RoleID: role.ID,
|
|
MenuID: menu.ID,
|
|
}).Error; err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|