domrichardson
40 lines
888 B
Go
40 lines
888 B
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"log"
|
|
"os/signal"
|
|
"syscall"
|
|
|
|
"github.com/mrhid6/keymanager/agent/internal/config"
|
|
agentsync "github.com/mrhid6/keymanager/agent/internal/sync"
|
|
)
|
|
|
|
var Version = "dev"
|
|
|
|
func main() {
|
|
genKey := flag.String("generate-key", "", "Generate SSH keypair and upload with this label")
|
|
flag.Parse()
|
|
|
|
cfg, err := config.Load()
|
|
if err != nil {
|
|
log.Fatalf("failed to load config: %v", err)
|
|
}
|
|
|
|
if *genKey != "" {
|
|
if err := agentsync.GenerateAndUpload(cfg, *genKey); err != nil {
|
|
log.Fatalf("key generation failed: %v", err)
|
|
}
|
|
return
|
|
}
|
|
|
|
ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
|
|
defer stop()
|
|
|
|
log.Printf("keymanager-agent %s starting (server=%s, poll=%s)", Version, cfg.ServerURL, cfg.PollInterval)
|
|
if err := agentsync.Run(ctx, cfg); err != nil {
|
|
log.Fatalf("agent error: %v", err)
|
|
}
|
|
}
|