-
Notifications
You must be signed in to change notification settings - Fork 6
/
import.go
73 lines (61 loc) · 1.43 KB
/
import.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
/*
Copyright 2021 Adevinta
*/
package cmd
import (
"github.com/spf13/cobra"
"github.com/adevinta/vulcan-api/cmd/vulcan-cli/cli"
)
var ImportCmd = &cobra.Command{
Use: `import <vulcanito_assets_dir>`,
Short: `Import assets in Vulcanito format into vulcan-api`,
Args: cobra.ExactArgs(1),
RunE: func(cmd *cobra.Command, args []string) error {
return runImport(args, apiClient)
},
}
func init() {
rootCmd.AddCommand(ImportCmd)
}
func runImport(args []string, apiClient *cli.CLI) error {
teams, err := cli.ReadVulcanitoTeams(args[0])
if err != nil {
return err
}
for _, t := range teams {
id, err := apiClient.CreateTeam(t)
if err != nil {
return err
}
if err := apiClient.AddRecipients(id, t.Recipients); err != nil { // nolint
return err
}
defaultGroupID, err := apiClient.CreateGroup(id, "Default")
if err != nil {
return err
}
sensitiveGroupID, err := apiClient.CreateGroup(id, "Sensitive")
if err != nil {
return err
}
for _, ac := range t.Collections {
for _, asset := range ac.Assets {
createdAssets, err := apiClient.CreateAsset(id, asset.Target, ac.AssetType, asset.Rolfp, asset.Alias)
if err != nil {
return err
}
group := defaultGroupID
if asset.Sensitive {
group = sensitiveGroupID
}
for _, a := range createdAssets {
err = apiClient.AssociateAsset(id, group, a.ID)
}
if err != nil {
return err
}
}
}
}
return nil
}