150 lines
3.2 KiB
Go
150 lines
3.2 KiB
Go
// Copyright © 2022 Roberto Hidalgo <joao@un.rob.mx>
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
package config
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"fmt"
|
|
|
|
op "github.com/1Password/connect-sdk-go/onepassword"
|
|
"gopkg.in/yaml.v3"
|
|
)
|
|
|
|
type Config struct {
|
|
Path string
|
|
Vault string
|
|
Name string
|
|
Tree *Entry
|
|
}
|
|
|
|
var redactOutput = false
|
|
|
|
func (cfg *Config) ToMap() map[string]interface{} {
|
|
ret := map[string]interface{}{}
|
|
for _, child := range cfg.Tree.Children {
|
|
ret[child.Key] = child.AsMap()
|
|
}
|
|
return ret
|
|
}
|
|
|
|
func (cfg *Config) ToOP() *op.Item {
|
|
annotationsSection := &op.ItemSection{
|
|
ID: "~annotations",
|
|
Label: "~annotations",
|
|
}
|
|
sections := []*op.ItemSection{annotationsSection}
|
|
fields := []*op.ItemField{
|
|
{
|
|
ID: "password",
|
|
Type: "CONCEALED",
|
|
Purpose: "PASSWORD",
|
|
Label: "password",
|
|
Value: "hash",
|
|
}, {
|
|
ID: "notesPlain",
|
|
Type: "STRING",
|
|
Purpose: "NOTES",
|
|
Label: "notesPlain",
|
|
Value: "flushed by joao",
|
|
},
|
|
}
|
|
|
|
for key, leaf := range cfg.Tree.Children {
|
|
if len(leaf.Children) == 0 {
|
|
fields = append(fields, leaf.ToOP(annotationsSection)...)
|
|
continue
|
|
}
|
|
|
|
if !leaf.isSequence {
|
|
sections = append(sections, &op.ItemSection{
|
|
ID: key,
|
|
Label: key,
|
|
})
|
|
} else {
|
|
fmt.Printf("Found sequence for %s", leaf.Key)
|
|
}
|
|
|
|
for _, child := range leaf.Children {
|
|
fields = append(fields, child.ToOP(annotationsSection)...)
|
|
}
|
|
}
|
|
|
|
return &op.Item{
|
|
Title: cfg.Name,
|
|
Sections: sections,
|
|
Vault: op.ItemVault{ID: "nidito-admin"},
|
|
Category: op.Password,
|
|
Fields: fields,
|
|
}
|
|
}
|
|
|
|
func ConfigFromYAML(data []byte) (*Config, error) {
|
|
cfg := &Config{
|
|
Vault: "vault",
|
|
Name: "title",
|
|
Tree: NewEntry("root"),
|
|
}
|
|
|
|
yaml.Unmarshal(data, cfg.Tree.Children)
|
|
|
|
for k, leaf := range cfg.Tree.Children {
|
|
leaf.SetKey(k, []string{})
|
|
}
|
|
|
|
return cfg, nil
|
|
}
|
|
|
|
func ConfigFromOP(item *op.Item) (*Config, error) {
|
|
cfg := &Config{
|
|
Vault: item.Vault.ID,
|
|
Name: item.Title,
|
|
Tree: NewEntry("root"),
|
|
}
|
|
|
|
err := cfg.Tree.FromOP(item.Fields)
|
|
return cfg, err
|
|
}
|
|
|
|
func (cfg *Config) MarshalYAML() (interface{}, error) {
|
|
return cfg.Tree.MarshalYAML()
|
|
}
|
|
|
|
func (cfg *Config) AsYAML(redacted bool) ([]byte, error) {
|
|
redactOutput = redacted
|
|
var out bytes.Buffer
|
|
enc := yaml.NewEncoder(&out)
|
|
enc.SetIndent(2)
|
|
err := enc.Encode(cfg)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("could not serialize config as yaml: %w", err)
|
|
}
|
|
return out.Bytes(), nil
|
|
}
|
|
|
|
func (cfg *Config) AsJSON(redacted bool, item bool) ([]byte, error) {
|
|
var repr interface{}
|
|
if item {
|
|
repr = cfg.ToOP()
|
|
} else {
|
|
redactOutput = redacted
|
|
repr = cfg.ToMap()
|
|
}
|
|
|
|
bytes, err := json.Marshal(repr)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("could not serialize config as json: %w", err)
|
|
}
|
|
return bytes, nil
|
|
}
|