57 lines
1.3 KiB
Go
57 lines
1.3 KiB
Go
package poller
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"golang.org/x/crypto/bcrypt"
|
|
"golang.org/x/term"
|
|
)
|
|
|
|
// PrintRawMetrics prints raw json from the UniFi Controller. This is currently
|
|
// tied into the -j CLI arg, and is probably not very useful outside that context.
|
|
func (u *UnifiPoller) PrintRawMetrics() (err error) {
|
|
split := strings.SplitN(u.Flags.DumpJSON, " ", 2)
|
|
filter := &Filter{Kind: split[0]}
|
|
|
|
// Allows you to grab a controller other than 0 from config.
|
|
if split2 := strings.Split(filter.Kind, ":"); len(split2) > 1 {
|
|
filter.Kind = split2[0]
|
|
filter.Unit, _ = strconv.Atoi(split2[1])
|
|
}
|
|
|
|
// Used with "other"
|
|
if len(split) > 1 {
|
|
filter.Path = split[1]
|
|
}
|
|
|
|
// As of now we only have one input plugin, so target that [0].
|
|
m, err := inputs[0].RawMetrics(filter)
|
|
fmt.Println(string(m))
|
|
|
|
return err
|
|
}
|
|
|
|
// PrintPasswordHash prints a bcrypt'd password. Useful for the web server.
|
|
func (u *UnifiPoller) PrintPasswordHash() (err error) {
|
|
pwd := []byte(u.Flags.HashPW)
|
|
|
|
if u.Flags.HashPW == "-" {
|
|
fmt.Print("Enter Password: ")
|
|
|
|
pwd, err = term.ReadPassword(int(os.Stdin.Fd()))
|
|
if err != nil {
|
|
return fmt.Errorf("reading stdin: %w", err)
|
|
}
|
|
|
|
fmt.Println() // print a newline.
|
|
}
|
|
|
|
hash, err := bcrypt.GenerateFromPassword(pwd, bcrypt.MinCost)
|
|
fmt.Println(string(hash))
|
|
|
|
return err //nolint:wrapcheck
|
|
}
|