mirror of
https://github.com/paralus/paralus.git
synced 2026-08-21 12:26:24 +00:00
Abin Simon
8184c23a8f
We should probably have the role creation moved into adminsrv, but since we are only creating new orgs like this as of now let us continue with this.
65 lines
1.7 KiB
Go
65 lines
1.7 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"log"
|
|
"path"
|
|
|
|
"github.com/RafaySystems/rcloud-base/internal/models"
|
|
"github.com/RafaySystems/rcloud-base/internal/persistence/provider/pg"
|
|
"github.com/uptrace/bun"
|
|
"github.com/uptrace/bun/dialect/pgdialect"
|
|
"github.com/uptrace/bun/driver/pgdriver"
|
|
)
|
|
|
|
func addResourcePermissions(dao pg.EntityDAO, basePath string) error {
|
|
var items []models.ResourcePermission
|
|
|
|
files, err := ioutil.ReadDir(basePath)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
for _, file := range files {
|
|
if !file.IsDir() { // probably not, but just in case
|
|
content, err := ioutil.ReadFile(path.Join(basePath, file.Name()))
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
// It has ResourceRefId, but that does not seem to be used in the old implementatino
|
|
// Also, why do we need two items?
|
|
var data models.ResourcePermission
|
|
err = json.Unmarshal(content, &data)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
items = append(items, data)
|
|
}
|
|
}
|
|
|
|
fmt.Println("Adding", len(items), "resource permissions")
|
|
_, err = dao.Create(context.Background(), &items)
|
|
return err
|
|
}
|
|
|
|
func main() {
|
|
dsn := "postgres://admindbuser:admindbpassword@localhost:5432/admindb?sslmode=disable"
|
|
sqldb := sql.OpenDB(pgdriver.NewConnector(pgdriver.WithDSN(dsn)))
|
|
db := bun.NewDB(sqldb, pgdialect.New())
|
|
dao := pg.NewEntityDAO(db)
|
|
|
|
// TODO: add option to update the existing list
|
|
err := dao.DeleteAll(context.Background(), &models.ResourcePermission{})
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
err = addResourcePermissions(dao, path.Join("scripts", "resourcepermissions", "data"))
|
|
if err != nil {
|
|
fmt.Println("Run from base directory")
|
|
log.Fatal(err)
|
|
}
|
|
}
|