syonad/two
2
0
Fork 0
Code Issues 6 Pull requests 1 Projects Releases 9 Packages Wiki Activity Actions

f-25: add load files

Browse source 
Signed-off-by: GnomeZworc <nicolas.boufidjeline@g3e.fr>
This commit is contained in:
GnomeZworc 2026-04-29 00:41:43 +02:00
commit f5707c343c
Signed by: nicolas.boufideline
GPG key ID: 4406BBBF8845D632
4 changed files with 25 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

14
internal/metadata/handle.go
View file

@ -3,18 +3,18 @@ package metadata
import ( import (
"fmt" "fmt"
configuration "git.g3e.fr/syonad/two/internal/config/agent"
"git.g3e.fr/syonad/two/pkg/systemd" "git.g3e.fr/syonad/two/pkg/systemd"
"github.com/dgraph-io/badger/v4"
) )
func StartMetadata(config NoCloudConfig, db *badger.DB, dryrun bool) error { func StartMetadata(config NoCloudConfig, cfg *configuration.Config, dryrun bool) error {
service, err := systemd.New() service, err := systemd.New()
if err != nil { if err != nil {
return fmt.Errorf("failed to connect to systemd: %w", err) return fmt.Errorf("failed to connect to systemd: %w", err)
} }
defer service.Close() defer service.Close()
LoadNcCloudInDB(config, db) LoadNcCloudInDB(config, cfg.Metadata.RunDir)
if !dryrun { if !dryrun {
if err := service.Start("metadata@" + config.Name + ".service"); err != nil { if err := service.Start("metadata@" + config.Name + ".service"); err != nil {
return fmt.Errorf("failed to start metadata@%s: %w", config.Name, err) return fmt.Errorf("failed to start metadata@%s: %w", config.Name, err)
@ -23,17 +23,17 @@ func StartMetadata(config NoCloudConfig, db *badger.DB, dryrun bool) error {
return nil return nil
} }
func StopMetadata(vm_name string, db *badger.DB, dryrun bool) error { func StopMetadata(vmName string, cfg *configuration.Config, dryrun bool) error {
service, err := systemd.New() service, err := systemd.New()
if err != nil { if err != nil {
return fmt.Errorf("failed to connect to systemd: %w", err) return fmt.Errorf("failed to connect to systemd: %w", err)
} }
defer service.Close() defer service.Close()
UnLoadNoCloudInDB(vm_name, db) UnLoadNoCloudInDB(vmName, cfg.Metadata.RunDir)
if !dryrun { if !dryrun {
if err := service.Stop("metadata@" + vm_name + ".service"); err != nil { if err := service.Stop("metadata@" + vmName + ".service"); err != nil {
return fmt.Errorf("failed to stop metadata@%s: %w", vm_name, err) return fmt.Errorf("failed to stop metadata@%s: %w", vmName, err)
} }
} }
return nil return nil

29
internal/metadata/render.go
View file

@ -3,10 +3,9 @@ package metadata
import ( import (
"bytes" "bytes"
"embed" "embed"
"os"
"path/filepath"
"text/template" "text/template"
"git.g3e.fr/syonad/two/pkg/db/kv"
"github.com/dgraph-io/badger/v4"
) )
//go:embed templates/*.tmpl //go:embed templates/*.tmpl
@ -26,21 +25,25 @@ func RenderConfig(path string, cfg NoCloudConfig) (string, error) {
return buf.String(), nil return buf.String(), nil
} }
func LoadNcCloudInDB(config NoCloudConfig, db *badger.DB) { func LoadNcCloudInDB(config NoCloudConfig, runDir string) {
meta_data, _ := RenderConfig("templates/meta-data.tmpl", config) meta_data, _ := RenderConfig("templates/meta-data.tmpl", config)
user_data, _ := RenderConfig("templates/user-data.tmpl", config) user_data, _ := RenderConfig("templates/user-data.tmpl", config)
network_config, _ := RenderConfig("templates/network-config.tmpl", config) network_config, _ := RenderConfig("templates/network-config.tmpl", config)
vendor_data, _ := RenderConfig("templates/vendor-data.tmpl", config) vendor_data, _ := RenderConfig("templates/vendor-data.tmpl", config)
kv.AddInDB(db, "metadata/"+config.Name+"/meta-data", meta_data) dir := filepath.Join(runDir, config.Name)
kv.AddInDB(db, "metadata/"+config.Name+"/user-data", user_data) if err := os.MkdirAll(dir, 0755); err != nil {
kv.AddInDB(db, "metadata/"+config.Name+"/network-config", network_config) return
kv.AddInDB(db, "metadata/"+config.Name+"/vendor-data", vendor_data) }
kv.AddInDB(db, "metadata/"+config.Name+"/vpc", config.VpcName) os.WriteFile(filepath.Join(dir, "meta-data"), []byte(meta_data), 0644)
kv.AddInDB(db, "metadata/"+config.Name+"/bind_ip", config.BindIP) os.WriteFile(filepath.Join(dir, "user-data"), []byte(user_data), 0644)
kv.AddInDB(db, "metadata/"+config.Name+"/bind_port", config.BindPort) os.WriteFile(filepath.Join(dir, "network-config"), []byte(network_config), 0644)
os.WriteFile(filepath.Join(dir, "vendor-data"), []byte(vendor_data), 0644)
os.WriteFile(filepath.Join(dir, "vpc"), []byte(config.VpcName), 0644)
os.WriteFile(filepath.Join(dir, "bind_ip"), []byte(config.BindIP), 0644)
os.WriteFile(filepath.Join(dir, "bind_port"), []byte(config.BindPort), 0644)
} }
func UnLoadNoCloudInDB(vm_name string, db *badger.DB) { func UnLoadNoCloudInDB(vmName string, runDir string) {
kv.DeleteInDB(db, "metadata/"+vm_name) os.RemoveAll(filepath.Join(runDir, vmName))
} }

2
internal/vm/create.go
View file

@ -45,7 +45,7 @@ func StartVM(db *badger.DB, name string, cfg *configuration.Config) error {
BindPort: d.metadataPort, BindPort: d.metadataPort,
Password: d.password, Password: d.password,
SSHKEY: d.sshkey, SSHKEY: d.sshkey,
}, db, false); err != nil { }, cfg, false); err != nil {
return fmt.Errorf("start metadata: %w", err) return fmt.Errorf("start metadata: %w", err)
} }

2
internal/vm/delete.go
View file

@ -57,7 +57,7 @@ func StopVM(db *badger.DB, name string, cfg *configuration.Config) error {
return fmt.Errorf("delete metadata redirect: %w", err) return fmt.Errorf("delete metadata redirect: %w", err)
} }
if err := metadata.StopMetadata(name, db, false); err != nil { if err := metadata.StopMetadata(name, cfg, false); err != nil {
return fmt.Errorf("stop metadata: %w", err) return fmt.Errorf("stop metadata: %w", err)
} }