145 lines
3.2 KiB
Go
145 lines
3.2 KiB
Go
package virtbuf
|
|
|
|
// functions to import and export the protobuf
|
|
// data to and from config files
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"google.golang.org/protobuf/encoding/protojson"
|
|
"google.golang.org/protobuf/encoding/prototext"
|
|
)
|
|
|
|
func WriteConfig(d *Droplets, h *Hypervisors, e *Events) bool {
|
|
if !d.WriteConfigJSON() {
|
|
return false
|
|
}
|
|
if !d.WriteConfigTEXT() {
|
|
return false
|
|
}
|
|
|
|
if e.WriteConfigJSON() {
|
|
return false
|
|
}
|
|
if e.WriteConfigTEXT() {
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
|
|
// export as json
|
|
func (e *Events) WriteConfigJSON() bool {
|
|
fullname := filepath.Join(os.Getenv("VIRTIGO_HOME"), "events.json")
|
|
cfgfile, err := os.OpenFile(fullname, os.O_RDWR|os.O_CREATE, 0666)
|
|
defer cfgfile.Close()
|
|
if err != nil {
|
|
fmt.Println("open config file :", err)
|
|
return false
|
|
}
|
|
text := e.FormatJSON()
|
|
fmt.Fprintln(cfgfile, text)
|
|
fmt.Println("Write:", fullname, "OK")
|
|
return true
|
|
}
|
|
|
|
// export as prototext
|
|
func (e *Events) WriteConfigTEXT() bool {
|
|
fullname := filepath.Join(os.Getenv("VIRTIGO_HOME"), "events.text")
|
|
cfgfile, err := os.OpenFile(fullname, os.O_RDWR|os.O_CREATE, 0666)
|
|
defer cfgfile.Close()
|
|
if err != nil {
|
|
fmt.Println("open config file :", err)
|
|
return false
|
|
}
|
|
text := e.FormatTEXT()
|
|
fmt.Fprintln(cfgfile, text)
|
|
fmt.Println("Write:", fullname, "OK")
|
|
return true
|
|
}
|
|
|
|
// export as json
|
|
func (d *Droplets) WriteConfigJSON() bool {
|
|
fullname := filepath.Join(os.Getenv("VIRTIGO_HOME"), "droplets.json")
|
|
cfgfile, err := os.OpenFile(fullname, os.O_RDWR|os.O_CREATE, 0666)
|
|
defer cfgfile.Close()
|
|
if err != nil {
|
|
fmt.Println("open config file :", err)
|
|
return false
|
|
}
|
|
text := d.FormatJSON()
|
|
fmt.Fprintln(cfgfile, text)
|
|
fmt.Println("Write:", fullname, "OK")
|
|
return true
|
|
}
|
|
|
|
// export as prototext
|
|
func (d *Droplets) WriteConfigTEXT() bool {
|
|
fullname := filepath.Join(os.Getenv("VIRTIGO_HOME"), "droplets.text")
|
|
cfgfile, err := os.OpenFile(fullname, os.O_RDWR|os.O_CREATE, 0666)
|
|
defer cfgfile.Close()
|
|
if err != nil {
|
|
fmt.Println("open config file :", err)
|
|
return false
|
|
}
|
|
text := d.FormatTEXT()
|
|
fmt.Fprintln(cfgfile, text)
|
|
fmt.Println("Write:", fullname, "OK")
|
|
return true
|
|
}
|
|
|
|
// human readable JSON
|
|
func (c *Cluster) FormatJSON() string {
|
|
return protojson.Format(c)
|
|
}
|
|
|
|
func (d *Droplets) FormatJSON() string {
|
|
return protojson.Format(d)
|
|
}
|
|
|
|
func (e *Events) FormatJSON() string {
|
|
return protojson.Format(e)
|
|
}
|
|
|
|
// apparently this isn't supposed to be used?
|
|
// https://protobuf.dev/reference/go/faq/#unstable-text
|
|
// this is a shame because this is much nicer output than JSON Format()
|
|
func (c *Cluster) FormatTEXT() string {
|
|
return prototext.Format(c)
|
|
}
|
|
|
|
func (d *Droplets) FormatTEXT() string {
|
|
return prototext.Format(d)
|
|
}
|
|
|
|
func (e *Events) FormatTEXT() string {
|
|
return prototext.Format(e)
|
|
}
|
|
|
|
// marshal
|
|
func (c *Cluster) MarshalJSON() ([]byte, error) {
|
|
return protojson.Marshal(c)
|
|
}
|
|
|
|
func (d *Droplets) MarshalJSON() ([]byte, error) {
|
|
return protojson.Marshal(d)
|
|
}
|
|
|
|
func (e *Events) MarshalJSON() ([]byte, error) {
|
|
return protojson.Marshal(e)
|
|
}
|
|
|
|
// unmarshal
|
|
func (c *Cluster) UnmarshalJSON(data []byte) error {
|
|
return protojson.Unmarshal(data, c)
|
|
}
|
|
|
|
func (d *Droplets) UnmarshalJSON(data []byte) error {
|
|
return protojson.Unmarshal(data, d)
|
|
}
|
|
|
|
func (e *Events) UnmarshalJSON(data []byte) error {
|
|
return protojson.Unmarshal(data, e)
|
|
}
|