60 lines
1.3 KiB
Go
60 lines
1.3 KiB
Go
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"bytes"
|
|
"fmt"
|
|
"os"
|
|
"os/exec"
|
|
"strings"
|
|
)
|
|
|
|
func main() {
|
|
s := bufio.NewScanner(os.Stdin)
|
|
if len(os.Args) != 1+1 {
|
|
fmt.Fprintf(os.Stderr, "not enough arguments\nusage: %s [ssh user]\n", os.Args[0])
|
|
return
|
|
}
|
|
user := os.Args[1]
|
|
|
|
fmt.Printf("[recorders]\n")
|
|
|
|
for s.Scan() {
|
|
host := s.Text()
|
|
if strings.HasPrefix(host, "#") {
|
|
// Is a comment
|
|
continue
|
|
}
|
|
if strings.HasSuffix(host, ":0") {
|
|
// Is a "system" host
|
|
continue
|
|
}
|
|
hostname, roomId, roomType, err := discover(host, user)
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "host %s discovery failed: %v\n", host, err)
|
|
}
|
|
fmt.Printf("%s ansible_host=%s ansible_user=%s room_id=%s room_type=%s\n", hostname, host, user, roomId, roomType)
|
|
}
|
|
}
|
|
|
|
func discover(host, user string) (hostname, roomId, roomType string, err error) {
|
|
cmd := exec.Command("/usr/bin/ssh", "-l", user, host, "sh", "-c", "hostname; cat room_type; cat room_id; exit 0")
|
|
fmt.Fprintf(os.Stderr, "command: %s\n", cmd)
|
|
var out bytes.Buffer
|
|
cmd.Stdout = &out
|
|
err = cmd.Run()
|
|
if err != nil {
|
|
return
|
|
}
|
|
hostname, err = out.ReadString('\n')
|
|
if err != nil {
|
|
return
|
|
}
|
|
hostname = strings.TrimSpace(hostname)
|
|
roomType, _ = out.ReadString('\n')
|
|
roomType = strings.TrimSpace(roomType)
|
|
roomId, _ = out.ReadString('\n')
|
|
roomId = strings.TrimSpace(roomId)
|
|
return
|
|
}
|