fankes/komari-agent
2
0
Fork 0
mirror of https://github.com/fankes/komari-agent.git synced 2025-10-18 18:49:23 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: 添加自动发现功能及相关配置

Browse Source
This commit is contained in:
Akizon77
2025-08-09 04:50:59 +00:00
parent 655a60ec98
commit 10f2548868
5 changed files with 200 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
cmd/root.go
View File

@@ -20,6 +20,14 @@ var RootCmd = &cobra.Command{
Run: func(cmd *cobra.Command, args []string) {
log.Println("Komari Agent", update.CurrentVersion)
log.Println("Github Repo:", update.Repo)
// Auto discovery
if flags.AutoDiscoveryKey != "" {
err := handleAutoDiscovery()
if err != nil {
log.Printf("Auto-discovery failed: %v", err)
os.Exit(1)
}
}
diskList, err := monitoring.DiskList()
if err != nil {
log.Println("Failed to get disk list:", err)
@@ -68,9 +76,10 @@ func Execute() {
func init() {
RootCmd.PersistentFlags().StringVarP(&flags.Token, "token", "t", "", "API token")
RootCmd.MarkPersistentFlagRequired("token")
//RootCmd.MarkPersistentFlagRequired("token")
RootCmd.PersistentFlags().StringVarP(&flags.Endpoint, "endpoint", "e", "", "API endpoint")
RootCmd.MarkPersistentFlagRequired("endpoint")
RootCmd.PersistentFlags().StringVar(&flags.AutoDiscoveryKey, "auto-discovery", "", "Auto discovery key for the agent")
RootCmd.PersistentFlags().BoolVar(&flags.DisableAutoUpdate, "disable-auto-update", false, "Disable automatic updates")
RootCmd.PersistentFlags().BoolVar(&flags.DisableWebSsh, "disable-web-ssh", false, "Disable remote control(web ssh and rce)")
RootCmd.PersistentFlags().BoolVar(&flags.MemoryModeAvailable, "memory-mode-available", false, "Report memory as available instead of used.")