feat(prompt): starship-style prompt layout and GetSessionCount RPC

Restructure the two-line prompt for both client and implant menus:
- Status line: name on hostname os/arch via pipeline age (group)
- Input line: sessionID ❯

Add lightweight GetSessionCount RPC to display accurate alive/total
session counts from memory and DB respectively.

Author: h3zh1

client/core/console.go

@@ -15,6 +15,7 @@ import (
 
 	"github.com/chainreactors/IoM-go/client"
 	"github.com/chainreactors/IoM-go/consts"
+	"github.com/chainreactors/IoM-go/proto/client/clientpb"
 	"github.com/chainreactors/logs"
 	"github.com/chainreactors/malice-network/client/repl"
 	"github.com/reeflective/console"
@@ -261,7 +262,7 @@ func (c *Console) GetPrompt() string {
 
 	session := c.ActiveTarget.Get()
 	if session != nil {
-		promptLine = tui.NewSessionColor(session.GroupName, session.SessionId[:8]) + " " + promptLine
+		promptLine = tui.DefaultNameStyle.Render(session.SessionId[:8]) + " " + promptLine
 	}
 
 	if statusLine == "" {
@@ -304,14 +305,7 @@ func (c *Console) getStatusLine() string {
 
 	session := c.ActiveTarget.Get()
 	if session == nil {
-		// Client menu: user on v0.5.0 sessions 3/5
-		var alive, total int
-		for _, s := range c.Sessions {
-			total++
-			if s.IsAlive {
-				alive++
-			}
-		}
+		// Client menu: user on v0.5.0 sessions alive/total
 		version := ""
 		if c.Info != nil {
 			version = c.Info.Version
@@ -320,32 +314,37 @@ func (c *Console) getStatusLine() string {
 		if c.Client != nil {
 			name = c.Client.Name
 		}
+		sessionInfo := fmt.Sprintf("%d", len(c.Sessions))
+		count, err := c.Rpc.GetSessionCount(context.Background(), &clientpb.Empty{})
+		if err == nil && count != nil {
+			sessionInfo = fmt.Sprintf("%d/%d", count.Alive, count.Total)
+		}
 		return fmt.Sprintf("%s %s %s %s %s",
 			tui.CyanFg.Render(name),
 			tui.DarkGrayFg.Render("on"),
 			tui.GreenFg.Render(version),
 			tui.DarkGrayFg.Render("sessions"),
-			tui.YellowFg.Render(fmt.Sprintf("%d/%d", alive, total)),
+			tui.YellowFg.Render(sessionInfo),
 		)
 	}
 
-	// Implant menu: [note] hostname os/arch via pipeline age
-	parts := make([]string, 0, 7)
-	if session.Note != "" {
-		parts = append(parts, tui.WhiteFg.Bold(true).Render(session.Note))
-	}
+	// Implant menu: name on hostname os/arch via pipeline age (group)
+	parts := make([]string, 0, 8)
 	hostname := ""
 	osInfo := ""
 	if session.Os != nil {
 		hostname = session.Os.Hostname
 		osInfo = session.Os.Name + "/" + session.Os.Arch
 	}
 	parts = append(parts,
+		tui.WhiteFg.Bold(true).Render(session.Name),
+		tui.DarkGrayFg.Render("on"),
 		tui.CyanFg.Render(hostname),
 		tui.GreenFg.Render(osInfo),
 		tui.DarkGrayFg.Render("via"),
 		tui.PurpleFg.Render(session.PipelineId),
 		tui.YellowFg.Render(formatCheckinAge(session.LastCheckin)),
+		tui.DarkGrayFg.Render("("+session.GroupName+")"),
 	)
 	return strings.Join(parts, " ")
 }

external/IoM-go

@@ -40,6 +40,18 @@ func (rpc *Server) GetSessions(ctx context.Context, req *clientpb.SessionRequest
 	return sessions, nil
 }
 
+func (rpc *Server) GetSessionCount(ctx context.Context, req *clientpb.Empty) (*clientpb.SessionCount, error) {
+	alive := int32(len(core.Sessions.All()))
+	total, err := db.NewSessionQuery().WhereRemoved(false).Count()
+	if err != nil {
+		return nil, err
+	}
+	return &clientpb.SessionCount{
+		Alive: alive,
+		Total: int32(total),
+	}, nil
+}
+
 func (rpc *Server) GetSession(ctx context.Context, req *clientpb.SessionRequest) (*clientpb.Session, error) {
 	session, err := core.Sessions.Get(req.SessionId)
 	if err == nil {