cscli: extract package 'crowdsec-cli/clientinfo'

cmd/crowdsec-cli/bouncers.go

@@ -17,6 +17,7 @@ import (
 	"github.com/spf13/cobra"
 
 	"github.com/crowdsecurity/crowdsec/cmd/crowdsec-cli/ask"
+	"github.com/crowdsecurity/crowdsec/cmd/crowdsec-cli/clientinfo"
 	"github.com/crowdsecurity/crowdsec/cmd/crowdsec-cli/cstable"
 	"github.com/crowdsecurity/crowdsec/cmd/crowdsec-cli/require"
 	middlewares "github.com/crowdsecurity/crowdsec/pkg/apiserver/middlewares/v1"
@@ -27,40 +28,6 @@ import (
 	"github.com/crowdsecurity/crowdsec/pkg/types"
 )
 
-type featureflagProvider interface {
-	GetFeatureflags() string
-}
-
-type osProvider interface {
-	GetOsname() string
-	GetOsversion() string
-}
-
-func getOSNameAndVersion(o osProvider) string {
-	ret := o.GetOsname()
-	if o.GetOsversion() != "" {
-		if ret != "" {
-			ret += "/"
-		}
-
-		ret += o.GetOsversion()
-	}
-
-	if ret == "" {
-		return "?"
-	}
-
-	return ret
-}
-
-func getFeatureFlagList(o featureflagProvider) []string {
-	if o.GetFeatureflags() == "" {
-		return nil
-	}
-
-	return strings.Split(o.GetFeatureflags(), ",")
-}
-
 type cliBouncers struct {
 	db  *database.Client
 	cfg configGetter
@@ -156,8 +123,8 @@ func newBouncerInfo(b *ent.Bouncer) bouncerInfo {
 		Version:      b.Version,
 		LastPull:     b.LastPull,
 		AuthType:     b.AuthType,
-		OS:           getOSNameAndVersion(b),
-		Featureflags: getFeatureFlagList(b),
+		OS:           clientinfo.GetOSNameAndVersion(b),
+		Featureflags: clientinfo.GetFeatureFlagList(b),
 	}
 }
 
@@ -463,10 +430,10 @@ func (cli *cliBouncers) inspectHuman(out io.Writer, bouncer *ent.Bouncer) {
 		{"Version", bouncer.Version},
 		{"Last Pull", lastPull},
 		{"Auth type", bouncer.AuthType},
-		{"OS", getOSNameAndVersion(bouncer)},
+		{"OS", clientinfo.GetOSNameAndVersion(bouncer)},
 	})
 
-	for _, ff := range getFeatureFlagList(bouncer) {
+	for _, ff := range clientinfo.GetFeatureFlagList(bouncer) {
 		t.AppendRow(table.Row{"Feature Flags", ff})
 	}
 

cmd/crowdsec-cli/clientinfo/clientinfo.go

@@ -0,0 +1,39 @@
+package clientinfo
+
+import (
+	"strings"
+)
+
+type featureflagProvider interface {
+	GetFeatureflags() string
+}
+
+type osProvider interface {
+	GetOsname() string
+	GetOsversion() string
+}
+
+func GetOSNameAndVersion(o osProvider) string {
+	ret := o.GetOsname()
+	if o.GetOsversion() != "" {
+		if ret != "" {
+			ret += "/"
+		}
+
+		ret += o.GetOsversion()
+	}
+
+	if ret == "" {
+		return "?"
+	}
+
+	return ret
+}
+
+func GetFeatureFlagList(o featureflagProvider) []string {
+	if o.GetFeatureflags() == "" {
+		return nil
+	}
+
+	return strings.Split(o.GetFeatureflags(), ",")
+}

cmd/crowdsec-cli/machines.go

@@ -20,6 +20,7 @@ import (
 	"gopkg.in/yaml.v3"
 
 	"github.com/crowdsecurity/crowdsec/cmd/crowdsec-cli/ask"
+	"github.com/crowdsecurity/crowdsec/cmd/crowdsec-cli/clientinfo"
 	"github.com/crowdsecurity/crowdsec/cmd/crowdsec-cli/cstable"
 	"github.com/crowdsecurity/crowdsec/cmd/crowdsec-cli/idgen"
 	"github.com/crowdsecurity/crowdsec/cmd/crowdsec-cli/require"
@@ -139,7 +140,7 @@ func (cli *cliMachines) listHuman(out io.Writer, machines ent.Machines) {
 			hb = emoji.Warning + " " + hb
 		}
 
-		t.AppendRow(table.Row{m.MachineId, m.IpAddress, m.UpdatedAt.Format(time.RFC3339), validated, m.Version, getOSNameAndVersion(m), m.AuthType, hb})
+		t.AppendRow(table.Row{m.MachineId, m.IpAddress, m.UpdatedAt.Format(time.RFC3339), validated, m.Version, clientinfo.GetOSNameAndVersion(m), m.AuthType, hb})
 	}
 
 	io.WriteString(out, t.Render() + "\n")
@@ -172,8 +173,8 @@ func newMachineInfo(m *ent.Machine) machineInfo {
 		Version:       m.Version,
 		IsValidated:   m.IsValidated,
 		AuthType:      m.AuthType,
-		OS:            getOSNameAndVersion(m),
-		Featureflags:  getFeatureFlagList(m),
+		OS:            clientinfo.GetOSNameAndVersion(m),
+		Featureflags:  clientinfo.GetFeatureFlagList(m),
 		Datasources:   m.Datasources,
 	}
 }
@@ -589,15 +590,15 @@ func (cli *cliMachines) inspectHuman(out io.Writer, machine *ent.Machine) {
 		{"Last Heartbeat", machine.LastHeartbeat},
 		{"Validated?", machine.IsValidated},
 		{"CrowdSec version", machine.Version},
-		{"OS", getOSNameAndVersion(machine)},
+		{"OS", clientinfo.GetOSNameAndVersion(machine)},
 		{"Auth type", machine.AuthType},
 	})
 
 	for dsName, dsCount := range machine.Datasources {
 		t.AppendRow(table.Row{"Datasources", fmt.Sprintf("%s: %d", dsName, dsCount)})
 	}
 
-	for _, ff := range getFeatureFlagList(machine) {
+	for _, ff := range clientinfo.GetFeatureFlagList(machine) {
 		t.AppendRow(table.Row{"Feature Flags", ff})
 	}