speatzle/go-passbolt-cli
1
0
Fork 0
mirror of https://github.com/passbolt/go-passbolt-cli.git synced 2025-05-12 02:58:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Initialize the CEL program in a util function

Browse source
This commit is contained in:
PiMaDaum 2023-02-05 22:19:54 +01:00
parent f30590588e
commit 0a542bfdba
5 changed files with 35 additions and 48 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
folder/filter.go
View file

@ -5,6 +5,7 @@ import (
"fmt" "fmt"
"github.com/google/cel-go/cel" "github.com/google/cel-go/cel"
"github.com/passbolt/go-passbolt-cli/util"
"github.com/passbolt/go-passbolt/api" "github.com/passbolt/go-passbolt/api"
) )
@ -23,24 +24,14 @@ func filterFolders(folders *[]api.Folder, celCmd string, ctx context.Context) ([
return *folders, nil return *folders, nil
} }
env, err := cel.NewEnv(celEnvOptions...) program, err := util.InitCELProgram(celCmd, celEnvOptions...)
if err != nil {
return nil, err
}
ast, issue := env.Compile(celCmd)
if issue.Err() != nil {
return nil, issue.Err()
}
program, err := env.Program(ast)
if err != nil { if err != nil {
return nil, err return nil, err
} }
filteredFolders := []api.Folder{} filteredFolders := []api.Folder{}
for _, folder := range *folders { for _, folder := range *folders {
val, _, err := program.ContextEval(ctx, map[string]any{ val, _, err := (*program).ContextEval(ctx, map[string]any{
"ID": folder.ID, "ID": folder.ID,
"FolderParentID": folder.FolderParentID, "FolderParentID": folder.FolderParentID,
"Name": folder.Name, "Name": folder.Name,

15
group/filter.go
View file

@ -5,6 +5,7 @@ import (
"fmt" "fmt"
"github.com/google/cel-go/cel" "github.com/google/cel-go/cel"
"github.com/passbolt/go-passbolt-cli/util"
"github.com/passbolt/go-passbolt/api" "github.com/passbolt/go-passbolt/api"
) )
@ -22,24 +23,14 @@ func filterGroups(groups *[]api.Group, celCmd string, ctx context.Context) ([]ap
return *groups, nil return *groups, nil
} }
env, err := cel.NewEnv(celEnvOptions...) program, err := util.InitCELProgram(celCmd, celEnvOptions...)
if err != nil {
return nil, err
}
ast, issue := env.Compile(celCmd)
if issue.Err() != nil {
return nil, issue.Err()
}
program, err := env.Program(ast)
if err != nil { if err != nil {
return nil, err return nil, err
} }
filteredGroups := []api.Group{} filteredGroups := []api.Group{}
for _, group := range *groups { for _, group := range *groups {
val, _, err := program.ContextEval(ctx, map[string]any{ val, _, err := (*program).ContextEval(ctx, map[string]any{
"ID": group.ID, "ID": group.ID,
"Name": group.Name, "Name": group.Name,
"CreatedTimestamp": group.Created.Time, "CreatedTimestamp": group.Created.Time,

15
resource/filter.go
View file

@ -7,6 +7,7 @@ import (
"github.com/google/cel-go/cel" "github.com/google/cel-go/cel"
"github.com/google/cel-go/common/types" "github.com/google/cel-go/common/types"
"github.com/google/cel-go/common/types/ref" "github.com/google/cel-go/common/types/ref"
"github.com/passbolt/go-passbolt-cli/util"
"github.com/passbolt/go-passbolt/api" "github.com/passbolt/go-passbolt/api"
"github.com/passbolt/go-passbolt/helper" "github.com/passbolt/go-passbolt/helper"
) )
@ -30,24 +31,14 @@ func filterResources(resources *[]api.Resource, celCmd string, ctx context.Conte
return *resources, nil return *resources, nil
} }
env, err := cel.NewEnv(celEnvOptions...) program, err := util.InitCELProgram(celCmd, celEnvOptions...)
if err != nil {
return nil, err
}
ast, issue := env.Compile(celCmd)
if issue.Err() != nil {
return nil, issue.Err()
}
program, err := env.Program(ast)
if err != nil { if err != nil {
return nil, err return nil, err
} }
filteredResources := []api.Resource{} filteredResources := []api.Resource{}
for _, resource := range *resources { for _, resource := range *resources {
val, _, err := program.ContextEval(ctx, map[string]any{ val, _, err := (*program).ContextEval(ctx, map[string]any{
"Id": resource.ID, "Id": resource.ID,
"FolderParentID": resource.FolderParentID, "FolderParentID": resource.FolderParentID,
"Name": resource.Name, "Name": resource.Name,

15
user/filter.go
View file

@ -5,6 +5,7 @@ import (
"fmt" "fmt"
"github.com/google/cel-go/cel" "github.com/google/cel-go/cel"
"github.com/passbolt/go-passbolt-cli/util"
"github.com/passbolt/go-passbolt/api" "github.com/passbolt/go-passbolt/api"
) )
@ -25,24 +26,14 @@ func filterUsers(users *[]api.User, celCmd string, ctx context.Context) ([]api.U
return *users, nil return *users, nil
} }
env, err := cel.NewEnv(celEnvOptions...) program, err := util.InitCELProgram(celCmd, celEnvOptions...)
if err != nil {
return nil, err
}
ast, issue := env.Compile(celCmd)
if issue.Err() != nil {
return nil, issue.Err()
}
program, err := env.Program(ast)
if err != nil { if err != nil {
return nil, err return nil, err
} }
filteredUsers := []api.User{} filteredUsers := []api.User{}
for _, user := range *users { for _, user := range *users {
val, _, err := program.ContextEval(ctx, map[string]any{ val, _, err := (*program).ContextEval(ctx, map[string]any{
"ID": user.ID, "ID": user.ID,
"Username": user.Username, "Username": user.Username,
"FirstName": user.Profile.FirstName, "FirstName": user.Profile.FirstName,

23
util/cel.go Normal file
View file

@ -0,0 +1,23 @@
package util
import "github.com/google/cel-go/cel"
// InitCELProgram - Initialize a CEL program with given CEL command and a set of environments
func InitCELProgram(celCmd string, options ...cel.EnvOption) (*cel.Program, error) {
env, err := cel.NewEnv(options...)
if err != nil {
return nil, err
}
ast, issue := env.Compile(celCmd)
if issue.Err() != nil {
return nil, issue.Err()
}
program, err := env.Program(ast)
if err != nil {
return nil, err
}
return &program, nil
}