chore: reorganize go code

This commit is contained in:
Aarnav Tale 2025-06-08 08:59:33 -04:00
parent 7a6ad8d2d5
commit 0f9bf73b82
No known key found for this signature in database
28 changed files with 796 additions and 220 deletions

37
cmd/hp_agent/hp_agent.go Normal file
View file

@ -0,0 +1,37 @@
package main
import (
_ "github.com/joho/godotenv/autoload"
"github.com/tale/headplane/agent/internal/config"
"github.com/tale/headplane/agent/internal/hpagent"
"github.com/tale/headplane/agent/internal/sshutil"
"github.com/tale/headplane/agent/internal/tsnet"
"github.com/tale/headplane/agent/internal/util"
)
type Register struct {
Type string
ID string
}
func main() {
log := util.GetLogger()
cfg, err := config.Load()
if err != nil {
log.Fatal("Failed to load config: %s", err)
}
log.SetDebug(cfg.Debug)
agent := tsnet.NewAgent(cfg)
agent.Connect()
defer agent.Shutdown()
log.Msg(&Register{
Type: "register",
ID: agent.ID,
})
sshutil.DispatchSSHStdin()
hpagent.FollowMaster(agent)
}